AWARe: Mitigating Catastrophic Forgetting via Activation-Weighted Adaptive REtention

Juncheng Liao, Jinfan Lv, Guoming Wang, Juepeng Zheng, Ling Xiao, Siliang Tang


Abstract
Multimodal Large Language Models (MLLMs) exhibit strong generalization and reasoning abilities due to large-scale multimodal pre-training. However, fine-tuning these models on downstream tasks often leads to catastrophic forgetting, where newly learned task-specific knowledge degrades previously acquired capabilities. This issue arises because gradient updates for new tasks overwrite parameters critical to prior knowledge, limiting the practical deployment of MLLMs. To address this challenge, we propose **Activation-Weighted Adaptive REtention (AWARe)**, a fine-tuning method that mitigates catastrophic forgetting by dynamically controlling parameter updates based on activation patterns. AWARe assigns activation-based importance scores to parameters, selectively freezing those essential for preserving prior capabilities while allowing less important parameters to adapt to new tasks. Importantly, AWARe operates without modifying model architectures, ensuring compatibility with existing inference engines. Extensive experiments demonstrate that AWARe effectively preserves upstream capabilities while achieving superior downstream performance compared to existing methods. Code is available at https://github.com/kaln27/AWARe.
Anthology ID:
2026.emnlp-main.604
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
13928–13944
Language:
URL:
https://aclanthology.org/2026.emnlp-main.604/
DOI:
Bibkey:
Cite (ACL):
Juncheng Liao, Jinfan Lv, Guoming Wang, Juepeng Zheng, Ling Xiao, and Siliang Tang. 2026. AWARe: Mitigating Catastrophic Forgetting via Activation-Weighted Adaptive REtention. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 13928–13944, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
AWARe: Mitigating Catastrophic Forgetting via Activation-Weighted Adaptive REtention (Liao et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.604.pdf