Riemann Tensor Neural Networks: Learning Conservative Systems with Physics-Constrained Networks

  • 2025-06-16 11:33:21
  • Anas Jnini, Lorenzo Breschi, Flavio Vella
  • 0

Abstract

Divergence-free symmetric tensors (DFSTs) are fundamental in continuummechanics, encoding conservation laws such as mass and momentum conservation.We introduce Riemann Tensor Neural Networks (RTNNs), a novel neuralarchitecture that inherently satisfies the DFST condition to machine precision,providing a strong inductive bias for enforcing these conservation laws. Weprove that RTNNs can approximate any sufficiently smooth DFST with arbitraryprecision and demonstrate their effectiveness as surrogates for conservativePDEs, achieving improved accuracy across benchmarks. This work is the first touse DFSTs as an inductive bias in neural PDE surrogates and to explicitlyenforce the conservation of both mass and momentum within a physics-constrainedneural architecture.

 

Quick Read (beta)

loading the full paper ...