CAST: Reconstruction-Coupled Acceleration of Interactive World Models
Organizations: Shanghai Jiao Tong University
Abstract
Interactive world models must respond quickly to controls while preserving scene consistency. Existing acceleration methods can miss heterogeneous control responses and spatial transport when recovering skipped features. We observe that interaction-induced feature changes correlate with approximation error, while low-frequency interpolation errors are phase-sensitive and show more predictable phase progression. These findings motivate CAST, a reconstruction-coupled inference framework. CAST selects anchors by interaction sensitivity and cross-layer coverage, reconstructs skipped residuals with frequency- and confidence-aware Phase-Aware Reconstruction (PAR), and coordinates historical KV routing according to downstream reconstruction responsibility. On Matrix-Game 3.0 and HY-World 1.5, CAST achieves 2.15x and 3.48x speedups, respectively, while maintaining visual quality close to Native (Figure 1). It also attains the highest VBench scores among compared methods and leads non-native baselines on seven and six of thirteen WorldMark dimensions, demonstrating a balance of generation speed, visual quality, and interactive responsiveness under real-time control. Code is available at https://github.com/lokiniuniu/CAST.
Figures & tables
| Action Dynamics | World Memory | Visual Quality | Efficiency | |||||||||||
| Method | Direction Accuracy | Direction Purity | Motion Stability | Response Latency | Local | Global | Revisit | Perceptual | Aesthetic | Speedup | ||||
| Trans. | Rot. | Trans. | Rot. | Trans. | Rot. | Trans. | Rot. | |||||||
| Matrix-Game 3.0 | ||||||||||||||
| Native | 98.053 | 87.203 | 83.605 | 82.966 | 79.832 | 98.184 | 93.636 | 96.296 | 76.763 | 47.755 | 80.124 | 77.174 | 58.688 | |
| SVG | 97.443 | 82.404 | 84.411 | 77.974 | 80.687 | 97.674 | 92.508 | 96.287 | 76.253 | 45.632 | 78.826 | 75.809 | 58.118 | |
| BSA | 97.306 | 86.877 | 84.061 | 82.903 | 82.822 | 98.427 | 93.748 | 96.268 | 76.502 | 44.809 | 80.591 | 77.362 | 59.192 | |
| Method | vs. Original | Self-Comparison | VBench | Speedup | ||||
| PSNR | SSIM | LPIPS | PSNR | SSIM | LPIPS | |||
| Matrix-Game 3.0 | ||||||||
| Native | — | — | — | 17.07 | 0.5264 | 0.3283 | 0.7504 | |
| SVG | 12.38 | 0.4170 | 0.5587 | 14.48 | 0.4949 | 0.4406 | 0.7511 | |
| BSA | 13.34 | 0.4228 | 0.5795 | 16.66 | 0.5326 | 0.4094 | 0.7336 | |
| MagCache | 15.02 | 0.4902 | 0.4504 | 14.06 | 0.5398 | 0.3837 | 0.7192 | |
| Variant | VBench | LPIPS | T-Motion | Local | Global | Revisit | Latency | Speedup |
| A: Uniform | 0.7218 | 0.5128 | 74.218 | 70.864 | 38.927 | 72.991 | 1,934 | |
| B: + CAFS | 0.7346 | 0.4974 | 76.482 | 72.943 | 41.586 | 75.648 | 1,951 | |
| C: + PAR | 0.7479 | 0.4776 | 78.219 | 75.046 | 44.387 | 79.206 | 1,976 | |
| Native | 0.7504 | — | 79.832 | 76.763 | 47.755 | 80.124 | 4,319 | |
| + CAFS | 0.7370 | 0.4960 | 76.800 | 74.000 | 43.200 | 76.700 | 2,800 | |
| + CAFS+ PAR | 0.7600 | 0.4550 | 80.200 | 76.900 | 47.500 | 82.000 | 2,825 |
| Variant | Core | Route | Recon. | Total | OH | Speedup |
| Matrix-Game 3.0 | ||||||
| Native | 4,319 | 0 | 0 | 4,319 | 0.0% | |
| Indep. (C) | 1,887 | 43 | 46 | 1,976 | 4.5% | |
| CAST | 1,891 | 74 | 42 | 2,007 | 5.8% | |
| HY-World 1.5 | ||||||
| Native | 8,713 | 0 | 0 | 8,713 | 0.0% | |
Appendix figures & tables12 assets
Supplementary material from the paper’s appendix.
Appendix
| 1 | Input: ordered anchors, omission scores , fixed candidates , interval coefficients, budget , boundary coefficients from Eq. ( 21 ), and directional-pass cap . |
|---|---|
| 2 | Initialize for every anchor; construct binary masks . |
| 3 | For directional refinement pass : |
| 4 | Set . |
| 5 | For in forward order if is odd, backward order otherwise: |
| 6 | Compute using the latest neighbor masks (Eq. ( 23 )). |
| 7 | Let . |
| Field | Matrix-Game 3.0 | HY-World 1.5 |
|---|---|---|
| Checkpoint and code provenance | Skywork/Matrix-Game-3.0 , base_distilled_model ; released code with CAST integration | tencent/HY-WorldPlay , ar_distilled_action_model ; HunyuanVideo 8B path |
| GPU model and count | RTX A6000 per rollout; batch 1 | RTX A6000 per rollout; model/tensor parallel inference; batch 1 |
| Software and attention backend | Python 3.12; PyTorch 2.6; CUDA 12.4; Triton 3.2; FlashAttention 2.7 for dense paths; released Light Interaction block-sparse execution backend | Python 3.10; PyTorch 2.6; CUDA 12.4; Triton 3.2; FlashAttention 2.7 for dense paths; released Light Interaction block-sparse execution backend |
| Model / attention / FFT precision | BF16 model and QKV; FP32 softmax and pooled reductions; FP32 / complex64 FFT; no INT8/FP8 | BF16 model and QKV; FP32 softmax and pooled reductions; FP32 / complex64 FFT; no FP8 |
| Resolution and native chunk | ; steady-state 40 decoded / 10 new latent frames; separate 57-frame startup | ; steady-state 16 decoded / 4 latent frames; first chunk has 13 decoded frames |
| Denoising schedule | 3 evaluations; retain the released distilled scheduler and its native timestep ordering | 4 evaluations; native few-step AR scheduler; no denoising cache |
| Field | Matrix-Game 3.0 | HY-World 1.5 |
|---|---|---|
| , , | , of , ; dynamic Top-5 over all current frames; boundary extrapolation uses the two nearest anchors | , of , ; dynamic Top-2 over all current frames; boundary extrapolation uses the two nearest anchors |
| Tiles and reliable spectral bins | spatial token tiles, disjoint; ; FP32 | spatial token tiles, disjoint; ; FP32 |
| ; principal phase branch; DC/Nyquist transport disabled | ; principal phase branch; DC/Nyquist transport disabled | |
| , , | ; target ; (full candidate pool); bypass if | ; target ; ; report realized block fraction |
| Projection and scaling | , native attention output-projection head slice; no additional learned/sketched projection; FP32 pooled reductions | Same native output-projection head slice and FP32 reductions; squared Euclidean norm |
| Routing weights / stopping | ; directional passes; accept | ; directional passes; same relative tolerance |
| WorldMark | ||||||
| Variant | VBench | LPIPS | T-Motion | Local | Global | Revisit |
| Frame selection; PAR and coupled routing fixed | ||||||
| Uniform/interleaved | 0.7437 | 0.4884 | 76.914 | 73.478 | 42.806 | 77.138 |
| Latent-motion magnitude | 0.7506 | 0.4769 | 79.312 | 75.087 | 44.578 | 79.341 |
| Control sensitivity | 0.7570 | 0.4657 | 79.830 | 76.531 | 46.645 | 81.514 |
| Reconstruction; control-aware selection and coupled routing fixed | ||||||
| KV routing | Anchor error | Interval error | VBench | Local | Global | Revisit |
|---|---|---|---|---|---|---|
| Independent (C) | 0.0831 | 0.1127 | 0.7479 | 75.046 | 44.387 | 79.206 |
| Coupled, (D) | 0.0774 | 0.1013 | 0.7570 | 76.531 | 46.645 | 81.514 |
| WorldMark | |||||||
|---|---|---|---|---|---|---|---|
| VBench | LPIPS | Global | Revisit | Latency | Speedup | ||
| 0.30 | 0.20 | 0.7441 | 0.4936 | 42.738 | 77.246 | 1,717 | |
| 0.50 | 0.10 | 0.7512 | 0.4788 | 42.911 | 77.926 | 1,909 | |
| 0.50 | 0.20 | 0.7570 | 0.4657 | 46.645 | 81.514 | 2,007 | |
| 0.50 | 0.30 | 0.7584 | 0.4608 | 47.386 | 82.263 | 2,078 | |
| 0.60 | 0.20 | 0.7601 | 0.4529 | 47.219 | 82.487 | 2,363 | |
| WorldMark | |||||||
|---|---|---|---|---|---|---|---|
| VBench | LPIPS | Global | Revisit | Interval NRMSE | Latency | Speedup | |
| 0 | 0.7479 | 0.4776 | 44.387 | 79.206 | 0.1127 | 1,976 | |
| 1 | 0.7549 | 0.4689 | 45.916 | 80.721 | 0.1058 | 1,991 | |
| 2 | 0.7570 | 0.4657 | 46.645 | 81.514 | 0.1013 | 2,007 | |
| 4 | 0.7574 | 0.4649 | 46.781 | 81.672 | 0.1008 | 2,038 | |