We gratefully acknowledge support from
the Simons Foundation and member institutions.
Full-text links:


Current browse context:


Change to browse by:


References & Citations

DBLP - CS Bibliography


(what is this?)
CiteULike logo BibSonomy logo Mendeley logo del.icio.us logo Digg logo Reddit logo ScienceWISE logo

Computer Science > Computation and Language

Title: Improved Text Classification via Contrastive Adversarial Training

Abstract: We propose a simple and general method to regularize the fine-tuning of Transformer-based encoders for text classification tasks. Specifically, during fine-tuning we generate adversarial examples by perturbing the word embeddings of the model and perform contrastive learning on clean and adversarial examples in order to teach the model to learn noise-invariant representations. By training on both clean and adversarial examples along with the additional contrastive objective, we observe consistent improvement over standard fine-tuning on clean examples. On several GLUE benchmark tasks, our fine-tuned BERT Large model outperforms BERT Large baseline by 1.7% on average, and our fine-tuned RoBERTa Large improves over RoBERTa Large baseline by 1.3%. We additionally validate our method in different domains using three intent classification datasets, where our fine-tuned RoBERTa Large outperforms RoBERTa Large baseline by 1-2% on average.
Subjects: Computation and Language (cs.CL)
Cite as: arXiv:2107.10137 [cs.CL]
  (or arXiv:2107.10137v2 [cs.CL] for this version)

Submission history

From: Lin Pan [view email]
[v1] Wed, 21 Jul 2021 15:14:15 GMT (103kb,D)
[v2] Thu, 17 Feb 2022 19:56:07 GMT (120kb,D)

Link back to: arXiv, form interface, contact.