mlsanityData to 14 Oct 2026

CURE: Controlled Unlearning for Robust Embeddings -- Mitigating Conceptual Shortcuts in Pre-Trained Language Models

arXiv 2509.05230 ↗First posted 5 Sep 2025cs.CL cs.AI cs.LGcited by 0

On the map: LLM safety and security → Machine unlearning

Abstract (arXiv)

Pre-trained language models have achieved remarkable success across diverse applications but remain susceptible to spurious, concept-driven correlations that impair robustness and fairness. In this work, we introduce CURE, a novel and lightweight framework that systematically disentangles and suppresses conceptual shortcuts while preserving essential content information. Our method first extracts concept-irrelevant representations via a dedicated content extractor reinforced by a reversal network, ensuring minimal loss of task-relevant information. A subsequent controllable debiasing module employs contrastive learning to finely adjust the influence of residual conceptual cues, enabling the model to either diminish harmful biases or harness beneficial correlations as appropriate for the target task. Evaluated on the IMDB and Yelp datasets using three pre-trained architectures, CURE achieves an absolute improvement of +10 points in F1 score on IMDB and +2 points on Yelp, while introducing minimal computational overhead. Our approach establishes a flexible, unsupervised blueprint for combating conceptual biases, paving the way for more reliable and fair language understanding systems.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Machine unlearning, one of 12 directions in LLM safety and security. Builds on 22 of its 43 references that are in the map. First posted on arXiv 5 Sep 2025.

Builds on

22 of 43 references.

References · 22

  1. 2017Attention Is All You Needcited by 60,361
  2. 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
  3. 2022Training language models to follow instructions with human feedbackcited by 15,971
  4. 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
  5. 2017A Unified Approach to Interpreting Model Predictionscited by 6,555
  6. 2015Character-level Convolutional Networks for Text Classificationcited by 2,330
  7. 2020GLU Variants Improve Transformercited by 1,804
  8. 2023ChatGPT Outperforms Crowd-Workers for Text-Annotation Taskscited by 870
  9. 2015Improving Neural Machine Translation Models with Monolingual Datacited by 788
  10. 2018Dear Sir or Madam, May I introduce the GYAFC Dataset: Corpus, Benchmarks and Metrics for Formality Style Transfercited by 211
  11. 2020An Empirical Study on Robustness to Spurious Correlations using Pre-trained Language Modelscited by 134
  12. 2019Unlearn Dataset Bias in Natural Language Inference by Fitting the Residualcited by 119
10 more
  1. 2018How Much Reading Does Reading Comprehension Require? A Critical Investigation of Popular Benchmarkscited by 88
  2. 2019Style Transformer: Unpaired Text Style Transfer without Disentangled Latent Representationcited by 84
  3. 2021Identifying and Mitigating Spurious Correlations for Improving Robustness in NLP Modelscited by 76
  4. 2020Identifying Spurious Correlations for Robust Text Classificationcited by 59
  5. 2023Explore Spurious Correlations at the Concept Level in Language Models for Text Classificationcited by 48
  6. 2021Unifying Heterogeneous Electronic Health Records Systems via Text-Based Code Embeddingcited by 24
  7. 2022Causal Intervention Improves Implicit Sentiment Analysiscited by 16
  8. 2019Topics to Avoid: Demoting Latent Confounds in Text Classificationcited by 14
  9. 2022Less Learn Shortcut: Analyzing and Mitigating Learning of Spurious Feature-Label Correlationcited by 13
  10. 2024RAZOR: Sharpening Knowledge by Cutting Bias with Unsupervised Text Rewritingcited by 5

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025Scalable and Robust LLM Unlearning by Correcting Responses with Retrieved Exclusions
  2. 2025CRISP: Persistent Concept Unlearning via Sparse Autoencoders
  3. 2025CURE: Concept Unlearning via Orthogonal Representation Editing in Diffusion Models
  4. 2024Espresso: Robust Concept Filtering in Text-to-Image Models
  5. 2026Selective Fine-Tuning for Targeted and Robust Concept Unlearning
  6. 2026Don't Forget Your Embeddings: Robust Knowledge Erasure via Precise Editing of Embeddings
  7. 2025Align-then-Unlearn: Embedding Alignment for LLM Unlearning
  8. 2024Erasing Conceptual Knowledge from Language Models