SSMBA: Self-Supervised Manifold Based Data Augmentation for Improving Out-of-Domain Robustness
Not on the map. Why?
Abstract (arXiv)
Models that perform well on a training domain often fail to generalize to out-of-domain (OOD) examples. Data augmentation is a common method used to prevent overfitting and improve OOD generalization. However, in natural language, it is difficult to generate new examples that stay on the underlying data manifold. We introduce SSMBA, a data augmentation method for generating synthetic training examples by using a pair of corruption and reconstruction functions to move randomly on a data manifold. We investigate the use of SSMBA in the natural language domain, leveraging the manifold assumption to reconstruct corrupted text with masked language models. In experiments on robustness benchmarks across 3 tasks and 9 datasets, SSMBA consistently outperforms existing data augmentation methods and baseline models on both in-domain and OOD data, achieving gains of 0.8% accuracy on OOD Amazon reviews, 1.8% accuracy on OOD MNLI, and 1.4 BLEU on in-domain IWSLT14 German-English.
Key facts. Cited by 94 papers in the map (data through 7 Oct 2026). First posted on arXiv 21 Sep 2020.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 19; 2022: 24; 2023: 27; 2024: 15; 2025: 7; 2026: 2.
94 papers.
Sort:
By area:
- 2021A Survey of Data Augmentation Approaches for NLPcited by 408
- 2021GPT3Mix: Leveraging Large-scale Language Models for Text Augmentationcited by 185
- 2023Increasing Diversity While Maintaining Accuracy: Text Data Generation with Large Language Models and Human Interventionscited by 121
- 2022State-of-the-art generalisation research in NLP: A taxonomy and reviewcited by 97
- 2023Revisiting Out-of-distribution Robustness in NLP: Benchmark, Analysis, and LLMs Evaluationscited by 96
- 2021Data Augmentation Approaches in Natural Language Processing: A Surveycited by 82
- 2021Towards Improving Adversarial Training of NLP Modelscited by 79
- 2021Understanding the Properties of Minimum Bayes Risk Decoding in Neural Machine Translationcited by 56
- 2022Perturbation Augmentation for Fairer NLPcited by 51
- 2021An Investigation of the (In)effectiveness of Counterfactually Augmented Datacited by 42
- 2021SSMix: Saliency-Based Span Mixup for Text Classificationcited by 39
- 2022DoCoGen: Domain Counterfactual Generation for Low Resource Domain Adaptationcited by 36
38 more
- 2021FlipDA: Effective and Robust Data Augmentation for Few-Shot Learningcited by 32
- 2022Improved Multi-label Classification under Temporal Concept Drift: Rethinking Group-Robust Algorithms in a Label-Wise Settingcited by 23
- 2023Mixture of Soft Prompts for Controllable Data Generationcited by 23
- 2022CoNT: Contrastive Neural Text Generationcited by 21
- 2021GOLD: Improving Out-of-Scope Detection in Dialogues using Data Augmentationcited by 20
- 2021Self-training Improves Pre-training for Few-shot Learning in Task-oriented Dialog Systemscited by 18
- 2024HarmAug: Effective Data Augmentation for Knowledge Distillation of Safety Guard Modelscited by 17
- 2022TreeMix: Compositional Constituency-based Data Augmentation for Natural Language Understandingcited by 15
- 2022Improving the Sample Efficiency of Prompt Tuning with Domain Adaptationcited by 15
- 2021Learning to Perturb Word Embeddings for Out-of-distribution QAcited by 13
- 2022Learning to Generalize to More: Continuous Semantic Augmentation for Neural Machine Translationcited by 13
- 2024What Variables Affect Out-of-Distribution Generalization in Pretrained Models?cited by 13
- 2023ASSERT: Automated Safety Scenario Red Teaming for Evaluating the Robustness of Large Language Modelscited by 12
- 2022Estimating Soft Labels for Out-of-Domain Intent Detectioncited by 11
- 2021Not Far Away, Not So Close: Sample Efficient Nearest Neighbour Data Augmentation via MiniMaxcited by 10
- 2021ALP: Data Augmentation using Lexicalized PCFGs for Few-Shot Text Classificationcited by 10
- 2021On-the-Fly Aligned Data Augmentation for Sequence-to-Sequence ASRcited by 9
- 2021Consistency Training with Virtual Adversarial Discrete Perturbationcited by 9
- 2021Exploiting Position Bias for Robust Aspect Sentiment Classificationcited by 9
- 2022Impact of Adversarial Training on Robustness and Generalizability of Language Modelscited by 9
- 2024CoDa: Constrained Generation based Data Augmentation for Low-Resource NLPcited by 9
- 2022GeoECG: Data Augmentation via Wasserstein Geodesic Perturbation for Robust Electrocardiogram Predictioncited by 8
- 2024Improving Black-box Robustness with In-Context Rewritingcited by 8
- 2024ABEX: Data Augmentation for Low-Resource NLU via Expanding Abstract Descriptionscited by 8
- 2022EPiDA: An Easy Plug-in Data Augmentation Framework for High Performance Text Classificationcited by 7
- 2023Feature Programming for Multivariate Time Series Predictioncited by 7
- 2024Connect Later: Improving Fine-tuning for Robustness with Targeted Augmentationscited by 7
- 2021LINDA: Unsupervised Learning to Interpolate in Natural Language Processingcited by 6
- 2022DSLOB: A Synthetic Limit Order Book Dataset for Benchmarking Forecasting Algorithms under Distributional Shiftcited by 6
- 2022When Chosen Wisely, More Data Is What You Need: A Universal Sample-Efficient Strategy For Data Augmentationcited by 5
- 2023Masked Language Model Based Textual Adversarial Example Detectioncited by 5
- 2023Neural Machine Translation Data Generation and Augmentation using ChatGPTcited by 5
- 2024An Effective Deployment of Diffusion LM for Data Augmentation in Low-Resource Sentiment Classificationcited by 5
- 2022Predicting Out-of-Domain Generalization with Neighborhood Invariancecited by 4
- 2023"A Tale of Two Movements": Identifying and Comparing Perspectives in #BlackLivesMatter and #BlueLivesMatter Movements-related Tweets using Weakly Supervised Graph-based Structured Predictioncited by 4
- 2024Controllable and Diverse Data Augmentation with Large Language Model for Low-Resource Open-Domain Dialogue Generationcited by 4
- 2024Targeted Augmentation for Low-Resource Event Extractioncited by 4
- 2025Exploring Gradient-Guided Masked Language Model to Detect Textual Adversarial Attackscited by 4
50 most cited of 94 All 94 (JSON)
- 2026KItCAT: Knowledge Injection via Input Corruption for Auto-regressive Training31 Aug 2026
- 2026Layering Virtual Try-On24 Jul 2026
- 2025D-GAP: Improving Out-of-Domain Robustness via Dataset-Agnostic and Gradient-Guided Augmentation in Frequency and Pixel Spaces14 Nov 2025
- 2025Filtering with Confidence: When Data Augmentation Meets Conformal Prediction25 Sep 2025
- 2025Improving Out-of-Domain Robustness with Targeted Augmentation in Frequency and Pixel Spaces18 May 2025
- 2025Detect, Explain, Escalate: Sustainable Dialogue Breakdown Management for LLM Agents26 Apr 2025
- 2025The Synthetic Imputation Approach: Generating Optimal Synthetic Texts For Underrepresented Categories In Supervised Classification Tasks21 Apr 2025
- 2025Exploring Gradient-Guided Masked Language Model to Detect Textual Adversarial Attacks8 Apr 2025
- 2025Unified Enhancement of the Generalization and Robustness of Language Models via Bi-Stage Optimization19 Mar 2025
- 2024Latent Paraphrasing: Perturbation on Layers Improves Knowledge Injection in Language Models1 Nov 2024
- 2024Relation-based Counterfactual Data Augmentation and Contrastive Learning for Robustifying Natural Language Inference Models28 Oct 2024
- 2024SimpleStrat: Diversifying Language Model Generation with Stratification11 Oct 2024
38 more
- 2024HarmAug: Effective Data Augmentation for Knowledge Distillation of Safety Guard Models2 Oct 2024
- 2024An Effective Deployment of Diffusion LM for Data Augmentation in Low-Resource Sentiment Classification5 Sep 2024
- 2024ABEX: Data Augmentation for Low-Resource NLU via Expanding Abstract Descriptions6 Jun 2024
- 2024Masked Language Modeling Becomes Conditional Density Estimation for Tabular Data Synthesis31 May 2024
- 2024What Variables Affect Out-of-Distribution Generalization in Pretrained Models?23 May 2024
- 2024Targeted Augmentation for Low-Resource Event Extraction14 May 2024
- 2024Enhance Robustness of Language Models Against Variation Attack through Graph Integration18 Apr 2024
- 2024CoDa: Constrained Generation based Data Augmentation for Low-Resource NLP30 Mar 2024
- 2024Controllable and Diverse Data Augmentation with Large Language Model for Low-Resource Open-Domain Dialogue Generation30 Mar 2024
- 2024Induced Model Matching: Restricted Models Help Train Full-Featured Models19 Feb 2024
- 2024Improving Black-box Robustness with In-Context Rewriting13 Feb 2024
- 2024Connect Later: Improving Fine-tuning for Robustness with Targeted Augmentations8 Jan 2024
- 2023RoAST: Robustifying Language Models via Adversarial Perturbation with Selective Training7 Dec 2023
- 2023Vicinal Risk Minimization for Few-Shot Cross-lingual Transfer in Abusive Language Detection3 Nov 2023
- 2023ASSERT: Automated Safety Scenario Red Teaming for Evaluating the Robustness of Large Language Models14 Oct 2023
- 2023"A Tale of Two Movements": Identifying and Comparing Perspectives in #BlackLivesMatter and #BlueLivesMatter Movements-related Tweets using Weakly Supervised Graph-based Structured Prediction11 Oct 2023
- 2023Unlock the Potential of Counterfactually-Augmented Data in Out-Of-Distribution Generalization10 Oct 2023
- 2023Robust Sentiment Analysis for Low Resource languages Using Data Augmentation Approaches: A Case Study in Marathi1 Oct 2023
- 2023Introducing "Forecast Utterance" for Conversational Data Science7 Sep 2023
- 2023Steering Language Generation: Harnessing Contrastive Expert Guidance and Negative Prompting for Coherent and Diverse Synthetic Data Generation15 Aug 2023
- 2023Feature-aware conditional GAN for category text generation2 Aug 2023
- 2023Improving the Reusability of Pre-trained Language Models in Real-world Applications19 Jul 2023
- 2023Neural Machine Translation Data Generation and Augmentation using ChatGPT11 Jul 2023
- 2023Synthetic Alone: Exploring the Dark Side of Synthetic Data for Grammatical Error Correction26 Jun 2023
- 2023Investigating Masking-based Data Generation in Language Models16 Jun 2023
- 2023Feature Programming for Multivariate Time Series Prediction9 Jun 2023
- 2023Revisiting Out-of-distribution Robustness in NLP: Benchmark, Analysis, and LLMs Evaluations7 Jun 2023
- 2023Increasing Diversity While Maintaining Accuracy: Text Data Generation with Large Language Models and Human Interventions7 Jun 2023
- 2023Simple Data Augmentation Techniques for Chinese Disease Normalization2 Jun 2023
- 2023MSMix:An Interpolation-Based Text Data Augmentation Method Manifold Swap Mixup31 May 2023
- 2023Improving Classifier Robustness through Active Generation of Pairwise Counterfactuals22 May 2023
- 2023Boosting Event Extraction with Denoised Structure-to-Text Augmentation16 May 2023
- 2023Progressive Translation: Improving Domain Robustness of Neural Machine Translation with Intermediate Sequences16 May 2023
- 2023ReMask: A Robust Information-Masking Approach for Domain Counterfactual Generation4 May 2023
- 2023Masked Language Model Based Textual Adversarial Example Detection18 Apr 2023
- 2023Mixture of Soft Prompts for Controllable Data Generation2 Mar 2023
- 2023BLIAM: Literature-based Data Synthesis for Synergistic Drug Combination Prediction14 Feb 2023
- 2023Coherence and Diversity through Noise: Self-Supervised Paraphrase Generation via Structure-Aware Denoising6 Feb 2023
50 newest of 94 All 94 (JSON)
- 2021Self-training Improves Pre-training for Few-shot Learning in Task-oriented Dialog Systemscited by 18
- 2023BLIAM: Literature-based Data Synthesis for Synergistic Drug Combination Predictioncited by 3
- 2023Steering Language Generation: Harnessing Contrastive Expert Guidance and Negative Prompting for Coherent and Diverse Synthetic Data Generationcited by 2
- 2024Masked Language Modeling Becomes Conditional Density Estimation for Tabular Data Synthesiscited by 1
- 2024Masked Language Modeling Becomes Conditional Density Estimation for Tabular Data Synthesis31 May 2024
- 2023Steering Language Generation: Harnessing Contrastive Expert Guidance and Negative Prompting for Coherent and Diverse Synthetic Data Generation15 Aug 2023
- 2023BLIAM: Literature-based Data Synthesis for Synergistic Drug Combination Prediction14 Feb 2023
- 2021Self-training Improves Pre-training for Few-shot Learning in Task-oriented Dialog Systems28 Aug 2021
- 2021A Survey of Data Augmentation Approaches for NLPcited by 408
- 2021GPT3Mix: Leveraging Large-scale Language Models for Text Augmentationcited by 185
- 2023Increasing Diversity While Maintaining Accuracy: Text Data Generation with Large Language Models and Human Interventionscited by 121
- 2022State-of-the-art generalisation research in NLP: A taxonomy and reviewcited by 97
- 2023Revisiting Out-of-distribution Robustness in NLP: Benchmark, Analysis, and LLMs Evaluationscited by 96
- 2021Data Augmentation Approaches in Natural Language Processing: A Surveycited by 82
- 2021Towards Improving Adversarial Training of NLP Modelscited by 79
- 2021Understanding the Properties of Minimum Bayes Risk Decoding in Neural Machine Translationcited by 56
- 2022Perturbation Augmentation for Fairer NLPcited by 51
- 2021An Investigation of the (In)effectiveness of Counterfactually Augmented Datacited by 42
- 2021SSMix: Saliency-Based Span Mixup for Text Classificationcited by 39
- 2022DoCoGen: Domain Counterfactual Generation for Low Resource Domain Adaptationcited by 36
12 most cited of 88 All 88 (JSON)
- 2026KItCAT: Knowledge Injection via Input Corruption for Auto-regressive Training31 Aug 2026
- 2026Layering Virtual Try-On24 Jul 2026
- 2025D-GAP: Improving Out-of-Domain Robustness via Dataset-Agnostic and Gradient-Guided Augmentation in Frequency and Pixel Spaces14 Nov 2025
- 2025Filtering with Confidence: When Data Augmentation Meets Conformal Prediction25 Sep 2025
- 2025Improving Out-of-Domain Robustness with Targeted Augmentation in Frequency and Pixel Spaces18 May 2025
- 2025The Synthetic Imputation Approach: Generating Optimal Synthetic Texts For Underrepresented Categories In Supervised Classification Tasks21 Apr 2025
- 2025Exploring Gradient-Guided Masked Language Model to Detect Textual Adversarial Attacks8 Apr 2025
- 2025Unified Enhancement of the Generalization and Robustness of Language Models via Bi-Stage Optimization19 Mar 2025
- 2024Latent Paraphrasing: Perturbation on Layers Improves Knowledge Injection in Language Models1 Nov 2024
- 2024Relation-based Counterfactual Data Augmentation and Contrastive Learning for Robustifying Natural Language Inference Models28 Oct 2024
- 2024SimpleStrat: Diversifying Language Model Generation with Stratification11 Oct 2024
- 2024HarmAug: Effective Data Augmentation for Knowledge Distillation of Safety Guard Models2 Oct 2024
12 newest of 88 All 88 (JSON)