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

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Kong, Zhenglun, Li, Yize, Zeng, Fanhu, Xin, Lei, Messica, Shvat, Lin, Xue, Zhao, Pu, Kellis, Manolis, Tang, Hao, Zitnik, Marinka
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866912819121750016
author Kong, Zhenglun
Li, Yize
Zeng, Fanhu
Xin, Lei
Messica, Shvat
Lin, Xue
Zhao, Pu
Kellis, Manolis
Tang, Hao
Zitnik, Marinka
author_facet Kong, Zhenglun
Li, Yize
Zeng, Fanhu
Xin, Lei
Messica, Shvat
Lin, Xue
Zhao, Pu
Kellis, Manolis
Tang, Hao
Zitnik, Marinka
contents 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, agentic framework design, and broader ML and scientific domains.
format Preprint
id arxiv_https___arxiv_org_abs_2505_18227
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Token Reduction Should Go Beyond Efficiency in Generative Models -- From Vision, Language to Multimodality
Kong, Zhenglun
Li, Yize
Zeng, Fanhu
Xin, Lei
Messica, Shvat
Lin, Xue
Zhao, Pu
Kellis, Manolis
Tang, Hao
Zitnik, Marinka
Machine Learning
Artificial Intelligence
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, agentic framework design, and broader ML and scientific domains.
title Token Reduction Should Go Beyond Efficiency in Generative Models -- From Vision, Language to Multimodality
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2505.18227