CRLoMTL: Conflict-Resolution through Low-rank matrices for Multi-Task Learning
Organizations: School of Engineering, Brown University, Providence, RI, USA
Abstract
Multi-task learning (MTL) improves model efficiency by sharing representations across related vision tasks, but it often suffers from negative transfer when task updates interfere with one another. Existing conflict-mitigation methods typically operate in gradient space, requiring explicit task-gradient manipulation and additional optimization steps. We propose Conflict-Resolution through Low-Rank Matrices for Multi-Task Learning (CRLoMTL), a parameter-efficient framework that mitigates task interference in the low-rank parameter-update space. CRLoMTL learns task-specific LoRA updates through task-specific passes, where each task updates its own low-rank matrices, and integrates these updates through global shared-subspace fusion. Specifically, we introduce a shared-subspace resolver based on singular value decomposition (SVD), referred to as the SVD resolver, which extracts a common low-dimensional orthonormal basis across task adapters and projects task updates into this shared space before merging them into a single deployment-time update. The resulting update can be folded into the backbone, preserving the inference structure of the original model without additional runtime branches. Experiments with Swin Transformer backbones on PASCAL-Context, NYUD-v2, and CelebA show that CRLoMTL achieves a favorable accuracy-parameter trade-off, improving over single-task baselines by up to 2.77% on PASCAL-Context, 5.03% on NYUD-v2, and 3.50% on CelebA, while training only a small fraction of the parameters used by single-task models.
Figures & tables
| Method | SemSeg | Human Parts | Saliency | Normals | Trainable | (%) |
|---|---|---|---|---|---|---|
| (mIoU ) | (mIoU ) | (mIoU ) | (RMSE ) | Parameters (M) | ||
| Single Task | 67.21 | 61.93 | 62.35 | 17.97 | 112.62 | 0 |
| Vanilla MTL | 67.56 | 60.24 | 65.21 | 16.64 | 29.47 | +2.23 |
| MTI-Net | 66.93 | 59.66 | 64.76 | 16.23 | 31.47 | +2.37 |
| MTLoRA | 67.90 | 59.84 | 65.40 | 16.60 | 8.34 | +2.55 |
| KnOTS | 67.90 | 60.26 | 64.68 | 16.81 | 30.29 | +2.12 |
| Method | SemSeg | Human Parts | Saliency | Normals | Trainable | (%) |
|---|---|---|---|---|---|---|
| (mIoU ) | (mIoU ) | (mIoU ) | (RMSE ) | Parameters (M) | ||
| Vanilla MTL | 70.20 | 63.30 | 66.09 | 16.17 | 50.78 | +5.66 |
| CRLoMTL | 70.78 | 63.60 | 66.28 | 16.17 | 6.75 | +6.08 |
| Model | Parameters (M) | Eval. Time (s) | Time / Image (ms) |
|---|---|---|---|
| CRLoMTL unmerged | 31.91 | ||
| CRLoMTL merged | 29.47 | ||
| Vanilla MTL | 29.47 |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Method | SemSeg | Human Parts | Saliency | Normals | |
| mIoU | mIoU | mIoU | RMSE | (%) | |
| CRLoMTL |
Multi-Task Learning
Multi-task learning trains models to solve related tasks together by sharing representations, parameters, or supervision.
See all 392 papers