在手机等设备上实现低延迟高准确的语音识别,模型体积缩小至670MB
Pushing the Limits of On-Device Streaming ASR: A Compact, High-Accuracy English Model for Low-Latency Inference
- 用ONNX Runtime重实现推理流程,结合量化与算子融合优化
- 模型从2.47GB压缩到0.67GB,WER仅比原版高1%以内
- 适合部署在无GPU的移动设备,实时性好,准确率领先
将高质量语音识别(ASR)部署到边缘设备需兼顾精度、延迟和内存占用,且仅依赖CPU运行。我们系统评估了当前主流的编码器-解码器、转换器及大语言模型类架构,在批处理、分块和流式推理模式下表现。通过对超过50种配置的全面基准测试,涵盖OpenAI Whisper、NVIDIA Nemotron、Parakeet TDT、Canary、Conformer Transducer和Qwen3-ASR,发现NVIDIA Nemotron Speech Streaming是资源受限硬件上实时英文流式识别的最佳候选。随后我们在ONNX Runtime中重新实现完整流式推理流程,并控制评估多种后训练量化策略,包括重要性加权k-量化、混合精度方案及四舍五入量化,结合图级算子融合。优化后模型从2.47 GB压缩至最低0.67 GB,同时保持词错误率(WER)在全精度PyTorch基线1%绝对误差内。推荐配置int4 k-量化版本在8个标准基准上平均达到8.20%流式WER,CPU上运行速度远超实时,算法延迟为0.56秒,确立了设备端流式ASR新的质量-效率帕累托前沿。
原文摘要 · Abstract (English)
Deploying high-quality automatic speech recognition (ASR) on edge devices requires models that jointly optimize accuracy, latency, and memory footprint while operating entirely on CPU without GPU acceleration. We conduct a systematic empirical study of state-of-the-art ASR architectures, encompassing encoder-decoder, transducer, and LLM-based paradigms, evaluated across batch, chunked, and streaming inference modes. Through a comprehensive benchmark of over 50 configurations spanning OpenAI Whisper, NVIDIA Nemotron, Parakeet TDT, Canary, Conformer Transducer, and Qwen3-ASR, we identify NVIDIA's Nemotron Speech Streaming as the strongest candidate for real-time English streaming on resource-constrained hardware. We then re-implement the complete streaming inference pipeline in ONNX Runtime and conduct a controlled evaluation of multiple post-training quantization strategies, including importance-weighted k-quant, mixed-precision schemes, and round-to-nearest quantization, combined with graph-level operator fusion. These optimizations reduce the model from 2.47 GB to as little as 0.67 GB while maintaining word error rate (WER) within 1% absolute of the full-precision PyTorch baseline. Our recommended configuration, the int4 k-quant variant, achieves 8.20% average streaming WER across eight standard benchmarks, running comfortably faster than real-time on CPU with 0.56 s algorithmic latency, establishing a new quality-efficiency Pareto point for on-device streaming ASR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。