arXiv:2606.02221cs.CVcs.LG2026-06中稿 · ICML

通过因果正交表示提升多任务学习的泛化能力

CORE-MTL: Rethinking Gradient Balancing via Causal Orthogonal Representations

论文配图:CORE-MTL: Rethinking Gradient Balancing via Causal Orthogonal Representations
图 1 · 摘自论文原文
  • 将共享表征分解为语义流与残差流,分离任务相关与干扰信息
  • 在视觉多任务基准上,分布内与分布外均优于现有方法
  • 无需梯度投影或重加权,理论保障更优泛化性能

多任务学习(MTL)旨在通过跨领域共享表征构建联合模型。现有优化导向方法或平衡任务梯度,或修改共享架构,但因忽略共享表征内容,无法区分任务相关结构与虚假上下文,导致负迁移和泛化能力差。为此,我们提出因果正交表示多任务学习(CORE-MTL),一种基于因果动机的表征中心框架,鼓励共享表征进行语义-残差结构化分解:任务相关结构集中于语义流,干扰变化归入残差流。在视觉领域中,利用物理先验建模结构场景,用统计约束处理属性。理论上,该方法比优化导向方法具有更紧的分布外泛化界,并在不显式梯度投影或重加权的情况下降低任务梯度干扰。实验表明,CORE-MTL在视觉多任务基准上,无论分布内还是分布外设置,均持续优于现有方法。代码已公开于 https://github.com/Hope-Rita/CORE-MTL。

原文摘要 · Abstract (English)

Multi-task learning (MTL) aims to construct a joint model for multiple tasks by sharing a common representation across domains. To achieve this goal, existing optimization-centric methods either balance task gradients or modify the shared architecture. However, as these approaches remain agnostic to the content of the shared representation, they fail to disentangle task-relevant structure from spurious context, leading to negative transfer and poor generalization. To overcome this limitation, we propose Causal Orthogonal Representations for Multi-Task Learning (CORE-MTL), a causally motivated representation-centric framework that encourages a structured semantic-residual factorization of the shared representation, concentrating task-relevant structure in the semantic stream while relegating nuisance variation to the residual stream. We instantiate this framework in the visual domain by leveraging physical priors for structured scenes and statistical constraints for attributes. Theoretically, our method enjoys a tighter out-of-distribution generalization bound than optimization-centric methods and reduces task gradient interference without explicit gradient projection or reweighting. Empirically, CORE-MTL consistently outperforms existing methods on visual multi-task benchmarks in both in-distribution and out-of-distribution settings. Code is publicly available at https://github.com/Hope-Rita/CORE-MTL.

多任务学习因果推理表征分解泛化能力

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。