arXiv:2510.09796cs.LGcs.NA2025-10被引 1

提出统一框架,解决深度网络训练中的梯度难题与非光滑激活问题。

A Unified Framework for Lifted Training and Inversion Approaches

  • 将训练转化为高维约束优化,用惩罚项替代直接约束。
  • 在图像任务中显著提升含近端激活网络的训练稳定性和效果。
  • 适合研究分布式优化、非光滑激活或逆问题求解的学者。

深度神经网络训练主要依赖基于梯度的优化和反向传播计算梯度。尽管成效显著,该方法仍面临梯度消失/爆炸、非光滑激活处理困难以及固有的串行结构限制并行化等问题。提升训练方法通过将嵌套优化问题重构为高维约束优化问题,使约束不再被直接强制,而是以惩罚项形式体现。本文提出一个统一框架,涵盖辅助坐标法、Fenchel 提升网络和提升 Bregman 训练等多种策略,并证明多层感知机、残差网络和近端网络等架构均可纳入此框架。借助凸优化工具(特别是 Bregman 距离),该框架支持分布式优化,可处理不可微的近端激活函数,并改善训练景观的条件性。通过块坐标下降策略实现,包括确定性方法(结合加速与自适应优化)及隐式随机梯度方法。此外,框架还可应用于逆问题,涵盖专用网络(如展开架构)的训练与预训练网络的稳定反演。标准成像任务的数值结果表明,相较于传统训练方式,提升 Bregman 方法在使用近端激活的架构上具有更强的有效性与稳定性。

原文摘要 · Abstract (English)

The training of deep neural networks predominantly relies on a combination of gradient-based optimisation and back-propagation for the computation of the gradient. While incredibly successful, this approach faces challenges such as vanishing or exploding gradients, difficulties with non-smooth activations, and an inherently sequential structure that limits parallelisation. Lifted training methods offer an alternative by reformulating the nested optimisation problem into a higher-dimensional, constrained optimisation problem where the constraints are no longer enforced directly but penalised with penalty terms. This chapter introduces a unified framework that encapsulates various lifted training strategies, including the Method of Auxiliary Coordinates, Fenchel Lifted Networks, and Lifted Bregman Training, and demonstrates how diverse architectures, such as Multi-Layer Perceptrons, Residual Neural Networks, and Proximal Neural Networks fit within this structure. By leveraging tools from convex optimisation, particularly Bregman distances, the framework facilitates distributed optimisation, accommodates non-differentiable proximal activations, and can improve the conditioning of the training landscape. We discuss the implementation of these methods using block-coordinate descent strategies, including deterministic implementations enhanced by accelerated and adaptive optimisation techniques, as well as implicit stochastic gradient methods. Furthermore, we explore the application of this framework to inverse problems, detailing methodologies for both the training of specialised networks (e.g., unrolled architectures) and the stable inversion of pre-trained networks. Numerical results on standard imaging tasks validate the effectiveness and stability of the lifted Bregman approach compared to conventional training, particularly for architectures employing proximal activations.

深度学习优化方法逆问题非光滑激活

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