HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

Self-supervised Character-to-Character Distillation for Text Recognition

Tongkun Guan; Wei Shen; Xue Yang; Qi Feng; Zekun Jiang; Xiaokang Yang

Self-supervised Character-to-Character Distillation for Text Recognition

Abstract

When handling complicated text images (e.g., irregular structures, low resolution, heavy occlusion, and uneven illumination), existing supervised text recognition methods are data-hungry. Although these methods employ large-scale synthetic text images to reduce the dependence on annotated real images, the domain gap still limits the recognition performance. Therefore, exploring the robust text feature representations on unlabeled real images by self-supervised learning is a good solution. However, existing self-supervised text recognition methods conduct sequence-to-sequence representation learning by roughly splitting the visual features along the horizontal axis, which limits the flexibility of the augmentations, as large geometric-based augmentations may lead to sequence-to-sequence feature inconsistency. Motivated by this, we propose a novel self-supervised Character-to-Character Distillation method, CCD, which enables versatile augmentations to facilitate general text representation learning. Specifically, we delineate the character structures of unlabeled real images by designing a self-supervised character segmentation module. Following this, CCD easily enriches the diversity of local characters while keeping their pairwise alignment under flexible augmentations, using the transformation matrix between two augmented views from images. Experiments demonstrate that CCD achieves state-of-the-art results, with average performance gains of 1.38% in text recognition, 1.7% in text segmentation, 0.24 dB (PSNR) and 0.0321 (SSIM) in text super-resolution. Code is available at https://github.com/TongkunGuan/CCD.

Code Repositories

tongkunguan/ccd
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
scene-text-recognition-on-cute80CCD-ViT-Base(ARD_2.8M)
Accuracy: 98.3
scene-text-recognition-on-cute80CCD-ViT-Small(ARD_2.8M)
Accuracy: 98.3
scene-text-recognition-on-cute80CCD-ViT-Tiny(ARD_2.8M)
Accuracy: 95.8
scene-text-recognition-on-hostCCD-ViT-Base
1:1 Accuracy: 77.3
scene-text-recognition-on-icdar2013CCD-ViT-Tiny(ARD_2.8M)
Accuracy: 97.5
scene-text-recognition-on-icdar2013CCD-ViT-Base(ARD_2.8M)
Accuracy: 98.3
scene-text-recognition-on-icdar2013CCD-ViT-Small(ARD_2.8M)
Accuracy: 98.3
scene-text-recognition-on-iiit5kCCD-ViT-Small(ARD_2.8M)
Accuracy: 98.0
scene-text-recognition-on-iiit5kCCD-ViT-Tiny(ARD_2.8M)
Accuracy: 97.1
scene-text-recognition-on-iiit5kCCD-ViT-Base(ARD_2.8M)
Accuracy: 98.0
scene-text-recognition-on-svtCCD-ViT-Base(ARD_2.8M)
Accuracy: 97.8
scene-text-recognition-on-svtCCD-ViT-Small(ARD_2.8M)
Accuracy: 96.4
scene-text-recognition-on-svtCCD-ViT-Tiny(ARD_2.8M)
Accuracy: 96.0
scene-text-recognition-on-svtpCCD-ViT-Base
Accuracy: 96.1
scene-text-recognition-on-svtpCCD-ViT-Small
Accuracy: 92.7
scene-text-recognition-on-svtpCCD-ViT-Tiny
Accuracy: 91.6
scene-text-recognition-on-wostCCD-ViT-Base
1:1 Accuracy: 86.0
self-supervised-scene-text-recognition-onCCD-ViT-Small
Average PSNR (dB): 21.84
SSIM: 0.7843
self-supervised-scene-text-recognition-on-1CCD-ViT-Small
IoU (%): 84.8
self-supervised-scene-text-recognition-on-2CCD-ViT-Small
Average Accuracy: 84.9

Build AI with AI

From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.

AI Co-coding
Ready-to-use GPUs
Best Pricing
Get Started

Hyper Newsletters

Subscribe to our latest updates
We will deliver the latest updates of the week to your inbox at nine o'clock every Monday morning
Powered by MailChimp
Self-supervised Character-to-Character Distillation for Text Recognition | Papers | HyperAI