Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Dataless Weight Disentanglement in Task Arithmetic via Kronecker-Factored Approximate Curvature

About

Task Arithmetic yields a modular, scalable way to adapt foundation models. Combining multiple task vectors, however, can lead to cross-task interference, causing representation drift and degraded performance. Representation drift regularization provides a natural remedy to disentangle task vectors; however, existing approaches typically require external task data, conflicting with modularity and data availability constraints (e.g., privacy requirements). We propose a dataless approach by framing regularization against representation drift as a curvature matrix approximation problem. This allows us to leverage well-established techniques; in particular, we adopt Kronecker-Factored Approximate Curvature and obtain a practical regularizer that achieves state-of-the-art results in task addition and negation. Our method has constant complexity in the number of tasks and promotes robustness to task vector rescaling, eliminating the need for held-out tuning.

Angelo Porrello, Pietro Buzzega, Felix Dangel, Thomas Sommariva, Riccardo Salami, Lorenzo Bonicelli, Simone Calderara• 2026

Related benchmarks

TaskDatasetResultRank
Task addition8 Vision
Absolute Score91.6
65
Task Negation8 Vision
Control Accuracy73.6
61
Task Negation14-Vision
Target Accuracy5.6
8
Task addition6NLI base (test)
Absolute Accuracy78.6
7
Showing 4 of 4 rows

Other info

Follow for update