INTERSPEECH2024 | SCDNet: 基于自监督学习表征的说话人转换检测

说话人转换检测（Speaker Change Detection, SCD）旨在识别对话中不同说话人之间的边界，并常被视为说话人日志（Speaker Diarization, SD）的子任务，具有广泛的应用。最近，微调 wav2vec 2.0 模型在 SCD 任务中达到了最先进的水平，但对其他经典的自监督学习（SSL）模型的探索仍然有限，并且直接对SSL模型进行微调需要一定量的数据和计算资源。

近期，西工大音频语音与语言处理研究组（ASLP@NPU）论文“SCDNet: Self-supervised Learning Feature based Speaker Change Detection”被语音研究顶级会议INTERSPEECH2024接收。该论文研究了SCD任务中的自监督学习 (SSL) 表征的应用。具体来说，该论文提出了一个名为 SCDNet 的模型，并基于该模型研究了各种主流SSL 模型，包括 Hubert、wav2vec 2.0 和 WavLm。同时，该论文还引入了一种基于微调的方法来进一步比较 SSL 模型在 SCD 任务中的特性。此外，该论文还提出了一种对比学习方法来减轻基于微调的方法和 SCDNet 在训练中的过度拟合趋势。实验展示了 WavLm 在 SCD 任务中的优势，也证明了 SCDNet 的良好设计。现对该论文进行简要的解读和分享。

作者列表：李越，王新升，张丽，谢磊

论文原文：https://arxiv.org/pdf/2406.08393

相关链接：NPU-ASLP实验室14篇论文被Interspeech2024 录用

说话人日志（SD）是语音处理中的一种关键方法，旨在回答涉及多个说话人的场景中“谁在何时说话”的问题[1]。相比之下，说话人转换检测（SCD）是寻找对话中的说话人转折点[2]，因此它可以被视为SD的子任务[3]，并且也具有广泛的应用，例如增强自动语音识别（ASR）准确性和切分字幕。

基于度量的方法是 SCD 任务的常见早期方法，其中通过比较两个连续语音窗口之间的分布来识别说话者变化点。随着 i-vector 和基于 DNN 的嵌入的出现，统一分割方案作为有效方法而受到欢迎。在这种方法中，目标音频被分割成一系列具有恒定窗口长度和重叠长度的片段。随后，比较各个片段的语音嵌入以确定说话者是否发生了变化。然而，由于窗口长度固定，语音嵌入的有效性和边界检测的准确性之间不可避免地需要进行权衡。

为了克服基于分段的方法的局限性，各种工作都致力于通过神经网络来预测帧级别的说话人变化点[1,4]。在这些方法中，模型通常使用真实的 SCD 标签进行训练，优化目标是最小化预测概率信号和线性模糊标记信号之间的距离。

除了用于帧级 SCD 的基于标签的方法之外，一些工作还探索了通过 ASR 技术利用文本转录进行单词级说话人转换检测。通过合并旨在表示说话者轮流的独特标记来增强用于训练 ASR 模型的转录。然后，使用增强转录来训练 ASR 模型，该模型不仅可以预测常规文本标记，还可以预测特殊的说话者轮流标记。虽然这种方法减轻了边界注释的必要性，但使用文本转录可能会更加复杂，特别是在以频繁中断和插入以及语调标记盛行为特征的对话场景中。另外，由于该方法中的预测边界是在字级上操作的，因此边界预测的精度可能不如基于帧级预测的边界预测那么高。

最近，研究者探索了最流行的 SSL 模型之一 wav2vec 2.0 [5] 在 SCD 任务上的有效性。在研究中，预训练的 wav2vec 2.0 以涉及多任务的端到端方式进行微调，即 SCD、重叠语音检测 (OSD) 和语音活动检测 (VAD)。这种 wav2vec 2.0 和基于多任务的方法展示了卓越的性能，在 SCD 任务中达到了最先进的 (SOTA) 水平。受这项研究的启发，我们正在对基于 SSL 的 SCD 端到端训练方法进行进一步研究。

