针对从零训练和微调设计差异化优化方法,提升模型性能
Neural Network Optimization Reimagined: Decoupled Techniques for Scratch and Fine-Tuning

- 分离优化策略:分别针对训练和微调设计专用技术
- 微调时引入权重回滚,减少知识遗忘,提升效果
- 适配不同任务动态调整参数,适合大规模模型优化
随着大数据时代的到来和预训练模型的兴起,神经网络优化常需区分从零训练与微调两种场景。现有优化器多聚焦于损失下降,未充分考虑两类范式的需求差异。本文提出DualOpt,一种分治式优化方法:对从零训练,引入实时层间权重衰减,增强收敛性与泛化能力;对微调,集成权重回滚机制,在每步更新中加入回滚项,保持上下游模型权重分布一致,有效缓解知识遗忘。进一步将层间权重衰减扩展为动态回滚调节,适应下游任务差异。在图像分类、目标检测、语义分割、实例分割等任务上广泛实验验证了DualOpt的普适性与先进性。代码已开源。
原文摘要 · Abstract (English)
With the accumulation of resources in the era of big data and the rise of pre-trained models in deep learning, optimizing neural networks for various tasks often involves different strategies for fine-tuning pre-trained models versus training from scratch. However, existing optimizers primarily focus on reducing the loss function by updating model parameters, without fully addressing the unique demands of these two major paradigms. In this paper, we propose DualOpt, a novel approach that decouples optimization techniques specifically tailored for these distinct training scenarios. For training from scratch, we introduce real-time layer-wise weight decay, designed to enhance both convergence and generalization by aligning with the characteristics of weight updates and network architecture. For more importantly fine-tuning, we integrate weight rollback with the optimizer, incorporating a rollback term into each weight update step. This ensures consistency in the weight distribution between upstream and downstream models, effectively mitigating knowledge forgetting and improving fine-tuning performance. Additionally, we extend the layer-wise weight decay to dynamically adjust the rollback levels across layers, adapting to the varying demands of different downstream tasks. Extensive experiments across diverse tasks, including image classification, object detection, semantic segmentation, and instance segmentation, demonstrate the broad applicability and state-of-the-art performance of DualOpt. Code is available at https://github.com/qklee-lz/OLOR-AAAI-2024.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。