arXiv:2607.02237cs.CV2026-07中稿 · ECCV

对比压缩与剪枝,发现剪枝在高压缩下更稳定。

When Token Compression Breaks: Structural Pruning vs. Token Reduction for Robust ViT Segmentation under High Compression

论文配图:When Token Compression Breaks: Structural Pruning vs. Token Reduction for Robust ViT Segmentation under High Compression
图 1 · 摘自论文原文
  • 用剪枝+轻度压缩的组合提升模型鲁棒性
  • 高压缩时剪枝比删令牌更少丢信息
  • 适合部署时需兼顾精度与抗干扰的场景

视觉变换器(ViT)是语义分割的强大主干网络,但其计算成本限制了实际部署。近期的令牌压缩方法通过减少令牌数量来降低计算开销,但现有评估多集中于低到中等压缩率,对极端压缩和受污染输入下的表现尚不明确。结构化剪枝则通过移除冗余模块实现效率提升,但很少与令牌压缩在统一框架下比较。本文在ADE20K和Cityscapes及其噪声变体ADE20K-C、Cityscapes-C上,对代表性令牌压缩与结构化剪枝方法在匹配FLOPs条件下进行基准测试。结果表明,在干净和受污染输入下,令牌压缩在轻度压缩时有效,但过度压缩导致性能急剧下降,源于信息严重丢失;而结构化剪枝呈现更平滑的退化曲线,高压缩下更稳定。受此启发,我们提出先剪枝再轻度合并的流水线,相同FLOPs下显著提升精度-鲁棒性权衡,为部署导向的ViT分割提供实用方案。代码已开源。

原文摘要 · Abstract (English)

Vision Transformers (ViTs) are strong backbones for semantic segmentation, but their computational cost limits deployment. Recent token compression methods for efficient transformer-based segmentation reduce this cost by decreasing the number of tokens. However, existing evaluations primarily focus on low-to-moderate compression, leaving their behavior under aggressive compression and corrupted inputs unclear. Meanwhile, structural pruning provides an orthogonal route to efficiency by removing redundant components in the ViT architecture, but is rarely compared to token compression under a unified protocol. To bridge this gap, we benchmark representative token compression and structural pruning methods for ViT-based semantic segmentation under matched FLOPs on ADE20K and Cityscapes, together with their common-corruption variants ADE20K-C and Cityscapes-C. Our results reveal a consistent trend on both clean and corrupted inputs: token compression is highly effective at mild reductions but degrades sharply when compression becomes severe, consistent with substantial information loss from overly aggressive token reduction. In contrast, structural pruning exhibits a smoother degradation curve and is more stable at high compression. Motivated by these findings, we study a prune-then-merge pipeline that applies moderate token compression on top of a moderately pruned backbone. At comparable FLOPs, this combined strategy consistently achieves a better accuracy-robustness trade-off at high compression, offering a practical recipe for deployment-oriented ViT segmentation. Code is available at https://github.com/phatnguyencs/vit-seg-compression.

ViT剪枝压缩分割

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。