arXiv:2605.11526math.OCcs.AI2026-05

提出可证明收敛的线性约束神经网络训练方法,支持高效端到端优化。

Efficient and provably convergent end-to-end training of deep neural networks with linear constraints

论文配图:Efficient and provably convergent end-to-end training of deep neural networks with linear constraints
图 1 · 摘自论文原文
  • 引入可高效计算的HS-Jacobian,解决投影层非光滑性问题。
  • 证明HS-Jacobian为保守映射,使Adam等优化器可直接使用。
  • 在金融、视觉、网络设计中表现优于现有方法,有理论保证。

许多现代数据驱动应用需要训练深层神经网络,使其部分层输出满足线性约束。虽然可通过在网络中加入投影层实现,但其端到端训练因缺乏严谨理论和高效反向传播算法而困难重重。关键难点在于投影层解映射的非光滑性。为此,本文提出一种可高效计算的HS-Jacobian,用于投影层。重要的是,我们证明了该HS-Jacobian是多面体集上投影算子的保守映射,从而可无缝集成至非光滑自动微分框架中进行反向传播。因此,包括Adam在内的多种高效算法均可用于线性约束深度神经网络的端到端训练。特别地,我们建立了基于HS-Jacobian的Adam算法在训练线性约束网络时的收敛性保证。在金融、计算机视觉及网络架构设计等多个重要应用上的大量实验结果表明,本方法性能显著优于现有主流方法。

原文摘要 · Abstract (English)

Training a deep neural network with the outputs of selected layers satisfying linear constraints is required in many contemporary data-driven applications. While this can be achieved by incorporating projection layers into the neural network, its end-to-end training remains challenging due to the lack of rigorous theory and efficient algorithms for backpropagation. A key difficulty in developing the theory and efficient algorithms for backpropagation arose from the nonsmoothness of the solution mapping of the projection layer. To address this bottleneck, we introduce an efficiently computable HS-Jacobian to the projection layer. Importantly, we prove that the HS-Jacobian is a conservative mapping for the projection operator onto the polyhedral set, enabling its seamless integration into the nonsmooth automatic differentiation framework for backpropagation. Therefore, many efficient algorithms, such as Adam, can be applied for end-to-end training of deep neural networks with linear constraints. Particularly, we establish convergence guarantees of the HS-Jacobian based Adam algorithm for training linearly constrained deep neural networks. Extensive experiment results on several important applications, including finance, computer vision, and network architecture design, demonstrate the superior performance of our method compared to other existing popular methods.

神经网络约束学习优化算法反向传播

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