Dataless Weight Disentanglement in Task Arithmetic via Kronecker-Factored Approximate Curvature
Not on the map. Why?
Abstract (arXiv)
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.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 49 of its 66 references that are in the map. First posted on arXiv 19 Feb 2026.
Builds on
49 of 66 references.
References · 49
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,491
- 2016Communication-Efficient Learning of Deep Networks from Decentralized Datacited by 7,592
- 2018GLUE: A Multi-Task Benchmark and Analysis Platform for Natural Language Understandingcited by 4,854
- 2019Advances and Open Problems in Federated Learningcited by 3,021
- 2016Deep Learning with Differential Privacycited by 2,682
- 2017A Broad-Coverage Challenge Corpus for Sentence Understanding through Inferencecited by 2,665
- 2018Neural Tangent Kernel: Convergence and Generalization in Neural Networkscited by 2,597
- 2020The Many Faces of Robustness: A Critical Analysis of Out-of-Distribution Generalizationcited by 1,842
- 2015A large annotated corpus for learning natural language inferencecited by 1,837
- 2017EuroSAT: A Novel Dataset and Deep Learning Benchmark for Land Use and Land Cover Classificationcited by 1,488
- 2022Editing Models with Task Arithmeticcited by 1,076
37 more
- 2023TIES-Merging: Resolving Interference When Merging Modelscited by 795
- 2017Remote Sensing Image Scene Classification: Benchmark and State of the Artcited by 791
- 2015Optimizing Neural Networks with Kronecker-factored Approximate Curvaturecited by 776
- 2019On Exact Computation with an Infinitely Wide Neural Netcited by 572
- 2014New insights and perspectives on the natural gradient methodcited by 435
- 2021Laplace Redux -- Effortless Bayesian Deep Learningcited by 359
- 2023Task Arithmetic in the Tangent Space: Improved Editing of Pre-Trained Modelscited by 218
- 2018Online Structured Laplace Approximations For Overcoming Catastrophic Forgettingcited by 197
- 2020Mixed-Privacy Forgetting in Deep Networkscited by 158
- 2024Task Singular Vectors: Reducing Task Interference in Model Mergingcited by 143
- 2017Practical Gauss-Newton Optimisation for Deep Learningcited by 139
- 2016A Kronecker-factored approximate Fisher matrix for convolution layerscited by 138
- 2022A Kernel-Based View of Language Model Fine-Tuningcited by 118
- 2024Model merging with SVD to tie the Knotscited by 118
- 2019EigenDamage: Structured Pruning in the Kronecker-Factored Eigenbasiscited by 98
- 2025No Task Left Behind: Isotropic Model Merging with Common and Task-Specific Subspacescited by 90
- 2019Harnessing the Power of Infinitely Wide Deep Nets on Small-data Taskscited by 87
- 2022More Than a Toy: Random Matrix Models Predict How Real-World Neural Representations Generalizecited by 87
- 2023Parameter Efficient Multi-task Model Fusion with Partial Linearizationcited by 64
- 2023Kronecker-Factored Approximate Curvature for Modern Neural Network Architecturescited by 49
- 2018Large-Scale Distributed Second-Order Optimization Using Kronecker-Factored Approximate Curvature for Deep Convolutional Neural Networkscited by 37
- 2020Gradients as Features for Deep Representation Learningcited by 30
- 2025Accurate and Efficient Low-Rank Model Merging in Core Spacecited by 28
- 2020LQF: Linear Quadratic Fine-Tuningcited by 25
- 2023Tangent Model Composition for Ensembling and Continual Fine-tuningcited by 25
- 2023How to prepare your task head for finetuningcited by 18
- 2024A Second-Order Perspective on Model Compositionality and Incremental Learningcited by 18
- 2025Update Your Transformer to the Latest Release: Re-Basin of Task Vectorscited by 17
- 2025Efficient Model Editing with Task-Localized Sparse Fine-tuningcited by 16
- 2024Fine-Tuning Attention Modules Only: Enhancing Weight Disentanglement in Task Arithmeticcited by 14
- 2019Modular Block-diagonal Curvature Approximations for Feedforward Architecturescited by 13
- 2023Tangent Transformers for Composition, Privacy and Removalcited by 13
- 2025Fishers for Free? Approximating the Fisher Information Matrix by Recycling the Squared Gradient Accumulatorcited by 13
- 2022DLCFT: Deep Linear Continual Fine-Tuning for General Incremental Learningcited by 11
- 2023Structured Inverse-Free Natural Gradient: Memory-Efficient & Numerically-Stable KFACcited by 9
- 2023Efficient Parametric Approximations of Neural Network Function Space Distancecited by 6
- 2025MASS: MoErging through Adaptive Subspace Selectioncited by 4
Built on by
Citing papers per year: 2026: 3.
3 papers.