AttentionMix: A Guided Text Data Augmentation Method Relying on Attention
摘要
The Mixup method has proven to be a powerful data augmentation technique in Computer Vision, with many successors that perform image mixing in a guided manner. One of the interesting research directions is transferring the underlying Mixup idea to other domains, e.g. Natural Language Processing (NLP). Even though there already exist several methods that apply Mixup to textual data, there is still room for new, improved approaches. In this work, we introduce AttentionMix, a novel mixing method that relies on attention-based information. While the paper focuses on the BERT attention mechanism, the proposed approach can technically be applied with any attention-based model. AttentionMix is evaluated on 3 standard sentiment classification datasets and in most of the tested setups outperforms two benchmark approaches that utilize Mixup mechanism, as well as the vanilla BERT method. These findings demonstrate that attention-based information can be effectively used for data augmentation in the NLP domain, indicating the relevance of further research along this line.