Think, Look, and Revise: Inconsistency-Aware Visual Self-Correction in MLLMs
Organizations: University of Edinburgh · NVIDIA Research
Abstract
Tool-augmented multimodal reasoning integrates external tools (e.g., object detection, depth estimation) into multimodal large language models (MLLMs) to address perceptual bottlenecks in complex visual tasks. However, existing approaches rarely verify tool outputs, limiting their ability to detect and recover from tool failures. We propose ReVISE, a framework that equips MLLMs with verification and dynamic error recovery for tool-augmented reasoning. ReVISE introduces (1) a curated training dataset that supervises reflective behaviors, enabling models to validate tool-derived evidence, reformulate queries when visual mismatches arise, and fall back to intrinsic grounding when external tools are unreliable; and (2) a reinforcement learning based targeted rewards that encourage internal reflection and penalize spatial misalignment. Experiments on several benchmarks demonstrate consistent improvements over existing methods, highlighting the importance of error detection and correction in tool-augmented multimodal reasoning.
Figures & tables
| Components | CountBench | CVBench | BLINK (sub) | MMVP | MMSTAR | BLINK-HARD | MMMU | MMBENCH | MathVista |
| REVPT Zhou et al. (2025) Cold-Start w/o RL* | 70.84 | 74.89 | 61.17 | 58.42 | 52.61 | 53.78 | 42.92 | 78.45 | 53.09 |
| Ours w/o RL | 72.41 | 77.89 | 61.57 | 60.48 | 47.53 | 54.67 | 42.46 | 78.62 | 53.21 |
| Ours w/o , | 73.15 | 79.24 | 64.32 | 62.85 | 49.60 | 56.80 | 44.10 | 80.50 | 54.10 |
| Ours w/o | 73.68 | 81.15 | 71.85 | 67.30 | 54.10 | 61.20 | 46.30 | 83.20 | 54.80 |
| Ours w/o | 74.45 | 82.10 | 67.15 | 64.90 | 51.50 | 58.60 | 45.20 | 81.90 | 54.50 |
| Ours | 74.92 | 83.83 | 74.28 | 69.41 | 56.23 | 63.89 | 47.31 | 85.74 | 55.37 |
Appendix figures & tables12 assets
Supplementary material from the paper’s appendix.
Appendix
| Param Name | SFT | RL |
| bf16 / tf32 | True | True |
| per_device_batch | 8 | 8 |
| grad_acc_steps | 2 | – |
| ppo_mini_batch | – | 128 |
| num_generation | – | 8 |
| kl_loss_coef | – | 1e-3 |
| System Prompt for Ground-Truth Bounding Box Generation (RL Reward) |
| You are an expert data annotator for Visual Question Answering. |
| Your goal is to identify and localize ALL objects that are necessary to reason about the user’s Question and arrive at the Correct Answer. |
| Rules for Object Selection: |
| 1. Identify objects explicitly mentioned in the Question (e.g., ’Where is the [chair] relative to the [table]?’). |
| 2. Identify objects mentioned in the Answer or Options . |
| 3. If the question involves comparison (e.g., ’closer’, ’left of’, ’bigger’), you MUST localize both objects involved in the comparison. |
| System Prompt for the ReVISE Agent (RL Training Phase) |
| You are a specialized multimodal agent. Your purpose is to solve visual question answering tasks by thinking step-by-step and using tools. |
| # Tools |
| You are provided with following tools: |
| 1. object_detection : Detects objects in an image. parameters: image_id, objects (list of object names). |
| 2. zoom_in : Zooms in on a specified bounding box in an image. parameters: image_id, bbox (bounding box coordinates), factor (zoom factor). |
| 3. edge_detection : Detects edges in an image. parameters: image_id. |
| Setting | CountBench | CVBench | BLINK-HARD | Avg. |
| w/o grounding reward | 73.68 | 81.15 | 61.20 | 72.01 |
| ReVISE-3B | 74.92 | 83.83 | 63.89 | 74.21 |
| 5% jitter | 74.30 (-0.62) | 83.08 (-0.75) | 63.37 (-0.52) | 73.58 (-0.63) |
| 10% jitter | 71.27 (-3.65) | 79.76 (-4.07) | 60.52 (-3.37) | 70.52 (-3.69) |
| Setting | CVBench | MMMU | MathVista |
| Qwen2.5-VL-3B-Instruct | 72.55 | 49.00 | 61.20 |
| ReVISE-3B | 83.83 | 47.31 | 55.37 |
| ReVISE-3B + mixed-domain | 83.97 (+0.14) | 49.02 (+1.71) | 59.72 (+4.35) |
| Setting | CountBench | CVBench |
| RL-only w/o cold-start SFT | 71.98 | 75.34 |
| ReVISE-3B w/o RL (SFT only) | 72.41 | 77.89 |
| ReVISE-3B (SFT + RL) | 74.92 | 83.83 |