用零阶优化提升稀疏训练效率,稳定收敛且省一半计算量。
ZO-SAM: Zero-Order Sharpness-Aware Minimization for Efficient Sparse Training
- 仅需一次反向传播估算梯度,降低计算开销
- 在高稀疏度下仍保持稳定训练,收敛更快
- 适合资源受限场景的高效稀疏模型训练
深度学习模型虽性能出色,但计算成本和内存需求高,在资源受限环境下应用受限。稀疏神经网络通过大幅减少参数量和计算开销缓解此问题。然而,现有稀疏训练方法常面临梯度信号混乱、噪声大等问题,尤其在高稀疏度下严重影响收敛与泛化。为此,本文提出零阶锐度感知最小化(ZO-SAM),将零阶优化融入SAM框架。相比传统SAM,ZO-SAM在扰动阶段仅需一次反向传播,通过零阶梯度估计实现计算成本减半,显著降低梯度方差并消除额外开销。借助SAM识别平坦极小值的能力,ZO-SAM稳定训练过程、加速收敛。该方法在稀疏训练中尤为重要,有效突破计算瓶颈。此外,使用ZO-SAM训练的模型在分布偏移下更具鲁棒性,增强实际部署可行性。
原文摘要 · Abstract (English)
Deep learning models, despite their impressive achievements, suffer from high computational costs and memory requirements, limiting their usability in resource-constrained environments. Sparse neural networks significantly alleviate these constraints by dramatically reducing parameter count and computational overhead. However, existing sparse training methods often experience chaotic and noisy gradient signals, severely hindering convergence and generalization performance, particularly at high sparsity levels. To tackle this critical challenge, we propose Zero-Order Sharpness-Aware Minimization (ZO-SAM), a novel optimization framework that strategically integrates zero-order optimization within the SAM approach. Unlike traditional SAM, ZO-SAM requires only a single backpropagation step during perturbation, selectively utilizing zero-order gradient estimations. This innovative approach reduces the backpropagation computational cost by half compared to conventional SAM, significantly lowering gradient variance and effectively eliminating associated computational overhead. By harnessing SAM's capacity for identifying flat minima, ZO-SAM stabilizes the training process and accelerates convergence. These efficiency gains are particularly important in sparse training scenarios, where computational cost is the primary bottleneck that limits the practicality of SAM. Moreover, models trained with ZO-SAM exhibit improved robustness under distribution shift, further broadening its practicality in real-world deployments.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。