Attending to Multimodal Generation One Token at a Time
Summary
This paper investigates token-level attention shifts in multimodal large language models during generation, revealing consistent patterns and proposing a simple test-time intervention that significantly improves task performance.
View Cached Full Text
Cached at: 07/09/26, 07:52 AM
Paper page - Attending to Multimodal Generation One Token at a Time
Source: https://huggingface.co/papers/2607.03738
Abstract
Multimodal large language models exhibit distinct attention patterns during generation, with attention to visual and textual modalities shifting based on semantic requirements, and these patterns can be leveraged to improve task performance through targeted interventions.
Multimodal large language models(MLLMs) generate responses autoregressively, integrating visual and linguistic information in an evolving context. Prior work on interpretability has focused on individual layers and circuits (where), leaving the token-level dynamics of multimodal computation during generation (when) underexplored. We address this gap and studyattention shiftsas persemantic role; tracking model attention to image, text, instruction, and previously generated tokens, One Token at a Time (OTaT). We introduce multimodal tasks that require explicit switching between visual and textual context within a single response. Across two mainstream model families and four open-weight MLLMs of varying sizes, we establish consistent patterns: attention to image peaks at tokens requiring image-derived information, instruction tokens are revisited during task transitions, and attention to previously generated tokens increases as the generation progresses.Causal attention blockinginterventions validate the functional role of these trends. We profile model behavior under disrupted attention and observe responses falling back to language priors, or exhibitingcross-modal leakage, denial, or recovery. Finally, informed of the attention dynamics through our novel analysis, we propose a simpletest-time interventionto boost attention to the relevant modality at the right time, significantly improving multimodal task performance.
View arXiv pageView PDFProject pageAdd to collection
Get this paper in your agent:
hf papers read 2607\.03738
Don’t have the latest CLI?curl \-LsSf https://hf\.co/cli/install\.sh \| bash
Models citing this paper0
No model linking this paper
Cite arxiv.org/abs/2607.03738 in a model README.md to link it from this page.
Datasets citing this paper0
No dataset linking this paper
Cite arxiv.org/abs/2607.03738 in a dataset README.md to link it from this page.
Spaces citing this paper0
No Space linking this paper
Cite arxiv.org/abs/2607.03738 in a Space README.md to link it from this page.
Collections including this paper0
No Collection including this paper
Add this paper to acollectionto link it from this page.
Similar Articles
Can Multimodal Large Language Models Understand OCT?
This paper introduces OCT-Bench, a comprehensive benchmark for evaluating multimodal large language models on OCT image understanding, comprising 10,076 questions across 20 fine-grained tasks in perception, cognition, and reasoning. Evaluation of 20 models shows current MLLMs fall short of reliable OCT understanding.
Squeezing Capacity from Multimodal Large Language Models for Subject-driven Generation
This paper proposes a novel approach that conditions diffusion models on Multimodal Large Language Models (MLLMs) for subject-driven image generation, using VAE-based identity conditioning and a Dual Layer Aggregation module to improve both semantic understanding and identity preservation while mitigating copy-paste artifacts.
OmniFocus: Query-Guided Modality-Balanced Token Compression for Omni-Modal Large Language Models
OmniFocus is a training-free, query-guided token compression method for omni-modal LLMs that independently estimates importance for video and audio to preserve modality-specific evidence while maintaining alignment, achieving significant inference speedups with minimal accuracy loss at low token retention ratios.
BitLM: Unlocking Multi-Token Language Generation with Bitwise Continuous Diffusion
This paper introduces BitLM, a language model that uses bitwise continuous diffusion to generate multiple tokens in parallel, aiming to overcome the sequential bottleneck of traditional autoregressive generation while preserving causal structure.
Token maxxing
Discusses strategies and techniques for maximizing token usage in large language models to improve efficiency and output quality.