HyperAIHyperAI

Command Palette

Search for a command to run...

3 months ago

ChunkFormer: Masked Chunking Conformer For Long-Form Speech Transcription

Khanh Le Tuan Vu Ho Dung Tran Duc Thanh Chau

ChunkFormer: Masked Chunking Conformer For Long-Form Speech Transcription

Abstract

Deploying ASR models at an industrial scale poses significant challenges in hardware resource management, especially for long-form transcription tasks where audio may last for hours. Large Conformer models, despite their capabilities, are limited to processing only 15 minutes of audio on an 80GB GPU. Furthermore, variable input lengths worsen inefficiencies, as standard batching leads to excessive padding, increasing resource consumption and execution time. To address this, we introduce ChunkFormer, an efficient ASR model that uses chunk-wise processing with relative right context, enabling long audio transcriptions on low-memory GPUs. ChunkFormer handles up to 16 hours of audio on an 80GB GPU, 1.5x longer than the current state-of-the-art FastConformer, while also boosting long-form transcription performance with up to 7.7% absolute reduction on word error rate and maintaining accuracy on shorter tasks compared to Conformer. By eliminating the need for padding in standard batching, ChunkFormer's masked batching technique reduces execution time and memory usage by more than 3x in batch processing, substantially reducing costs for a wide range of ASR systems, particularly regarding GPU resources for models serving in real-world applications.

Code Repositories

khanld/chunkformer
Official
pytorch

Benchmarks

BenchmarkMethodologyMetrics
speech-recognition-on-common-voice-vikhanhld/chunkformer-large-vie
Test WER: 6.66
speech-recognition-on-vivoskhanhld/chunkformer-large-vie
Test WER: 4.18

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