Parameter-, Memory-, Time-Efficient Multi-Task Dense Vision Adaptation
DOI:
https://doi.org/10.1609/aaai.v40i14.38171Abstract
While adapting pretrained vision models to downstream dense prediction tasks is widely used, current methods often overlook adaptation efficiency, especially in the context of multi-task learning (MTL). Although parameter-efficient fine-tuning (PEFT) methods can enhance parameter efficiency, broader aspects such as GPU memory and training time efficiency remain underexplored. In this paper, we propose a new paradigm that simultaneously achieves efficiency in Parameters, GPU Memory, and Training Time for Multi-Task Dense Vision Adaptation. Specifically, we propose a dual-branch framework, in which a frozen pretrained backbone serves as the generic main branch, and the proposed Bi-Directional Task Adaptation (BDTA) modules are integrated in parallel to form a task bypass branch that extracts adaptation features required by multiple specific tasks. This adaptation module is lightweight, efficient, and does not require backpropagation through the large pre-trained backbone, thus avoiding resource-intensive gradient computations. Moreover, a Mixture of Task Experts mechanism (MoTE) is further proposed to integrate adaptation features across tasks and scales, thereby obtaining more robust representations tailored for dense prediction tasks. On the PASCAL-Context benchmark, our method achieves over 2× relative performance improvement compared to the best prior multi-task PEFT method, while using only ~30% of the parameters, ~50% of the memory, and ~60% of the training time, demonstrating superior overall adaptation efficiency.Published
2026-03-14
How to Cite
Yao, H., Luo, W., Chen, Q., Liao, J., & You, W. (2026). Parameter-, Memory-, Time-Efficient Multi-Task Dense Vision Adaptation. Proceedings of the AAAI Conference on Artificial Intelligence, 40(14), 11848–11856. https://doi.org/10.1609/aaai.v40i14.38171
Issue
Section
AAAI Technical Track on Computer Vision XI