Improved Text Classification via Contrastive Adversarial Training
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 38 papers in the map (data through 7 Oct 2026). Builds on 46 of its 51 references that are in the map. First posted on arXiv 21 Jul 2021.
Builds on
46 of 51 references.
References · 46
- 2017Attention Is All You Needcited by 59,932
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,204
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2018Representation Learning with Contrastive Predictive Codingcited by 7,662
- 2019Momentum Contrast for Unsupervised Visual Representation Learningcited by 7,482
- 2014Explaining and Harnessing Adversarial Examplescited by 7,287
- 2017Towards Deep Learning Models Resistant to Adversarial Attackscited by 6,326
- 2013Intriguing properties of neural networkscited by 5,114
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,854
- 2020Supervised Contrastive Learningcited by 2,803
- 2017A Broad-Coverage Challenge Corpus for Sentence Understanding through Inferencecited by 2,665
34 more
- 2021SimCSE: Simple Contrastive Learning of Sentence Embeddingscited by 2,168
- 2020DeBERTa: Decoding-enhanced BERT with Disentangled Attentioncited by 2,157
- 2015A large annotated corpus for learning natural language inferencecited by 1,837
- 2020Don't Stop Pretraining: Adapt Language Models to Domains and Taskscited by 1,548
- 2018Obfuscated Gradients Give a False Sense of Security: Circumventing Defenses to Adversarial Examplescited by 1,259
- 2020MPNet: Masked and Permuted Pre-training for Language Understandingcited by 973
- 2015The Limitations of Deep Learning in Adversarial Settingscited by 784
- 2019Adversarial Training for Free!cited by 638
- 2017One pixel attack for fooling deep neural networkscited by 579
- 2020Efficient Intent Detection with Dual Sentence Encoderscited by 419
- 2019How to Fine-Tune BERT for Text Classification?cited by 409
- 2019An Evaluation Dataset for Intent Classification and Out-of-Scope Predictioncited by 378
- 2017Adversarial Examples: Attacks and Defenses for Deep Learningcited by 350
- 2016Adversarial Training Methods for Semi-Supervised Text Classificationcited by 280
- 2017Adversarial Examples for Semantic Segmentation and Object Detectioncited by 275
- 2020Supervised Contrastive Learning for Pre-trained Language Model Fine-tuningcited by 275
- 2019FreeLB: Enhanced Adversarial Training for Natural Language Understandingcited by 261
- 2018Adversarial Attacks and Defences: A Surveycited by 259
- 2020DeCLUTR: Deep Contrastive Learning for Unsupervised Textual Representationscited by 249
- 2020InfoXLM: An Information-Theoretic Framework for Cross-Lingual Language Model Pre-Trainingcited by 237
- 2020CLEAR: Contrastive Learning for Sentence Representationcited by 186
- 2020CERT: Contrastive Self-supervised Learning for Language Understandingcited by 169
- 2019Benchmarking Natural Language Understanding Services for building Conversational Agentscited by 166
- 2018Physical Adversarial Examples for Object Detectorscited by 163
- 2018ShapeShifter: Robust Physical Adversarial Attack on Faster R-CNN Object Detectorcited by 159
- 2017On the Robustness of Semantic Segmentation Models to Adversarial Attackscited by 126
- 2021COCO-LM: Correcting and Contrasting Text Sequences for Language Model Pretrainingcited by 126
- 2020Self-training Improves Pre-training for Natural Language Understandingcited by 96
- 2015Distributional Smoothing with Virtual Adversarial Trainingcited by 61
- 2020On Learning Universal Representations Across Languagescited by 58
- 2020Multilingual BERT Post-Pretraining Alignmentcited by 30
- 2020Attention Meets Perturbations: Robust and Interpretable Attention with Adversarial Trainingcited by 12
- 2020Benchmarking Commercial Intent Detection Services with Practice-Driven Evaluationscited by 8
- 2021Making Attention Mechanisms More Robust and Interpretable with Virtual Adversarial Trainingcited by 8
Built on by
Citing papers per year: 2021: 1; 2022: 10; 2023: 9; 2024: 14; 2025: 4.
38 papers.
Sort:
By area:
- 2024Latent Adversarial Training Improves Robustness to Persistent Harmful Behaviors in LLMscited by 158
- 2024Black-Box Access is Insufficient for Rigorous AI Auditscited by 123
- 2024Efficient Adversarial Training in LLMs with Continuous Attackscited by 104
- 2024Soft Prompt Threats: Attacking Safety Alignment and Unlearning in Open-Source LLMs through the Embedding Spacecited by 88
- 2024Defending Against Unforeseen Failure Modes with Latent Adversarial Trainingcited by 82
- 2022Incorporating Hierarchy into Text Encoder: a Contrastive Learning Approach for Hierarchical Text Classificationcited by 52
- 2023ConDA: Contrastive Domain Adaptation for AI-generated Text Detectioncited by 37
- 2023Improving In-context Learning of Multilingual Generative Language Models with Cross-lingual Alignmentcited by 33
- 2021Improving Neural Cross-Lingual Summarization via Employing Optimal Transport Distance for Knowledge Distillationcited by 22
- 2023Towards Viewpoint-Invariant Visual Recognition via Adversarial Trainingcited by 19
- 2024Surgical-VQLA++: Adversarial Contrastive Learning for Calibrated Robust Visual Question-Localized Answering in Robotic Surgerycited by 17
- 2024Towards Interpretable Hate Speech Detection using Large Language Model-extracted Rationalescited by 16
26 more
- 2022Supervised Contrastive Learning with Hard Negative Samplescited by 13
- 2024Sim-CLIP: Unsupervised Siamese Adversarial Fine-Tuning for Robust and Semantically-Rich Vision-Language Modelscited by 12
- 2024Is Contrasting All You Need? Contrastive Learning for the Detection and Attribution of AI-generated Textcited by 11
- 2022Impact of Adversarial Training on Robustness and Generalizability of Language Modelscited by 9
- 2023HateProof: Are Hateful Meme Detection Systems really Robust?cited by 7
- 2024Adversarial Training: A Surveycited by 6
- 2022Cyberbullying Classifiers are Sensitive to Model-Agnostic Perturbationscited by 5
- 2023Towards a Holistic Understanding of Mathematical Questions with Contrastive Pre-trainingcited by 5
- 2023Expand BERT Representation with Visual Information via Grounded Language Learning with Multimodal Partial Alignmentcited by 5
- 2023VendorLink: An NLP approach for Identifying & Linking Vendor Migrants & Potential Aliases on Darknet Marketscited by 3
- 2024Vision HgNN: An Electron-Micrograph is Worth Hypergraph of Hypernodescited by 3
- 2022Two-Level Supervised Contrastive Learning for Response Selection in Multi-Turn Dialoguecited by 2
- 2022A Novel Approach to Train Diverse Types of Language Models for Health Mention Classification of Tweetscited by 2
- 2022A Benchmark Study of Contrastive Learning for Arabic Social Meaningcited by 2
- 2023Proxy-based Zero-Shot Entity Linking by Effective Candidate Retrievalcited by 2
- 2024RobustSentEmbed: Robust Sentence Embeddings Using Adversarial Self-Supervised Contrastive Learningcited by 2
- 2024Multi-label Sequential Sentence Classification via Large Language Modelcited by 2
- 2025Tougher Text, Smarter Models: Raising the Bar for Adversarial Defence Benchmarkscited by 2
- 2025SECRET: Semi-supervised Clinical Trial Document Similarity Searchcited by 2
- 2022Improving Health Mentioning Classification of Tweets using Contrastive Adversarial Trainingcited by 1
- 2022Contrastive Learning of Sociopragmatic Meaning in Social Mediacited by 1
- 2022Zero-shot stance detection based on cross-domain feature enhancement by contrastive learningcited by 1
- 2023DEFTri: A Few-Shot Label Fused Contextual Representation Learning For Product Defect Triage in e-Commercecited by 0
- 2024QLSC: A Query Latent Semantic Calibrator for Robust Extractive Question Answeringcited by 0
- 2025A Simple Graph Contrastive Learning Framework for Short Text Classificationcited by 0
- 2025Indirect Attention: Turning Context Misalignment into a Featurecited by 0
- 2025Indirect Attention: Turning Context Misalignment into a Feature30 Sep 2025
- 2025SECRET: Semi-supervised Clinical Trial Document Similarity Search16 May 2025
- 2025A Simple Graph Contrastive Learning Framework for Short Text Classification16 Jan 2025
- 2025Tougher Text, Smarter Models: Raising the Bar for Adversarial Defence Benchmarks5 Jan 2025
- 2024Multi-label Sequential Sentence Classification via Large Language Model23 Nov 2024
- 2024Adversarial Training: A Survey19 Oct 2024
- 2024Vision HgNN: An Electron-Micrograph is Worth Hypergraph of Hypernodes21 Aug 2024
- 2024Surgical-VQLA++: Adversarial Contrastive Learning for Calibrated Robust Visual Question-Localized Answering in Robotic Surgery9 Aug 2024
- 2024Latent Adversarial Training Improves Robustness to Persistent Harmful Behaviors in LLMs22 Jul 2024
- 2024Sim-CLIP: Unsupervised Siamese Adversarial Fine-Tuning for Robust and Semantically-Rich Vision-Language Models20 Jul 2024
- 2024Is Contrasting All You Need? Contrastive Learning for the Detection and Attribution of AI-generated Text12 Jul 2024
- 2024Efficient Adversarial Training in LLMs with Continuous Attacks24 May 2024
26 more
- 2024QLSC: A Query Latent Semantic Calibrator for Robust Extractive Question Answering30 Apr 2024
- 2024Towards Interpretable Hate Speech Detection using Large Language Model-extracted Rationales19 Mar 2024
- 2024RobustSentEmbed: Robust Sentence Embeddings Using Adversarial Self-Supervised Contrastive Learning17 Mar 2024
- 2024Defending Against Unforeseen Failure Modes with Latent Adversarial Training8 Mar 2024
- 2024Soft Prompt Threats: Attacking Safety Alignment and Unlearning in Open-Source LLMs through the Embedding Space14 Feb 2024
- 2024Black-Box Access is Insufficient for Rigorous AI Audits25 Jan 2024
- 2023Expand BERT Representation with Visual Information via Grounded Language Learning with Multimodal Partial Alignment4 Dec 2023
- 2023Improving In-context Learning of Multilingual Generative Language Models with Cross-lingual Alignment14 Nov 2023
- 2023ConDA: Contrastive Domain Adaptation for AI-generated Text Detection7 Sep 2023
- 2023DEFTri: A Few-Shot Label Fused Contextual Representation Learning For Product Defect Triage in e-Commerce21 Jul 2023
- 2023Towards Viewpoint-Invariant Visual Recognition via Adversarial Training16 Jul 2023
- 2023VendorLink: An NLP approach for Identifying & Linking Vendor Migrants & Potential Aliases on Darknet Markets4 May 2023
- 2023HateProof: Are Hateful Meme Detection Systems really Robust?11 Feb 2023
- 2023Proxy-based Zero-Shot Entity Linking by Effective Candidate Retrieval30 Jan 2023
- 2023Towards a Holistic Understanding of Mathematical Questions with Contrastive Pre-training18 Jan 2023
- 2022Impact of Adversarial Training on Robustness and Generalizability of Language Models10 Nov 2022
- 2022A Benchmark Study of Contrastive Learning for Arabic Social Meaning22 Oct 2022
- 2022Zero-shot stance detection based on cross-domain feature enhancement by contrastive learning7 Oct 2022
- 2022Supervised Contrastive Learning with Hard Negative Samples31 Aug 2022
- 2022A Novel Approach to Train Diverse Types of Language Models for Health Mention Classification of Tweets13 Apr 2022
- 2022Contrastive Learning of Sociopragmatic Meaning in Social Media15 Mar 2022
- 2022Incorporating Hierarchy into Text Encoder: a Contrastive Learning Approach for Hierarchical Text Classification8 Mar 2022
- 2022Improving Health Mentioning Classification of Tweets using Contrastive Adversarial Training3 Mar 2022
- 2022Two-Level Supervised Contrastive Learning for Response Selection in Multi-Turn Dialogue1 Mar 2022
- 2022Cyberbullying Classifiers are Sensitive to Model-Agnostic Perturbations17 Jan 2022
- 2021Improving Neural Cross-Lingual Summarization via Employing Optimal Transport Distance for Knowledge Distillation7 Dec 2021
- 2024Black-Box Access is Insufficient for Rigorous AI Auditscited by 123
- 2022Incorporating Hierarchy into Text Encoder: a Contrastive Learning Approach for Hierarchical Text Classificationcited by 52
- 2023ConDA: Contrastive Domain Adaptation for AI-generated Text Detectioncited by 37
- 2023Improving In-context Learning of Multilingual Generative Language Models with Cross-lingual Alignmentcited by 33
- 2021Improving Neural Cross-Lingual Summarization via Employing Optimal Transport Distance for Knowledge Distillationcited by 22
- 2023Towards Viewpoint-Invariant Visual Recognition via Adversarial Trainingcited by 19
- 2024Surgical-VQLA++: Adversarial Contrastive Learning for Calibrated Robust Visual Question-Localized Answering in Robotic Surgerycited by 17
- 2024Towards Interpretable Hate Speech Detection using Large Language Model-extracted Rationalescited by 16
- 2022Supervised Contrastive Learning with Hard Negative Samplescited by 13
- 2024Is Contrasting All You Need? Contrastive Learning for the Detection and Attribution of AI-generated Textcited by 11
- 2022Impact of Adversarial Training on Robustness and Generalizability of Language Modelscited by 9
- 2023HateProof: Are Hateful Meme Detection Systems really Robust?cited by 7
12 most cited of 33 All 33 (JSON)
- 2025Indirect Attention: Turning Context Misalignment into a Feature30 Sep 2025
- 2025SECRET: Semi-supervised Clinical Trial Document Similarity Search16 May 2025
- 2025A Simple Graph Contrastive Learning Framework for Short Text Classification16 Jan 2025
- 2025Tougher Text, Smarter Models: Raising the Bar for Adversarial Defence Benchmarks5 Jan 2025
- 2024Multi-label Sequential Sentence Classification via Large Language Model23 Nov 2024
- 2024Adversarial Training: A Survey19 Oct 2024
- 2024Vision HgNN: An Electron-Micrograph is Worth Hypergraph of Hypernodes21 Aug 2024
- 2024Surgical-VQLA++: Adversarial Contrastive Learning for Calibrated Robust Visual Question-Localized Answering in Robotic Surgery9 Aug 2024
- 2024Is Contrasting All You Need? Contrastive Learning for the Detection and Attribution of AI-generated Text12 Jul 2024
- 2024QLSC: A Query Latent Semantic Calibrator for Robust Extractive Question Answering30 Apr 2024
- 2024Towards Interpretable Hate Speech Detection using Large Language Model-extracted Rationales19 Mar 2024
- 2024RobustSentEmbed: Robust Sentence Embeddings Using Adversarial Self-Supervised Contrastive Learning17 Mar 2024
12 newest of 33 All 33 (JSON)