OMNIFLOW: A Physics-Grounded Multimodal Agent for Generalized Scientific Reasoning
Organizations: Tsinghua University · Tencent · Shenzhen Loop Area Institute · Ocean University of China ·
Abstract
Large Language Models (LLMs) have demonstrated exceptional logical reasoning capabilities but frequently struggle with the continuous spatiotemporal dynamics governed by Partial Differential Equations (PDEs), often resulting in non-physical hallucinations. Existing approaches typically resort to costly, domain-specific fine-tuning, which severely limits cross-domain generalization and interpretability. To bridge this gap, we propose OMNIFLOW, a neuro-symbolic architecture designed to ground frozen multimodal LLMs in fundamental physical laws without requiring domain-specific parameter updates. OMNIFLOW introduces a novel Semantic-Symbolic Alignment mechanism that projects high-dimensional flow tensors into topological linguistic descriptors, enabling the model to perceive physical structures rather than raw pixel values. Furthermore, we construct a Physics-Guided Chain-of-Thought (PG-CoT) workflow that orchestrates reasoning through dynamic constraint injection (e.g., mass conservation) and iterative reflexive verification. We evaluate OMNIFLOW on a comprehensive benchmark spanning microscopic turbulence, theoretical Navier-Stokes equations, and macroscopic global weather forecasting. Empirical results demonstrate that OMNIFLOW significantly outperforms traditional deep learning baselines in zero-shot generalization and few-shot adaptation tasks. Crucially, it offers transparent, physically consistent reasoning reports, marking a paradigm shift from black-box fitting to interpretable scientific reasoning.
Figures & tables
| 2D Turbulence (Micro) | ERA5 (Global) | SEVIR (Regional) | |||||||
| Model | RMSE | SSIM | PSNR | RMSE | SSIM | PSNR | RMSE | SSIM | PSNR |
| Monolithic Foundation Models (Zero-shot Image Generation) | |||||||||
| Banana Pro | 118.4 5.2 | 0.142 .02 | 8.12 .4 | 102.5 4.8 | 0.194 .03 | 8.85 .3 | 108.6 5.5 | 0.165 .02 | 8.40 .5 |
| Seedream 4.5 | 98.2 4.1 | 0.265 .03 | 9.85 .5 | 85.2 3.9 | 0.352 .04 | 10.15 .4 | 90.4 4.2 | 0.312 .03 | 9.92 .4 |
| ChatGPT-Images | 112.5 6.3 | 0.185 .02 | 8.70 .6 | 96.4 5.1 | 0.228 .03 | 9.42 .5 | 101.3 5.8 | 0.205 .02 | 9.15 .6 |
| OMNIFLOW | 64.32 2.5 | 0.552 .04 | 11.45 .3 | 59.10 1.8 | 0.685 .03 | 12.70 .2 | 52.45 1.2 | 0.712 .04 | 13.82 .3 |
Appendix figures & tables3 assets
Supplementary material from the paper’s appendix.
Appendix
| Dataset | Physical Regime | Resolution | Temporal Horizon | Key Variables | Evaluation Scale |
| 2D Turbulence | Microscopic Flow | 100 timesteps | Vorticity | 1,280 sequences | |
| SEVIR | Regional Weather | 5-min intervals | VIL, IR, VIS, GLM | Standard Events | |
| ERA5 | Global Climate | 200-day forecast | 21 variables (T, Z, U/V, Q) | Long-term Validation |
| Category | Component / Parameter | Specifications / Values |
| Architecture | Reasoning Core | Gemini 3 Flash (Agentic Brain) |
| Numerical Simulator (NES) | Diffusion Transformer (DiT) | |
| Inference | 2D Turbulence Strategy | Training; Recursive Testing |
| ERA5 Strategy | 200-day Autoregressive Forecast | |
| Ensemble Size ( ) | 8 – 32 (Adaptive) | |
| Hyperparameters | Perturbation Factor ( ) | 0.01 – 0.05 |
Scientific Reasoning in Language Models
Scientific reasoning in language models studies how AI systems use evidence, domain knowledge, and tools to solve problems and support discovery.
See all 144 papers