一方面，由于SSL模型中的参数通常较多，直接对其进行微调需要一定的数据和计算资源阈值。另一方面，尽管可以通过多任务处理（例如 OSD 和 VAD）来增强 SCD 性能，但所有这些任务都是帧级二元分类任务，在训练时存在过度拟合的风险由于简单的学习范式而导致复杂的模型。此外，除了 wav2vec 2.0 之外，其他 SSL 模型（例如 Hubert [6] 和 WavLm [7]）也在各种下游任务中获得了极大的关注。然而，这些模型在 SCD 中的性能尚未得到探索。

为了解决这些问题，我们提出了一种基于 Conformer 架构的创新的端到端 SCD 模型，称为 SCDNet。SCDNet 利用现成的特征作为输入，并经过端到端的训练来完成 SCD 任务。此外，我们提出了一种对比学习方法来训练面向 SCD 的模型，以解决与帧级二元分类任务相关的过度拟合趋势。此外，我们通过 SCDNet 和基于微调的方法探索各种 SSL 功能的性能。

SCDNet 是一个基于 Conformer 的模型，以语音表示作为输入实现帧级二元分类。除了分类损失之外，还提出了对比损失来缓解简单的二元学习方式造成的过拟合倾向。这种对比损失还用于微调 SCD 任务的预训练 SSL 模型。

问题表述

网络结构

图1 SCDNet 的架构（左）和对比学习的采样策略（右）

自监督表征的使用

微调自监督模型

除了现成的基于表征的SCDNet之外，我们还通过微调来评估各种SSL模型在SCD任务中的表现。在微调过程中只更新来自变压器层和决策层的参数，并使用我们所提出的损失函数。这种微调方法具有双重目的：它比较了在SCD任务中微调不同SSL模型的性能，并能够直接比较[2]中基于多任务的损失和所提出的损失函数。

数据集和实验配置

实验采用四个真实数据集，包括 AMI [7]、AliMeeting [8]、AISHELL-4 [9] 和 DIHARD3 [10]。对于 AMI 数据集，使用“耳机混合”录音。分别采用AliMeeting的远端通道0和AISHELL-4的通道0。除了上述真实数据集之外，还使用随机混合的模拟过程，从 LibriSpeech [11]的“train-other-500”子集创建了一个人工数据集。考虑到AMI在SCD任务中的广泛使用，在AMI数据集上与其他方法进行比较，同时使用其他数据集进一步验证SCDNet的鲁棒性并证明对比学习方法的有效性。按照[2]，采用纯度（Pur）和覆盖率（Cov）分数作为SCD任务的评估指标，F1表示这两者的调和平均值。Python 库 pyannote.metrics 用于计算相应的指标。

SCDNet 包含一个 3 层 Conformer 块（N = 3），隐藏维度为 384。方程中的参数 α。5设置为0.05。在推理过程中，采用0.35的阈值对模型生成的说话人变化点的预测概率进行二值化。

SSL表征分析

基于 SCDNet 的 SSL 探索从加权融合策略开始，以检查给定 SSL 模型的哪一层表示对 SCD 任务影响最大。图 2 说明了可学习的权重值对应于 SSL 模型的不同转换器层 (l)。层的值越高表示在 SCD 任务的上下文中，该特定层的表示对最终表示的贡献越大。

图2 加权表示融合方法中不同层的权重值

与 SOTA 方法的比较

值得注意的是，[2]中之前的 SOTA 性能基于对 wav2vec2-base 模型的微调，与表 3 中所示的模型相同。然而，我们的结果是通过与提出的对比微调 wav2vec-base 来实现的学习方法明显优于[2]中的方法。这种优越性强调了所提出的对比学习方法的有效性。这种优越性的进一步证据将在以下消融研究中探讨。

对比学习损失分析

泛化性分析