Papers
Topics
Authors
Recent
Search
2000 character limit reached

Centrifugal Token Pruning Paradigm

Updated 9 December 2025
  • Centrifugal token pruning is a novel sparsification strategy that selects tokens in a near-to-far spatial order, balancing redundancy reduction with local detail preservation.
  • It employs a Buffering for Spatial Sparsity (BSS) criterion to modulate token similarities with spatial distances, ensuring contiguous token selection and efficient feature aggregation.
  • Empirical implementations like VLM-Pruner and SPViT demonstrate high accuracy retention and resource efficiency in vision-language and transformer models even at high pruning rates.

The centrifugal token pruning paradigm is a model sparsification strategy in which token selection proceeds in a near-to-far (or inside-to-outside) spatial order, fundamentally balancing redundancy reduction with fine-grained spatial coverage. It was proposed to address limitations of traditional importance-based and redundancy-minimizing pruning schemes that overlook local spatial structure in visual transformer models and multi-modal vision-LLMs. The paradigm is characterized by explicit control over the spatial expansion of selected token sets, often through a spatial buffer or weighting mechanism, and often incorporates downstream information fusion from pruned tokens. The centrifugal paradigm has been instantiated by frameworks such as VLM-Pruner for vision-LLMs and the soft-pruning SPViT for Vision Transformers, and demonstrates strong empirical advantages in accuracy-retention and resource efficiency under high pruning rates (Wu et al., 2 Dec 2025, Kong et al., 2021).

1. Paradigm Definition and Foundations

The centrifugal token pruning paradigm aims to select a token subset SS of size RR from the ground set VV (V=N|V|=N) that both maximizes informational diversity (by minimizing representational redundancy among selected tokens) and enforces a progressive, near-to-far spatial order in selection. In formal terms, given reduced hidden representations and learned spatial coordinates pip_i for tokens ii, centrifugal selection penalizes the early selection of spatially distant tokens through a Buffering for Spatial Sparsity (BSS) criterion. The BSS-weighted token similarity is defined as

M~ij=Mij(1+λδˉi(S))\widetilde M_{ij} = M_{ij} \cdot (1 + \lambda\, \bar{\delta}_i(S))

where MijM_{ij} is the cosine similarity in projected hidden space, λ\lambda is a positive buffer weight, and δˉi(S)=minjSDij(sp)Dmax\bar{\delta}_i(S) = \min_{j\in S} \frac{D^{(sp)}_{ij}}{D_{max}} is the normalized spatial distance from candidate RR0 to the nearest selected token, with RR1 the Euclidean distance in token grid coordinates and RR2 the maximal observed grid distance.

The selection maximizes the objective

RR3

where RR4 are current candidates. This objective is approximately submodular; greedy maximization yields an effective approximation (Wu et al., 2 Dec 2025).

2. Algorithmic Implementations

VLM-Pruner

VLM-Pruner realizes the centrifugal paradigm via a training-free, three-stage algorithm:

  1. Pivot Initialization: Select RR5 spatially and semantically diverse initial pivots in key space using a max-min strategy.
  2. Buffered Parallel Greedy Selection: Iteratively add tokens to RR6 with high non-duplication scores and low BSS-modulated similarities to RR7, processing candidates in batches and annealing selection thresholds. Candidate similarities are recalculated with the spatial buffer, prioritizing contiguous local expansion and penalizing early selection of remote tokens.
  3. Similarity-Weighted Aggregation: After selection, for each discarded token RR8, identify the most similar retained token RR9, cluster discarded tokens accordingly, and aggregate their hidden features back into the kept tokens using normalized similarity weights and a fusion coefficient VV0 (typically 0.3 for aggregation, 0.7 for preservation of the original feature).

The computational complexity per image includes VV1 channel screening, VV2 similarity matrix computation, VV3 spatial distance computation, and VV4 post-selection aggregation (Wu et al., 2 Dec 2025).

SPViT

SPViT applies a dynamic, attention-based, multi-head token selector as part of a soft-pruning pipeline:

  • Per-Head Scoring: For input VV5, each head VV6 encodes local/global features, producing soft keep/prune probabilities via scoring MLPs.
  • Head-Weighted Fusion: Final keep/prune score per token is an attention-weighted average across heads.
  • Soft Pruning/Package Token: Instead of hard removal, dropped tokens' features are aggregated into a "package token" by weighted averaging, ensuring residual information is retained and available in downstream layers.
  • Latency-Aware Training: Training incorporates a latency-sparsity loss and explicit per-block latency lookup to enforce target resource constraints, following a progressive, layer-to-phase selector scheduling for optimal accuracy-speed trade-off (Kong et al., 2021).

3. Buffering for Spatial Sparsity (BSS)

