DREAM-R: Multimodal Speculative Reasoning with RL-Based Refined Drafting, Precise Verification, and Fully Parallel Execution
Abstract
Lay Summary
Large multimodal models are powerful at complex reasoning tasks but often generate responses slowly due to expensive step-by-step reasoning. Speculative reasoning accelerates this process by using a smaller draft model to propose reasoning steps that are later verified by a stronger target model. However, existing approaches often suffer from mismatches between draft and target reasoning, leading to unstable verification and error propagation. We propose DREAM-R, a framework that improves both the efficiency and reliability of speculative reasoning. DREAM-R trains draft models with a reinforcement learning objective to better align speculative reasoning with target-model behavior, introduces a stable threshold-based verification mechanism to reduce incorrect reasoning acceptance, and parallelizes drafting and verification across multiple reasoning steps. Experiments on challenging multimodal reasoning benchmarks show that DREAM-R achieves up to 2.49× speedup while preserving target-model accuracy and reasoning quality.