⚡ 前沿动态

NVIDIA Nemotron 3 Diarization:1亿参数开源端到端说话人分离模型‌

🕐 2026.09.29 星期二 来源 · 媒体 🔥 158 次点击

NVIDIA Nemotron 3 Diarization是英伟达2026年9月23日正式发布的‌1亿参数开源权重说话人分离模型‌,核心能力是精准回答音频场景下“谁在什么时候说话”的核心问题,支持最多8位说话人同时发声的实时流音频与离线录音处理,在Voice Arena权威基准测试中登顶性能榜首,大幅降低了多说话人语音分析的商用落地门槛。

说话人分离(diarization)是"谁在什么时候说话"这一步,长期是转写服务中做得不够好的中间件:会议、呼叫中心、法务录音、问诊记录都需要它,但它通常作为付费 API 的一部分存在。

NVIDIA Nemotron 3 Diarization 免费开放权重(Hugging Face),明显定位为 Parakeet 生态的引流件。

一、核心规格与基础定位

这是一款专门面向说话人分离(Speaker Diarization)任务优化的Transformer架构模型,核心基础参数如下:

  • 模型规模‌:100M(1亿)参数,31层Transformer编码器搭配RoPE旋转位置编码,轻量化设计兼顾性能与部署效率
  • 支持能力‌:单模型同时覆盖离线录音与实时流式推理,最多可稳定追踪8位说话人,支持重叠语音场景下的精准识别
  • 输入输出‌:原生支持16kHz单声道音频输入,输出形状为[T, 8]的浮点张量,对应8个说话人的逐帧活动概率,默认输出步长10ms,可灵活配置为10ms的任意整数倍
  • 商用授权‌:采用OpenMDW 1.1开源许可,完全允许商业场景下免费使用,无版权风险
  • 部署适配‌:可通过NVIDIA NeMo框架在Linux系统运行,兼容Ampere、Ada Lovelace、Hopper、Blackwell全系列NVIDIA GPU,单张RTX PRO 6000即可完成全量部署

参数表listAIDeepT↑hink.cn

参数 数值
参数量 约 1 亿(100M)
最大说话人数 8
重叠语音检测 支持
音频缓冲档位 4 档,30.4 秒 → 0.32 秒
实时/离线 均支持
权重 免费开放(Hugging Face)
榜单 VoiceArena Diarization-Bench v1 第一,DER 14.72%

二、核心技术突破:端到端架构替代传统多阶段流水线

传统说话人分离方案普遍采用“语音活动检测+说话人特征提取+聚类+后处理”的多阶段串联流水线,环节多、调优成本高,而Nemotron 3 Diarization用单模型端到端架构彻底替代了这套复杂流程:

1. 到达时间排序机制‌

沿用Sortformer的排列消解方案,8个输出通道严格按照说话人首次出现的时间排序,首个新声音自动占据第一通道,次个占据第二通道,彻底避免了传统聚类方案中每段音频重新排列组合的计算开销,保证匿名说话人标签(如speaker_2)的全程稳定性。listAIDeepT↑hink.cn

2. 三层极简处理管线‌

  • 特征堆叠:将10ms的Mel频谱特征按8倍因子堆叠,形成80ms的编码帧,大幅提升编码器处理效率
  • Transformer编码:用31层带RoPE的编码器处理堆叠后的80ms帧,输出帧级说话人特征表示
  • Conv1D上采样:通过一层一维卷积将预测结果上采样回原始10ms的输入特征分辨率,保证时间戳精度

3. 双记忆流式缓存机制‌

通过“到达顺序说话人缓存(AOSC)”保留早期片段的说话人信息,搭配FIFO先进先出队列维护当前片段前的最近帧上下文,彻底消除传统流式推理中“长音频跨片段说话人身份漂移”的通病。listAIDeepT↑hink.cn

三、训练体系:9万小时数据+两阶段A100集群训练

为了实现高精度泛化能力,英伟达搭建了工业级的训练数据与训练流程体系:listAIDeepT↑hink.cn

1. 训练数据总规模‌

累计约9.3万小时多说话人音频,分为三大类:1万小时真实多说话人对话(覆盖会议、电话、多语言等14个真实数据源)、2.8万小时单说话人原料音频、8.2万小时通过FastMSS工具合成的1-8人虚拟会议混音数据,覆盖21种语言。

2. 训练硬件‌

在8节点×8卡A100-SXM4-80GB集群上完成全流程训练,模型初始化自NEST自监督预训练检查点。

3. 两阶段训练流程‌

第一阶段在8.2万小时合成混音数据上完成离线预训练,学习基础的多说话人区分能力;第二阶段在真实对话数据上完成微调,大幅提升复杂真实场景下的鲁棒性。

四、性能表现:权威基准测试登顶

在Voice Arena Diarization-Bench v1版本严苛基准测试中,该模型以‌14.72%的说话人错误率(DER)‌位居榜首,大幅领先第二名系统的19.3%,相对错误率降低约24% 。

与前代Streaming Sortformer相比,在1.04秒音频缓冲区的8个测试场景下,平均错误率直接降低41%,同时将支持的最大说话人数从4人翻倍提升至8人,吞吐量也实现显著提升 。

为了灵活适配不同业务场景,模型提供4档可动态配置的音频输入缓冲区:30.4秒、1.04秒、0.64秒、0.32秒,缓冲区越大上下文信息越充分、识别精度越高,缓冲区越小端到端延迟越低,最低可实现0.32秒的超低实时响应,理论延迟下限可达80ms,用户可根据自身业务的延迟-精度需求灵活切换 。listAIDeepT↑hink.cn

五、落地适配与典型应用场景

该模型可与NVIDIA Parakeet、Canary,以及OpenAI Whisper等几乎所有主流ASR语音识别系统无缝对接,将输出的说话人时间戳与ASR转录文本结合,直接生成带明确说话人归属的完整对话转录结果,典型落地场景包括:

  • 实时会议记录:自动区分参会人发言,生成带发言人标签的会议纪要,精准提取不同人的承诺、异议与行动项
  • 呼叫中心通话分析:自动区分客服与用户对话,批量分析服务话术质量、用户诉求分布
  • 播客/访谈内容生产:自动完成多嘉宾对话的说话人标注,大幅降低人工剪辑整理成本
  • 多角色语音Agent记忆系统:精准区分不同用户的指令来源,避免多用户场景下的记忆混淆
    需要注意的是,该模型仅输出匿名的说话人编号标签,不直接识别说话人的真实身份,若需要关联具体人员信息,可在下游应用层对接声纹验证模型完成映射 。

需要我为你整理这套模型的‌快速部署实操步骤清单‌,帮你快速在本地GPU上跑通基础推理吗?

声明:文中数据及部分观点来源于网络,部分观点来源于个人分析和推理,仅供参考,不构成投资或商业决策建议。
资料来源
  1. Huggingface
← 返回 [前沿动态]