Nüwa: Mending the Spatial Integrity Torn by VLM Token Pruning

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Huang, Yihong, Ma, Fei, Shao, Yihua, Guo, Jingcai, Yu, Zitong, Cui, Laizhong, Tian, Qi
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911417811075072
author Huang, Yihong
Ma, Fei
Shao, Yihua
Guo, Jingcai
Yu, Zitong
Cui, Laizhong
Tian, Qi
author_facet Huang, Yihong
Ma, Fei
Shao, Yihua
Guo, Jingcai
Yu, Zitong
Cui, Laizhong
Tian, Qi
contents Vision token pruning has proven to be an effective acceleration technique for the efficient Vision Language Model (VLM). However, existing pruning methods demonstrate excellent performance preservation in visual question answering (VQA) and suffer substantial degradation on visual grounding (VG) tasks. Our analysis of the VLM's processing pipeline reveals that strategies utilizing global semantic similarity and attention scores lose the global spatial reference frame, which is derived from the interactions of tokens' positional information. Motivated by these findings, we propose $\text{Nüwa}$, a two-stage token pruning framework that enables efficient feature aggregation while maintaining spatial integrity. In the first stage, after the vision encoder, we apply three operations, namely separation, alignment, and aggregation, which are inspired by swarm intelligence algorithms to retain information-rich global spatial anchors. In the second stage, within the LLM, we perform text-guided pruning to retain task-relevant visual tokens. Extensive experiments demonstrate that $\text{Nüwa}$ achieves SOTA performance on multiple VQA benchmarks (from 94% to 95%) and yields substantial improvements on visual grounding tasks (from 7% to 47%).
format Preprint
id arxiv_https___arxiv_org_abs_2602_02951
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Nüwa: Mending the Spatial Integrity Torn by VLM Token Pruning
Huang, Yihong
Ma, Fei
Shao, Yihua
Guo, Jingcai
Yu, Zitong
Cui, Laizhong
Tian, Qi
Computer Vision and Pattern Recognition
Artificial Intelligence
Computation and Language
Vision token pruning has proven to be an effective acceleration technique for the efficient Vision Language Model (VLM). However, existing pruning methods demonstrate excellent performance preservation in visual question answering (VQA) and suffer substantial degradation on visual grounding (VG) tasks. Our analysis of the VLM's processing pipeline reveals that strategies utilizing global semantic similarity and attention scores lose the global spatial reference frame, which is derived from the interactions of tokens' positional information. Motivated by these findings, we propose $\text{Nüwa}$, a two-stage token pruning framework that enables efficient feature aggregation while maintaining spatial integrity. In the first stage, after the vision encoder, we apply three operations, namely separation, alignment, and aggregation, which are inspired by swarm intelligence algorithms to retain information-rich global spatial anchors. In the second stage, within the LLM, we perform text-guided pruning to retain task-relevant visual tokens. Extensive experiments demonstrate that $\text{Nüwa}$ achieves SOTA performance on multiple VQA benchmarks (from 94% to 95%) and yields substantial improvements on visual grounding tasks (from 7% to 47%).
title Nüwa: Mending the Spatial Integrity Torn by VLM Token Pruning
topic Computer Vision and Pattern Recognition
Artificial Intelligence
Computation and Language
url https://arxiv.org/abs/2602.02951