HyperAIHyperAI

Command Palette

Search for a command to run...

5 months ago

HTR-VT: Handwritten Text Recognition with Vision Transformer

Li Yuting ; Chen Dexiong ; Tang Tinglong ; Shen Xi

HTR-VT: Handwritten Text Recognition with Vision Transformer

Abstract

We explore the application of Vision Transformer (ViT) for handwritten textrecognition. The limited availability of labeled data in this domain poseschallenges for achieving high performance solely relying on ViT. Previoustransformer-based models required external data or extensive pre-training onlarge datasets to excel. To address this limitation, we introduce adata-efficient ViT method that uses only the encoder of the standardtransformer. We find that incorporating a Convolutional Neural Network (CNN)for feature extraction instead of the original patch embedding and employSharpness-Aware Minimization (SAM) optimizer to ensure that the model canconverge towards flatter minima and yield notable enhancements. Furthermore,our introduction of the span mask technique, which masks interconnectedfeatures in the feature map, acts as an effective regularizer. Empirically, ourapproach competes favorably with traditional CNN-based models on small datasetslike IAM and READ2016. Additionally, it establishes a new benchmark on the LAMdataset, currently the largest dataset with 19,830 training text lines. Thecode is publicly available at: https://github.com/YutingLi0606/HTR-VT.

Code Repositories

yutingli0606/htr-vt
Official
pytorch
Mentioned in GitHub

Benchmarks

BenchmarkMethodologyMetrics
handwritten-text-recognition-on-iamHTR-VT(line-level)
CER: 4.7
WER: 14.9
handwritten-text-recognition-on-iam-lineHTR-VT
Test CER: 4.7
Test WER: 14.9
handwritten-text-recognition-on-lam-lineHTR-VT
Test CER: 2.8
Test WER: 7.4
handwritten-text-recognition-on-read-2016HTR-VT(line-level)
CER (%): 3.9
WER (%): 16.5
handwritten-text-recognition-on-read2016-lineHTR-VT
Test CER: 3.9
Test WER: 16.5

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