When Less is Enough: Adaptive Token Reduction for Efficient Image Representation

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Allakhverdov, Eduard, Goncharova, Elizaveta, Kuznetsov, Andrey
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866917964450627584
author Allakhverdov, Eduard
Goncharova, Elizaveta
Kuznetsov, Andrey
author_facet Allakhverdov, Eduard
Goncharova, Elizaveta
Kuznetsov, Andrey
contents Vision encoders typically generate a large number of visual tokens, providing information-rich representations but significantly increasing computational demands. This raises the question of whether all generated tokens are equally valuable or if some of them can be discarded to reduce computational costs without compromising quality. In this paper, we introduce a new method for determining feature utility based on the idea that less valuable features can be reconstructed from more valuable ones. We implement this concept by integrating an autoencoder with a Gumbel-Softmax selection mechanism, that allows identifying and retaining only the most informative visual tokens. To validate our approach, we compared the performance of the LLaVA-NeXT model, using features selected by our method with randomly selected features. We found that on OCR-based tasks, more than 50% of the visual context can be removed with minimal performance loss, whereas randomly discarding the same proportion of features significantly affects the model capabilities. Furthermore, in general-domain tasks, even randomly retaining only 30% of tokens achieves performance comparable to using the full set of visual tokens. Our results highlight a promising direction towards adaptive and efficient multimodal pruning that facilitates scalable and low-overhead inference without compromising performance.
format Preprint
id arxiv_https___arxiv_org_abs_2503_16660
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle When Less is Enough: Adaptive Token Reduction for Efficient Image Representation
Allakhverdov, Eduard
Goncharova, Elizaveta
Kuznetsov, Andrey
Computer Vision and Pattern Recognition
68T10, 68T30, 68T45
I.2.10
Vision encoders typically generate a large number of visual tokens, providing information-rich representations but significantly increasing computational demands. This raises the question of whether all generated tokens are equally valuable or if some of them can be discarded to reduce computational costs without compromising quality. In this paper, we introduce a new method for determining feature utility based on the idea that less valuable features can be reconstructed from more valuable ones. We implement this concept by integrating an autoencoder with a Gumbel-Softmax selection mechanism, that allows identifying and retaining only the most informative visual tokens. To validate our approach, we compared the performance of the LLaVA-NeXT model, using features selected by our method with randomly selected features. We found that on OCR-based tasks, more than 50% of the visual context can be removed with minimal performance loss, whereas randomly discarding the same proportion of features significantly affects the model capabilities. Furthermore, in general-domain tasks, even randomly retaining only 30% of tokens achieves performance comparable to using the full set of visual tokens. Our results highlight a promising direction towards adaptive and efficient multimodal pruning that facilitates scalable and low-overhead inference without compromising performance.
title When Less is Enough: Adaptive Token Reduction for Efficient Image Representation
topic Computer Vision and Pattern Recognition
68T10, 68T30, 68T45
I.2.10
url https://arxiv.org/abs/2503.16660