用强化学习做脑部MRI异常检测,少标注也能准。
Semi-Supervised Anomaly Detection in Brain MRI Using a Domain-Agnostic Deep Reinforcement Learning Approach
- 用深度强化学习融合特征表示,解决标注少、数据多和过拟合问题。
- 在脑MRI上像素级准确率88.7%,图像级96.7%,优于现有方法。
- 跨域能力强,在工业缺陷数据集上表现优异,适合临床部署。
为应对大规模数据、过拟合和类别不平衡等挑战,提出一种领域无关的半监督异常检测框架,结合深度强化学习(DRL)处理脑部MRI体积数据。回顾性研究使用2005至2021年间公开的脑部MRI数据集:IXI数据集提供581例T1加权与578例T2加权健康影像用于训练;BraTS 2021提供251例用于验证,1000例用于测试(含胶质母细胞瘤患者)。预处理包括归一化、去颅骨和配准至统一体素大小。实验在T1和T2模态上进行,并在工业数据集上开展额外实验与消融分析。所提方法通过整合DRL与特征表示,有效应对标签稀缺、大规模数据及过拟合问题。评估指标包括AUROC与Dice分数。在脑部MRI数据集上,像素级与图像级AUROC分别达到88.7%与96.7%,优于当前最先进方法。在工业表面数据集(MVTec AD)上亦表现优异,像素级与图像级AUROC分别为99.8%与99.3%,显示强跨域泛化能力。分析表明,随着异常样本增加,AUROC单调上升,无过拟合迹象且计算成本未显著增加。该领域无关的半监督方法在医学与工业场景中均具潜力,具备鲁棒性、通用性与高效性,适用于真实临床应用。
原文摘要 · Abstract (English)
To develop a domain-agnostic, semi-supervised anomaly detection framework that integrates deep reinforcement learning (DRL) to address challenges such as large-scale data, overfitting, and class imbalance, focusing on brain MRI volumes. This retrospective study used publicly available brain MRI datasets collected between 2005 and 2021. The IXI dataset provided 581 T1-weighted and 578 T2-weighted MRI volumes (from healthy subjects) for training, while the BraTS 2021 dataset provided 251 volumes for validation and 1000 for testing (unhealthy subjects with Glioblastomas). Preprocessing included normalization, skull-stripping, and co-registering to a uniform voxel size. Experiments were conducted on both T1- and T2-weighted modalities. Additional experiments and ablation analyses were also carried out on the industrial datasets. The proposed method integrates DRL with feature representations to handle label scarcity, large-scale data and overfitting. Statistical analysis was based on several detection and segmentation metrics including AUROC and Dice score. The proposed method achieved an AUROC of 88.7% (pixel-level) and 96.7% (image-level) on brain MRI datasets, outperforming State-of-The-Art (SOTA) methods. On industrial surface datasets, the model also showed competitive performance (AUROC = 99.8% pixel-level, 99.3% image-level) on MVTec AD dataset, indicating strong cross-domain generalization. Studies on anomaly sample size showed a monotonic increase in AUROC as more anomalies were seen, without evidence of overfitting or additional computational cost. The domain-agnostic semi-supervised approach using DRL shows significant promise for MRI anomaly detection, achieving strong performance on both medical and industrial datasets. Its robustness, generalizability and efficiency highlight its potential for real-world clinical applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。