通过迭代解码3D脑MRI的掩码区域,提升异常检测与分割精度。
IterMask3D: Unsupervised Anomaly Detection and Segmentation with Test-Time Iterative Mask Refinement in 3D Brain MR
- 用迭代掩码重构策略逐步释放正常区域信息,减少误报
- 在真实和合成图像上均实现更优的异常检测与病灶分割性能
- 适合需要高精度3D医学图像分析的研究者或临床应用
无监督异常检测与分割方法通过学习训练数据中的正常分布来识别测试阶段的异常。现有方法通常对图像进行破坏并训练模型重建,但破坏过程会损失正常区域信息,导致重建不佳和误报增加。为此,本文提出IterMask3D,一种针对3D脑部MRI的迭代空间掩码精炼策略:通过逐步掩码并重建图像区域,根据重建误差缩小掩码范围,将正常区域信息逐步释放给模型,从而提升重建准确性。同时,引入高频图像内容作为额外结构信息以指导遮蔽区域重建。在多种磁共振序列上的实验表明,该方法在合成与真实成像伪影检测、各类病理性病变分割任务中均表现优异。代码已开源。
原文摘要 · Abstract (English)
Unsupervised anomaly detection and segmentation methods train a model to learn the training distribution as `normal'. In the testing phase, they identify patterns that deviate from this normal distribution as `anomalies'. To learn the `normal' distribution, prevailing methods corrupt the images and train a model to reconstruct them. During testing, the model attempts to reconstruct corrupted inputs based on the learned `normal' distribution. Deviations from this distribution lead to high reconstruction errors, which indicate potential anomalies. However, corrupting an input image inevitably causes information loss even in normal regions, leading to suboptimal reconstruction and an increased risk of false positives. To alleviate this, we propose $\rm{IterMask3D}$, an iterative spatial mask-refining strategy designed for 3D brain MRI. We iteratively spatially mask areas of the image as corruption and reconstruct them, then shrink the mask based on reconstruction error. This process iteratively unmasks `normal' areas to the model, whose information further guides reconstruction of `normal' patterns under the mask to be reconstructed accurately, reducing false positives. In addition, to achieve better reconstruction performance, we also propose using high-frequency image content as additional structural information to guide the reconstruction of the masked area. Extensive experiments on the detection of both synthetic and real-world imaging artifacts, as well as segmentation of various pathological lesions across multiple MRI sequences, consistently demonstrate the effectiveness of our proposed method. Code is available at https://github.com/ZiyunLiang/IterMask3D.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。