Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Text is Text, No Matter What: Unifying Text Recognition using Knowledge Distillation

About

Text recognition remains a fundamental and extensively researched topic in computer vision, largely owing to its wide array of commercial applications. The challenging nature of the very problem however dictated a fragmentation of research efforts: Scene Text Recognition (STR) that deals with text in everyday scenes, and Handwriting Text Recognition (HTR) that tackles hand-written text. In this paper, for the first time, we argue for their unification -- we aim for a single model that can compete favourably with two separate state-of-the-art STR and HTR models. We first show that cross-utilisation of STR and HTR models trigger significant performance drops due to differences in their inherent challenges. We then tackle their union by introducing a knowledge distillation (KD) based framework. This is however non-trivial, largely due to the variable-length and sequential nature of text sequences, which renders off-the-shelf KD techniques that mostly works with global fixed-length data inadequate. For that, we propose three distillation losses all of which are specifically designed to cope with the aforementioned unique characteristics of text recognition. Empirical evidence suggests that our proposed unified model performs on par with individual models, even surpassing them in certain cases. Ablative studies demonstrate that naive baselines such as a two-stage framework, and domain adaption/generalisation alternatives do not work as well, further verifying the appropriateness of our design.

Ayan Kumar Bhunia, Aneeshan Sain, Pinaki Nath Chowdhury, Yi-Zhe Song• 2021

Related benchmarks

TaskDatasetResultRank
Text RecognitionIIIT, SVT, IC13, IC15, SVTP, CT
IIIT Acc92.3
37
Scene Text RecognitionICDAR 2015
Accuracy (No Lexicon)76.9
35
Scene Text RecognitionICDAR 2013
Accuracy93.3
27
Scene Text RecognitionSVT Perspective (645)
Accuracy84.4
22
Scene Text RecognitionStreet View Text 647
Accuracy89.9
22
Scene Text RecognitionIIIT5K-Words (3000)
Accuracy92.3
22
Scene Text RecognitionCUTE80 288
Accuracy86.3
20
Showing 7 of 7 rows

Other info

Follow for update