Parameter-, Memory-, Time-Efficient Multi-Task Dense Vision Adaptation

Authors

  • Haiming Yao Tsinghua University Advanced Computing and Storage Lab, Huawei Technologies Co. Ltd
  • Wei Luo Tsinghua University Advanced Computing and Storage Lab, Huawei Technologies Co. Ltd
  • Qiyu Chen Institute of Automation, Chinese Academy of Sciences Advanced Computing and Storage Lab, Huawei Technologies Co. Ltd
  • Jianxing Liao Advanced Computing and Storage Lab, Huawei Technologies Co. Ltd
  • Wei You Advanced Computing and Storage Lab, Huawei Technologies Co. Ltd

DOI:

https://doi.org/10.1609/aaai.v40i14.38171

Abstract

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.

Downloads

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