DynaSolver

A Neural Operator for Autoregressive 4D Flow Prediction with Dynamic Boundaries

Rui Qian1,*, Shibo Dai1,*, Mengjun Lin1,*, Yuk-Kit Hou2, Saining Zhang3, Bohan Li4, Junhao Chen1, Hao Zhao1, Chaojian Li5,†

1Tsinghua University 2Peking University 3Nanyang Technological University 4Shanghai Jiao Tong University 5The Hong Kong University of Science and Technology

* Equal contribution.   † Corresponding author.

Video

Overview of RoboCFD-4D and DynaSolver (1:16).

Abstract

Neural operators provide efficient surrogates for computational fluid dynamics (CFD), but most are designed for static geometries and are not equipped to model unsteady flows induced by continuously moving boundaries. We study dynamic-boundary 4D CFD prediction for articulated moving geometries and introduce RoboCFD-4D, a temporally resolved robotic CFD benchmark spanning diverse morphologies and flow conditions. To address the computational and autoregressive challenges of this setting, we propose DynaSolver, a neural operator that explicitly conditions on boundary evolution and causal flow history. DynaSolver employs an efficient physical-token formulation that amortizes point-to-token projection across the operator stack, substantially reducing repeated point-space spatiotemporal computation while retaining accurate point-level reconstruction. We further introduce Replay-Augmented Parallel Training, which exposes the model to autoregressively generated histories while preserving parallel dense supervision, narrowing the gap between ground-truth-conditioned training and autoregressive rollout. Experiments on RoboCFD-4D show that DynaSolver achieves the best full-rollout accuracy among a broad range of learned CFD surrogates, with its advantage increasing as rollout proceeds, indicating substantially reduced autoregressive error accumulation.

Experiments

Quantitative comparison

On RoboCFD-4D Base we report relative L2 (%) of velocity and pressure at selected rollout steps, plus full-rollout averages. Persistence is a no-learning baseline and is excluded from ranking. DynaSolver is not the strongest at the first step, but it is best from mid-horizon onward, so the full-rollout overall error is lowest.

Method Per-frame Velocity ↓ Per-frame Pressure ↓ Avg. Vel. ↓ Avg. Pre. ↓ Overall ↓
1612182429 1612182429
GNOT11.3821.1428.4734.5840.0143.9934.0872.8993.4998.66102.2199.1830.5687.8431.55
GAOT14.4619.5423.5826.9329.6631.1940.9657.7066.5275.1884.0890.6724.6070.2525.32
CNO9.3219.3624.2728.4935.2547.0434.2477.0987.2998.14108.89120.7127.1291.5228.28
P3D9.3217.0722.6226.7030.4432.3632.5562.6676.0385.2596.86102.0823.7278.8224.61
MGN9.4919.7023.6925.8827.6628.5529.3265.0279.2586.5596.7899.8623.4879.4424.51
DPOT13.6622.6526.7032.7739.3944.9842.4578.12103.77133.25166.67187.9730.30121.1032.08
AB-UPT14.7519.6023.3625.6827.9229.3842.2977.4791.3096.3097.5496.2323.8386.8425.03
UNO11.7118.1722.8226.9134.5362.3134.1358.3671.4381.76104.69156.9927.3281.5528.40
U-Net7.4816.7320.9923.3025.0526.0827.9061.6478.4185.4196.0495.7520.8177.9921.86
MSPT9.2819.1523.1825.8928.1429.9427.7061.1873.9086.65101.71110.8823.4279.1424.39
Transformergalerkin11.8919.4923.0728.4335.6742.7438.6169.1396.66115.08136.97168.6726.86104.6428.11
DilResNet8.5517.9922.0823.9025.0425.3029.0061.8778.0083.8086.8484.9921.4474.5622.30
PhysGTO9.4118.5121.8523.6825.6926.9730.5260.9770.1278.4089.1092.7921.8172.9622.99
GNS9.6719.7424.3127.7230.7032.7429.6961.9278.2489.2899.89106.8324.9780.9126.09
DeepONet16.6821.7426.2028.0528.2928.6760.0773.6486.3791.3492.4990.6325.5184.6526.38
GraphViT8.3317.0520.4122.1823.6124.4428.2857.9869.3671.1675.6176.2020.1566.0521.00
BSMS8.4417.3021.2024.1226.6628.8527.0154.2061.5768.2177.8481.2421.7863.9222.51
FNO10.7018.0422.0726.5433.9542.8131.1352.0260.8566.8371.3076.0625.4961.6926.08
LinearNO10.2616.8119.7421.6622.9223.8429.1452.0560.8165.8272.0772.5519.8261.0020.58
MWT9.5516.5018.6720.2622.0323.3532.0557.1970.5883.2897.58106.5218.9776.6420.11
Transolver10.1916.2419.1020.7322.3823.4529.7754.2761.5166.9974.5375.6419.2362.7619.99
Persistence14.1420.7819.8020.4220.9221.3540.1760.1954.1654.6358.7558.2120.1456.0920.95
DynaSolver (Ours) 11.6116.2717.3717.8618.0618.36 34.0253.1957.4859.1760.3761.61 17.0756.3517.71

