Search papers, labs, and topics across Lattice.
This paper introduces RLMesh, a reinforcement learning (RL) framework for optimizing mesh configurations in PDE simulations to accelerate the training of deep surrogate models. By adaptively allocating mesh grid points based on solution sensitivity, RLMesh reduces the number of fine-grid simulations needed for training, addressing the data-hungry nature of deep PDE surrogates. Experiments on PDE benchmarks demonstrate that RLMesh achieves comparable accuracy to baselines with significantly fewer simulation queries, highlighting the benefits of solver-level spatial adaptivity.
Dramatically cut the simulation budget for PDE surrogate training by using RL to intelligently allocate mesh points.
Deep surrogate models for parametric partial differential equations (PDEs) can deliver high-fidelity approximations but remain prohibitively data-hungry: training often requires thousands of fine-grid simulations, each incurring substantial computational cost. To address this challenge, we introduce RLMesh, an end-to-end framework for efficient surrogate training under limited simulation budget. The key idea is to use reinforcement learning (RL) to adaptively allocate mesh grid points non-uniformly within each simulation domain, focusing numerical resolution in regions most critical for accurate PDE solutions. A lightweight proxy model further accelerates RL training by providing efficient reward estimates without full surrogate retraining. Experiments on PDE benchmarks demonstrate that RLMesh achieves competitive accuracy to baselines but with substantially fewer simulation queries. These results show that solver-level spatial adaptivity can dramatically improve the efficiency of surrogate training pipelines, enabling practical deployment of learning-based PDE surrogates across a wide range of problems.
