Papers
arxiv:2602.17385

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

Published on Feb 23
Authors:
,
,
,
,
,
,

Abstract

Task arithmetic adapts foundation models through modular combinations of task vectors, but suffers from representation drift that degrades performance; a dataless regularization approach using curvature matrix approximation addresses this issue with improved robustness and scalability.

AI-generated summary

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.

Community

Sign up or log in to comment

Get this paper in your agent:

hf papers read 2602.17385
Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash

Models citing this paper 1

Datasets citing this paper 0

No dataset linking this paper

Cite arxiv.org/abs/2602.17385 in a dataset README.md to link it from this page.

Spaces citing this paper 0

No Space linking this paper

Cite arxiv.org/abs/2602.17385 in a Space README.md to link it from this page.

Collections including this paper 0

No Collection including this paper

Add this paper to a collection to link it from this page.