Capacity and Redundancy Trade-offs in Multi-Task Learning
Abstract
In multi-task learning (MTL) negative transfer is often considered as an optimization artifact, but it can also be viewed as a consequence of limited shared capacity and weak task redundancy. We investigate this effect through a Capacity--Redundancy (CR) identity that decomposes the sum of per-task predictive informations into joint predictive information that includes label redundancy defined via total correlation (TC), and a residual coupling term that quantifies interference left unresolved by the shared representation. Additionally, we show two key results: (i) a clustering-gap decomposition that gives a necessary and sufficient condition for clustered sharing to outperform global sharing, and (ii) a gradient--TC bridge in a Gaussian multi-task model that formally justifies gradient cosine similarity as a proxy for redundancy ordering. Empirically, we estimate the residual coupling from validation residual correlations, showing that clustered LoRA substantially reduces , outperforms size-matched random partitions, and results in statistically significant gains with multi-seed confidence intervals.
Cite
@article{arxiv.2607.16554,
title = {Capacity and Redundancy Trade-offs in Multi-Task Learning},
author = {Asif Khan},
journal= {arXiv preprint arXiv:2607.16554},
year = {2026}
}
Comments
Accepted in 42nd Conference on Uncertainty in Artificial Intelligence (UAI) 2026