CST-Bench: Benchmarking Counterfactual Spatial Transition Reasoning with Causal Transition On-Policy Self-Distillation
Organizations: School of Artificial Intelligence and Computer Science, North China University of Technology, Beijing, China · School of Artificial Intelligence, Jilin University, Changchun, China · ReLER Lab, Zhejiang University, Hangzhou, China
Abstract
Spatial reasoning is a fundamental capability of multimodal large language models (MLLMs), yet existing benchmarks mainly focus on static scene understanding. When objects are moved or swapped, models must infer not only spatial relations but also how interventions change the scene state. To address this problem, we introduce CST-Bench, a benchmark for counterfactual spatial reasoning. CST-Bench contains 7,082 examples covering 6 intervention types and 12 transformation operators, with dense causal annotations including intervention sources, affected relations, expected effects, invariant relations, and metric changes. We benchmark general-purpose MLLMs on CST-Bench and further compare post-training strategies on CST-Bench and standard spatial benchmarks. Results show that methods effective for ordinary spatial reasoning can still fail under counterfactual interventions. We further formulate counterfactual spatial reasoning as causal state-transition modeling and propose CT-OPSD, which uses CST-Bench annotations to guide a teacher model in learning counterfactual state transitions and causal reasoning chains. These chains are then distilled into a student model that reasons only from real observations. Overall, experiment results establish CST-Bench as a challenging benchmark and reusable supervision resource for counterfactual spatial reasoning. CT-OPSD further provides an initial demonstration of how its annotations can support model improvement.
Figures & tables
| Transition Families | Aggregate Results | |||||||
| Model | Mirror | Swap | Metric | Translation | Depth | Viewpoint | Macro Avg. | Micro Acc. |
| General-purpose Open-source MLLMs | ||||||||
| Qwen2.5-VL-3B | 15.5 | 15.1 | 17.8 | 20.7 | 11.6 | 17.0 | 16.3 | 17.0 |
| Qwen3-VL-4B | 42.5 | 25.0 | 35.6 | 96.3 | 37.5 | 30.0 | 44.5 | 49.4 |
| Qwen2.5-VL-7B | 27.6 | 19.2 | 22.3 | 36.3 | 12.5 | 20.6 | 23.1 | 25.1 |
| LLaVA-OneVision-7B | 26.3 | 35.0 | 39.4 | 90.0 | 32.5 | 27.5 | 41.8 | 46.6 |
| CST-Bench | 3DSRBench | CV-Bench | BLINK val | |||||
| Model | Micro Acc. | Acc. | 2D | 3D | Avg. | Spatial Relation | Relative Depth | Avg. |
| Post-training Strategies with Qwen2.5-VL-3B | ||||||||
| Qwen2.5-VL-3B | 17.0 | 46.0 | 64.0 | 66.0 | 65.0 | 83.0 | 64.0 | 73.5 |
| Qwen2.5-VL-3B + SFT | 50.3 | 38.0 | 63.0 | 68.0 | 65.5 | 70.0 | 67.0 | 68.5 |
| Qwen2.5-VL-3B + GRPO | 46.7 | 52.9 | 71.0 | 76.3 | 73.7 | 81.8 | 66.9 | 74.4 |
| Qwen2.5-VL-3B + OPSD | 53.3 | 51.8 | 73.8 | 78.3 | 76.1 | 82.5 | 67.2 | 74.9 |
| Variant | CST W-Avg. | Ans. Acc. | Src. | Eff. | Inv. | Cons. |
|---|---|---|---|---|---|---|
| OPSD | 24.0 | 53.3 | 34.1 | 0.7 | 0.3 | 22.5 |
| CT w/o Aux. | 24.7 | 52.7 | 40.1 | 0.9 | 0.4 | 30.0 |
| CT w/o Bridge | 24.5 | 52.1 | 40.0 | 1.0 | 0.7 | 30.6 |
| CT w/o Eff. | 24.0 | 50.4 | 40.0 | 1.2 | 1.0 | 28.5 |
| CT w/o Inv. | 24.6 | 51.7 | 41.2 | 1.3 | 0.5 | 31.2 |
| CT w/o Metric | 24.1 | 50.4 | 41.3 | 1.2 | 0.5 | 28.5 |
Appendix figures & tables6 assets
Supplementary material from the paper’s appendix.
Appendix
| Family | Samples | Share |
|---|---|---|
| Global Mirror Transformation | 1,600 | 22.6% |
| Object Position Swap | 1,600 | 22.6% |
| Metric Distance Transition | 1,600 | 22.6% |
| Translation-Based Crossing | 1,300 | 18.4% |
| Depth Transition | 300 | 4.2% |
| Viewpoint Reversal | 682 | 9.6% |
| Transition operator | Samples | Transition operator | Samples |
| Pairwise Left–Right Flip | 1,300 | Lateral Relation Crossing | 386 |
| Multi-object Horizontal Order Flip | 300 | Above–Below Crossing | 614 |
| Pairwise Relation Swap | 300 | Vertical Near–Far Crossing | 300 |
| Third-object Relation Update | 1,300 | Depth Order Change | 300 |
| Metric Distance Change (Relation Preserved) | 1,300 | Front-to-Back Viewpoint Shift | 382 |
| Distance-to-Reference Near–Far Crossing | 300 | Side-to-Side Viewpoint Shift | 300 |
| Family | Operator | Precondition | Intervention | Affected / expected | Principal invariants |
|---|---|---|---|---|---|
| Global mirror | Pairwise Left–Right Flip | Clear horizontal pair | Mirror image coordinates | left right | identity, count, vertical relation |
| Global mirror | Multi-object Horizontal Order Flip | Three ordered, distinct objects | Mirror all selected objects | reverse all horizontal pair orders | identity, count, vertical relations |
| Object swap | Pairwise Relation Swap | Two distinct objects | Exchange their positions | queried-pair relation reverses | identity, count, attributes |
| Object swap | Third-object Relation Update | Swapped pair plus unchanged reference | Exchange two positions | update relation to third object | reference object, identity, count |
| Metric distance | Metric Distance Change (Relation Preserved) | Clear near/far pair | Change subject–reference distance | metric distance changes; label preserved | direction, identity, image-plane order |
| Metric distance | Distance-to-Reference Near–Far Crossing | Clear pair with valid offset | Change distance without crossing reference object | near far | side of reference, identity, orthogonal relation |
| Field | Meaning |
|---|---|
| sample_uid | Stable sample identifier. |
| image_id | Source-image identifier used for grouped splitting. |
| operator | Intervention family and fine-grained transition operator. |
| question | Intervention-conditioned natural-language query. |
| options , answer | Answer choices and target choice. |
| source_state | Relevant source objects and spatial relations. |
| Component | Role |
|---|---|
| Observable input | Image, question, intervention description, and answer options. |
| Causal teacher | Predicts source state, intervention consequences, invariants, target state, and answer. |
| Student rollout | Samples a trajectory from the observation-only student policy. |
| On-policy alignment | Matches the teacher and student distributions on the shared student prefix. |
| Answer supervision | Anchors the final response to the gold answer . |
| Inference student | Uses only the image and question after teacher removal. |
Spatial Reasoning Benchmarks
Spatial reasoning benchmarks test whether AI systems can perceive, represent, and act on geometric, relational, and viewpoint-dependent information.
See all 168 papers