cs.CLOct 21, 2026

C2^2U in RAG: Compositional Concept Unlearning in Retrieval-Augmented Generation

Authors: Yiyang Duan, Fan Wu, Tiantong Wu, Che Wang, Zhiren Gong, Wei Guo, Yang Cao, Wei Yang Bryan Lim

Organizations: Midori Lab, Nanyang Technological University · Beihang University · Institute of Science Tokyo (formerly Tokyo Institute of Technology)

Abstract

Retrieval-augmented generation can expose unwanted concepts through retrieved evidence even when the generator is moderated. We study query-side concept suppression with a fixed encoder and document index. Compositional Concept Unlearning (C2^2U) routes a profile-derived direction through an augmented atom bank, adding only a bounded edited-minus-original delta to the raw query. Across five datasets, four encoders, and three seeds, the validation-selected configuration reduces pooled binary target Hit@5 by 4.25 percentage points with non-positive control degradation and target-exposure changes. Under the evaluated query transforms, it suppresses target density without increasing budgeted control-query exposure. Separately, ArXiv four-concept composition improves over an averaged-direction comparator after independent validation gates and first-use test evaluation.

Figures & tables

Appendix figures & tables16 assets

Supplementary material from the paper’s appendix.

Appendix

Explore similar work

CardsList
  1. RAGSieve: Self-Referenced Local Contrast for Knowledge-Poison Detection in Retrieval-Augmented Generation

    Aug 13, 2026Xinlong Xu, Yoshua Y. LiRAG Poisoning AttacksRAG Security

  2. Know Before You Fetch: Calibrated Retrieval-Budget Allocation for Retrieval-Augmented Generation

    Jun 29, 2026Zhe Dong, Fang Qin, Manish Shah +1Retrieval-Augmented GenerationSelective Prediction

  3. Retrieval Augmented Generation Compositionality

    Oct 21, 2026Ido Levy, Segev Shlomov, Leshem ChoshenRetrieval-Augmented GenerationRAG Evaluation