Skip to content
arXiv· arXiv:2601.22030· Chengyi Cai, Zesheng Ye, Jiangchao Yao, Jianzhong Qi, Bo Han, Xiaolu Zhang, Feng Liu, Jun Zhou·Indexed by this site · Originally published

Per-Parameter Task Arithmetic for Large Language Model Unlearning

Per-parameter Task Arithmetic for Unlearning in Large Language Models

AI summary

Researchers propose per-parameter task arithmetic (PerTA), which mitigates over-forgetting in large language model unlearning by scaling task vectors according to parameter importance. The method estimates each parameter's importance relative to forgetting and retention, and computes scaling weights using gradients (PerTA-grad) or a diagonal Fisher information approximation (PerTA-fisher). Experiments show that PerTA consistently outperforms standard task vectors in forgetting effectiveness and overall model utility, and in many cases also exceeds commonly used training-based unlearning methods, while retaining the efficiency of task arithmetic.

Read originalarxiv.org