FlowForge: A Staged Local Rollout Engine for Flow-Field Prediction
For researchers and engineers using deep learning surrogates for CFD, FlowForge offers a more robust and efficient alternative to global models, particularly under noisy or incomplete data.
FlowForge introduces a staged local rollout engine for flow-field prediction that compiles a locality-preserving update schedule and executes it with a shared lightweight predictor, matching or improving accuracy on PDEBench, CFDBench, and BubbleML while reducing per-step latency and improving robustness to noise.
Deep learning surrogates for CFD flow-field prediction often rely on large, complex models, which can be slow and fragile when data are noisy or incomplete. We introduce FlowForge, a staged local rollout engine that predicts future flow fields by compiling a locality-preserving update schedule and executing it with a shared lightweight local predictor. Rather than producing the next frame in a single global pass, FlowForge rewrites spatial sites stage by stage so that each update conditions only on bounded local context exposed by earlier stages. This compile-execute design aligns inference with short-range physical dependence, keeps latency predictable, and limits error amplification from global mixing. Across PDEBench, CFDBench, and BubbleML, FlowForge matches or improves upon strong baselines in pointwise accuracy, delivers consistently better robustness to noise and missing observations, and maintains stable multi-step rollout behavior while reducing per-step latency.