arXiv:2409.18542eess.AScs.AI2024-09被引 6

用扩散模型生成逼真异常机器声音,提升故障检测系统评估效果。

MIMII-Gen: Generative Modeling Approach for Simulated Evaluation of Anomalous Sound Detection System

  • 基于编码器-解码器架构与Flan-T5,实现条件化异常声波生成
  • 生成音频的FAD得分优于现有方法,真实度高且可复现原始系统性能
  • 适合做工业声学检测系统测试与数据增强的研究者使用

由于实际异常样本稀缺,构建和验证稳健的机器声音异常检测系统面临挑战。为此,我们提出一种基于潜在扩散模型的新方法,通过集成编码器-解码器框架,在音轨元数据描述基础上生成多样化异常声音。该方法利用Flan-T5对元数据进行编码,并通过精心设计的U-Net架构实现条件生成,使音频信号在EnCodec潜在空间中合成,保证上下文相关性与音质。我们采用弗雷歇音频距离(FAD)等指标客观评估生成音频质量,结果表明该方法在生成真实感机器异常音频方面优于现有模型。使用生成数据对异常检测系统进行评估,其受试者工作特征曲线下面积(AUC)仅比原系统低4.8%,验证了生成数据的有效性。该成果展示了在多样且未见过的条件下增强异常检测系统评估与鲁棒性的潜力。音频样本可访问:https://hpworkhub.github.io/MIMII-Gen.github.io/

原文摘要 · Abstract (English)

Insufficient recordings and the scarcity of anomalies present significant challenges in developing and validating robust anomaly detection systems for machine sounds. To address these limitations, we propose a novel approach for generating diverse anomalies in machine sound using a latent diffusion-based model that integrates an encoder-decoder framework. Our method utilizes the Flan-T5 model to encode captions derived from audio file metadata, enabling conditional generation through a carefully designed U-Net architecture. This approach aids our model in generating audio signals within the EnCodec latent space, ensuring high contextual relevance and quality. We objectively evaluated the quality of our generated sounds using the Fréchet Audio Distance (FAD) score and other metrics, demonstrating that our approach surpasses existing models in generating reliable machine audio that closely resembles actual abnormal conditions. The evaluation of the anomaly detection system using our generated data revealed a strong correlation, with the area under the curve (AUC) score differing by 4.8\% from the original, validating the effectiveness of our generated data. These results demonstrate the potential of our approach to enhance the evaluation and robustness of anomaly detection systems across varied and previously unseen conditions. Audio samples can be found at \url{https://hpworkhub.github.io/MIMII-Gen.github.io/}.

异常检测声音生成扩散模型工业诊断

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。