LAK-LoRA: Layer-Wise Adaptive Expert Allocation and Principal Component–Based Feature Separation for Medical Multi-Task Learning

Ye Lyu, Chuan Li, Chengyu Wang, Mingyuan Fan, Cen Chen


Abstract
Multi-task learning in the medical domain remains a significant challenge for large language models (LLMs), plagued by conflicting task objectives and substantial computational overhead associated with full fine-tuning. Although parameter-efficient tuning methods help reduce these resource demands, their effectiveness is limited by two critical issues: (i) rigid allocation of expert capacity across layers, and (ii) lack of regularization mechanisms to partition shared components from task-specific signals. To address these limitations, we propose LAK-LoRA, a novel architecture that enables adaptive, depth-aware scaling of expert capacity, allowing more efficient resource allocation throughout the model. Furthermore, to resolve conflicting task objectives, LAK-LoRA divides experts into two distinct groups dedicated to modeling general and task-specific features, respectively. This separation is implemented via a feature separator module, which disentangles representations into shared and task-specific subspaces before assigning them to the appropriate experts. To foster diversity and minimize representational redundancy, orthogonality constraints are imposed among specialized experts. Extensive experiments on benchmark medical datasets demonstrate that LAK-LoRA achieves the best average performance.
Anthology ID:
2026.emnlp-main.1407
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
32150–32171
Language:
URL:
https://aclanthology.org/2026.emnlp-main.1407/
DOI:
Bibkey:
Cite (ACL):
Ye Lyu, Chuan Li, Chengyu Wang, Mingyuan Fan, and Cen Chen. 2026. LAK-LoRA: Layer-Wise Adaptive Expert Allocation and Principal Component–Based Feature Separation for Medical Multi-Task Learning. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 32150–32171, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
LAK-LoRA: Layer-Wise Adaptive Expert Allocation and Principal Component–Based Feature Separation for Medical Multi-Task Learning (Lyu et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.1407.pdf