English

SymPlex: A Structure-Aware Transformer for Symbolic PDE Solving

Machine Learning 2026-02-04 v1

Abstract

We propose SymPlex, a reinforcement learning framework for discovering analytical symbolic solutions to partial differential equations (PDEs) without access to ground-truth expressions. SymPlex formulates symbolic PDE solving as tree-structured decision-making and optimizes candidate solutions using only the PDE and its boundary conditions. At its core is SymFormer, a structure-aware Transformer that models hierarchical symbolic dependencies via tree-relative self-attention and enforces syntactic validity through grammar-constrained autoregressive decoding, overcoming the limited expressivity of sequence-based generators. Unlike numerical and neural approaches that approximate solutions in discretized or implicit function spaces, SymPlex operates directly in symbolic expression space, enabling interpretable and human-readable solutions that naturally represent non-smooth behavior and explicit parametric dependence. Empirical results demonstrate exact recovery of non-smooth and parametric PDE solutions using deep learning-based symbolic methods.

Keywords

Cite

@article{arxiv.2602.03816,
  title  = {SymPlex: A Structure-Aware Transformer for Symbolic PDE Solving},
  author = {Yesom Park and Annie C. Lu and Shao-Ching Huang and Qiyang Hu and Y. Sungtaek Ju and Stanley Osher},
  journal= {arXiv preprint arXiv:2602.03816},
  year   = {2026}
}

Comments

27 pages

R2 v1 2026-07-01T09:34:45.939Z