Enforcing Analytic Constraints in Neural Networks Emulating Physical Systems

Phys Rev Lett. 2021 Mar 5;126(9):098302. doi: 10.1103/PhysRevLett.126.098302.

Abstract

Neural networks can emulate nonlinear physical systems with high accuracy, yet they may produce physically inconsistent results when violating fundamental constraints. Here, we introduce a systematic way of enforcing nonlinear analytic constraints in neural networks via constraints in the architecture or the loss function. Applied to convective processes for climate modeling, architectural constraints enforce conservation laws to within machine precision without degrading performance. Enforcing constraints also reduces errors in the subsets of the outputs most impacted by the constraints.