Learning to Focus and Precise Cropping: A Reinforcement Learning Framework with Information Gaps and Grounding Loss for MLLMs
Organizations: School of Cyber Science and Technology, University of Science and Technology of China · Anhui Province Key Laboratory of Digital Security · Independent Researcher
Abstract
To enhance the perception and reasoning capabilities of multimodal large language models in complex visual scenes, recent research has introduced agent-based workflows. In these works, MLLMs autonomously utilize image cropping tool to analyze regions of interest for question answering. While existing training strategies, such as those employing supervised fine-tuning and reinforcement learning, have made significant progress, our empirical analysis reveals a key limitation. We demonstrate the model's strong reliance on global input and its weak dependence on the details within the cropped region. To address this issue, we propose a novel two-stage reinforcement learning framework that does not require trajectory supervision. In the first stage, we introduce the ``Information Gap" mechanism by adjusting the granularity of the global image. This mechanism trains the model to answer questions by focusing on cropped key regions, driven by the information gain these regions provide. The second stage further enhances cropping precision by incorporating a grounding loss, using a small number of bounding box annotations. Experiments show that our method significantly enhances the model's attention to cropped regions, enabling it to achieve state-of-the-art performance on high-resolution visual question-answering benchmarks. Our method provides a more efficient approach for perceiving and reasoning fine-grained details in MLLMs. Code is available at: https://github.com/XuanPu-Z/LFPC.
Figures & tables
| Model | HR-Bench 8k | HR-Bench 4k | |||||||||||||
| FSP | FCP | Acc | FSP | FCP | Acc | Attr | Spatial | Acc | |||||||
| Acc | IoU | Acc | IoU | Acc | IoU | Acc | IoU | Acc | IoU | Acc | IoU | ||||
| Baseline | 65.2 | 28.5 | 56.8 | 40.1 | 61.0 | 75.0 | 30.3 | 60.8 | 45.6 | 67.9 | 72.2 | 39.1 | 72.4 | 30.3 | 72.3 |
| Stage-I | 77.7 | 29.8 | 62.0 | 39.3 | 69.9 | 85.3 | 32.3 | 63.5 | 46.5 | 74.4 | 80.0 | 25.7 | 75.0 | 21.0 | 78.0 |
| Stage-II | 83.3 | 34.5 | 60.8 | 44.4 | 72.1 | 86.8 | 37.8 | 63.7 | 51.0 | 75.3 | 82.6 | 37.4 | 77.6 | 30.8 | 80.6 |
| Model | HR-Bench 8k | HR-Bench 4k | |||||||
| FSP | FCP | Overall | FSP | FCP | Overall | Attr | Spatial | Overall | |
| Stage-I | 77.7 | 62.0 | 69.9 | 85.3 | 63.5 | 74.4 | 80.0 | 75.0 | 78.0 |
| + Original Data | 76.7 | 59.5 | 68.1 | 82.5 | 62.0 | 72.3 | 79.1 | 75.0 | 77.5 |
| + VP Data | 81.8 | 63.5 | 72.7 | 83.8 | 62.0 | 72.9 | 77.4 | 75.0 | 76.4 |
| + VP Data & L1 Reward | 83.3 | 60.8 | 72.1 | 86.8 | 63.7 | 75.3 | 82.6 | 77.6 | 80.6 |
| Model | HR-Bench 8k | HR-Bench 4k | |
| Baseline | 4.5 | 0.5 | 3.2 |
| Stage-I | 18.5 | 10.5 | 11.0 |
| Model | HR-Bench 8k | HR-Bench 4k | ||
| Acc | Time (s) | Acc | Time (s) | |
| CoF-sft | 69.4 | 6.0 | 73.0 | 5.5 |
| Pixel Reasoner | 66.9 | 5.0 | 73.6 | 8.4 |
| Mini-o3 | 65.6 | 27.8 | 69.0 | 21.4 |
| DeepEyes | 69.5 | 12.4 | 72.9 | 9.3 |
| Ours 1,024 | 72.1 | 2.8 | 75.3 | 2.6 |
| Model | HR-Bench 8k | HR-Bench 4k | |||||||
| FSP | FCP | Overall | FSP | FCP | Overall | Attr | Spatial | Overall | |
| Mix Data | 77.5 | 59.0 | 68.3 | 85.0 | 61.0 | 73.0 | 82.6 | 73.7 | 79.1 |
| Stage-II | 83.3 | 60.8 | 72.1 | 86.8 | 63.7 | 75.3 | 82.6 | 77.6 | 80.6 |
| Model | HR-Bench 8k | HR-Bench 4k | V Star |
| Hard | 60.5 | 63.2 | 71.1 |
| Random | 59.0 | 64.0 | 72.4 |
| Answer | 62.0 | 63.5 | 75.0 |
| Design | FSP | FCP | Acc | ||
| Acc | IoU | Acc | IoU | ||
| IoU Reward | 81.8 | 39.3 | 63.5 | 47.4 | 72.7 |
| - IoU Thresh | 79.0 | 29.3 | 64.2 | 39.0 | 71.6 |
| - IoU Reward | 79.0 | 30.6 | 63.7 | 37.7 | 71.2 |
| Method | Time (A100-hours) | |
| DeepEyes | 480.0 | |
| Ours | Stage-I | 320.0 |
| Stage-II | 96.0 | |
| Offline Res. Selection | 2.5 | |
| Total | 418.5 | |
| Parameter | Value |
| train batch size | 256 |
| rollout num per sample | 16 |
| ppo mini batch size | 32 |
| ppo micro batch size per gpu | 2 |
| rollout log prob micro batch size per gpu | 4 |
| ref log prob micro batch size per gpu | 4 |
Visual Question Answering
Visual Question Answering (VQA) develops systems that answer natural-language questions about images, videos, and visual documents.
See all 253 papers