paper-with-me

홈 › Papers

Improving LLM Unlearning Robustness via Random Perturbations

2025-01-31 · Dang Huu-Tien, Hoang Thanh-Tung, Anh Bui, Le-Minh Nguyen, Naoya Inoue

In this paper, we show that current state-of-the-art LLM unlearning methods inherently reduce models' robustness, causing them to misbehave even when a single non-adversarial forget-token is in the retain-query. Toward understanding underlying causes, we reframe the unlearning process as backdoor attacks and defenses: forget-tokens act as backdoor triggers that, when activated in retain-queries, cause disruptions in unlearned models' behaviors, similar to successful backdoor attacks. To mitigate this vulnerability, we propose Random Noise Augmentation (RNA) -- a plug-and-play, model and method agnostic approach with theoretical guarantees for improving the robustness of unlearned models. Extensive experiments demonstrate that RNA significantly improves the robustness of unlearned models, maintains unlearning performances while introducing no additional computational overhead.

📄 PDF Abstract BibTeX arXiv:2501.19202

Code (1)

rebelsnlu-jaist/llmu-robustness 공식 구현 pytorch

Tasks

Language ModelingLanguage ModellingLarge Language Model

Similar Papers 제목 키워드 기반

Downgrade to Upgrade: Optimizer Simplification Enhances Robustness in LLM Unlearning

2025-10-01 · Yicheng Lang, Yihua Zhang, Chongyu Fan, Changsheng Wang 외 arxiv

Large language model (LLM) unlearning aims to surgically remove the influence of undesired data or knowledge from an existing model while preserving its utility on unrelated tasks. This paradigm has shown promise in addr…

Protecting the Neural Networks against FGSM Attack Using Machine Unlearning

2025-11-03 · Amir Hossein Khorasani, Ali Jahanian, Maryam Rastgarpour arxiv

Machine learning is a powerful tool for building predictive models. However, it is vulnerable to adversarial attacks. Fast Gradient Sign Method (FGSM) attacks are a common type of adversarial attack that adds small pertu…

Image ClassificationAdversarial Attack

SoK: Unlearnability and Unlearning for Model Dememorization

2026-05-12 · Mengying Zhang, Derui Wang, Ruoxi Sun, Xiaoyu Xia 외 arxiv

Advanced model dememorization methods, including availability poisoning (unlearnability) and machine unlearning, are emerging as key safeguards against data misuse in machine learning (ML). At the training stage, unlearn…

Abstract Gradient Training: A Unified Certification Framework for Data Poisoning, Unlearning, and Differential Privacy

2025-11-12 · Philip Sosnin, Matthew Wicker, Josh Collyer, Calvin Tsay arxiv

The impact of inference-time data perturbation (e.g., adversarial attacks) has been extensively studied in machine learning, leading to well-established certification techniques for adversarial robustness. In contrast, c…

Adversarial Robustness

GraphMU: Repairing Robustness of Graph Neural Networks via Machine Unlearning

2024-06-19 · Tao Wu, Xinwen Cao, Chao Wang, Shaojie Qiao 외

Graph Neural Networks (GNNs) have demonstrated significant application potential in various fields. However, GNNs are still vulnerable to adversarial attacks. Numerous adversarial defense methods on GNNs are proposed to …

Adversarial AttackAdversarial DefenseMachine Unlearning