PI-MFM: Physics-informed multimodal foundation model for solving partial differential equations
By: Min Zhu , Jingmin Sun , Zecheng Zhang and more
Partial differential equations (PDEs) govern a wide range of physical systems, and recent multimodal foundation models have shown promise for learning PDE solution operators across diverse equation families. However, existing multi-operator learning approaches are data-hungry and neglect physics during training. Here, we propose a physics-informed multimodal foundation model (PI-MFM) framework that directly enforces governing equations during pretraining and adaptation. PI-MFM takes symbolic representations of PDEs as the input, and automatically assembles PDE residual losses from the input expression via a vectorized derivative computation. These designs enable any PDE-encoding multimodal foundation model to be trained or adapted with unified physics-informed objectives across equation families. On a benchmark of 13 parametric one-dimensional time-dependent PDE families, PI-MFM consistently outperforms purely data-driven counterparts, especially with sparse labeled spatiotemporal points, partially observed time domains, or few labeled function pairs. Physics losses further improve robustness against noise, and simple strategies such as resampling collocation points substantially improve accuracy. We also analyze the accuracy, precision, and computational cost of automatic differentiation and finite differences for derivative computation within PI-MFM. Finally, we demonstrate zero-shot physics-informed fine-tuning to unseen PDE families: starting from a physics-informed pretrained model, adapting using only PDE residuals and initial/boundary conditions, without any labeled solution data, rapidly reduces test errors to around 1% and clearly outperforms physics-only training from scratch. These results show that PI-MFM provides a practical and scalable path toward data-efficient, transferable PDE solvers.
Similar Papers
Towards a Foundation Model for Partial Differential Equations Across Physics Domains
Machine Learning (CS)
Predicts how things move and change using math.
Paving the way for scientific foundation models: enhancing generalization and robustness in PDEs with constraint-aware pre-training
Machine Learning (CS)
Teaches computers to solve hard math problems faster.
Missing Physics Discovery through Fully Differentiable Finite Element-Based Machine Learning
Computational Engineering, Finance, and Science
Learns missing science rules for better computer predictions.