Papers
arxiv:2505.18227

Token Reduction Should Go Beyond Efficiency in Generative Models -- From Vision, Language to Multimodality

Published on May 23
ยท Submitted by TonyK on May 29
Authors:
,
,
,
,
,
,
,
,

Abstract

Token reduction in Transformer models, beyond efficiency, enhances multimodal integration, reduces hallucinations, and improves training stability in generative modeling.

AI-generated summary

In Transformer architectures, tokens\textemdash discrete units derived from raw data\textemdash are formed by segmenting inputs into fixed-length chunks. Each token is then mapped to an embedding, enabling parallel attention computations while preserving the input's essential information. Due to the quadratic computational complexity of transformer self-attention mechanisms, token reduction has primarily been used as an efficiency strategy. This is especially true in single vision and language domains, where it helps balance computational costs, memory usage, and inference latency. Despite these advances, this paper argues that token reduction should transcend its traditional efficiency-oriented role in the era of large generative models. Instead, we position it as a fundamental principle in generative modeling, critically influencing both model architecture and broader applications. Specifically, we contend that across vision, language, and multimodal systems, token reduction can: (i) facilitate deeper multimodal integration and alignment, (ii) mitigate "overthinking" and hallucinations, (iii) maintain coherence over long inputs, and (iv) enhance training stability, etc. We reframe token reduction as more than an efficiency measure. By doing so, we outline promising future directions, including algorithm design, reinforcement learning-guided token reduction, token optimization for in-context learning, and broader ML and scientific domains. We highlight its potential to drive new model architectures and learning strategies that improve robustness, increase interpretability, and better align with the objectives of generative modeling.

Community

Paper submitter
โ€ข
edited 3 days ago

Due to the quadratic computational complexity of Transformer self-attention, token reduction has been widely used as an efficiency strategy to balance compute cost, memory usage, and inference latency in vision, language, and multimodal models. However, viewing token reduction purely through the lens of efficiency is fundamentally limiting.

This paper posits token reduction as a core design principle in generative modeling, deeply integrated with both training and inference, to prioritize tokens that enhance downstream performance while preserving semantic integrity.

โญ Explore the Github repo: https://lnkd.in/er6i4pEQ
๐Ÿ“š A detailed list of papers organized by modality can be found in this Google Sheet, including a brief introduction of the task, token reduction type, contribution, and methodology for each paper.

timeline.png

awesome๏ผ๐Ÿ‘๐Ÿ‘๐Ÿ‘

Sign up or log in to comment

Models citing this paper 0

No model linking this paper

Cite arxiv.org/abs/2505.18227 in a model README.md to link it from this page.

Datasets citing this paper 0

No dataset linking this paper

Cite arxiv.org/abs/2505.18227 in a dataset README.md to link it from this page.

Spaces citing this paper 0

No Space linking this paper

Cite arxiv.org/abs/2505.18227 in a Space README.md to link it from this page.

Collections including this paper 0

No Collection including this paper

Add this paper to a collection to link it from this page.