无需标注数据,通过重建图像位置来定位脑部异常病灶
Patch2Loc: Learning to Localize Patches for Unsupervised Brain Lesion Detection
- 用正常脑部图像块训练模型预测其原始位置
- 异常区域因预测误差大而被识别,生成热力图
- 适用于无标注数据的脑肿瘤检测,适合医学影像研究者
脑部病灶(如肿瘤、畸形)的检测对诊断和治疗至关重要。放射科医生可借助基于机器学习的计算机视觉系统,将正常组织与异常组织分割开来。传统监督方法需标注病灶,本文提出一种新方法 Patch2Loc,采用无监督方式,仅使用结构化 MRI 中的正常图像块进行训练。模型学习将局部图像块映射回其在脑体积切片中的空间位置。推理时,异常图像块因位置预测误差或方差较大而被识别,生成热力图,可用于细粒度分割。我们在 BraTS2021 和 MSLUB 的 T2 加权图像,以及 ATLAS 和 WMH 的 T1 加权图像上测试该方法,成功实现脑肿瘤检测,性能优于现有无监督分割方法。代码已开源。
原文摘要 · Abstract (English)
Detecting brain lesions as abnormalities observed in magnetic resonance imaging (MRI) is essential for diagnosis and treatment. In the search of abnormalities, such as tumors and malformations, radiologists may benefit from computer-aided diagnostics that use computer vision systems trained with machine learning to segment normal tissue from abnormal brain tissue. While supervised learning methods require annotated lesions, we propose a new unsupervised approach (Patch2Loc) that learns from normal patches taken from structural MRI. We train a neural network model to map a patch back to its spatial location within a slice of the brain volume. During inference, abnormal patches are detected by the relatively higher error and/or variance of the location prediction. This generates a heatmap that can be integrated into pixel-wise methods to achieve finer-grained segmentation. We demonstrate the ability of our model to segment abnormal brain tissues by applying our approach to the detection of tumor tissues in MRI on T2-weighted images from BraTS2021 and MSLUB datasets and T1-weighted images from ATLAS and WMH datasets. We show that it outperforms the state-of-the art in unsupervised segmentation. The implementation for this work can be found on our \href{https://github.com/bakerhassan/Patch2Loc}{GitHub page}. This paper has been accepted at AISTATS 2026.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。