White-Box Physics-AI for Reaction–Diffusion Systems: EnVarA-Guided Constitutive Learning and Stage-Consistent Time Integration
Min-Jhe Lu, Institute of Computational and Modeling Science, National Tsing Hua University
We present a white-box physics-AI framework for learning reaction–diffusion dynamics while retaining the mathematical and numerical structure of the governing equations. Guided by the energetic variational approach (EnVarA), we separate prescribed transport kinematics from model-dependent constitutive physics.
For systems of the form ∂tρ + ∇·(ρu) = ρr, the density is factorized as ρ = MI, where I describes local compression and M follows material trajectories while accumulating reaction. This decomposition provides a transparent interface between learned velocity and reaction laws and a conventional numerical solver: a finite-volume method updates the compression factor, while trajectory-based transport and reaction splitting update the mass factor.
The resulting neural solver is tested across seven PDE families, including diffusion, phase separation, porous-medium dynamics, reaction variants, and a coupled two-species system. The numerical experiments demonstrate transfer across spatial grids without retraining, coherent evolution beyond the training interval, and stable long-time pattern formation.
We then address a temporal-accuracy issue that becomes important when the constitutive law depends on the evolving density. At every Runge–Kutta stage, the density must first be reconstructed and the constitutive law evaluated again at that stage state. A matched comparison between an updated law and a reused law, together with a fixed-grid Fourier calculation, explains the observed split between second- and first-order time accuracy. The rigorous analysis covers prescribed smooth fields; extensions involving nonlinear learned feedback require additional regularity and stage-wise control.
This work is carried out jointly with Professor Shih-Hsuan Hung and master’s student Shang-Ke Chen from the Department of Computer Science, master’s student Kuan-Yu Chen from the Institute of Computational and Modeling Science, all at National Tsing Hua University, and Chao-Shun Zhan and Johnson Sun from the NVIDIA AI Technology Center, NVIDIA Corporation, Taipei, Taiwan.