KGA: A General Machine Unlearning Framework Based on Knowledge Gap Alignment
Open accessAbstract
Recent legislation of the "right to be forgotten" has led to the interest in machine unlearning, where the learned models are endowed with the function to forget information about specific training instances as if they have never existed in the training set.Previous work mainly focuses on computer vision scenarios and largely ignores the essentials of unlearning in NLP field, where text data contains more explicit and sensitive personal information than images.In this paper, we propose a general unlearning framework called KGA to induce forgetfulness.Different from previous work that tries to recover gradients or forces models to perform close to one specific distribution, KGA maintains distribution differences (i.e., knowledge gap).This relaxes the distribution assumption.Furthermore, we first apply the unlearning method to various NLP tasks (i.e., classification, translation, response generation) and propose several unlearning evaluation metrics with pertinence.Experiments on large-scale datasets show that KGA yields comprehensive improvements over baselines, where extensive analyses further validate the effectiveness of KGA and provide insight into unlearning for NLP tasks 1 .
Integrity signals
Registry
Cites
Statistics
Claims the paper makes
Claims other papers make about this one
Each is checked against this paper’s own text: does it say what the citing paper says it does?
The main advantage of KGA lies in its generic nature, which allows it to be applied to various models and tasks.
“The main advan -tage of this method lies in its generic nature, which allows it to be applied to various models and tasks.”
From Digital forgetting in large language models: a survey of unlearning methods (2025) · cites this paper as Wang et al (2023)
A few authors use the success of a Membership Inference Attack (MIA) as a metric.
“Membership Inference Attack (MIA) : A few authors (Wang et al 2023; Chen and Yang 2023; Pawelczyk et al 2023) use the success of a MIA as a metric.”
From Digital forgetting in large language models: a survey of unlearning methods (2025) · cites this paper as Wang et al 2023
Wang et al (2023) measure output similarity with the text to be forgotten using JSD, LPD, and PDLP, while Yao et al (2023) use BLEU.
“For instance, Wang et al (2023) measure output similarity with the text to be forgotten by using the Jensen-Shannon Divergence (JSD), the Language model Probability Distance (LPD), and the Proportion of instances with Decreased Language model Probability (PDLP), while Yao et al (2023) leverage the BiLingual Evaluation Understudy (BLEU) score.”
From Digital forgetting in large language models: a survey of unlearning methods (2025) · cites this paper as Wang et al (2023)
Eldan and Russinovich (2023), Wang et al (2023), and Jang et al (2022) leverage well-established LLM benchmarks to measure retained utility.
“General benchmarks . Eldan and Russinovich (2023), Wang et al (2023), and Jang et al (2022) leverage well-established LLM benchmarks, such as those used for comparing LLMs (Sakaguchi et al 2020; Zellers et al 2019; Paperno et al 2016; Bisk et al 2020).”
From Digital forgetting in large language models: a survey of unlearning methods (2025) · cites this paper as Wang et al (2023)
Reported runtime values of unlearning methods range from minutes to hours or even longer than a day.
“Reported values range from minutes (Wang et al 2023; Yao et al 2023 ; Limisiewicz and Mareček 2022; Jang et al 2022; Yu et al 2023) to hours (Chen and Yang 2023) or even longer than a day (Lu et al 2022 ).”
From Digital forgetting in large language models: a survey of unlearning methods (2025) · cites this paper as Wang et al 2023
Chen and Yang (2023), Wang et al (2023), Pawelczyk et al (2023), Ni et al (2023), and Pochinkov and Schoots (2023) evaluate retaining on remaining samples of the same dataset used for unlearning.
“Same dataset but different samples . Chen and Yang (2023); Wang et al (2023); Pawelczyk et al (2023); Ni et al (2023); Pochinkov and Schoots (2023) evaluate retaining on the remaining samples of the same dataset used for unlearning.”
From Digital forgetting in large language models: a survey of unlearning methods (2025) · cites this paper as Wang et al (2023)
Cites
Adam: A Method for Stochastic Optimization
Kingma, D. P., & Ba, J. (2014) · arXiv · doi:10.48550/arxiv.1412.6980 · Cited by 98,946 · see citing works
BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding
Devlin, J., Chang, M.-W., Lee, K., & Toutanova, K. (2019) · Proceedings of the 2019 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies, Volume 1 (Long and Short Papers) · doi:10.18653/v1/n19-1423 · Cited by 48,293 · see citing works
(2014) · Cited by 33,419 · see citing works
Lewis, M., Liu, Y., Goyal, N., Ghazvininejad, M., Mohamed, A., Levy, O., Stoyanov, V., & Zettlemoyer, L. (2020) · Proceedings of the 58th Annual Meeting of the Association for Computational Linguistics · doi:10.18653/v1/2020.acl-main.703 · Cited by 8,821 · see citing works
DistilBERT, a distilled version of BERT: smaller, faster, cheaper and lighter
Sanh, V., Debut, L., Chaumond, J., & Wolf, T. (2019) · arXiv · doi:10.48550/arxiv.1910.01108 · Cited by 4,653 · see citing works
Understanding Black-box Predictions via Influence Functions
Koh, P. W., & Liang, P. (2017) · arXiv · doi:10.48550/arxiv.1703.04730 · Cited by 1,514 · see citing works
Personalizing Dialogue Agents: I have a dog, do you have pets too?
Zhang, S., Dinan, E., Urbanek, J., Szlam, A., Kiela, D., & Weston, J. (2018) · Proceedings of the 56th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers) · doi:10.18653/v1/p18-1205 · Cited by 1,177 · see citing works
Salem, A., Zhang, Y., Humbert, M., Berrang, P., Fritz, M., & Backes, M. (2019) · Proceedings 2019 Network and Distributed System Security Symposium · doi:10.14722/ndss.2019.23119 · Cited by 852 · see citing works
Towards Making Systems Forget with Machine Unlearning
Cao, Y., & Yang, J. (2015) · 2015 IEEE Symposium on Security and Privacy · doi:10.1109/sp.2015.35 · Cited by 622 · see citing works
Eternal Sunshine of the Spotless Net: Selective Forgetting in Deep Networks
Golatkar, A., Achille, A., & Soatto, S. (2020) · 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR) · doi:10.1109/cvpr42600.2020.00932 · Cited by 394 · see citing works
Multiple Incremental Decremental Learning of Support Vector Machines
Karasuyama, M., & Takeuchi, I. (2009) · IEEE Transactions on Neural Networks · doi:10.1109/tnn.2010.2048039 · Cited by 176 · see citing works
Making AI Forget You: Data Deletion in Machine Learning
Ginart, A., Guan, M. Y., Valiant, G., & Zou, J. (2019) · arXiv · doi:10.48550/arxiv.1907.05012 · Cited by 175 · see citing works
Certified Data Removal from Machine Learning Models
Guo, C., Goldstein, T., Hannun, A., & van der Maaten, L. (2019) · arXiv · doi:10.48550/arxiv.1911.03030 · Cited by 160 · see citing works
Can Bad Teaching Induce Forgetting? Unlearning in Deep Networks Using an Incompetent Teacher
Chundawat, V. S., Tarun, A. K., Mandal, M., & Kankanhalli, M. (2023) · Proceedings of the AAAI Conference on Artificial Intelligence · doi:10.1609/aaai.v37i6.25879 · Cited by 129 · see citing works
Golatkar, A., Achille, A., & Soatto, S. (2020) · Lecture Notes in Computer Science · doi:10.1007/978-3-030-58526-6_23 · Cited by 122 · see citing works
Challenges in Detoxifying Language Models
Welbl, J., Glaese, A., Uesato, J., Dathathri, S., Mellor, J., Hendricks, L. A., Anderson, K., Kohli, P., Coppin, B., & Huang, P.-S. (2021) · Findings of the Association for Computational Linguistics: EMNLP 2021 · doi:10.18653/v1/2021.findings-emnlp.210 · Cited by 112 · see citing works
Deep Unlearning via Randomized Conditionally Independent Hessians
Mehta, R., Pal, S., Singh, V. P., & Ravi, S. N. (2022) · IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings - IEEE Computer Society Conference on Computer Vision and Pattern Recognition/Proceedings · doi:10.1109/cvpr52688.2022.01017 · Cited by 62 · see citing works
LEDGAR : a large-scale multi-label corpus for text classification of legal provisions in contracts
Tuggener, D., von Däniken, P., Peetz, T., & Cieliebak, M. (2020) · Zürcher Hochschule für Angewandte Wissenschaften digital collection (Zurich University of Applied Sciences) · doi:10.21256/zhaw-20087 · Cited by 45 · see citing works
Incremental and Decremental Learning for Linear Support Vector Machines
Romero, E., Barrio, I., & Belanche, L. (2007) · International Conference on Artificial Neural Networks · doi:10.1007/978-3-540-74690-4_22 · Cited by 33 · see citing works
Knowledge-Adaptation PriorsPreprint
Khan, M. E., & Swaroop, S. (2021) · arXiv (Cornell University) · doi:10.48550/arxiv.2106.08769 · Cited by 14 · see citing works
Flagged first, then by citations
1 of the 21 references couldn’t be matched to a known paper.
Cited by
Sorted by citations
Rethinking machine unlearning for large language models
Liu, S., Yao, Y., Jia, J., Casper, S., Baracaldo, N., Hase, P., Yao, Y., Liu, C. Y., Xu, X., Li, H., et al. (2025) · Nature Machine Intelligence · doi:10.1038/s42256-025-00985-0 · Cited by 173 · see citing works
Machine Unlearning: Taxonomy, Metrics, Applications, Challenges, and Prospects
Li, N., Zhou, C., Gao, Y., Chen, H., Zhang, Z., Kuang, B., & Fu, A. (2025) · IEEE Transactions on Neural Networks and Learning Systems · doi:10.1109/tnnls.2025.3530988 · Cited by 53 · see citing works
Digital forgetting in large language models: a survey of unlearning methods
Blanco-Justicia, A., Jebreel, N., Manzanares-Salor, B., Sánchez, D., Domingo-Ferrer, J., Collell, G., & Eeik Tan, K. (2025) · Artificial Intelligence Review · doi:10.1007/s10462-024-11078-6 · Cited by 31 · see citing works
The Frontier of Data Erasure: A Survey on Machine Unlearning for Large Language Models
Qu, Y., Ding, M., Sun, N., Thilakarathna, K., Zhu, T., & Niyato, D. T. (2025) · Computer · doi:10.1109/mc.2024.3405397 · Cited by 29 · see citing works
Kumar, R., Prakash, R., Kumar, N., Chinara, S., & Mohammed, M. A. (2025) · Archives of Computational Methods in Engineering · doi:10.1007/s11831-025-10436-z · Cited by 14 · see citing works
Federated TrustChain: Blockchain-Enhanced LLM Training and Unlearning
Zuo, X., Wang, M., Zhu, T., Zhang, L., Ye, D., Yu, S., & Zhou, W. (2026) · IEEE Transactions on Dependable and Secure Computing · doi:10.1109/tdsc.2026.3665277 · Cited by 14 · see citing works
MultiDelete for Multimodal Machine Unlearning
Cheng, J., & Amiri, H. (2024) · Lecture notes in computer science · doi:10.1007/978-3-031-72940-9_10 · Cited by 11 · see citing works
A survey on large language models unlearning: taxonomy, evaluations, and future directions
Le-Khac, U. N., & Truong, V. (2025) · Artificial Intelligence Review · doi:10.1007/s10462-025-11376-7 · Cited by 9 · see citing works
Yuan, X., Liu, J., Wang, B., Chen, G., Xu, X., Wang, J., Li, T., & Wang, W. (2025) · IEEE Transactions on Information Forensics and Security · doi:10.1109/tifs.2025.3583231 · Cited by 5 · see citing works
Privacy dilemmas and opportunities in large language models: a brief review
Li, H., Ye, J., & Wu, J. (2025) · Frontiers of Computer Science · doi:10.1007/s11704-024-40583-8 · Cited by 6 · see citing works
Unlearning in Large Language Models: We Are Not There Yet
Blanco-Justicia, A., Domingo‐Ferrer, J., Jebreel, N., Manzanares-Salor, B., & Sánchez, D. (2025) · Computer · doi:10.1109/mc.2024.3468588 · Cited by 4 · see citing works
A Study Regarding Machine Unlearning on Facial Attribute Data
Gündoğdu, E., Unal, A., & Ünal, G. (2024) · International Conference on Automatic Face and Gesture Recognition · doi:10.1109/fg59268.2024.10581972 · Cited by 4 · see citing works
Iqbal, S., Zhong, X., Khan, M. A., Wu, Z., Almujally, N. A., Liu, W., & Hussain, A. (2025) · Information Processing & Management · doi:10.1016/j.ipm.2025.104417 · Cited by 3 · see citing works
From Forgetting to Future: A Survey of Machine Unlearning Approaches
Sehgal, S., Verma, A., & Bansal, H. (2026) · Wiley Interdisciplinary Reviews Data Mining and Knowledge Discovery · doi:10.1002/widm.70082 · Cited by 1 · see citing works
Few-Shot Unlearning for Large Language Models Through Pretrained Soft Prompts
Tang, Z. H., & Yeh, M.-Y. (2025) · Lecture notes in computer science · doi:10.1007/978-981-96-8186-0_14
KaRe: Towards Flexible and Effective Machine Unlearning with Knowledge Alignment and Repair
Chai, Z., Liu, Q., Hu, Q., & Li, P. (2025) · Lecture notes of the Institute for Computer Sciences, Social Informatics and Telecommunications Engineering · doi:10.1007/978-3-031-93251-9_16
MiniUn: A Machine Unlearning Method to Minimize Dependency on Original Training Data
Youn, S., & Kim, C. (2026) · IEEE Access · doi:10.1109/access.2026.3653817
Learn while Unlearn: An Iterative Unlearning Framework for Generative Language Models
Tang, H., Liu, Y., Zhao, X., Liu, X., Zhang, Y., Zhang, K., Zhou, X., & Chen, E. (2025) · Proceedings · doi:10.1109/icdm65498.2025.00082 · Cited by 1 · see citing works
Zhang, Y., Cz, Z., Li, Z., Xu, T., Zhang, K., Gao, Z., Li, M., Zhang, X., Qi, Q., & Chen, B. (2026) · IEEE Transactions on Consumer Electronics · doi:10.1109/tce.2026.3670687
Empirical Analysis of Loss Functions for Machine Unlearning Across Minority and Majority Classes
Ruthi Niyenthri, S., Nithya Leela, R., Kumar, M. A., Anjali, R., & Karpagam, G. R. (2026) · International Conference on Intelligent Systems for Communication, IoT and Security (ICISCoIS) · doi:10.1109/iciscois62701.2026.11447967
Mahale, N. J., & Sangole, T. R. (2026) · International Conference on Emerging Trends and Innovations in ICT (ICEI) · doi:10.1109/icei65890.2026.11448092
The Right to be Forgotten in the Age of AI: Legal, Philosophical, and Technical Challenges
Bucci, A., Domingo‐Ferrer, J., Monreale, A., & Naretto, F. (2026) · Communications in computer and information science · doi:10.1007/978-3-032-19105-2_42
Forgetting Similar Samples: Can Machine Unlearning Do it Better?
Xu, H., Zhu, T., Ye, D., Zhang, L., Wang, L., & Zhou, W. (2026) · IEEE Transactions on Dependable and Secure Computing · doi:10.1109/tdsc.2026.3714774
Self-citationForgotten Horizons in Concept Erasure: Safeguarding Close-Proximity Concepts in Text-to-Image Models
Li, J. F., Liu, Y., Yu, R., Zhang, C., Qi, G., Bi, S., Liu, F., Geng, J., & Karray, F. O. (2026) · IEEE Transactions on Information Forensics and Security · doi:10.1109/tifs.2026.3714133
Turning Over a New Leaf: Machine Unlearning for Malicious Code Removal in LLMsPreprint
Nguyen, M. X., Bui, A. M. T., Vu, D. H., Nguyen, P. T., Rocco, J. D., Sipio, C. D., & Ruscio, D. D. (2026) · SSRN Electronic Journal · doi:10.2139/ssrn.7412271
- Authors
- Wang, L., Chen, T., Yuan, W., Zeng, X., Wong, K.-F., & Yin, H.
- Affiliations
- Chinese University of Hong Kong (HK)
- The University of Queensland (AU)
- Venue
- Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
- Published
- 2023
- Pages
- 13264-13276
- ISSN
- 0736-587X (ISSN-L)
- OpenAlex
- W4385571042
- Cited by
- 25
- References
- 21
- Citations by year
- 2026: 10 · 2025: 13 · 2024: 2
- FWCI
- 4.43
- Type
- conference paper
- Language
- en
- Open access
- hybrid
- doi.org
- Licence
- cc-by
- Locations
- Proceedings of the conference - Association for Computational Linguistics. Meeting (publishedVersion, cc-by)
- arXiv (Cornell University) (submittedVersion)
- arXiv (Cornell University)
- Funders
- Chinese University of Hong Kong
- Australian Research Council (FT210100624, DE230101033, DP190101985)
- Topics
- Physical Sciences › Computer Science › Artificial Intelligence › Privacy-Preserving Technologies in Data
- Physical Sciences › Computer Science › Artificial Intelligence › Adversarial Robustness in Machine Learning
- Physical Sciences › Computer Science › Artificial Intelligence › Domain Adaptation and Few-Shot Learning
- Keywords
- machine unlearning, right to be forgotten, NLP tasks, text classification, machine translation, response generation
About these checks
Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.
Statistics and arithmetic are recomputed, not judged.
Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.
What counts as a claim ›How we measure confidence ›How we check this ›
Are you an author, or do you see a problem with this page? Request a review