提出二阶优化解决物理神经网络中的梯度冲突问题
Gradient Alignment in Physics-informed Neural Networks: A Second-Order Optimization Perspective
- 用二阶优化隐式对齐多任务梯度方向,缓解损失函数冲突
- SOAP方法在10个复杂偏微分方程上达到顶尖性能,湍流模拟精度提升2-10倍
- 引入多梯度相似性评分工具,可分析优化过程中的梯度对齐状态
通过复合损失函数的多任务学习是现代深度学习的基础,但优化相互竞争的目标仍具挑战。本文从理论与实践两方面提出新方法,解决损失项间的方向冲突问题,尤其在物理信息神经网络(PINNs)中效果显著。理论分析表明,一阶优化受限于梯度冲突,而二阶优化可通过隐式梯度对齐自然化解。我们证明,近期提出的准牛顿方法SOAP能高效近似海森矩阵预处理器,在10个具有挑战性的偏微分方程基准测试中实现领先性能,首次成功应用于雷诺数高达10,000的湍流模拟,相比现有方法精度提升2-10倍。此外,我们提出一种新的梯度对齐评分,将余弦相似性推广至多梯度场景,为分析优化动态提供实用工具。研究结果建立了理解与解决梯度冲突的框架,对科学计算之外的优化领域亦具广泛意义。
原文摘要 · Abstract (English)
Multi-task learning through composite loss functions is fundamental to modern deep learning, yet optimizing competing objectives remains challenging. We present new theoretical and practical approaches for addressing directional conflicts between loss terms, demonstrating their effectiveness in physics-informed neural networks (PINNs) where such conflicts are particularly challenging to resolve. Through theoretical analysis, we demonstrate how these conflicts limit first-order methods and show that second-order optimization naturally resolves them through implicit gradient alignment. We prove that SOAP, a recently proposed quasi-Newton method, efficiently approximates the Hessian preconditioner, enabling breakthrough performance in PINNs: state-of-the-art results on 10 challenging PDE benchmarks, including the first successful application to turbulent flows with Reynolds numbers up to 10,000, with 2-10x accuracy improvements over existing methods. We also introduce a novel gradient alignment score that generalizes cosine similarity to multiple gradients, providing a practical tool for analyzing optimization dynamics. Our findings establish frameworks for understanding and resolving gradient conflicts, with broad implications for optimization beyond scientific computing.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。