arXiv:2412.17586eess.IVcs.CV2024-12被引 11

用模型与度量组合提升脑部MRI异常检测精度

Enhancing Reconstruction-Based Out-of-Distribution Detection in Brain MRI with Model and Metric Ensembles

  • 采用自编码器结合多模型多度量集成进行无监督异常检测
  • 在像素级PR曲线下达0.66的AUPR,对全局异常检测效果更佳
  • 适合关注医疗影像安全性的研究人员和临床应用开发者

为确保自动化医学图像分析系统的安全部署,需有效识别分布外(OOD)图像。本文针对脑部MRI中合成伪影的无监督检测问题,研究了基于重构的自编码器方法。评估了模型泛化能力、训练轮次选择及重构度量的影响,探索了模型与度量集成的潜力,并在包含多样伪影的数据集上测试。结果表明,SSIM的对比度分量与LPIPS在检测均匀圆形异常时表现最优。结合两个收敛良好的模型,使用LPIPS与对比度作为度量,实现像素级精度-召回曲线下面积0.66。在更真实的OOD数据集中,局部伪影较难检测,而全局伪影检测效果更好。研究强调了度量与模型配置选择的重要性,指出通用深度学习方法难以满足特定的OOD检测需求。

原文摘要 · Abstract (English)

Out-of-distribution (OOD) detection is crucial for safely deploying automated medical image analysis systems, as abnormal patterns in images could hamper their performance. However, OOD detection in medical imaging remains an open challenge, and we address three gaps: the underexplored potential of a simple OOD detection model, the lack of optimization of deep learning strategies specifically for OOD detection, and the selection of appropriate reconstruction metrics. In this study, we investigated the effectiveness of a reconstruction-based autoencoder for unsupervised detection of synthetic artifacts in brain MRI. We evaluated the general reconstruction capability of the model, analyzed the impact of the selected training epoch and reconstruction metrics, assessed the potential of model and/or metric ensembles, and tested the model on a dataset containing a diverse range of artifacts. Among the metrics assessed, the contrast component of SSIM and LPIPS consistently outperformed others in detecting homogeneous circular anomalies. By combining two well-converged models and using LPIPS and contrast as reconstruction metrics, we achieved a pixel-level area under the Precision-Recall curve of 0.66. Furthermore, with the more realistic OOD dataset, we observed that the detection performance varied between artifact types; local artifacts were more difficult to detect, while global artifacts showed better detection results. These findings underscore the importance of carefully selecting metrics and model configurations, and highlight the need for tailored approaches, as standard deep learning approaches do not always align with the unique needs of OOD detection.

医学影像异常检测自编码器脑部MRI

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