English

RL-PINNs: Reinforcement Learning-Driven Adaptive Sampling for Efficient Training of PINNs

Machine Learning 2025-04-18 v1 Numerical Analysis Numerical Analysis

Abstract

Physics-Informed Neural Networks (PINNs) have emerged as a powerful framework for solving partial differential equations (PDEs). However, their performance heavily relies on the strategy used to select training points. Conventional adaptive sampling methods, such as residual-based refinement, often require multi-round sampling and repeated retraining of PINNs, leading to computational inefficiency due to redundant points and costly gradient computations-particularly in high-dimensional or high-order derivative scenarios. To address these limitations, we propose RL-PINNs, a reinforcement learning(RL)-driven adaptive sampling framework that enables efficient training with only a single round of sampling. Our approach formulates adaptive sampling as a Markov decision process, where an RL agent dynamically selects optimal training points by maximizing a long-term utility metric. Critically, we replace gradient-dependent residual metrics with a computationally efficient function variation as the reward signal, eliminating the overhead of derivative calculations. Furthermore, we employ a delayed reward mechanism to prioritize long-term training stability over short-term gains. Extensive experiments across diverse PDE benchmarks, including low-regular, nonlinear, high-dimensional, and high-order problems, demonstrate that RL-PINNs significantly outperforms existing residual-driven adaptive methods in accuracy. Notably, RL-PINNs achieve this with negligible sampling overhead, making them scalable to high-dimensional and high-order problems.

Keywords

Cite

@article{arxiv.2504.12949,
  title  = {RL-PINNs: Reinforcement Learning-Driven Adaptive Sampling for Efficient Training of PINNs},
  author = {Zhenao Song},
  journal= {arXiv preprint arXiv:2504.12949},
  year   = {2025}
}
R2 v1 2026-06-28T23:02:04.412Z