ThanoRA: Task Heterogeneity-Aware Multi-Task Low-Rank Adaptation

Jian Liang, Wenke Huang, Xianda Guo, Guancheng Wan, Bo Du +1 more
5/24/2025
cs.LGcs.AI

Official code: LiangJian24/ThanoRA (12 stars, last commit 2025-10-01).

Abstract

Low-Rank Adaptation (LoRA) is widely adopted for downstream fine-tuning of foundation models due to its efficiency and zero additional inference cost. Many real-world applications require foundation models to specialize in several specific tasks simultaneously, motivating the need for efficient multi-task downstream adaptation. To address this need, existing studies have primarily explored two directions: Model Merging with LoRA, which shows advantages in training-free scenarios but still lags behind multi-task training in overall performance; and MoE-based LoRA approaches, which improve multi-task learning performance but introduce routers that hinder the mergeability of LoRA parameters and incur considerable inference overhead, thereby limiting real-world deployment practicality. To this end, we propose ThanoRA, a Task Heterogeneity-Aware Multi-Task Low-Rank Adaptation framework that enables effective, efficient and unified multi-task downstream adaptation without introducing additional structure. ThanoRA performs multi-task learning by tailoring subspace allocation at initialization and enforcing diversity preservation throughout training: it allocates varying dimensions to construct task-specific low-rank subspaces driven by inter-task heterogeneity, enabling fine-grained knowledge injection, while diversity-preserving regularization mitigates task interference and subspace collapse, thereby fully exploiting the low-rank capacity. Extensive experiments across multimodal and text-only benchmarks under varying multi-task mixtures demonstrate that ThanoRA consistently outperforms strong baselines, surpassing even separate task-specific fine-tuning, while introducing no additional structures or inference overhead. Our code will be publicly available at: https://github.com/LiangJian24/ThanoRA.

View on arXivView PDF

Code Implementations(1)

ThanoRA: Task Heterogeneity-Aware Multi-Task Low-Rank Adaptation

120May 24, 20251 years ago

Similar Papers

Cite this paper

@article{liang2025thanora,
  title  = {ThanoRA: Task Heterogeneity-Aware Multi-Task Low-Rank Adaptation},
  author = {Jian Liang and Wenke Huang and Xianda Guo and Guancheng Wan and Bo Du and Mang Ye},
  year   = {2025},
  eprint = {2505.18640},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2505.18640v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2505.18640.svg)](https://codeofpaper.com/paper/2505.18640)

Discussion