在极低内存下实现模型高效微调,让手机端模型自适应新任务。
Memory Constrained Dynamic Subnetwork Update for Transfer Learning
- 按重要性动态筛选层和通道,节省内存
- 在仅几百kB内存下仍保持高精度性能
- 适合资源受限的移动端设备使用
设备端神经网络训练面临严峻的内存限制,难以将预训练模型适配到下游任务。我们提出 MeDyate,一个理论支撑的内存约束动态子网络更新框架。其核心创新包括:LaRa(层重要性排序),一种改进的层重要性度量方法,可合理预选关键层;以及动态通道采样策略,利用微调过程中通道重要性分布的时间稳定性。MeDyate 在各轮次间根据重要性加权概率动态重采样通道,确保充分探索参数空间的同时严格遵守内存预算。大量实验表明,在多种任务与架构下,该方法在极端内存约束下达到当前最优性能,持续优于现有静态与动态方法,且计算效率高。本方法显著推进了设备端高效学习的发展,首次在仅数百kB RAM内存下实现有效微调。
原文摘要 · Abstract (English)
On-device neural network training faces critical memory constraints that limit the adaptation of pre-trained models to downstream tasks. We present MeDyate, a theoretically-grounded framework for memory-constrained dynamic subnetwork adaptation. Our approach introduces two key innovations: LaRa (Layer Ranking), an improved layer importance metric that enables principled layer pre-selection, and a dynamic channel sampling strategy that exploits the temporal stability of channel importance distributions during fine-tuning. MeDyate dynamically resamples channels between epochs according to importance-weighted probabilities, ensuring comprehensive parameter space exploration while respecting strict memory budgets. Extensive evaluation across a large panel of tasks and architectures demonstrates that MeDyate achieves state-of-the-art performance under extreme memory constraints, consistently outperforming existing static and dynamic approaches while maintaining high computational efficiency. Our method represents a significant step towards enabling efficient on-device learning by demonstrating effective fine-tuning with memory budgets as low as a few hundred kB of RAM.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。