arXiv:2601.10378cs.CVcs.AI2026-01被引 1

用视觉编码压缩文本,实现推理阶段也提速的端到端文字模型

Global Context Compression with Interleaved Vision-Text Transformation

  • 将文本分块转为草图图像,与原文交替输入,仅用视觉信息预测下一个词
  • 4倍压缩下,首词生成快3倍,内存减少77%,计算量降74%
  • 适合长文本生成、低延迟部署场景,尤其对资源受限设备友好

近期视觉语言模型在端到端OCR中的进展表明,可通过视觉编码实现文本信息的低损耗压缩。已有方法将Transformer输入转为图像以减少令牌数量,缓解注意力计算的二次增长问题,但仅在预填充阶段有效,无法降低逐令牌推理时的开销。本文提出全局上下文压缩方案,在预填充和推理阶段均节省令牌。为此设计VIST2:将文本块与其视觉编码交错输入,仅依赖前文视觉令牌预测下一文本令牌分布。通过分阶段训练——从课程调度的光学语言建模预训练,到模态交错指令微调——构建0.6B至8B规模的VIST2系列模型。实验显示,4倍压缩比下,模型在长文本任务上显著优于基线,平均实现首词生成速度提升3倍,内存占用减少77%,计算量(FLOPS)降低74%。代码与数据集将公开,支持后续研究。

原文摘要 · Abstract (English)

Recent achievements of vision-language models in end-to-end OCR point to a new avenue for low-loss compression of textual information. This motivates earlier works that render the Transformer's input into images for prefilling, which effectively reduces the number of tokens through visual encoding, thereby alleviating the quadratically increased Attention computations. However, this partial compression fails to save computational or memory costs at token-by-token inference. In this paper, we investigate global context compression, which saves tokens at both prefilling and inference stages. Consequently, we propose VIST2, a novel Transformer that interleaves input text chunks alongside their visual encoding, while depending exclusively on visual tokens in the pre-context to predict the next text token distribution. Around this idea, we render text chunks into sketch images and train VIST2 in multiple stages, starting from curriculum-scheduled pretraining for optical language modeling, followed by modal-interleaved instruction tuning. We conduct extensive experiments using VIST2 families scaled from 0.6B to 8B to explore the training recipe and hyperparameters. With a 4$\times$ compression ratio, the resulting models demonstrate significant superiority over baselines on long writing tasks, achieving, on average, a 3$\times$ speedup in first-token generation, 77% reduction in memory usage, and 74% reduction in FLOPS. Our codes and datasets will be public to support further studies.

视觉编码文本压缩Transformer优化长文本生成

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