4 个月前

StrucTexTv2:文档图像预训练中的掩码视觉-文本预测

StrucTexTv2:文档图像预训练中的掩码视觉-文本预测

摘要

本文介绍了StrucTexTv2,一种通过执行掩码视觉-文本预测的有效文档图像预训练框架。该框架包含两个自监督预训练任务:基于文本区域级图像掩码的掩码图像建模和掩码语言建模。所提出的方法根据文本单词的边界框坐标随机掩码一些图像区域。我们的预训练任务的目标是同时重建被掩码的图像区域的像素及其对应的被掩码的标记。因此,与通常预测被掩码图像块的掩码图像建模相比,预训练编码器可以捕捉更多的文本语义。与依赖图像和文本两种模态的文档图像理解多模态掩码建模方法相比,StrucTexTv2仅对纯图像输入进行建模,并且在无需OCR预处理的情况下,能够应对更多应用场景。在主流文档图像理解基准上的广泛实验表明了StrucTexTv2的有效性。在端到端场景下,它在各种下游任务中(如图像分类、布局分析、表格结构识别、文档OCR和信息提取)取得了具有竞争力甚至最新的最先进性能。

基准测试

基准方法指标
document-image-classification-on-rvl-cdipStrucTexTv2 (small)
Accuracy: 93.4%
Parameters: 28M
document-image-classification-on-rvl-cdipStrucTexTv2 (large)
Accuracy: 94.62%
Parameters: 238M
semantic-entity-labeling-on-funsdStrucTexTv2 (large)
F1: 91.82
semantic-entity-labeling-on-funsdStrucTexTv2 (small)
F1: 89.23
table-recognition-on-wtwStrucTexTv2 (small)
F1: 78.9%

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
StrucTexTv2:文档图像预训练中的掩码视觉-文本预测 | 论文 | HyperAI超神经