Principled Acceleration of Iterative Numerical Methods Using Machine Learning
Not on the map. Why?
Abstract (arXiv)
Iterative methods are ubiquitous in large-scale scientific computing applications, and a number of approaches based on meta-learning have been recently proposed to accelerate them. However, a systematic study of these approaches and how they differ from meta-learning is lacking. In this paper, we propose a framework to analyze such learning-based acceleration approaches, where one can immediately identify a departure from classical meta-learning. We show that this departure may lead to arbitrary deterioration of model performance. Based on our analysis, we introduce a novel training method for learning-based acceleration of iterative methods. Furthermore, we theoretically prove that the proposed method improves upon the existing methods, and demonstrate its significant advantage and versatility through various numerical applications.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Builds on 26 of its 51 references that are in the map. First posted on arXiv 17 Jun 2022.
Builds on
26 of 51 references.
References · 26
- 2014Adam: A Method for Stochastic Optimizationcited by 37,240
- 2017Model-Agnostic Meta-Learning for Fast Adaptation of Deep Networkscited by 4,897
- 2020Fourier Neural Operator for Parametric Partial Differential Equationscited by 2,908
- 2017Sigmoid-Weighted Linear Units for Neural Network Function Approximation in Reinforcement Learningcited by 950
- 2020Meta-Learning in Neural Networks: A Surveycited by 950
- 2020Learning Mesh-Based Simulation with Graph Networkscited by 868
- 2019Understanding Straight-Through Estimator in Training Activation Quantized Neural Netscited by 231
- 2020Solver-in-the-Loop: Learning from Differentiable Physics to Interact with Iterative PDE-Solverscited by 205
- 2019Learning Neural PDE Solvers with Convergence Guaranteescited by 77
- 2020Learning Algebraic Multigrid Using Graph Neural Networkscited by 54
- 2020Path Planning using Neural A* Searchcited by 52
- 2021Meta-learning PINN loss functionscited by 51
14 more
- 2019SciPy 1.0--Fundamental Algorithms for Scientific Computing in Pythoncited by 49
- 2021Meta-Auto-Decoder for Solving Parametric Partial Differential Equationscited by 36
- 2021A novel meta-learning initialization method for physics-informed neural networkscited by 35
- 2022Multigrid-augmented deep learning preconditioners for the Helmholtz equationcited by 25
- 2019Poisson CNN: Convolutional neural networks for the solution of the Poisson equation on a Cartesian meshcited by 22
- 2021Optimization-Based Algebraic Multigrid Coarsening Using Reinforcement Learningcited by 21
- 2019Meta-learning Pseudo-differential Operators with Deep Neural Networkscited by 17
- 2020A Helmholtz equation solver using unsupervised learning: Application to transcranial ultrasoundcited by 12
- 2022A Deep Conjugate Direction Method for Iteratively Solving Linear Systemscited by 12
- 2021Neural Fixed-Point Acceleration for Convex Optimizationcited by 10
- 2021Using neural networks to solve the 2D Poisson equation for electric field computation in plasma fluid simulationscited by 10
- 2021Personalized Algorithm Generation: A Case Study in Learning ODE Integratorscited by 9
- 2021Accelerating GMRES with Deep Learning in Real-Timecited by 8
- 2021Structural engineering from an inverse problems perspectivecited by 7
Built on by
Citing papers per year: 2023: 1; 2024: 2; 2025: 2; 2026: 1.
6 papers.
- 2025Leveraging Operator Learning to Accelerate Convergence of the Preconditioned Conjugate Gradient Methodcited by 6
- 2026PRISM: Distribution-free Adaptive Computation of Matrix Functions for Accelerating Neural Network Trainingcited by 4
- 2025Learning to accelerate distributed ADMM using graph neural networkscited by 3
- 2024Mitigating distribution shift in machine learning-augmented hybrid simulationcited by 2
- 2024Accelerating Legacy Numerical Solvers by Non-intrusive Gradient-based Meta-solvingcited by 2
- 2023Learning to Relax: Setting Solver Parameters Across a Sequence of Linear System Instancescited by 0
- 2026PRISM: Distribution-free Adaptive Computation of Matrix Functions for Accelerating Neural Network Training29 Jan 2026
- 2025Learning to accelerate distributed ADMM using graph neural networks5 Sep 2025
- 2025Leveraging Operator Learning to Accelerate Convergence of the Preconditioned Conjugate Gradient Method31 Jul 2025
- 2024Accelerating Legacy Numerical Solvers by Non-intrusive Gradient-based Meta-solving5 May 2024
- 2024Mitigating distribution shift in machine learning-augmented hybrid simulation17 Jan 2024
- 2023Learning to Relax: Setting Solver Parameters Across a Sequence of Linear System Instances3 Oct 2023