arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.08573cs.CV

HTR-VT:基于视觉Transformer的手写文本识别

HTR-VT: Handwritten Text Recognition with Vision Transformer

China Three Gorges University(三峡大学);Max Planck Institute of Biochemistry(马克斯·普朗克生物化学研究所);Intellindust(英特工业科技公司)

Yuting Li, Dexiong Chen, Tinglong Tang, Xi Shen

AI总结:

针对手写文本识别领域标注数据有限的问题,提出仅用Transformer编码器、CNN特征提取结合SAM优化器和跨度掩码的高效ViT方法,在小数据集性能媲美传统CNN模型,还在最大的LAM数据集上刷新基准。

摘要:

我们探索了视觉Transformer(Vision Transformer,ViT)在手写文本识别中的应用。该领域标注数据有限,仅依靠ViT难以实现高性能。此前基于Transformer的模型需要外部数据或在大型数据集上进行大规模预训练才能取得优异效果。为解决这一局限,我们提出了一种数据高效的ViT方法,该方法仅使用标准Transformer的编码器。我们发现,用卷积神经网络(Convolutional Neural Network,CNN)替代原始的补丁嵌入进行特征提取,并采用锐度感知最小化(Sharpness-Aware Minimization,SAM)优化器,可确保模型收敛至更平坦的最小值,带来显著性能提升。此外,我们引入的跨度掩码技术可对特征图中的互联特征进行掩码,是一种有效的正则化手段。实验结果表明,我们的方法在IAM、READ2016等小型数据集上的表现与传统基于CNN的模型相当,并且在LAM数据集(目前最大的数据集,包含19830条训练文本行)上创下了新的基准。代码已公开于:https://github.com/YutingLi0606/HTR-VT。

英文摘要:

We explore the application of Vision Transformer (ViT) for handwritten text recognition. The limited availability of labeled data in this domain poses challenges for achieving high performance solely relying on ViT. Previous transformer-based models required external data or extensive pre-training on large datasets to excel. To address this limitation, we introduce a data-efficient ViT method that uses only the encoder of the standard transformer. We find that incorporating a Convolutional Neural Network (CNN) for feature extraction instead of the original patch embedding and employ Sharpness-Aware Minimization (SAM) optimizer to ensure that the model can converge towards flatter minima and yield notable enhancements. Furthermore, our introduction of the span mask technique, which masks interconnected features in the feature map, acts as an effective regularizer. Empirically, our approach competes favorably with traditional CNN-based models on small datasets like IAM and READ2016. Additionally, it establishes a new benchmark on the LAM dataset, currently the largest dataset with 19,830 training text lines. The code is publicly available at: https://github.com/YutingLi0606/HTR-VT.

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

↑