š§ Remember-R1: Our fix for MLLMs forgetting the image during long reasoning
We noticed a frustrating problem: when multimodal models reason over long chains, they gradually stop looking at the imageāand start hallucinating based on their own text.
So we built RememberāR1, a simple RL framework that directly supervises visual attention on the original reasoning trajectoryāno inference overhead, no proxy tasks.
We use three complementary rewards: coverage, persistence, and focus. They encourage the model to keep attending to relevant visual evidence even in later reasoning steps.
Results across 7 benchmarks and 2 model sizes: better reasoning, andāmore importantlyāvisual attention decays much more slowly during generation.
No extra cost at inference, just cleaner supervision where it counts.
š§ Remember-R1: Our fix for MLLMs forgetting the image during long reasoning
We noticed a frustrating problem: when multimodal models reason over long chains, they gradually stop looking at the imageāand start hallucinating based on their own text.
So we built RememberāR1, a simple RL framework that directly supervises visual attention on the original reasoning trajectoryāno inference overhead, no proxy tasks.
We use three complementary rewards: coverage, persistence, and focus. They encourage the model to keep attending to relevant visual evidence even in later reasoning steps.
Results across 7 benchmarks and 2 model sizes: better reasoning, andāmore importantlyāvisual attention decays much more slowly during generation.
No extra cost at inference, just cleaner supervision where it counts.
IQA-T1: EvidenceāBased Image Quality Assessment with MLLMs
Most MLLMs are blind to lowālevel degradationsānoise, blur, compression artifacts look the same as clean images in their internal representations. That leads to quality scores based on semantic āgut feelingā rather than real perceptual evidence.
IQA-T1 changes that. We equip the model with a toolbox of 15 perceptual tools (noise residual maps, Fourier spectra, gradient maps, etc.) that generate structured visual evidence on demand. The model learns how to use tools via supervised fineātuning on our QāTool dataset (11k evidenceāgrounded reasoning chains), and when to call them via GRPO reinforcement learning that balances accuracy, tool count, and redundancy.
The result: SOTA performance across 7 benchmarks (avg PLCC 0.795), using only 2.34 tools per image on average. Every predicted score is now interpretable and backed by hard visual evidence.
All code, weights, dataset, and demo are open. Check them out and give it a spin!