Deciphering and Optimizing Multi-Task Learning: a Random Matrix Approach
Malik Tiomoko, Hafiz Tiomoko Ali, Romain Couillet
Abstract
The article proposes and theoretically analyses a computationally efficient multi-task learning (MTL) extension of popular principal component analysis (PCA)-based supervised learning schemes (Barshan et al., 2011; Bair et al., 2006) . The analysis reveals that (i) by default, learning may dramatically fail by suffering from negative transfer, but that (ii) simple counter-measures on data labels avert negative transfer and necessarily result in improved performances. Supporting experiments on synthetic and real data benchmarks show that the proposed method achieves comparable performance with state-of-the-art MTL methods but at a significantly reduced computational cost. c MD 1 2 c ( Ỹ ỸT ) 1 2 .
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext 26ff78f3-b267-45c8-ac70-dde807d76cf3Cited by top-tier papers2
- Diffusion Models for Multi-Task Generative ModelingChangyou Chen, Han Ding, Bunyamin Sisman, Yi Xu et al.ICLR 2024 · 11 citations
- Random Matrix Analysis to Balance between Supervised and Unsupervised Learning under the Low Density Separation AssumptionVasilii Feofanov, Malik Tiomoko, Aladin VirmauxICML 2023 · 8 citations
Builds on2
- Random Matrix Theory Proves that Deep Learning Representations of GAN-data Behave as Gaussian MixturesMohamed El Amine Seddik, Cosme Louart, Mohamed Tamaazousti, Romain CouilletICML 2020 · 78 citations
- Two-way kernel matrix puncturing: towards resource-efficient PCA and spectral clusteringRomain Couillet, Florent Chatelain, Nicolas Le BihanICML 2021 · 11 citations
Related papers
- PCA-based Multi-Task Learning: a Random Matrix ApproachMalik Tiomoko, Romain Couillet, Frédéric PascalICML 2023 · 6 citations
- NTKMTL: Mitigating Task Imbalance in Multi-Task Learning from Neural Tangent Kernel PerspectiveXiaohan Qin, Xiaoxing Wang, Ning Liao, Junchi YanNeurIPS 2025 · 3 citations
- When Does Aggregating Multiple Skills with Multi-Task Learning Work? A Case Study in Financial NLPJingwei Ni, Zhijing Jin, Qian Wang, Mrinmaya Sachan et al.ACL 2023 · 2 citations
- Augmentation Component Analysis: Modeling Similarity via the Augmentation OverlapsLu Han, Han-Jia Ye, De-Chuan ZhanICLR 2023
- Distribution Matching for Multi-Task Learning of Classification Tasks: A Large-Scale Study on Faces & BeyondDimitrios Kollias, Viktoriia Sharmanska, Stefanos ZafeiriouAAAI 2024 · 83 citations
