通过建模文本间关系与图像掩码,提升场景文字识别的自监督表征能力。
Relational Contrastive Learning and Masked Image Modeling for Scene Text Recognition
- 引入关系重排模块生成新文本关系,增强对比学习的多样性。
- 在多个下游任务上超越现有自监督方法,性能显著提升。
- 适合研究自监督视觉表征与场景文字识别的学者使用。
上下文感知方法通过利用词汇语义先验,在监督式场景文字识别中取得了显著进展。考虑到文字与背景的异质性,本文提出将此类上下文先验重新诠释为文本元素间的关联关系,作为有效的自监督标签用于表示学习。然而,由于词汇依赖性,文本关系受限于数据集规模,易导致过拟合,影响表征质量。为此,本文提出统一框架 RCMSTR(Relational Contrastive Learning and Masked Image Modeling for STR),显式建模丰富文本关系。在 RCL 分支中,首先引入关系重排模块实时生成新关系,并基于此进行层级内与层级间关系的对比学习(针对帧、子词和词)。另一方面,MIM 可通过掩码自然增强上下文信息,实验发现块掩码策略对 STR 更有效。为有效融合 RCL 与 MIM,提出一种新型解耦设计,减轻掩码图像对对比学习的影响。此外,为提升 MIM 与 CNN 的兼容性,采用稀疏卷积并共享稠密卷积权重进行训练。RCMSTR 在多种评估协议下均优于现有自监督 STR 技术,消融实验与定性分析进一步验证其有效性。代码与预训练模型将开源于 https://github.com/ThunderVVV/RCMSTR。
原文摘要 · Abstract (English)
Context-aware methods have achieved remarkable advancements in supervised scene text recognition by leveraging semantic priors from words. Considering the heterogeneity of text and background in STR, we propose that such contextual priors can be reinterpreted as the relations between textual elements, serving as effective self-supervised labels for representation learning. However, textual relations are restricted to the finite size of the dataset due to lexical dependencies, which causes over-fitting problem, thus compromising the representation quality. To address this, our work introduces a unified framework of Relational Contrastive Learning and Masked Image Modeling for STR (RCMSTR), which explicitly models the enriched textual relations. For the RCL branch, we first introduce the relational rearrangement module to cultivate new relations on the fly. Based on this, we further conduct relational contrastive learning to model the intra- and inter-hierarchical relations for frames, sub-words and words. On the other hand, MIM can naturally boost the context information via masking, where we find that the block masking strategy is more effective for STR. For the effective integration of RCL and MIM, we also introduce a novel decoupling design aimed at mitigating the impact of masked images on contrastive learning. Additionally, to enhance the compatibility of MIM with CNNs, we propose the adoption of sparse convolutions and directly sharing the weights with dense convolutions in training. The proposed RCMSTR demonstrates superior performance in various evaluation protocols for different STR-related downstream tasks, outperforming the existing state-of-the-art self-supervised STR techniques. Ablation studies and qualitative experimental results further validate the effectiveness of our method. The code and pre-trained models will be available at https://github.com/ThunderVVV/RCMSTR .
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。