用伪标签和知识蒸馏提升无标注音频的和弦识别效果
Enhancing Automatic Chord Recognition via Pseudo-Labeling and Knowledge Distillation
- 先用预训练模型生成海量无标注音频的伪标签,训练学生模型
- 在仅有伪标签时,学生模型达到教师模型97%以上性能,后续持续学习更优
- 特别擅长识别罕见和弦,适合数据少但需高精度的音乐分析场景
自动和弦识别受制于对齐和弦标注稀缺,其获取成本高昂。同时,开放权重的预训练模型比其私有训练数据更易获取。本文提出一种两阶段训练流程,结合预训练模型与无标注音频。第一阶段,以预训练的BTC模型为教师,为超过1000小时多样化的无标注音频生成伪标签,并仅用这些伪标签训练学生模型。第二阶段,随着真实标注逐步可用,学生模型持续训练。为防止第一阶段学得的表示被遗忘,采用选择性知识蒸馏作为正则化。实验中使用BTC和2E1D两个模型作为学生。第一阶段仅用伪标签,BTC学生模型达到教师约99%性能,2E1D达约97%(基于七个标准mir_eval指标)。第二阶段持续学习真实标签后,最终的BTC学生模型在所有指标上均优于传统监督基线和原始教师模型;2E1D学生也超越监督基线,接近教师水平,且在罕见和弦识别上均有显著提升。
原文摘要 · Abstract (English)
Automatic Chord Recognition (ACR) is constrained by the scarcity of aligned chord annotations, which are costly to acquire. At the same time, open-weight pre-trained models are more accessible than their proprietary training data. In this work, we present a two-stage training pipeline that leverages pre-trained models together with unlabeled audio. The proposed method decouples training into two stages. In the first stage, we use the pre-trained BTC model as a teacher to generate pseudo-labels for over 1,000 hours of diverse unlabeled audio and train a student model solely on these pseudo-labels. In the second stage, the student is continually trained on ground-truth labels as they become available. To prevent catastrophic forgetting of the representations learned in the first stage, we apply selective knowledge distillation (KD) from the teacher as a regularizer. In our experiments, two models (BTC, 2E1D) were used as students. In Stage 1, using only pseudo-labels, the BTC student achieves about 99% of the teacher's performance, while the 2E1D model achieves about 97% of the teacher's performance across seven standard mir_eval metrics. After continual training with labeled data in Stage 2, the resulting BTC student model consistently surpasses both the traditional supervised learning baseline and the original pre-trained teacher model across all metrics. The resulting 2E1D student model also outperforms the supervised baseline and approaches teacher-level performance, with both models demonstrating substantial gains on rare chord qualities.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。