通过结构重参数化提升手写段落识别的精度与速度
RVAFM: Re-parameterizing Vertical Attention Fusion Module for Handwritten Paragraph Text Recognition
- 训练时用多分支学习,推理时转为单分支加速
- 在IAM数据集上达CER 4.44%、WER 14.37%
- 适合追求高精度与高效推理的手写文本识别场景
手写段落文本识别(HPTR)是计算机视觉中的挑战性任务,需将富含手写文字的段落图像转换为文本编码序列。当前先进模型垂直注意力网络(VAN)采用垂直注意力模块(VAM)隐式分割段落图像为文本行以降低识别难度。然而,VAM为单分支结构,相比多分支结构学习能力较弱。本文提出重参数化垂直注意力融合模块(RVAFM),利用结构重参数化技术,在训练阶段使用多分支结构增强学习能力,推理阶段转为单分支结构提升速度。通过专用融合方法(重参数化融合,RF)将多分支学习特征无损融合至单分支结构。实验表明,在IAM段落级测试集上,该方法达到4.44%的字符错误率(CER)和14.37%的词错误率(WER),且推理速度略优于VAN。
原文摘要 · Abstract (English)
Handwritten Paragraph Text Recognition (HPTR) is a challenging task in Computer Vision, requiring the transformation of a paragraph text image, rich in handwritten text, into text encoding sequences. One of the most advanced models for this task is Vertical Attention Network (VAN), which utilizes a Vertical Attention Module (VAM) to implicitly segment paragraph text images into text lines, thereby reducing the difficulty of the recognition task. However, from a network structure perspective, VAM is a single-branch module, which is less effective in learning compared to multi-branch modules. In this paper, we propose a new module, named Re-parameterizing Vertical Attention Fusion Module (RVAFM), which incorporates structural re-parameterization techniques. RVAFM decouples the structure of the module during training and inference stages. During training, it uses a multi-branch structure for more effective learning, and during inference, it uses a single-branch structure for faster processing. The features learned by the multi-branch structure are fused into the single-branch structure through a special fusion method named Re-parameterization Fusion (RF) without any loss of information. As a result, we achieve a Character Error Rate (CER) of 4.44% and a Word Error Rate (WER) of 14.37% on the IAM paragraph-level test set. Additionally, the inference speed is slightly faster than VAN.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。