arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2210.03347cs.CLcs.CV
收藏

Pix2Struct:将截图解析作为视觉语言理解的预训练任务

Pix2Struct: Screenshot Parsing as Pretraining for Visual Language Understanding

Google Research(谷歌研究院);University of Cambridge(剑桥大学)

Kenton Lee, Mandar Joshi, Iulia Turc, Hexiang Hu, Fangyu Liu, Julian Eisenschlos, Urvashi Khandelwal, Peter Shaw, Ming-Wei Chang, Kristina Toutanova

AI总结:

本研究提出预训练图像到文本模型Pix2Struct,以网页掩码截图解析为简化HTML为预训练目标,结合可变分辨率输入与灵活的多模态融合方式,首次实现单模型在四个领域9项视觉语言任务中6项达SOTA。

摘要:

视觉情境语言无处不在——其来源涵盖带图表的教科书、含图片与表格的网页,到带按钮和表单的移动应用。或许正是由于这种多样性,以往研究通常依赖特定领域的方案,底层数据、模型架构与目标的共享程度十分有限。我们提出Pix2Struct,这是一款面向纯视觉语言理解的预训练图像到文本模型,可在包含视觉情境语言的任务上进行微调。Pix2Struct的预训练方式是学习将网页的掩码截图解析为简化版HTML。网页拥有丰富的视觉元素,且这些元素能清晰反映在HTML结构中,为预训练提供了大量适配下游任务多样性的数据来源。直观来看,该预训练目标涵盖了OCR(光学字符识别)、语言建模、图像描述等常见预训练信号。除了这种新颖的预训练策略,我们还提出了可变分辨率输入表示,以及更灵活的语言与视觉输入融合方式——即把问题这类语言提示直接渲染在输入图像上方。我们首次证明,单个预训练模型能在文档、插图、用户界面、自然图像四个领域的9项任务中,取得6项的当前最优结果。

英文摘要:

Visually-situated language is ubiquitous -- sources range from textbooks with diagrams to web pages with images and tables, to mobile apps with buttons and forms. Perhaps due to this diversity, previous work has typically relied on domain-specific recipes with limited sharing of the underlying data, model architectures, and objectives. We present Pix2Struct, a pretrained image-to-text model for purely visual language understanding, which can be finetuned on tasks containing visually-situated language. Pix2Struct is pretrained by learning to parse masked screenshots of web pages into simplified HTML. The web, with its richness of visual elements cleanly reflected in the HTML structure, provides a large source of pretraining data well suited to the diversity of downstream tasks. Intuitively, this objective subsumes common pretraining signals such as OCR, language modeling, image captioning. In addition to the novel pretraining strategy, we introduce a variable-resolution input representation and a more flexible integration of language and vision inputs, where language prompts such as questions are rendered directly on top of the input image. For the first time, we show that a single pretrained model can achieve state-of-the-art results in six out of nine tasks across four domains: documents, illustrations, user interfaces, and natural images.

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

↑