VLMs benefit from tool calls as they allow for "thinking with images".
But can a VLM simply internalise the effects of calling correct tools in the latent space and not need to explicitly call them during inference? ๐งต
Presenting tomorrow at 11 AM @COLM_conf !
VLMs benefit from tool calls as they allow for "thinking with images".
But can a VLM simply internalise the effects of calling correct tools in the latent space and not need to explicitly call them during inference? ๐งต
Presenting tomorrow at 11 AM @COLM_conf !
With zero tool calls, PEARL improves over SFT and matches complex alternatives (avg. acc., Qwen2.5-VL-7B):
- vs SFT on the same data: 59.7 โ 64.1 (Latent Visual Reasoning), 51.0 โ 64.7 (ThinkMorphData-SFT)
- vs LVR: 63.1 โ 64.1
- vs PixelReasoner, which calls tools: 61.8 โ 63.1
Unlike prior latent methods, it handles multiple tool calls, and the gains carry over to Qwen2.5-VL-3B and Qwen3-VL-4B.
And its predictive embeddings are descriptive: closer to the correct trajectory than to a visually or textually corrupted one.
Inspired by LLM-JEPA, PEARL treats image + question (view 1) and the expert's tool-use trajectory (view 2) as two views to learn predictive embeddings to internalise tool calls.
In order to get descriptive representations of the views we introduce learning to generate hidden representations that are belief state that can aid in avoiding collapse when using jepa-loss.
We also maintain cross-entropy loss on the answer tokens to allow the model to maintain the text-generation capabilities.
Prior approaches to multimodal latent reasoning use reconstruction loss to generate latent tokens that mimic the outputs of the tool calls.
However, we find identify several drawbacks to them:
1. Auto-regressively generating latent tokens doesn't actually generate images owing to training-inference mismatch of number of latent tokens used (most work use only a small fixed number of latent tokens during inference but were trained with longer latent reasoning traces).
2. They are limited to internalising only single-step trajectories, but what about trajectories that contain multiple sequential tool calls?
3. More latents during inference don't help, and they barely tell the right trajectory from a corrupted one! โ ๏ธ
As a result, they aren't reconstructing but simply embedding! So why not just embed directly?
Did you know training on your own incorrect samples can beat distilling from a much larger teacher?
Traditional RFT only reinforces a narrow set of approaches. Strategically sampling for approach-level diversity gives better pass@k, test-time scaling and RL initialization!
https://t.co/pONJCHEO58
๐งต