Table 1. Quantitative comparison across rollout horizons on RoboCFD-4D Base. Best in green; second-best underlined. Persistence is not ranked.

Rollout error trends

The same Base evaluation as curves: overall, velocity, and pressure relative L2 over the full autoregressive horizon. DynaSolver’s gap widens as rollout proceeds, especially for velocity, matching the late-horizon numbers in Table 1 (18.36% velocity and 61.61% pressure at step 29).

Overall
Velocity
Pressure

Figure 1. Rollout error evolution on RoboCFD-4D Base; curves are drawn frame by frame along the rollout horizon.

Accuracy versus training cost

Training cost versus overall rollout error
17.71%lowest overall error
−11.4%vs. next best, 19.99%
21learned baselines

Each point is one learned surrogate, placed by its training time and full-rollout error on RoboCFD-4D Base. The dashed line marks Persistence, a no-learning baseline that copies the last state.

Most baselines trade one for the other: fast models stay near the Persistence line, while accurate ones need several times longer to train.

Two choices avoid this trade-off. The token-space operator skips repeated point-space work, and Replay-Augmented Parallel Training avoids slow sequential unrolling.

Figure 2. Training wall-clock versus overall rollout error on RoboCFD-4D Base.

Qualitative comparison

Thirty autoregressive rollout frames on three robots (A1, G1, and H1-2). The first panel is the ground-truth velocity field; the rest are point-wise velocity error of DynaSolver and five baselines. Click the board to pause or resume.

Robot

RoboCFD-4D Dataset

RoboCFD-4D is a temporally resolved CFD benchmark of articulated robots in motion. Below are twenty-two robots at u = 6 m/s and yaw 0°. Clips are muted loops; click a clip to pause it.

Method

1Efficient 4D Sequence Modeling

DynaSolver predicts the next velocity and pressure field from the robot geometry, its boundary motion, and a short causal history of flow states. It is designed to do this cheaply and with explicit temporal reasoning.

Overview of DynaSolver

Figure 3. DynaSolver pipeline. A single Fast Slice / Fast Deslice pair encloses the whole spatio-temporal operator stack.

Efficiency

WhyEvery transition carries thousands of surface and volume points. Point-space operators re-project points to tokens and run point-wise feed-forward layers in every layer, so a temporal window multiplies this heavy cost by both depth and window length.

HowWe slice once: lifted point features are encoded into compact surface and volume physical tokens, all self-, cross-, causal attention and feed-forward layers operate in token space, and a single deslice with a light MLP decodes the point-wise residual field.

ResultFor a window of W transitions and L operator blocks, point–token projection runs O(W) rather than O(LW) times, so deeper temporal reasoning adds only token-space cost while outputs stay point-accurate.

Temporal modeling

1Boundary evolution injection. Each transition is conditioned on the boundary motion between adjacent frames, giving the operator explicit information on how the geometry moves instead of inferring it from the flow.

2Causal sliding window. A short history of up to W transitions supplies the memory that a single, partially observed flow state lacks; the window grows from the initial state and then slides forward.

3Causal temporal attention. A block-causal mask lets each transition aggregate information only from itself and earlier transitions, so the model reasons over history exactly as it will during rollout and predicts a residual on top of the previous state.

2Replay-Augmented Parallel Training

Replay-Augmented Parallel Training

Figure 4. (a–b) sequential unroll and pushforward; (c) parallel teacher forcing; (d) our replay-augmented parallel training.

Training an autoregressive solver is a case of you can’t have your cake and eat it: fast training or robust rollout.

Unroll and Pushforward feed the model its own predictions, so rollout is robust, but they unroll sequentially and take about 16.5 h. Parallel teacher forcing supervises every transition in one block-causal pass (1.8 h), yet it only ever sees ground-truth histories, so the error grows to 18.18% once the model rolls out on its own outputs.

We close this train–inference gap without giving up parallelism. A frozen previous-epoch model rolls out replay histories without gradients, and the current model is trained on a mix of replay and clean clips in a single parallel pass with ground-truth targets.

Both at once: the lowest rollout error, 16.14%, in just 2.313 h, beating Unroll while training about 7.2× faster.

Training strategyOverall rollout L2 (%)Training time (h)

RoboCFD-4D Mini, same DynaSolver architecture. Error bars start at 15%.

Citation

@misc{qian2026dynasolverneuraloperatorautoregressive,
  title         = {DynaSolver: A Neural Operator for Autoregressive 4D Flow Prediction with Dynamic Boundaries},
  author        = {Rui Qian and Shibo Dai and Mengjun Lin and Yuk-Kit Hou and Saining Zhang and Bohan Li and Junhao Chen and Hao Zhao and Chaojian Li},
  year          = {2026},
  eprint        = {2611.03595},
  archivePrefix = {arXiv},
  primaryClass  = {cs.LG},
  url           = {https://arxiv.org/abs/2611.03595},
}