arXiv:2509.06415cs.CVcs.AI2025-09中稿 · ICLR被引 1

通过轻量剪枝保留关键文本区域,显著降低文档理解的计算开销。

Index-Preserving Lightweight Token Pruning for Efficient Document Understanding in Vision-Language Models

  • 用二值分类器剔除图像中非文本区域,减少冗余信息
  • 通过最大池化修复断裂文本区域,保持空间连贯性
  • 在真实文档数据集上实现高精度与低计算成本的平衡

视觉语言模型(VLMs)在文档理解任务中取得显著进展,但其高计算需求仍是主要挑战。为缓解计算负担,我们提出一种轻量级标记剪枝框架,在VLM处理前过滤文档图像中的非信息背景区域。该方法采用二值块级分类器移除非文本区域,并通过最大池化精炼步骤恢复断裂的文本区域,提升空间连贯性。在真实文档数据集上的实验表明,该方法显著降低计算成本,同时保持相近的准确率。

原文摘要 · Abstract (English)

Recent progress in vision-language models (VLMs) has led to impressive results in document understanding tasks, but their high computational demands remain a challenge. To mitigate the compute burdens, we propose a lightweight token pruning framework that filters out non-informative background regions from document images prior to VLM processing. A binary patch-level classifier removes non-text areas, and a max-pooling refinement step recovers fragmented text regions to enhance spatial coherence. Experiments on real-world document datasets demonstrate that our approach substantially lowers computational costs, while maintaining comparable accuracy.

文档理解轻量化视觉语言模型

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