MDiff4STR: Mask Diffusion Model for Scene Text Recognition

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Du, Yongkun, Zhao, Miaomiao, Fan, Songlin, Chen, Zhineng, Jia, Caiyan, Jiang, Yu-Gang
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866912740739645440
author Du, Yongkun
Zhao, Miaomiao
Fan, Songlin
Chen, Zhineng
Jia, Caiyan
Jiang, Yu-Gang
author_facet Du, Yongkun
Zhao, Miaomiao
Fan, Songlin
Chen, Zhineng
Jia, Caiyan
Jiang, Yu-Gang
contents Mask Diffusion Models (MDMs) have recently emerged as a promising alternative to auto-regressive models (ARMs) for vision-language tasks, owing to their flexible balance of efficiency and accuracy. In this paper, for the first time, we introduce MDMs into the Scene Text Recognition (STR) task. We show that vanilla MDM lags behind ARMs in terms of accuracy, although it improves recognition efficiency. To bridge this gap, we propose MDiff4STR, a Mask Diffusion model enhanced with two key improvement strategies tailored for STR. Specifically, we identify two key challenges in applying MDMs to STR: noising gap between training and inference, and overconfident predictions during inference. Both significantly hinder the performance of MDMs. To mitigate the first issue, we develop six noising strategies that better align training with inference behavior. For the second, we propose a token-replacement noise mechanism that provides a non-mask noise type, encouraging the model to reconsider and revise overly confident but incorrect predictions. We conduct extensive evaluations of MDiff4STR on both standard and challenging STR benchmarks, covering diverse scenarios including irregular, artistic, occluded, and Chinese text, as well as whether the use of pretraining. Across these settings, MDiff4STR consistently outperforms popular STR models, surpassing state-of-the-art ARMs in accuracy, while maintaining fast inference with only three denoising steps. Code: https://github.com/Topdu/OpenOCR.
format Preprint
id arxiv_https___arxiv_org_abs_2512_01422
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle MDiff4STR: Mask Diffusion Model for Scene Text Recognition
Du, Yongkun
Zhao, Miaomiao
Fan, Songlin
Chen, Zhineng
Jia, Caiyan
Jiang, Yu-Gang
Computer Vision and Pattern Recognition
Mask Diffusion Models (MDMs) have recently emerged as a promising alternative to auto-regressive models (ARMs) for vision-language tasks, owing to their flexible balance of efficiency and accuracy. In this paper, for the first time, we introduce MDMs into the Scene Text Recognition (STR) task. We show that vanilla MDM lags behind ARMs in terms of accuracy, although it improves recognition efficiency. To bridge this gap, we propose MDiff4STR, a Mask Diffusion model enhanced with two key improvement strategies tailored for STR. Specifically, we identify two key challenges in applying MDMs to STR: noising gap between training and inference, and overconfident predictions during inference. Both significantly hinder the performance of MDMs. To mitigate the first issue, we develop six noising strategies that better align training with inference behavior. For the second, we propose a token-replacement noise mechanism that provides a non-mask noise type, encouraging the model to reconsider and revise overly confident but incorrect predictions. We conduct extensive evaluations of MDiff4STR on both standard and challenging STR benchmarks, covering diverse scenarios including irregular, artistic, occluded, and Chinese text, as well as whether the use of pretraining. Across these settings, MDiff4STR consistently outperforms popular STR models, surpassing state-of-the-art ARMs in accuracy, while maintaining fast inference with only three denoising steps. Code: https://github.com/Topdu/OpenOCR.
title MDiff4STR: Mask Diffusion Model for Scene Text Recognition
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2512.01422