通过解耦解剖结构与成像信息,提升脑部MRI异常检测的泛化性与准确性。
Unsupervised Anomaly Detection in Brain MRI via Disentangled Anatomy Learning
- 将脑MRI解耦为成像信息与不变解剖结构,聚焦于解剖重建。
- 在9个数据集上实现AP+18.32%、DSC+13.64%的性能提升。
- 适合多中心、多模态脑MRI异常检测任务,尤其关注泛化能力。
脑部MRI中各类病灶检测具有重要临床意义,但因病灶多样性及成像条件差异而极具挑战。现有无监督方法主要通过正常样本学习,将异常图像重建为伪健康图像(PHI),并分析差异进行检测。然而,这些方法存在两大局限:受限于正常数据中的特定成像信息,难以泛化至多中心、多模态MRI;且异常残差从输入传播至重建的PHI,限制性能。为此,本文提出两个新模块,构建新型PHI重建框架。首先,解耦表示模块通过引入脑解剖先验和可微分独热编码算子,将脑部MRI解耦为成像信息与成像不变的解剖结构,确保重建聚焦解剖。其次,边缘到图像恢复模块利用解剖图像的高频边缘信息重建高质量的健康图像,并重新融合解耦的成像信息。该模块通过仅输入边缘信息,减少异常像素输入,抑制异常残差,同时保留结构细节以高效重建正常区域。在包含4,443例患者、来自多个中心的九个公开数据集上评估,本方法超越17种最先进方法,在平均精度(AP)和骰子系数(DSC)上分别提升+18.32%和+13.64%。
原文摘要 · Abstract (English)
Detection of various lesions in brain MRI is clinically critical, but challenging due to the diversity of lesions and variability in imaging conditions. Current unsupervised learning methods detect anomalies mainly through reconstructing abnormal images into pseudo-healthy images (PHIs) by normal samples learning and then analyzing differences between images. However, these unsupervised models face two significant limitations: restricted generalizability to multi-modality and multi-center MRIs due to their reliance on the specific imaging information in normal training data, and constrained performance due to abnormal residuals propagated from input images to reconstructed PHIs. To address these limitations, two novel modules are proposed, forming a new PHI reconstruction framework. Firstly, the disentangled representation module is proposed to improve generalizability by decoupling brain MRI into imaging information and essential imaging-invariant anatomical images, ensuring that the reconstruction focuses on the anatomy. Specifically, brain anatomical priors and a differentiable one-hot encoding operator are introduced to constrain the disentanglement results and enhance the disentanglement stability. Secondly, the edge-to-image restoration module is designed to reconstruct high-quality PHIs by restoring the anatomical representation from the high-frequency edge information of anatomical images, and then recoupling the disentangled imaging information. This module not only suppresses abnormal residuals in PHI by reducing abnormal pixels input through edge-only input, but also effectively reconstructs normal regions using the preserved structural details in the edges. Evaluated on nine public datasets (4,443 patients' MRIs from multiple centers), our method outperforms 17 SOTA methods, achieving absolute improvements of +18.32% in AP and +13.64% in DSC.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。