Reflect to Inform: Boosting Multimodal Reasoning via Information-Gain-Driven Verification
Authors: Shuai Lv, Chang Liu, Feng Tang, Yujie Yuan, Aojun Zhou, Kui Zhang, Xi Yang, Yangqiu Song
Organizations: University of Science and Technology of China, Hefei, China · Huawei Foundation Model Department, Shanghai, China · The Chinese University of Hong Kong, Hong Kong, China · Hong Kong University of Science and Technology, Hong Kong, China
Multimodal Large Language Models (MLLMs) achieve strong multimodal reasoning performance, yet we identify a recurring failure mode in long-form generation: as outputs grow longer, models progressively drift away from image evidence and fall back on textual priors, resulting in ungrounded reasoning and hallucinations. Interestingly, Based on attention analysis, we find that MLLMs have a latent capability for late-stage visual verification that is present but not consistently activated. Motivated by this observation, we propose Visual Re-Examination (VRE), a self-evolving training framework that enables MLLMs to autonomously perform visual introspection during reasoning without additional visual inputs. Rather than distilling visual capabilities from a stronger teacher, VRE promotes iterative self-improvement by leveraging the model itself to generate reflection traces, making visual information actionable through information gain. Extensive experiments across diverse multimodal benchmarks demonstrate that VRE consistently improves reasoning accuracy and perceptual reliability, while substantially reducing hallucinations, especially in long-chain settings. Code is available at https://github.com/Xiaobu-USTC/VRE.
Figures & tables
Figure 1: Overall pipeline of the proposed Visual Reflection Enhancement (VRE) framework. Left: Reflection data construction, where difficulty-aware filtering and information-gain-guided reflection synthesis generate high-quality SFT data. Right: Model optimization, where the cold-start model is iteratively improved through rejection sampling and reinforcement learning under a structured reward scheme comprising format, accuracy, and reflection rewards.
Model
Tool Use
Param Size
MathVista
MathVerse
MathVision
WeMath
LogicVista
MMMU
Closed-Source MLLMs
GPT-4o [ 1 ]
✗
-
63.8
50.8
30.4
69.0
53.2
69.1
Claude 3.7 [ 3 ]
✗
-
66.8
52.0
41.3
72.6
-
-
Open-Source General MLLMs
LLaVA-OneVision [ 25 ]
✗
7B
58.6
19.3
18.3
20.9
33.3
48.8
Qwen2.5-VL [ 5 ]
✗
7B
68.2
49.2
23.7 ∗
62.1
45.9
51.2 ∗
Table 1: Performance on Mathematical and Logical Reasoning Benchmarks. We compare our VRE model with representative closed-source, open-source general, and open-source reasoning MLLMs across mathematical and logical reasoning Benchmarks. ∗ denotes the results are reproduced by ourselves.
Model
Tool Use
Param Size
V∗ -Bench
ChartQA
OCRBench_v2
RealworldQA
Attr
Spatial
Overall
Human
Aug
Overall
en
zh
-
Closed-Source MLLMs
GPT-4o [ 1 ]
✗
-
60.5
67.5
79.5
91.9
85.7
86.7
46.5
32.2
75.5
Open-Source General MLLMs
LLaVA-OneVision [ 25 ]
✗
7B
75.7
75.0
75.4
-
-
80.0
-
-
66.3
InternVL2.5 [ 10 ]
✗
8B
-
-
-
-
-
84.8
49.8
52.1
70.1
Table 2: Performance on High-Resolution Perception, Document Understanding, and Real-world QA. We compare our VRE model with representative closed-source, open-source general, and open-source reasoning MLLMs across high-resolution perception, document understanding, and Real-world QA. ∗ denotes the results are reproduced by ourselves.
Setting
Cold-SFT
RLVR
Iter-SFT
V∗ -Bench
MathVista
MathVerse
ChartQA
OCRBench_v2 (en/zh)
Base Model
✗
✗
✗
76.4
68.2
49.2
83.7
56.3/57.2
SFT Only
✓
✗
✗
80.1
69.2
49.6
87.8
59.6 / 62.3
SFT + RL
✓
✓
✗
83.8
71.0
52.5
88.7
62.1 / 64.3
VRE
✓
✓
✓
84.3
71.2
53.1
88.8
62.6 / 64.7
Table 3: Component Ablation. Performance on select benchmarks when progressively adding VRE components.
Figure 2: Visualizing Implicit Visual Re-Examination via Attention. The heatmaps display attention weights on visual tokens across inferences steps. (Left) The Base Model exhibits visual decay, where attention to the image vanishes as the textual context grows, leading to ungrounded hallucinations. (Right) The VRE Model demonstrates a spontaneous attention resurgence. Inside the <reflection> block, the model sharply re-allocates attention back to the visual features, proving that the mechanism actively triggers a re-examination behavior to extract missing visual evidence.
Figure 3: Visualizing the Re-examination Mechanism: From Blindness to Grounding. Phase 1: Initial Reasoning. During the initial pass, the model’s attention is dispersed over the background, failing to locate the dustpan. Phase 2: Visual Re-examination. Triggered by the reflection token, the model performs an active visual search. The attention map shows a sharp, targeted focus on the dustpan (red region), successfully retrieving the correct visual evidence to fix the answer.
Figure 4: Evolution of Reflection Paradigms. The stacked bar charts detail the distribution of introspection types across training stages.
Appendix figures & tables19 assets
Supplementary material from the paper’s appendix.
Appendix
Data Source
Cold-Start SFT
RLVR
Post-RL SFT
Thyme-SFT [ 64 ]
∼ 80k
-
-
MM-Eureka [ 36 ]
-
∼ 20k
∼ 4k
V∗ Dataset [ 51 ]
-
∼ 20k
∼ 4k
ViRL39K [ 46 ]
-
∼ 20k
∼ 4k
Total Data
∼ 80k
∼ 60k
∼ 12k
Appendix
Table 4: Detailed Composition of Training Data. This table summarizes the data sources and the approximate number of samples in our VRE Framework.
Figure 5: System prompt template for information gains.
Figure 6: Final accuracy evaluation prompt. Dynamic inputs {question} (Question), {ground_truth} (Standard Answer) and {predict_str} (Model_answer) are inserted at inference time.
Figure 7: Prompt Template for Elaborative Reflection. Dynamic inputs {question} (Question), {initial_reasoning} (Initial Reasoning) are inserted at inference time.
Figure 8: Prompt for evaluating whether reflection blocks demonstrate genuine visual re-examination. Dynamic inputs {question} (Question) and {predict_str} (Model_answer) are inserted at inference time.
Figure 9: Prompt Template for Corrective Reflection. Dynamic inputs {question} (Question), {initial_reasoning} (Initial Reasoning) are inserted at inference time.
Figure 10: Prompt Template for Recheck Reflection. Dynamic inputs {question} (Question), {initial_reasoning} (Initial Reasoning) are inserted at inference time.
Figure 11: Scaling Laws in VRE Training. (a) Cold-Start SFT Data Scaling. There is no significant performance improvement after 50k samples, and the performance is approaching saturation. (b) RL Training Step Scaling. The policy achieves optimal reasoning performance at 100 steps. Prolonged training ( >150 steps) induces severe over-optimization and capability degradation.
Models
Evaluation Set
All Correct ( ↑ )
Mixed
All Wrong ( ↓ )
Cold-Start
Full Set
30.8%
56.5%
12.7%
RL Model
Full Set
55.6%
38.4%
6.0%
Post-RL SFT
Full Set
58.1%
36.2%
5.7%
Post-RL SFT
Cold-Start “All Wrong” Only
9.2%
58.9%
31.9%
Appendix
Table 7: Distributions of Rollout Trajectories Across Training Stages. Full Set denotes the complete dataset used for RL optimization. Cold-Start “All Wrong” Only refers to the specific subset of “hard queries” where the Cold-Start model failed across all sampled trajectories. This subset highlights the model’s ability to breakthrough its initial cognitive boundaries.
Figure 12: KL Divergence Analysis on V∗ -Bench. (a) Sample-wise average KL divergence. (b) A specific case study showing token-level KL divergence over inference steps.
Figure 13: Information Entropy Comparison on V∗ -Bench. VRE consistently maintains higher conditional entropy than the Semantic-back model. The shift toward higher entropy values validates that VRE’s reflection mechanism actively retrieves novel, non-redundant visual evidence.
Figure 14: Training Dynamics of GRPO. (a) The reward score steadily increases and converges around step 50, reflecting stable policy optimization. (b) The average response length stabilizes at ∼ 600 tokens, indicating that the model extracts visual evidence efficiently without degenerating into verbose patterns.