描述
NVIDIA Parakeet是一个先进的自动语音识别(ASR)模型,具有6亿个参数,旨在提供高质量的英语转录。它支持自动标点、大小写和准确的单词级时间戳预测,使其适用于各种应用,如对话AI、语音助手和转录服务。
阅读更多
该模型基于FastConformer架构,集成了TDT解码器,并采用全注意力训练。这使得它能够在一次传递中高效转录长达24分钟的音频片段。它在HF-Open-ASR排行榜上以128的批量大小达到了3380的RTFx,尽管性能可能因数据集音频时长和批量大小而有所不同。
NVIDIA Parakeet已准备好用于商业和非商业用途,受CC-BY-4.0许可证的约束。它经过优化,能够在NVIDIA GPU加速系统上运行,利用GPU核心和CUDA库等硬件和软件框架,相比仅使用CPU的解决方案,提供更快的训练和推理时间。
该模型旨在为开发者、研究人员、学术界和需要语音转文本能力的行业构建应用。它使用NeMo工具包进行训练,从FastConformer SSL检查点初始化,并在高质量的人类转录数据上进行微调。训练数据集包括来自各种来源的约120,000小时的英语语音数据。
NVIDIA Parakeet在不同的信噪比和电话风格音频格式下表现出强大的性能。它使用Huggingface Open ASR排行榜数据集进行评估,平均达到6.05%的字错误率(WER)。该模型旨在具有抗噪声能力,并在各种类型的音频输入中表现良好。
- 类别AI 模型与大语言模型
- 定价模式开源
- 部署云端
- 平台Linux
NVIDIA Parakeet ASR模型亮点
6亿参数的ASR模型
自动标点和大小写
准确的单词级时间戳预测
在口语数字和歌词上的强大表现
具有TDT解码器的FastConformer架构
支持长达24分钟的音频片段
在HF-Open-ASR排行榜上达到3380的RTFx
针对NVIDIA GPU加速系统进行了优化
商业和非商业用途
CC-BY-4.0许可证
NVIDIA Parakeet ASR模型入门
访问页面:访问huggingface.co/nvidia/parakeet-tdt-0.6b-v2
加载模型:使用NeMo工具包加载预训练模型
配置环境:安装NVIDIA NeMo和PyTorch
集成:使用API或CLI进行转录任务
微调:使用额外的数据集自定义模型
NVIDIA Parakeet ASR模型的使用案例
- 对话AI
- 语音助手
- 转录服务
- 字幕生成
- 语音分析






