arXiv:2409.04095cs.CV2024-09NeurIPS被引 15

一个模型同时搞定图像和文本识别,还能保持原图识能力

UNIT: Unifying Image and Text Recognition in One Vision Encoder

  • 在预训练视觉编码器上加轻量解码器,统一图文识别
  • 在文档任务上显著超越现有方法,自然图像性能不变
  • 无需修改架构,推理部署零成本,适合多模态应用

当前视觉编码器如视觉变换器(ViTs)通常擅长图像识别任务,但无法同时支持文本识别,如同人类视觉系统。为解决这一局限,我们提出UNIT——一种新型训练框架,旨在将图像与文本识别统一于单一模型中。基于已通过图像识别任务预训练的视觉编码器,UNIT引入轻量级语言解码器以预测文本输出,并加入轻量级视觉解码器以防止原始图像编码能力的灾难性遗忘。训练过程分为两个阶段:内部尺度预训练与跨尺度微调。在内部尺度预训练中,UNIT从多尺度输入中学习统一表征,其中图像与文档采用常用分辨率,以建立基础识别能力;在跨尺度微调阶段,模型引入尺度交换数据,即图像与文档在非常用分辨率下的样本,以增强其尺度鲁棒性。值得注意的是,UNIT保留了原始视觉编码器架构,因此在推理与部署方面成本为零。多个基准测试结果表明,该方法在文档相关任务(如OCR、DocQA)上显著优于现有方法,同时保持了在自然图像上的性能,证明其可在不牺牲核心图像识别能力的前提下大幅提升文本识别能力。

原文摘要 · Abstract (English)

Currently, vision encoder models like Vision Transformers (ViTs) typically excel at image recognition tasks but cannot simultaneously support text recognition like human visual recognition. To address this limitation, we propose UNIT, a novel training framework aimed at UNifying Image and Text recognition within a single model. Starting with a vision encoder pre-trained with image recognition tasks, UNIT introduces a lightweight language decoder for predicting text outputs and a lightweight vision decoder to prevent catastrophic forgetting of the original image encoding capabilities. The training process comprises two stages: intra-scale pretraining and inter-scale finetuning. During intra-scale pretraining, UNIT learns unified representations from multi-scale inputs, where images and documents are at their commonly used resolution, to enable fundamental recognition capability. In the inter-scale finetuning stage, the model introduces scale-exchanged data, featuring images and documents at resolutions different from the most commonly used ones, to enhance its scale robustness. Notably, UNIT retains the original vision encoder architecture, making it cost-free in terms of inference and deployment. Experiments across multiple benchmarks confirm that our method significantly outperforms existing methods on document-related tasks (e.g., OCR and DocQA) while maintaining the performances on natural images, demonstrating its ability to substantially enhance text recognition without compromising its core image recognition capabilities.

多模态视觉编码器文本识别OCR

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