用混合精度量化+交互式专家网络,让小模型高效迁移学习
MP-ISMoE: Mixed-Precision Interactive Side Mixture-of-Experts for Efficient Transfer Learning

- 用低比特量化减少内存占用,保留高精度特征
- 通过交互式专家选择机制提升性能,比现有方法准确率更高
- 适合资源受限场景下的高效模型微调,尤其视频与语言任务
参数高效迁移学习(PETL)虽能大幅减少可训练参数,但微调时梯度反传带来巨大内存开销。现有内存高效迁移学习(METL)通过轻量侧网络绕过主干梯度计算,却因内存限制严重制约侧网络容量,影响性能。为此,我们提出混合精度交互式侧向专家架构(MP-ISMoE)。首先设计高斯噪声扰动迭代量化(GNP-IQ)方案,将权重压缩至低比特并有效降低量化误差;利用节省出的内存,引入交互式侧向专家(ISMoE),在不牺牲内存效率的前提下扩展侧网络规模。不同于传统MoE,ISMoE通过与冻结主干提取的显著特征交互,动态选择最优专家,缓解知识遗忘并提升性能。在多种视觉-语言与纯语言任务上的实验表明,MP-ISMoE显著优于当前最优METL方法,同时保持相当的参数与内存效率。
原文摘要 · Abstract (English)
Parameter-efficient transfer learning (PETL) has emerged as a pivotal paradigm for adapting pre-trained foundation models to downstream tasks, significantly reducing trainable parameters yet suffering from substantial memory overhead caused by gradient backpropagation during fine-tuning. While memory-efficient transfer learning (METL) circumvents this challenge by bypassing backbone gradient computation via lightweight small side networks, its stringent memory constraint severely limits learning capacity of side networks, thereby significantly compromising performance. To address these limitations, we propose a novel Mixed-Precision Interactive Side Mixture-of-Experts framework (MP-ISMoE). Specifically, we first propose a Gaussian Noise Perturbed Iterative Quantization (GNP-IQ) scheme to quantize weights into lower-bits while effectively decreasing quantization errors. By leveraging memory conserved from GNP-IQ, we subsequently employ Interactive Side Mixture-of-Experts (ISMoE) to scaling up side networks without sacrificing overall memory efficiency. Different from conventional mixture-of-experts, ISMoE learns to select optimal experts by interacting with salient features from frozen backbones, thus suppressing knowledge forgetting and boosting performance. Extensive experiments across diverse vision-language and language-only tasks demonstrate that MP-ISMoE remarkably promotes accuracy compared to state-of-the-art METL approaches, while maintaining comparable parameter and memory efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。