arXiv:2507.21164cs.LGcs.AI2025-07被引 5

用OCSVM指导特征学习,让无监督异常检测更准更鲁棒。

OCSVM-Guided Representation Learning for Unsupervised Anomaly Detection

  • 通过自定义损失函数,让特征空间直接对齐OCSVM决策边界。
  • 在脑MRI小病灶检测中表现优异,优于传统方法。
  • 适合医学影像等真实场景的异常检测任务。

无监督异常检测(UAD)旨在无需标注数据的情况下识别异常,在异常样本稀少或不可得的机器学习应用中至关重要。现有方法主要分为重建类和解耦表示学习两类,前者常过度重构异常,后者可能陷入次优特征空间。尽管部分新方法尝试耦合特征学习与异常检测,但多依赖代理目标、受限核函数或引入近似,影响表达能力与鲁棒性。为此,本文提出一种新方法,通过定制化损失函数,将表示学习与可解析求解的一类支持向量机(OCSVM)直接耦合,使隐层特征精准对齐其决策边界。模型在两个任务上评估:基于MNIST-C的新基准,以及具有挑战性的脑部MRI微小病灶检测任务。不同于多数方法聚焦图像级大而高信号的病灶,本方法成功定位小且非高信号的病灶,并采用体素级指标,更具临床意义。两项实验均评估了对领域偏移的鲁棒性,包括MNIST-C中的各类损坏,以及MRI中的纹理或人群年龄差异。结果表明,所提模型在性能与鲁棒性方面均表现优越,凸显其在通用无监督异常检测及真实医学影像应用中的潜力。源代码见 https://github.com/Nicolas-Pinon/uad_ocsvm_guided_repr_learning。

原文摘要 · Abstract (English)

Unsupervised anomaly detection (UAD) aims to detect anomalies without labeled data, a necessity in many machine learning applications where anomalous samples are rare or not available. Most state-of-the-art methods fall into two categories: reconstruction-based approaches, which often reconstruct anomalies too well, and decoupled representation learning with density estimators, which can suffer from suboptimal feature spaces. While some recent methods attempt to couple feature learning and anomaly detection, they often rely on surrogate objectives, restrict kernel choices, or introduce approximations that limit their expressiveness and robustness. To address this challenge, we propose a novel method that couples representation learning with an analytically solvable One-Class SVM (OCSVM), through a custom loss formulation that directly aligns latent features with the OCSVM decision boundary. The model is evaluated on two tasks: a \deleted{new} benchmark based on MNIST-C, and a challenging brain MRI \deleted{subtle} lesion detection task. Unlike most methods that focus on large, hyperintense lesions at the image level, our approach succeeds to target small, non-hyperintense lesions, while we evaluate voxel-wise metrics, addressing a more clinically relevant scenario. Both experiments evaluate a form of robustness to domain shifts, including corruption types in MNIST-C and texture or population age variations in MRI. Results demonstrate performance and robustness of our proposed model, highlighting its potential for general UAD and real-world medical imaging applications. The source code is available at https://github.com/Nicolas-Pinon/uad_ocsvm_guided_repr_learning.

异常检测OCSVM医学影像无监督学习

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