arrow
Return

Multi-Objective Loss Balancing for Physics-Informed Deep Learning

delete2025-05-01
delete1
delete
OA
AI
R
Rafael Bischof
M
Michael Kraus *
DOI:10.1016/j.cma.2025.117914delete
deleteOriginal
deleteShare
deleteSave
View PDF
Abstract

Abstract

En 中文
Physics-Informed Neural Networks (PINN) are deep learning algorithms that leverage physical laws by including partial differential equations together with a respective set of boundary and initial conditions as penalty terms in their loss function. In this work, we observe the significant role of correctly weighting the combination of multiple competitive loss functions for training PINNs effectively. To this end, we implement and evaluate different methods aiming at balancing the contributions of multiple terms of the PINN's loss function and their gradients. After reviewing three existing loss scaling approaches (Learning Rate Annealing, GradNorm and SoftAdapt), we propose a novel self-adaptive loss balancing scheme for PINNs named ReLoBRaLo (Relative Loss Balancing with Random Lookback). We extensively evaluate the performance of the aforementioned balancing schemes by solving both forward as well as inverse problems on three benchmark PDEs for PINNs: Burgers' equation, Kirchhoff's plate bending equation, Helmholtz's equation and over 20 PDEs from the PINNacle collection. The results show that ReLoBRaLo is able to consistently outperform the baseline of existing scaling methods in terms of accuracy while also inducing significantly less computational overhead for a variety of PDE classes.
Keywords:
Partial differential equations
Scientific machine learning
Helmholtz equation
Burgers equation
PINNacle
Loss balancing
Multi-objective optimisation
AI Summary

AI Summary

Key information extracted from the uploaded paper, including a brief overview, abstract, background, key highlights, visual analysis, and future outlook.

Journal

Computer Methods in Applied Mechanics and Engineering cover
Computer Methods in Applied Mechanics and Engineering
IF:
7.3
Papers:
1.3W
Citations:
5.6W

Organization

S
Swiss Fed Inst Technol
Scholars:
2.3K
Papers: 1.1K
Citations: 517
T
Tech Univ Darmstadt
Scholars:
586
Papers: 283
Citations: 90