用扩散模型生成少数故障数据,让工业设备诊断更可靠。
FaultDiffusion: Few-Shot Fault Time Series Generation with Diffusion Model
- 基于扩散模型,利用正常数据差异合成故障时序
- 在少样本下生成真实多样故障数据,超越传统方法
- 适合工业故障诊断与数据稀缺场景的研究者
工业设备监测中,故障诊断对保障系统可靠性与实现预测性维护至关重要。然而,由于故障事件罕见且标注成本高,故障数据极度稀缺,严重制约了数据驱动方法的应用。现有时间序列生成模型虽针对大量正常数据优化,但在少样本场景下难以捕捉故障分布,生成样本缺乏真实性和多样性,源于正常与故障域间巨大差距及故障内部高度变异性。为此,我们提出一种基于扩散模型的新型少样本故障时序生成框架。该方法采用正负样本差异适配器,利用预训练的正常数据分布建模正常与故障域的差异,实现精准故障合成;同时引入多样性损失,通过样本间差异正则化防止模式坍塌,促进多样故障样本生成。实验表明,该模型在真实性与多样性上显著优于传统方法,在关键基准测试中达到当前最优性能。
原文摘要 · Abstract (English)
In industrial equipment monitoring, fault diagnosis is critical for ensuring system reliability and enabling predictive maintenance. However, the scarcity of fault data, due to the rarity of fault events and the high cost of data annotation, significantly hinders data-driven approaches. Existing time-series generation models, optimized for abundant normal data, struggle to capture fault distributions in few-shot scenarios, producing samples that lack authenticity and diversity due to the large domain gap and high intra-class variability of faults. To address this, we propose a novel few-shot fault time-series generation framework based on diffusion models. Our approach employs a positive-negative difference adapter, leveraging pre-trained normal data distributions to model the discrepancies between normal and fault domains for accurate fault synthesis. Additionally, a diversity loss is introduced to prevent mode collapse, encouraging the generation of diverse fault samples through inter-sample difference regularization. Experimental results demonstrate that our model significantly outperforms traditional methods in authenticity and diversity, achieving state-of-the-art performance on key benchmarks.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。