Co-Adaptive Multi-Task LoRA: Transfer-Aware, Label-Free Control of Domain Participation
Not on the map. Why?
Abstract (arXiv)
Fine-tuning a single low-rank adapter on many domains at once is multi-task learning: the domains must be co-learned, and how they share the adapter decides whether they help or hurt one another. Most efficient fine-tuning pipelines ignore this and train on a fixed, uniform mixture, leaving two coupled questions unanswered: how much should each domain participate, and which domains should be co-trained given that some transfer positively and others interfere? We show that both answers can be read off cheaply and without labels. A forward pass of the current shared adapter over a small unlabeled probe yields, per domain, a competence signal whose level tracks remaining headroom and whose trajectory tracks learning speed; the drift of these probe representations yields a signed cross-domain affinity that predicts pairwise transfer. We fold both into CoDA, a co-adaptive controller that solves a small entropy-regularized quadratic program on the simplex to set each domain's participation -- jointly its loss weight and its share of the sampled data -- rewarding high-headroom, still-learning, mutually synergistic domains and damping interfering ones. The controller is forward-only, adds no trainable parameters, and wraps any multi-task LoRA pipeline. Across five heterogeneous domains and two backbones, CoDA improves the average over uniform mixing, learned mixtures, gradient-surgery multi-task optimizers, and online data selection while using half the data, and lowers cross-domain gradient conflict. We prove that the competence signal tracks domain risk, that the participation program has a unique fixed point reached by a contraction, and that its solution performs transfer-aware water-filling; analysis, ablations, and controls corroborate each claim.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 48 of its 63 references that are in the map. First posted on arXiv 3 Jul 2026.
Builds on
48 of 63 references.
References · 48
- 2021Training Verifiers to Solve Math Word Problemscited by 9,743
- 2021Evaluating Large Language Models Trained on Codecited by 8,053
- 2020Measuring Massive Multitask Language Understandingcited by 7,768
- 2021Finetuned Language Models Are Zero-Shot Learnerscited by 3,747
- 2016A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networkscited by 2,362
- 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,928
- 2022Challenging BIG-Bench Tasks and Whether Chain-of-Thought Can Solve Themcited by 1,767
- 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
- 2017Multi-Task Learning Using Uncertainty to Weigh Losses for Scene Geometry and Semanticscited by 1,345
- 2020What Disease does this Patient Have? A Large-scale Open Domain Question Answering Dataset from Medical Examscited by 1,324
- 2020Gradient Surgery for Multi-Task Learningcited by 1,210
- 2023LIMA: Less Is More for Alignmentcited by 1,080
36 more
- 2017GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networkscited by 816
- 2018Multi-Task Learning as Multi-Objective Optimizationcited by 809
- 2023TIES-Merging: Resolving Interference When Merging Modelscited by 795
- 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 679
- 2021Merging Models with Fisher-Weighted Averagingcited by 613
- 2021Deep Learning on a Data Diet: Finding Important Examples Early in Trainingcited by 529
- 2022Beyond neural scaling laws: beating power law scaling via data pruningcited by 519
- 2024LESS: Selecting Influential Data for Targeted Instruction Tuningcited by 514
- 2021Conflict-Averse Gradient Descent for Multi-task Learningcited by 441
- 2022Dataless Knowledge Fusion by Merging Weights of Language Modelscited by 384
- 2020Multi-Task Learning with Deep Neural Networks: A Surveycited by 374
- 2019Which Tasks Should Be Learned Together in Multi-task Learning?cited by 367
- 2023LoraHub: Efficient Cross-Task Generalization via Dynamic LoRA Compositioncited by 359
- 2023What Makes Good Data for Alignment? A Comprehensive Study of Automatic Data Selection in Instruction Tuningcited by 354
- 2019Selection via Proxy: Efficient Data Selection for Deep Learningcited by 337
- 2018Not All Samples Are Created Equal: Deep Learning with Importance Samplingcited by 323
- 2019Massively Multilingual Neural Machine Translation in the Wild: Findings and Challengescited by 253
- 2023Magicoder: Empowering Code Generation with OSS-Instructcited by 251
- 2022Prioritized Training on Points that are Learnable, Worth Learning, and Not Yet Learntcited by 233
- 2021Efficiently Identifying Task Groupings for Multi-Task Learningcited by 218
- 2023Task Arithmetic in the Tangent Space: Improved Editing of Pre-Trained Modelscited by 218
- 2024Data Mixing Laws: Optimizing Data Mixtures by Predicting Language Modeling Performancecited by 179
- 2024QuRating: Selecting High-Quality Data for Training Language Modelscited by 153
- 2024MixLoRA: Enhancing Large Language Models Fine-Tuning with LoRA-based Mixture of Expertscited by 144
- 2022Leveraging Unlabeled Data to Predict Out-of-Distribution Performancecited by 141
- 2015Online Batch Selection for Faster Training of Neural Networkscited by 121
- 2023Skill-it! A Data-Driven Skills Framework for Understanding and Training Language Modelscited by 120
- 2024Model merging with SVD to tie the Knotscited by 118
- 2021Predicting with Confidence on Unseen Distributionscited by 116
- 2023Parameter Efficient Multi-task Model Fusion with Partial Linearizationcited by 64
- 2022Predicting Out-of-Distribution Error with the Projection Normcited by 50
- 2025FlyLoRA: Boosting Task Decoupling and Parameter Efficiency via Implicit Rank-Wise Mixture-of-Expertscited by 34
- 2025Unraveling LoRA Interference: Orthogonal Subspaces for Robust Model Mergingcited by 18
- 2025Decouple and Orthogonalize: A Data-Free Framework for LoRA Mergingcited by 17
- 2025Utility-Diversity Aware Online Batch Selection for LLM Supervised Fine-tuningcited by 14
- 2026Predicting Mergeability of Parameter-Efficient Fine-Tuning Updatescited by 2
Built on by
Not cited yet.