Search papers, labs, and topics across Lattice.
The paper identifies temporal drift in parameter importance during supervised fine-tuning (SFT) of large language models, showing that parameters critical early in training may become less important later. To address this, they introduce Evolving Parameter Isolation (EPI), a framework that dynamically updates parameter isolation masks based on online gradient-based estimates of parameter importance. Experiments on multi-task benchmarks demonstrate that EPI reduces task interference and catastrophic forgetting compared to static isolation methods and standard fine-tuning.
Parameter importance isn't forever: dynamically adapting which parameters are frozen during fine-tuning significantly improves generalization and reduces forgetting in LLMs.
Supervised Fine-Tuning (SFT) of large language models often suffers from task interference and catastrophic forgetting. Recent approaches alleviate this issue by isolating task-critical parameters during training. However, these methods represent a static solution to a dynamic problem, assuming that parameter importance remains fixed once identified. In this work, we empirically demonstrate that parameter importance exhibits temporal drift over the course of training. To address this, we propose Evolving Parameter Isolation (EPI), a fine-tuning framework that adapts isolation decisions based on online estimates of parameter importance. Instead of freezing a fixed subset of parameters, EPI periodically updates isolation masks using gradient-based signals, enabling the model to protect emerging task-critical parameters while releasing outdated ones to recover plasticity. Experiments on diverse multi-task benchmarks demonstrate that EPI consistently reduces interference and forgetting compared to static isolation and standard fine-tuning, while improving overall generalization. Our analysis highlights the necessity of synchronizing isolation mechanisms with the evolving dynamics of learning diverse abilities.