mlsanityData to 7 Oct 2026

Co-Adaptive Multi-Task LoRA: Transfer-Aware, Label-Free Control of Domain Participation

arXiv 2607.03522 ↗First posted 3 Jul 2026cs.LGcited by 0

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

  1. 2021Training Verifiers to Solve Math Word Problemscited by 9,743
  2. 2021Evaluating Large Language Models Trained on Codecited by 8,053
  3. 2020Measuring Massive Multitask Language Understandingcited by 7,768
  4. 2021Finetuned Language Models Are Zero-Shot Learnerscited by 3,747
  5. 2016A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networkscited by 2,362
  6. 2024Gemma 2: Improving Open Language Models at a Practical Sizecited by 1,928
  7. 2022Challenging BIG-Bench Tasks and Whether Chain-of-Thought Can Solve Themcited by 1,767
  8. 2022Model soups: averaging weights of multiple fine-tuned models improves accuracy without increasing inference timecited by 1,366
  9. 2017Multi-Task Learning Using Uncertainty to Weigh Losses for Scene Geometry and Semanticscited by 1,345
  10. 2020What Disease does this Patient Have? A Large-scale Open Domain Question Answering Dataset from Medical Examscited by 1,324
  11. 2020Gradient Surgery for Multi-Task Learningcited by 1,210
  12. 2023LIMA: Less Is More for Alignmentcited by 1,080
36 more
  1. 2017GradNorm: Gradient Normalization for Adaptive Loss Balancing in Deep Multitask Networkscited by 816
  2. 2018Multi-Task Learning as Multi-Objective Optimizationcited by 809
  3. 2023TIES-Merging: Resolving Interference When Merging Modelscited by 795
  4. 2023Language Models are Super Mario: Absorbing Abilities from Homologous Models as a Free Lunchcited by 679
  5. 2021Merging Models with Fisher-Weighted Averagingcited by 613
  6. 2021Deep Learning on a Data Diet: Finding Important Examples Early in Trainingcited by 529
  7. 2022Beyond neural scaling laws: beating power law scaling via data pruningcited by 519
  8. 2024LESS: Selecting Influential Data for Targeted Instruction Tuningcited by 514
  9. 2021Conflict-Averse Gradient Descent for Multi-task Learningcited by 441
  10. 2022Dataless Knowledge Fusion by Merging Weights of Language Modelscited by 384
  11. 2020Multi-Task Learning with Deep Neural Networks: A Surveycited by 374
  12. 2019Which Tasks Should Be Learned Together in Multi-task Learning?cited by 367
  13. 2023LoraHub: Efficient Cross-Task Generalization via Dynamic LoRA Compositioncited by 359
  14. 2023What Makes Good Data for Alignment? A Comprehensive Study of Automatic Data Selection in Instruction Tuningcited by 354
  15. 2019Selection via Proxy: Efficient Data Selection for Deep Learningcited by 337
  16. 2018Not All Samples Are Created Equal: Deep Learning with Importance Samplingcited by 323
  17. 2019Massively Multilingual Neural Machine Translation in the Wild: Findings and Challengescited by 253
  18. 2023Magicoder: Empowering Code Generation with OSS-Instructcited by 251
  19. 2022Prioritized Training on Points that are Learnable, Worth Learning, and Not Yet Learntcited by 233
  20. 2021Efficiently Identifying Task Groupings for Multi-Task Learningcited by 218
  21. 2023Task Arithmetic in the Tangent Space: Improved Editing of Pre-Trained Modelscited by 218
  22. 2024Data Mixing Laws: Optimizing Data Mixtures by Predicting Language Modeling Performancecited by 179
  23. 2024QuRating: Selecting High-Quality Data for Training Language Modelscited by 153
  24. 2024MixLoRA: Enhancing Large Language Models Fine-Tuning with LoRA-based Mixture of Expertscited by 144
  25. 2022Leveraging Unlabeled Data to Predict Out-of-Distribution Performancecited by 141
  26. 2015Online Batch Selection for Faster Training of Neural Networkscited by 121
  27. 2023Skill-it! A Data-Driven Skills Framework for Understanding and Training Language Modelscited by 120
  28. 2024Model merging with SVD to tie the Knotscited by 118
  29. 2021Predicting with Confidence on Unseen Distributionscited by 116
  30. 2023Parameter Efficient Multi-task Model Fusion with Partial Linearizationcited by 64
  31. 2022Predicting Out-of-Distribution Error with the Projection Normcited by 50
  32. 2025FlyLoRA: Boosting Task Decoupling and Parameter Efficiency via Implicit Rank-Wise Mixture-of-Expertscited by 34
  33. 2025Unraveling LoRA Interference: Orthogonal Subspaces for Robust Model Mergingcited by 18
  34. 2025Decouple and Orthogonalize: A Data-Free Framework for LoRA Mergingcited by 17
  35. 2025Utility-Diversity Aware Online Batch Selection for LLM Supervised Fine-tuningcited by 14
  36. 2026Predicting Mergeability of Parameter-Efficient Fine-Tuning Updatescited by 2

Built on by

Not cited yet.