Per-parameter Task Arithmetic for Unlearning in Large Language Models
On the map: LLM safety and security → Machine unlearning
Abstract (arXiv)
In large language model (LLM) unlearning, private information is required to be removed. Task arithmetic unlearns by subtracting a specific task vector (TV)--defined as the parameter difference between a privacy-information-tuned model and the original model. While efficient, it can cause over-forgetting by disrupting parameters essential for retaining other information. Motivated by the observation that each parameter exhibits different importance for forgetting versus retention, we propose a per-parameter task arithmetic (PerTA) mechanism to rescale the TV, allowing per-parameter adjustment. These weights quantify the relative importance of each parameter for forgetting versus retention, estimated via gradients (i.e., PerTA-grad) or the diagonal Fisher information approximation (i.e., PerTA-fisher). Moreover, we discuss the effectiveness of PerTA, extend it to a more general form, and provide further analysis. Extensive experiments demonstrate that PerTA consistently improves upon standard TV, and in many cases surpasses widely used training-based unlearning methods in both forgetting effectiveness and overall model utility. By retaining the efficiency of task arithmetic while mitigating over-forgetting, PerTA offers a principled and practical framework for LLM unlearning.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Placed in Machine unlearning, one of 12 directions in LLM safety and security. Builds on 50 of its 57 references that are in the map. First posted on arXiv 29 Jan 2026.
Builds on
50 of 57 references.
Closest ancestors · 22
Same direction, newest first
- 2025OpenUnlearning: Accelerating LLM Unlearning via Unified Benchmarking of Methods and Metrics14 Jun
- 2025RULE: Reinforcement UnLEarning Achieves Forget-Retain Pareto Optimality8 Jun
- 2025Exploring Criteria of Loss Reweighting to Enhance LLM Unlearning17 May
- 2025GRU: Mitigating the Trade-off between Unlearning and Retention for LLMs12 Mar
- 2025Rethinking LLM Unlearning Objectives: A Gradient Perspective and Go Beyond26 Feb
- 2025LLM Unlearning via Neural Activation Redirection11 Feb
- 2025Towards LLM Unlearning Resilient to Relearning Attacks: A Sharpness-Aware Minimization Perspective and Beyond7 Feb
- 2024Catastrophic Failure of LLM Unlearning via Quantization21 Oct
- 2024Simplicity Prevails: Rethinking Negative Preference Optimization for LLM Unlearning9 Oct
- 2024MUSE: Machine Unlearning Six-Way Evaluation for Language Models8 Jul
- 2024Split, Unlearn, Merge: Leveraging Data Attributes for More Effective Unlearning in LLMs17 Jun
- 2024Towards Effective Evaluations and Comparisons for LLM Unlearning Methods13 Jun
10 more
- 2024SOUL: Unlocking the Power of Second-Order Optimization for LLM Unlearning28 Apr
- 2024Negative Preference Optimization: From Catastrophic Collapse to Effective Unlearning8 Apr
- 2024Rethinking Machine Unlearning for Large Language Models13 Feb
- 2024TOFU: A Task of Fictitious Unlearning for LLMs11 Jan
- 2023Large Language Model Unlearning14 Oct
- 2023In-Context Unlearning: Language Models as Few Shot Unlearners11 Oct
- 2022Knowledge Unlearning for Mitigating Privacy Risks in Language Models4 Oct
- 2022Quark: Controllable Text Generation with Reinforced Unlearning26 May
- 2022Continual Learning and Private Unlearning24 Mar
- 2021Fast Yet Effective Machine Unlearning17 Nov
Other references · 28
- 2023LLaMA: Open and Efficient Foundation Language Modelscited by 14,132
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
- 2020Extracting Training Data from Large Language Modelscited by 2,189
- 2014On the Properties of Neural Machine Translation: Encoder-Decoder Approachescited by 1,304
- 2022Editing Models with Task Arithmeticcited by 1,120
- 2019Machine Unlearningcited by 1,061
- 2023TIES-Merging: Resolving Interference When Merging Modelscited by 818
- 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 694
- 2024The WMDP Benchmark: Measuring and Reducing Malicious Use With Unlearningcited by 509
- 2019Making AI Forget You: Data Deletion in Machine Learningcited by 465
- 2014New insights and perspectives on the natural gradient methodcited by 443
- 2023A Survey on Large Language Model (LLM) Security and Privacy: The Good, the Bad, and the Uglycited by 387
16 more
- 2018Generating Wikipedia by Summarizing Long Sequencescited by 335
- 2023AdaMerging: Adaptive Model Merging for Multi-Task Learningcited by 252
- 2019Analyzing the Structure of Attention in a Transformer Language Modelcited by 243
- 2023Task Arithmetic in the Tangent Space: Improved Editing of Pre-Trained Modelscited by 221
- 2024Knowledge Fusion of Large Language Modelscited by 115
- 2023Copyright Violations and Large Language Modelscited by 108
- 2024MetaGPT: Merging Large Language Models Using Model Exclusive Task Arithmeticcited by 53
- 2023Gradient-based Parameter Selection for Efficient Fine-Tuningcited by 40
- 2024Knowledge Composition using Task Vectors with Learned Anisotropic Scalingcited by 37
- 2023Beyond Size: How Gradients Shape Pruning Decisions in Large Language Modelscited by 30
- 2018Fisher Information and Natural Gradient Learning of Random Deep Networkscited by 28
- 2025AdaMMS: Model Merging for Heterogeneous Multimodal Large Language Models with Unsupervised Coefficient Optimizationcited by 25
- 2024Model Composition for Multimodal Large Language Modelscited by 12
- 2025Dynamic Fisher-weighted Model Merging via Bayesian Optimizationcited by 12
- 2024SPP: Sparsity-Preserved Parameter-Efficient Fine-Tuning for Large Language Modelscited by 11
- 2024Knowledge Fusion of Chat LLMs: A Preliminary Technical Reportcited by 6
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2023Unlearn What You Want to Forget: Efficient Unlearning for LLMs
- 2025iShumei-Chinchunmei at SemEval-2025 Task 4: A balanced forgetting and retention multi-task framework using effective unlearning loss
- 2024Machine Unlearning of Pre-trained Large Language Models
- 2025Unlearning That Lasts: Utility-Preserving, Robust, and Almost Irreversible Forgetting in LLMs
- 2024Digital Forgetting in Large Language Models: A Survey of Unlearning Methods
- 2026Not Every Change Is Necessary: Recoverable Drift in Large Language Model Unlearning
- 2025Unlearning Isn't Deletion: Investigating Reversibility of Machine Unlearning in LLMs
- 2025Not Every Token Needs Forgetting: Selective Unlearning to Limit Change in Utility in Large Language Model Unlearning