用空间感知Mamba模型,无监督检测医学图像异常
SP-Mamba: Spatial-Perception State Space Model for Unsupervised Medical Anomaly Detection
- 引入窗口滑动原型学习与环形希尔伯特扫描,捕捉医学图像结构规律
- 在三个数据集上达到最新最佳性能,异常定位更精准
- 适合医学图像分析、无监督异常检测研究者使用
放射影像检查针对特定解剖区域,导致患者间图像具有高度一致性及重复性结构模式。近年来,基于CNN和Transformer的方法在医学异常检测中表现良好,但CNN难以捕捉长程依赖,而Transformer存在二次计算复杂度问题。相比之下,Mamba模型凭借优越的长程建模能力、结构特征提取及线性计算效率,成为有前景的替代方案。为利用医学图像的内在结构规律,本文提出SP-Mamba,一种面向无监督医学异常检测的空间感知Mamba框架。通过引入窗口滑动原型学习和基于环形希尔伯特扫描的Mamba结构,更好地挖掘一致的解剖模式并利用空间信息提升异常检测效果。此外,我们挖掘异常图的集中度与对比度特性以优化检测性能。在三个不同医学异常检测基准上的大量实验验证了该方法的先进性与鲁棒性。代码已开源:https://github.com/Ray-RuiPan/SP-Mamba。
原文摘要 · Abstract (English)
Radiography imaging protocols target on specific anatomical regions, resulting in highly consistent images with recurrent structural patterns across patients. Recent advances in medical anomaly detection have demonstrated the effectiveness of CNN- and transformer-based approaches. However, CNNs exhibit limitations in capturing long-range dependencies, while transformers suffer from quadratic computational complexity. In contrast, Mamba-based models, leveraging superior long-range modeling, structural feature extraction, and linear computational efficiency, have emerged as a promising alternative. To capitalize on the inherent structural regularity of medical images, this study introduces SP-Mamba, a spatial-perception Mamba framework for unsupervised medical anomaly detection. The window-sliding prototype learning and Circular-Hilbert scanning-based Mamba are introduced to better exploit consistent anatomical patterns and leverage spatial information for medical anomaly detection. Furthermore, we excavate the concentration and contrast characteristics of anomaly maps for improving anomaly detection. Extensive experiments on three diverse medical anomaly detection benchmarks confirm the proposed method's state-of-the-art performance, validating its efficacy and robustness. The code is available at https://github.com/Ray-RuiPan/SP-Mamba.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。