用扩散模型检测图像和音频中的异常,效果优于现有方法。
Research on Anomaly Detection Methods Based on Diffusion Models
- 通过反向扩散重建数据,结合重构误差与语义差异识别异常
- 在MVTec AD和UrbanSound8K上达到领先性能,鲁棒性强
- 适合工业故障诊断、医疗监测等需要高精度异常检测的场景
异常检测是机器学习与数据挖掘中的基础任务,在网络安全、工业故障诊断和临床疾病监控中有重要应用。传统方法如统计建模和基于机器学习的方法在处理复杂高维数据分布时面临挑战。本文探索扩散模型在异常检测中的潜力,提出一种新框架,利用扩散概率模型(DPMs)有效识别图像和音频中的异常。该方法通过扩散过程建模正常数据分布,并通过反向扩散重建输入数据,以重构误差与语义差异作为异常指标。为提升性能,引入多尺度特征提取、注意力机制和小波域表示,使模型能捕捉数据的细粒度结构与全局依赖关系。在MVTec AD和UrbanSound8K等基准数据集上的大量实验表明,该方法在多种数据模态下均优于当前最优技术,具备更高的准确率与鲁棒性。本研究验证了扩散模型在异常检测中的有效性,为实际应用提供了高效可靠的解决方案。
原文摘要 · Abstract (English)
Anomaly detection is a fundamental task in machine learning and data mining, with significant applications in cybersecurity, industrial fault diagnosis, and clinical disease monitoring. Traditional methods, such as statistical modeling and machine learning-based approaches, often face challenges in handling complex, high-dimensional data distributions. In this study, we explore the potential of diffusion models for anomaly detection, proposing a novel framework that leverages the strengths of diffusion probabilistic models (DPMs) to effectively identify anomalies in both image and audio data. The proposed method models the distribution of normal data through a diffusion process and reconstructs input data via reverse diffusion, using a combination of reconstruction errors and semantic discrepancies as anomaly indicators. To enhance the framework's performance, we introduce multi-scale feature extraction, attention mechanisms, and wavelet-domain representations, enabling the model to capture fine-grained structures and global dependencies in the data. Extensive experiments on benchmark datasets, including MVTec AD and UrbanSound8K, demonstrate that our method outperforms state-of-the-art anomaly detection techniques, achieving superior accuracy and robustness across diverse data modalities. This research highlights the effectiveness of diffusion models in anomaly detection and provides a robust and efficient solution for real-world applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。