arXiv:2604.09088cs.CV2026-04中稿 · ed

提出新方法让迁移学习更省内存且推理更快,无需额外开销。

Memory-Efficient Transfer Learning with Fading Side Networks via Masked Dual Path Distillation

论文配图:Memory-Efficient Transfer Learning with Fading Side Networks via Masked Dual Path Distillation
图 1 · 摘自论文原文
  • 用双路径知识蒸馏提升模型性能,训练时保留轻量侧网络
  • 推理时完全丢弃侧网络,速度提升至少25.2%且不损失精度
  • 适用于视觉、语言及多模态任务,适合追求高效部署的研究者

内存高效的迁移学习(METL)方法在将预训练模型适配到下游任务时取得了良好表现。它们避免在大型主干网络中进行梯度反向传播,显著减少了可训练参数数量和微调时的高内存消耗。然而,由于通常采用轻量且可学习的侧网络,这些方法在推理阶段不可避免地引入额外的内存与时间开销,违背了高效迁移学习的初衷。为此,我们提出一种名为掩码双路径蒸馏(MDPD)的新方法,在保持微调阶段参数与内存效率的同时加速推理过程,并采用渐退侧网络。具体而言,MDPD通过在微调中相互蒸馏冻结主干与可学习侧网络来增强性能,并在推理阶段彻底舍弃侧网络而不牺牲准确率。此外,我们设计了一种针对多层编码器结构的基于特征的知识蒸馏方法。在多种主干网络上,涵盖纯视觉、纯语言以及视觉-语言任务的大量实验表明,该方法不仅使推理速度提升至少25.2%,同时保持参数与内存消耗相当,且显著优于当前最优方法。源代码已开源:https://github.com/Zhang-VKk/MDPD。

原文摘要 · Abstract (English)

Memory-efficient transfer learning (METL) approaches have recently achieved promising performance in adapting pre-trained models to downstream tasks. They avoid applying gradient backpropagation in large backbones, thus significantly reducing the number of trainable parameters and high memory consumption during fine-tuning. However, since they typically employ a lightweight and learnable side network, these methods inevitably introduce additional memory and time overhead during inference, which contradicts the ultimate goal of efficient transfer learning. To address the above issue, we propose a novel approach dubbed Masked Dual Path Distillation (MDPD) to accelerate inference while retaining parameter and memory efficiency in fine-tuning with fading side networks. Specifically, MDPD develops a framework that enhances the performance by mutually distilling the frozen backbones and learnable side networks in fine-tuning, and discard the side network during inference without sacrificing accuracy. Moreover, we design a novel feature-based knowledge distillation method for the encoder structure with multiple layers. Extensive experiments on distinct backbones across vision/language-only and vision-and-language tasks demonstrate that our method not only accelerates inference by at least 25.2\% while keeping parameter and memory consumption comparable, but also remarkably promotes the accuracy compared to SOTA approaches. The source code is available at https://github.com/Zhang-VKk/MDPD.

迁移学习知识蒸馏高效推理多模态

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。