arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2409.04095cs.CV
收藏

UNIT:在一个视觉编码器中统一图像与文本识别

UNIT: Unifying Image and Text Recognition in One Vision Encoder

Hong Kong University of Science and Technology(香港科技大学)

Yi Zhu, Yanpeng Zhou, Chunwei Wang, Yang Cao, Jianhua Han, Lu Hou, Hang Xu

AI总结:

本文提出 UNIT 框架,通过轻量语言/视觉解码器与两阶段多尺度训练,在不改变原视觉编码器推理成本的前提下统一图像与文本识别。

摘要:

当前,Vision Transformers(ViTs)等视觉编码器模型通常擅长图像识别任务,却无法像人类视觉识别那样同时支持文本识别。为解决这一局限,我们提出 UNIT,这是一种旨在在单一模型内统一图像与文本识别的新型训练框架。UNIT 从一个经过图像识别任务预训练的视觉编码器出发,引入一个用于预测文本输出的轻量级语言解码器,以及一个用于防止原始图像编码能力发生灾难性遗忘的轻量级视觉解码器。训练过程包括两个阶段:尺度内预训练和尺度间微调。在尺度内预训练期间,UNIT 从多尺度输入中学习统一表示,其中图像和文档均采用其常用分辨率,以获得基础识别能力。在尺度间微调阶段,模型引入尺度交换数据,其特点是图像和文档采用不同于最常用分辨率的尺度,以增强其尺度鲁棒性。值得注意的是,UNIT 保留了原始视觉编码器架构,因此在推理和部署方面无需额外成本。在多个基准上的实验证实,我们的方法在文档相关任务(例如 OCR 和 DocQA)上显著优于现有方法,同时保持在自然图像上的性能,表明其能够在不损害核心图像识别能力的情况下大幅增强文本识别能力。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