低ランク行列によるマルチタスク学習の競合解消
CRLoMTL: Conflict-Resolution through Low-rank matrices for Multi-Task Learning
タスクごとのLoRA更新をSVDで共通低次元部分空間に射影・統合し、推論時の追加分岐なしでマルチタスクの負の転移を抑える手法を提案。
詳しい要約
1. どんなもの?
2. 先行研究と比べてどこがすごい?
3. 技術・手法の肝は?
4. どうやって有効だと検証した?
5. 議論はある?
6. 次に読むべき論文は?
※ AIが要旨から生成した要約です。正確性は原文をご確認ください。
著者: Ahmed Agiza, Tasneem Shaffee, Rahil Badkul, Sherief Reda
分類: cs.CV
原文アブストラクト
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.
関連論文
- ReSteer: マルチタスクロボットポリシーの操縦可能性の定量化と改善マルチタスク学習
- 不均衡データに対するロボット知覚のマルチタスク学習マルチタスク学習
- 学習進捗とエネルギー消費に基づくタスク交互マルチタスク学習マルチタスク学習
- DenseMTL: 密なマルチタスク学習のためのクロスタスク注意機構マルチタスク学習
- マルチタスク学習における転移の測定と活用マルチタスク学習