arXiv:2605.30046cs.LGcs.AI2026-05被引 1

用掩码扩散模型检测离群数据,无需反向采样。

Masked Diffusion Modeling for Anomaly Detection

论文配图:Masked Diffusion Modeling for Anomaly Detection
图 1 · 摘自论文原文
  • 通过随机掩码并重建来评估样本异常程度。
  • 在14个表格和4个文本数据集上表现优于12种基线方法。
  • 理论保证错误率,适合安全关键场景的离群检测。

异常检测旨在识别偏离正常数据分布的样本,对诸多安全关键应用至关重要。然而,针对分类、混合类型及离散序列数据的有效异常检测方法仍具挑战且研究较少。掩码扩散模型可通过学习从可见上下文中恢复被掩码值来自然建模此类数据。本文提出基于掩码扩散模型的异常检测方法(MaskDiff-AD),仅使用正常数据训练,采用前向过程。测试时,通过计算随机掩码坐标重建难度生成内容敏感的异常分数,直接作用于离散状态空间,避免反向时间采样。我们还提出非参数变体,并在固定检测阈值下提供类型Ⅰ和Ⅱ误差的理论保证。在来自ADBench与UADAD的14个分类与混合类型表格数据集,以及来自NLP-ADBench的4个文本异常检测数据集上的实验表明,MaskDiff-AD性能媲美经典、扩散模型及近期基线方法。尤为突出的是,其整体平均排名最优,超越所有12种表格类基线方法。

原文摘要 · Abstract (English)

Anomaly detection aims to identify samples that deviate from the nominal data distribution and is central to many safety-critical applications. However, developing effective anomaly detection methods for categorical, mixed-type, and discrete sequence data remains challenging and relatively underexplored. Masked diffusion models provide a natural way to model such data by learning to recover masked values from the remaining visible context. In this paper, we propose Masked Diffusion for Anomaly Detection (MaskDiff-AD), a forward-only method based on masked diffusion models trained only on nominal data. Given a test sample, MaskDiff-AD constructs anomaly scores from the difficulty of reconstructing randomly masked coordinates, yielding a content-sensitive score that operates directly on discrete state spaces while avoiding reverse-time sampling. We also develop a non-parametric variant of MaskDiff-AD and provide theoretical guarantees by characterizing Type-I and Type-II errors under a fixed detection threshold. Experiments on fourteen categorical and mixed-type tabular datasets from ADBench and UADAD, as well as four text anomaly detection datasets from NLP-ADBench, show that MaskDiff-AD achieves competitive performance against classical, diffusion-based, and recent tabular/text anomaly detection baselines. Notably, MaskDiff-AD achieves the best overall average rank, outperforming all twelve tabular baseline methods.

异常检测扩散模型离散数据无监督

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