NeFut Logo NeFut
Admin Login

[CS.AI] ST-Veto: Spatio-Temporal Token Management for Diffusion MLLMs

Published at: 2026-07-22 22:00 Last updated: 2026-07-23 12:33
#AI #Machine Learning #optimization

Abstract

Vision Language Models (VLMs) achieve strong reasoning with Chain-of-Thought (CoT) prompting but incur high sequential-generation cost, error accumulation, and limited self-correction. Diffusion Multimodal Large Language Models (dMLLMs) unmask tokens in an order-agnostic process, improving efficiency and enabling iterative refinement, yet their reasoning and how to enhance it remain underexplored.

We propose a training-free method, Spatio-Temporal Token Veto (ST-Veto), which leverages the ability to observe all token positions at each diffusion step. Rather than relying only on current-step confidence, ST-Veto vetoes temporally unstable tokens via second-order Taylor prediction of confidence dynamics and filters weakly grounded tokens using image-attention mass, swapping them with safer candidates.

Across multiple dMLLMs and multimodal reasoning benchmarks, ST-Veto consistently outperforms standard decoding policies and prior VLM reasoning methods, improving accuracy by up to 9% with no additional training or generation cost. Analyses show that ST-Veto steers generation toward higher-confidence, better-grounded paths.

Blogger's Review: The ST-Veto method effectively enhances the reasoning efficiency of diffusion multimodal large language models through innovative Taylor prediction and visual grounding mechanisms. It demonstrates the potential to improve model performance without additional training, providing a fresh perspective and methodology for future multimodal learning research, which is worth exploring and applying.

Original Source: https://arxiv.org/abs/2607.17884

[h] Back to Home