Multi-Task Representation Alignment on Language Understanding: A Mutual Information Perspective

Dou Hu, Lingwei Wei, Hongjiang Xiao, Songlin Hu, Yuan Zhang


Abstract
Multi-task learning (MTL) enables joint learning over multiple tasks based on shared representations, but suffers from task interference issue during optimization. Existing works mainly focus on task balancing or probabilistic modeling but fail to address the issue since they struggle to learn sufficient representations for all target tasks. To address this, we propose a multi-task representation alignment (MTRA) framework to achieve task-specific alignment and self-alignment on the shared representations from a mutual information perspective. MTRA ensures that the learned representations contain task-relevant features while mitigating the negative effects of task-irrelevant features. First, we design a task-specific alignment objective to align the shared representations and task-specific representations with the expected targets of all tasks via information maximization. Besides, we design a self-alignment objective to eliminate task-irrelevant features via conditional information minimization. Experiments on two multi-task language benchmarks show that MTRA outperforms 13 representative MTL methods under the same settings, particularly under label-noisy and data-constrained conditions. Further analysis shows that the learned shared representations exhibit sufficient task informativeness and superior alignment properties.
Anthology ID:
2026.acl-long.2120
Volume:
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Month:
July
Year:
2026
Address:
San Diego, California, United States
Editors:
Maria Liakata, Viviane P. Moreira, Jiajun Zhang, David Jurgens
Venue:
ACL
SIG:
Publisher:
Association for Computational Linguistics
Note:
Pages:
45722–45740
Language:
URL:
https://preview.aclanthology.org/ingest-acl/2026.acl-long.2120/
DOI:
Bibkey:
Cite (ACL):
Dou Hu, Lingwei Wei, Hongjiang Xiao, Songlin Hu, and Yuan Zhang. 2026. Multi-Task Representation Alignment on Language Understanding: A Mutual Information Perspective. In Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers), pages 45722–45740, San Diego, California, United States. Association for Computational Linguistics.
Cite (Informal):
Multi-Task Representation Alignment on Language Understanding: A Mutual Information Perspective (Hu et al., ACL 2026)
Copy Citation:
PDF:
https://preview.aclanthology.org/ingest-acl/2026.acl-long.2120.pdf
Checklist:
 2026.acl-long.2120.checklist.pdf