# Self-test loss functions for learning weak-form operators and gradient flows

arXiv: https://arxiv.org/abs/2412.03506 · HTML: https://mlsanity.com/paper/2412.03506/

## Key facts

- Cited by 4 papers in the map (data through 14 Oct 2026).
- Builds on 21 of its 51 references that are in the map.
- First posted on arXiv 4 Dec 2024.

## Builds on (most cited first, 21 of 21)

- [Score-Based Generative Modeling through Stochastic Differential Equations](https://mlsanity.com/paper/2011.13456/) (2020)
- [DGM: A deep learning algorithm for solving partial differential equations](https://mlsanity.com/paper/1708.07469/) (2017)
- [The Deep Ritz method: A deep learning-based numerical algorithm for solving variational problems](https://mlsanity.com/paper/1710.00211/) (2017)
- [PDE-Net: Learning PDEs from Data](https://mlsanity.com/paper/1710.09668/) (2017)
- [hp-VPINNs: Variational Physics-Informed Neural Networks With Domain Decomposition](https://mlsanity.com/paper/2003.05385/) (2020)
- [Weak SINDy For Partial Differential Equations](https://mlsanity.com/paper/2007.02848/) (2020)
- [Variational Physics-Informed Neural Networks For Solving Partial Differential Equations](https://mlsanity.com/paper/1912.00873/) (2019)
- [Physics-informed graph neural Galerkin networks: A unified framework for solving PDE-governed forward and inverse problems](https://mlsanity.com/paper/2107.12146/) (2021)
- [Kernel Flows: from learning kernels from data into the abyss](https://mlsanity.com/paper/1808.04475/) (2018)
- [Nonparametric inference of interaction laws in systems of agents from trajectory data](https://mlsanity.com/paper/1812.06003/) (2018)
- [Numerical Solution of Inverse Problems by Weak Adversarial Networks](https://mlsanity.com/paper/2002.11340/) (2020)
- [wPINNs: Weak Physics informed neural networks for approximating entropy solutions of hyperbolic conservation laws](https://mlsanity.com/paper/2207.08483/) (2022)
- [Learning interaction kernels in heterogeneous systems of agents from multiple trajectories](https://mlsanity.com/paper/1910.04832/) (2019)
- [Learning interaction kernels in stochastic systems of interacting particles from multiple trajectories](https://mlsanity.com/paper/2007.15174/) (2020)
- [Learning dynamical systems from data: a simple cross-validation perspective](https://mlsanity.com/paper/2007.05074/) (2020)
- [Learning interaction kernels in mean-field equations of 1st-order systems of interacting particles](https://mlsanity.com/paper/2010.15694/) (2020)
- [Weak-PDE-LEARN: A Weak Form Based Approach to Discovering PDEs From Noisy, Limited Data](https://mlsanity.com/paper/2309.04699/) (2023)
- [Particle-based Energetic Variational Inference](https://mlsanity.com/paper/2004.06443/) (2020)
- [Identifiability of interaction kernels in mean-field equations of interacting particles](https://mlsanity.com/paper/2106.05565/) (2021)
- [The Weak Form Is Stronger Than You Think](https://mlsanity.com/paper/2409.06751/) (2024)
- [Data adaptive RKHS Tikhonov regularization for learning kernels in operators](https://mlsanity.com/paper/2203.03791/) (2022)

## Built on by (most cited first, 4 of 4)

- [Learning interacting particle systems from unlabeled data](https://mlsanity.com/paper/2604.02581/) (2026)
- [Moment Estimate and Variational Approach for Learning Generalized Diffusion with Non-gradient Structures](https://mlsanity.com/paper/2508.01854/) (2025)
- [Structure-Aware Variational Learning of a Class of Generalized Diffusions](https://mlsanity.com/paper/2604.20188/) (2026)
- [Moment Estimates and DeepRitz Methods on Learning Diffusion Systems with Non-gradient Drifts](https://mlsanity.com/paper/2509.10495/) (2025)

## Abstract (arXiv)

The construction of loss functions presents a major challenge in data-driven modeling involving weak-form operators in PDEs and gradient flows, particularly due to the need to select test functions appropriately. We address this challenge by introducing self-test loss functions, which employ test functions that depend on the unknown parameters, specifically for cases where the operator depends linearly on the unknowns. The proposed self-test loss function conserves energy for gradient flows and coincides with the expected log-likelihood ratio for stochastic differential equations. Importantly, it is quadratic, facilitating theoretical analysis of identifiability and well-posedness of the inverse problem, while also leading to efficient parametric or nonparametric regression algorithms. It is computationally simple, requiring only low-order derivatives or even being entirely derivative-free, and numerical experiments demonstrate its robustness against noisy and discrete data.
