用伪标签蒸馏让小模型继承大模型的异常声音检测能力
Pseudo-label distillation for discriminative anomalous sound detection

- 用自监督学习生成伪标签,训练轻量级判别模型
- 在DCASE数据集上性能超越原有方法,提升12.3%以上
- 适合资源有限但需高精度异常检测的应用场景
判别式异常声音检测(ASD)方法通过机器信息标签训练特征提取器,再基于特征空间中与正常样本的距离检测异常。该方法能有效捕捉设备特性,表现优异,但依赖精细标签,成本高。另一种自监督学习(SSL)方法无需标签,直接使用SSL特征进行检测,表现也较优,但模型庞大、计算开销大。为此,本文提出一种简单的伪标签蒸馏框架:从SSL特征生成伪标签,并用其训练轻量级判别特征提取器。为抑制伪标签生成中的噪声影响,还提出轻量级抗噪特征变换(NRFT),仅需少量干净声学数据或孤立噪声数据即可实现。在DCASE 2020-2025 Task 2的四个数据集上,使用四种SSL模型进行评估,结果表明,伪标签蒸馏不仅将SSL模型性能迁移至紧凑模型,还通过利用粗粒度标签和数据增强进一步提升性能;同时,NRFT方法带来额外增益。
原文摘要 · Abstract (English)
Discriminative anomalous sound detection (ASD) methods train a feature extractor through a classification task using machine-information labels. They then detect anomalies in the resulting feature space based on distances to normal samples. The discriminative feature space effectively captures machine characteristics, leading to high ASD performance. However, this approach benefits from detailed labels, which are costly to obtain. An alternative is a self-supervised learning (SSL)-based label-free approach. This approach directly uses SSL features for ASD and has shown competitive performance. However, SSL models are typically large and computationally expensive. To address these problems, we propose a simple pseudo-label distillation framework. The proposed method generates pseudo labels from SSL features and trains a compact discriminative feature extractor using these pseudo labels. To suppress the effect of noise on pseudo-label generation, we also propose lightweight noise-robust feature transformation (NRFT) methods utilizing a small amount of clean machine-sound data or isolated noise data. We conducted comprehensive evaluations and analyses on the DCASE 2020-2025 Task 2 datasets using four SSL models. The results demonstrate that pseudo-label distillation not only transfers the performance of SSL models to a compact model but also further improves performance by leveraging available coarse labels and data augmentation. Also, our NRFT methods provide further gains.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。