# Enhancement of shock-capturing methods via machine learning

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

## Key facts

- Cited by 16 papers in the map (data through 7 Oct 2026).
- First posted on arXiv 6 Feb 2020.

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

- [Enhancing Computational Fluid Dynamics with Machine Learning](https://mlsanity.com/paper/2110.02085/) (2021)
- [Weak baselines and reporting biases lead to overoptimism in machine learning for fluid-related partial differential equations](https://mlsanity.com/paper/2407.07218/) (2024)
- [JAX-FLUIDS: A fully-differentiable high-order computational fluid dynamics solver for compressible two-phase flows](https://mlsanity.com/paper/2203.13760/) (2022)
- [Physics-Guided, Physics-Informed, and Physics-Encoded Neural Networks in Scientific Computing](https://mlsanity.com/paper/2211.07377/) (2022)
- [Machine Learning for Partial Differential Equations](https://mlsanity.com/paper/2303.17078/) (2023)
- [Applying Machine Learning to Study Fluid Mechanics](https://mlsanity.com/paper/2110.02083/) (2021)
- [Invariant preservation in machine learned PDE solvers via error correction](https://mlsanity.com/paper/2303.16110/) (2023)
- [Machine Learning Applications to Computational Plasma Physics and Reduced-Order Plasma Modeling: A Perspective](https://mlsanity.com/paper/2409.02349/) (2024)
- [A fully-differentiable compressible high-order computational fluid dynamics solver](https://mlsanity.com/paper/2112.04979/) (2021)
- [Data-Driven Adaptive Gradient Recovery for Unstructured Finite Volume Computations](https://mlsanity.com/paper/2507.16571/) (2025)
- [Deep smoothness WENO scheme for two-dimensional hyperbolic conservation laws: A deep learning approach for learning smoothness indicators](https://mlsanity.com/paper/2309.10117/) (2023)
- [Learning WENO for entropy stable schemes to solve conservation laws](https://mlsanity.com/paper/2403.14848/) (2024)
- [An unsupervised machine-learning-based shock sensor for high-order supersonic flow solvers](https://mlsanity.com/paper/2308.00086/) (2023)
- [Rational-WENO: A lightweight, physically-consistent three-point weighted essentially non-oscillatory scheme](https://mlsanity.com/paper/2409.09217/) (2024)
- [Physics-Constrained Neural Networks for Improved Short-Term Weather Forecasting: A Case Study over the South Pacific](https://mlsanity.com/paper/2606.17659/) (2026)
- [Attention Is All You Need (to Avoid Spurious Oscillations)](https://mlsanity.com/paper/2609.13531/) (2026)

## Abstract (arXiv)

In recent years, machine learning has been used to create data-driven solutions to problems for which an algorithmic solution is intractable, as well as fine-tuning existing algorithms. This research applies machine learning to the development of an improved finite-volume method for simulating PDEs with discontinuous solutions. Shock capturing methods make use of nonlinear switching functions that are not guaranteed to be optimal. Because data can be used to learn nonlinear relationships, we train a neural network to improve the results of a fifth-order WENO method. We post-process the outputs of the neural network to guarantee that the method is consistent. The training data consists of the exact mapping between cell averages and interpolated values for a set of integrable functions that represent waveforms we would expect to see while simulating a PDE. We demonstrate our method on linear advection of a discontinuous function, the inviscid Burgers' equation, and the 1-D Euler equations. For the latter, we examine the Shu-Osher model problem for turbulence-shockwave interactions. We find that our method outperforms WENO in simulations where the numerical solution becomes overly diffused due to numerical viscosity.