用自模型生成伪标签,提升语音自监督学习效果
BiRQ: Bi-Level Self-Labeling Random Quantization for Self-Supervised Speech Recognition
- 用随机投影量化器从中间表示生成增强伪标签
- 在多个数据集上超越BEST-RQ,960小时LibriSpeech提升1.2% CER
- 无需外部编码器,适合资源受限场景的语音识别
语音是丰富信号,但标注音视频对成本高昂,自监督学习对可扩展表征学习至关重要。语音自监督学习的核心挑战在于生成既信息丰富又高效的伪标签:如HuBERT使用的强标签虽性能好,但依赖外部编码器和多阶段流程;而高效方法如BEST-RQ则因标签较弱而受限。本文提出BiRQ,一种双层自标注随机量化框架,结合BEST-RQ的效率与HuBERT式标签优化的优势。核心思想是复用模型自身作为伪标签生成器:通过随机投影量化器对中间表示进行离散化,生成增强标签;同时直接从原始输入获取锚定标签以稳定训练并防止坍塌。训练被建模为高效的首阶双层优化问题,通过可微的Gumbel-Softmax选择端到端求解。该设计无需外部标签编码器,降低内存开销,并支持端到端迭代标签优化。实验在多个数据集上验证:包括960小时LibriSpeech、150小时AMI会议和5,000小时YODAS,结果表明BiRQ持续优于BEST-RQ,且保持低复杂度与计算效率。
原文摘要 · Abstract (English)
Speech is a rich signal, and labeled audio-text pairs are costly, making self-supervised learning essential for scalable representation learning. A core challenge in speech SSL is generating pseudo-labels that are both informative and efficient: strong labels, such as those used in HuBERT, improve downstream performance but rely on external encoders and multi-stage pipelines, while efficient methods like BEST-RQ achieve simplicity at the cost of weaker labels. We propose BiRQ, a bilevel SSL framework that combines the efficiency of BEST-RQ with the refinement benefits of HuBERT-style label enhancement. The key idea is to reuse part of the model itself as a pseudo-label generator: intermediate representations are discretized by a random-projection quantizer to produce enhanced labels, while anchoring labels derived directly from the raw input stabilize training and prevent collapse. Training is formulated as an efficient first-order bilevel optimization problem, solved end-to-end with differentiable Gumbel-softmax selection. This design eliminates the need for external label encoders, reduces memory cost, and enables iterative label refinement in an end-to-end fashion. BiRQ consistently improves over BEST-RQ while maintaining low complexity and computational efficiency. We validate our method on various datasets, including 960-hour LibriSpeech, 150-hour AMI meetings and 5,000-hour YODAS, demonstrating consistent gains over BEST-RQ.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。