提出解耦多预测器优化方法,让模型在少算时仍保持高精度。
Decoupled Multi-Predictor Optimization for Inference-Efficient Model Tuning
- 通过轻量旁路模块分解浅层特征,分离表征与判别能力
- 两阶段损失权重分配使深层先学表征,浅层后学判别
- 适用于需要低推理开销的场景,如移动端部署
大规模预训练模型微调取得显著进展,推理效率对实际部署愈发关键。结合多阶段预测器与早期退出策略,并配合参数高效微调,可实现高效推理。然而核心挑战在于:早期阶段如何同时为深层提供基础特征,又自身具备判别能力?为此,我们提出解耦多预测器优化(DMPO)方法,从架构与优化两方面解耦早期阶段的低层表征能力与高层判别能力。在架构上,引入轻量旁路模块,从早期阶段分解浅层特征;设计基于高阶统计的预测器以增强早期判别能力。在训练上,采用分阶段损失权重分配策略:初期强化早期阶段的表征能力以促进深层判别学习,后期则最大化推动判别能力向早期阶段迁移。实验表明,在多个数据集和预训练骨干网络上,DMPO 在降低计算成本的同时显著优于现有方法。
原文摘要 · Abstract (English)
Recently, remarkable progress has been made in large-scale pre-trained model tuning, and inference efficiency is becoming more crucial for practical deployment. Early exiting in conjunction with multi-stage predictors, when cooperated with a parameter-efficient fine-tuning strategy, offers a straightforward way to achieve an inference-efficient model. However, a key challenge remains unresolved: How can early stages provide low-level fundamental features to deep stages while simultaneously supplying high-level discriminative features to early-stage predictors? To address this problem, we propose a Decoupled Multi-Predictor Optimization (DMPO) method to effectively decouple the low-level representative ability and high-level discriminative ability in early stages. First, in terms of architecture, we introduce a lightweight bypass module into multi-stage predictors for functional decomposition of shallow features from early stages, while a high-order statistics-based predictor is developed for early stages to effectively enhance their discriminative ability. To reasonably train our multi-predictor architecture, a decoupled optimization is proposed to allocate two-phase loss weights for multi-stage predictors during model tuning, where the initial training phase enables the model to prioritize the acquisition of discriminative ability of deep stages via emphasizing representative ability of early stages, and the latter training phase drives discriminative ability towards earlier stages as much as possible. As such, our DMPO can effectively decouple representative and discriminative abilities in early stages in terms of architecture design and model optimization. Experiments across various datasets and pre-trained backbones demonstrate that DMPO clearly outperforms its counterparts when reducing computational cost.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。