arXiv:2410.19055cs.LG2024-10NeurIPS被引 2

用二阶信息优化难训练的损失函数,提升排序与最短路径算法性能

Newton Losses: Using Curvature Information for Learning with Differentiable Algorithms

  • 通过经验费雪和海森矩阵提取损失函数二阶信息
  • 在8个不同可微算法上实现显著或一致性能提升
  • 无需复杂二阶优化,仅改损失函数即可加速收敛

训练具有自定义目标(如排序损失、最短路径损失)的神经网络时,常面临目标本身不可微的问题。常用方法是连续松弛目标以获得梯度,但此类可微松弛往往非凸,易产生梯度消失或爆炸,使优化困难。本文提出牛顿损失(Newton Losses),利用损失函数的二阶信息(基于经验费雪矩阵和海森矩阵),将其替换为更平滑的牛顿损失,而仍使用梯度下降训练网络。该方法计算高效,适用于8种可微排序与最短路径算法,在欠优化算法上取得显著提升,对已优化算法也实现稳定改进。

原文摘要 · Abstract (English)

When training neural networks with custom objectives, such as ranking losses and shortest-path losses, a common problem is that they are, per se, non-differentiable. A popular approach is to continuously relax the objectives to provide gradients, enabling learning. However, such differentiable relaxations are often non-convex and can exhibit vanishing and exploding gradients, making them (already in isolation) hard to optimize. Here, the loss function poses the bottleneck when training a deep neural network. We present Newton Losses, a method for improving the performance of existing hard to optimize losses by exploiting their second-order information via their empirical Fisher and Hessian matrices. Instead of training the neural network with second-order techniques, we only utilize the loss function's second-order information to replace it by a Newton Loss, while training the network with gradient descent. This makes our method computationally efficient. We apply Newton Losses to eight differentiable algorithms for sorting and shortest-paths, achieving significant improvements for less-optimized differentiable algorithms, and consistent improvements, even for well-optimized differentiable algorithms.

优化方法二阶信息可微算法损失函数

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