中文

TriNet:在 ASR 上稳定自监督学习以避免完全或缓慢坍缩

音频与语音处理 2023-03-15 v2 计算与语言 机器学习

摘要

自监督学习(SSL)模型面临信息 abrupt 坍缩或缓慢维度坍缩的挑战。我们提出 TriNet,其引入了一种新颖的三分支架构,用于防止坍缩并稳定预训练。TriNet 学习 SSL 潜在嵌入空间,并将其整合到更高层空间以预测由冻结教师生成的伪目标向量。我们的实验结果表明,所提方法显著稳定并加速了预训练,并且在下游基准 ASR 任务上相比最先进的(SOTA)Data2vec 实现了 6.06% 的相对词错误率降低(WERR)。我们将在 https://github.com/tencent-ailab/ 发布我们的代码。

关键词

引用

@article{arxiv.2301.00656,
  title  = {TriNet: stabilizing self-supervised learning from complete or slow collapse on ASR},
  author = {Lixin Cao and Jun Wang and Ben Yang and Dan Su and Dong Yu},
  journal= {arXiv preprint arXiv:2301.00656},
  year   = {2023}
}

备注

Accepted by ICASSP 2023