Skip to main navigation Skip to search Skip to main content

Multitask Causal Contrastive Learning

  • Chaoyang Li
  • , Heyan Chai
  • , Yan Jia
  • , Ning Hu
  • , Qing Liao*
  • *Corresponding author for this work
  • Harbin Institute of Technology Shenzhen
  • Peng Cheng Laboratory
  • Chinese University of Hong Kong

Research output: Contribution to journalArticlepeer-review

Abstract

Multitask learning (MTL) aims to improve the performance of multiple tasks by sharing knowledge among multiple different tasks, which has attracted increasing interest and shown success in various fields. However, MTL often suffers from negative transfer since the model may utilize useless features and face interference among tasks' optimization objectives. The utilization of useless features can be attributed to the confounding factors in multitask features, while the interference among tasks' optimization objectives is due to inadequate measurement of the relationship among tasks. This article proposes a novel multitask causal contrastive learning (MT-CCL) approach to address the above problem. First, we propose a multitask causal inference method, which removes the confounding factors in features via task-aware causal intervention (TACI) and measures the relationship among tasks from a novel causal perspective by quantifying the intertask causal affinity. Then, we build a dual contrastive learning objective to help the model better learn useful features via intratask contrast (Intra-TCS) and mitigate interference among tasks' optimization objectives via intertask contrast (Inter-TCS). Experiments demonstrate that MT-CCL achieves improved performance over state-of-the-art methods on Multi-MNIST, NYU-v2, CityScapes, and CelebA, verifying the effectiveness of intertask causal affinity.

Original languageEnglish
Pages (from-to)17322-17335
Number of pages14
JournalIEEE Transactions on Neural Networks and Learning Systems
Volume36
Issue number9
DOIs
StatePublished - 2025
Externally publishedYes

Keywords

  • Causal inference
  • contrastive learning
  • multitask learning (MTL)
  • negative transfer

Fingerprint

Dive into the research topics of 'Multitask Causal Contrastive Learning'. Together they form a unique fingerprint.

Cite this