arXiv:2606.12054cs.LG2026-06中稿 · IJCAI

简单噪声注入就能让SGD训练更有效,无需复杂设计。

Simplicity Suffices for Parameter Noise Injection in Stochastic Gradient Descent

  • 用线性层分布特性实现每样本独立加噪,不破坏批量计算。
  • 单次前向传播+各向同性噪声在CIFAR100上效果接近复杂方案。
  • 适合追求高效训练的工程师,尤其看重实用性的研究者。

在深度神经网络训练中,向优化过程注入噪声是提升训练效果和泛化能力的成熟方法。尽管已有多种技术,但哪些设计选择真正重要仍不明确。本文聚焦随机梯度下降中的参数噪声注入,解决两个核心问题:如何在小批量训练中高效为每个样本配置独立扰动,以及复杂的噪声参数化或多样本梯度平均是否优于简单方案。针对第一个问题,我们利用线性层的分布恒等式,实现了无需破坏批处理计算的每样本噪声注入。针对第二个问题,我们在CIFAR100上系统比较了多种对角高斯参数化与各向同性基线,在不同噪声水平下进行测试。结果一致表明,简单轻量的策略——单次扰动前向传播、各向同性噪声——已能恢复多数复杂方案的优势。这说明,对于参数噪声注入,简洁足以,实践者无需采用繁琐的扰动设计即可获得优化与泛化收益。

原文摘要 · Abstract (English)

Injecting noise into the optimization process is a well-established technique for improving the training and generalization of deep neural networks. Yet, despite the breadth of existing approaches, it remains unclear which design choices truly matter in practice. In this work, we investigate parameter noise injection for stochastic gradient descent, focusing on two key questions: how to efficiently pair each training example with its own perturbation in mini-batch training, and whether sophisticated noise parameterizations or multi-sample gradient averaging yield meaningful gains over simpler alternatives. To address the first question, we leverage a distributional identity for linear layers that allows per-example noise injection without breaking batched computation. To address the second, we systematically compare several diagonal Gaussian parameterizations against an isotropic baseline across varying noise levels on CIFAR100. Our results consistently show that simple, lightweight strategies, isotropic noise with a single perturbed forward pass per update step, recover most of the benefit of more complex schemes. These findings suggest that simplicity suffices for parameter noise injection, and that practitioners need not resort to elaborate perturbation designs to reap the optimization and generalization benefits of noisy SGD.

优化器噪声注入SGD简洁设计

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。