The BSS criterion serves as the central mechanism for spatial control in centrifugal pruning. By modulating hidden-state similarity with normalized spatial distance, BSS defers the selection of spatially distant tokens until local regions are sufficiently represented. Formally, for each candidate VV7, the normalized spatial distance to VV8 is computed as

VV9

The BSS-modulated similarity V=N|V|=N0 amplifies similarity for distant candidates, biasing the greedy selection towards spatially proximate tokens and enforcing an inside-out growth pattern. As V=N|V|=N1, BSS reduces to classical redundancy-based selection; higher V=N|V|=N2 enforces stronger near-to-far expansion, improving local detail retention at the expense of global diversity (Wu et al., 2 Dec 2025).

4. Comparison with Alternative Pruning Strategies

The centrifugal paradigm contrasts with token importance-based and classical redundancy-aware pruning (e.g., top-V=N|V|=N3 importance or min-sum diversity). Importance-only schemes disregard inter-token redundancy, potentially retaining many spatially redundant tokens, while non-spatial diversity methods can scatter selections over the input, missing fine object details. Centrifugal buffering provides explicit coverage control, balancing diversity with local completeness.

Soft pruning frameworks such as SPViT incorporate a centrifugal progression by adaptively increasing pruning across subsequent layers or blocks. Their soft aggregation mechanisms, such as the "package token," complement the centrifugal principle by ensuring that even pruned information influences final representations, further mitigating information loss under high sparsity (Kong et al., 2021).

5. Empirical Results and Efficiency

Empirical evaluations demonstrate that centrifugal token pruning achieves high sparsity with minimal performance degradation. For VLM-Pruner, with 88.9% pruning (keeping V=N|V|=N4 of V=N|V|=N5 tokens), the system retains 95.61% of full-model accuracy on LLaVA-1.5-7B across 9 benchmarks, outperforming DivPrune, DART, and FastV, which achieve 93–94% retention. Computationally, VLM-Pruner achieves up to 1.4V=N|V|=N6 end-to-end speedup and reduces FLOPs by V=N|V|=N778% at this sparsity level. Metrics for GQA, POPE, OCRBench, and OK-VQA exhibit superior or comparable performance at identical pruning rates. Similar trends are observed across larger-scale vision-language and video-oriented models (Wu et al., 2 Dec 2025).

SPViT exhibits complementary results on standard vision transformer backbones. For DeiT-T, SPViT reduces GFLOPs by 31%, mobile-CPU latency by 41%, and FPGA inference time by 36%, while incurring only a 0.1% top-1 accuracy loss on ImageNet (Kong et al., 2021).

Model Pruning Rate Retain Acc. Speedup Backbone
VLM-Pruner (LLaVA) 88.9% 95.61% 1.4× LLaVA-1.5-7B
SPViT (DeiT-T) ~31–43% V=N|V|=N80.5% loss 1.4–1.6× DeiT-T/S, Swin-T

6. Applications and Deployment Considerations

Centrifugal token pruning is particularly suited to vision-LLMs, vision transformers, and edge deployment scenarios where computational resource constraints are acute. The approach enables real-time inference by reducing the number of tokens processed at each transformer stage without sacrificing critical spatial detail, essential for fine-grained recognition and localization tasks.

Runtime efficiencies are achieved via straightforward matrix operations without requiring unsupported operators (e.g., fast top-V=N|V|=N9 or complex sorting), facilitating implementation on mobile devices and FPGAs. Selector and aggregation overheads remain negligible compared to core transformer computation. Parameter configurations for image-based VLMs typically use pip_i0 tokens (pip_i1), pip_i2 principal channels, buffer coefficient pip_i3, and fusion coefficient pip_i4 (Wu et al., 2 Dec 2025, Kong et al., 2021).

7. Limitations, Extensions, and Future Directions

While centrifugal token pruning achieves strong results at high sparsity, coverage-vs-detail trade-offs remain. Higher buffer strengths (pip_i5) improve local detail but may marginally lower global coverage metrics (e.g., POPE F1). Extensions integrating spatial and semantic constraints, adaptive stage-wise aggregation, or learning-based selection/fusion criteria represent plausible directions for improved flexibility. Broadening applicability to non-visual modalities where spatial locality or analogous structured relationships exist is also supported by the generalized definition of BSS-weighted similarity.

The empirical coverage gaps of alternative pruning methods under severe sparsity, as exposed in head-to-head comparisons, suggest that centrifugal selection, with explicit spatial buffering and downstream fusion, sets a new state of the art for high-efficiency transformer inference in both unimodal and multimodal domains (Wu et al., 2 Dec 2025, Kong et al., 2021).

Topic to Video (Beta)

No one has generated a video about this topic yet.

Whiteboard

No one has generated a whiteboard explanation for this topic yet.

Follow Topic

Get notified by email when new papers are published related to Centrifugal Token Pruning Paradigm.