用扩散模型自适应检测医学影像分割质量,跨器官通用且鲁棒。
Diffusion-Based Quality Control of Medical Image Segmentations across Organs
- 基于扩散生成框架,双专家独立编码空间与解剖信息
- 在严重退化或缺失掩码下仍保持高精度,优于现有方法
- 无需重新训练,可适配不同器官、数据集和模态
深度学习在医学图像分割中推动了大规模人群研究的自动化分析流程发展。然而,当前先进方法易产生幻觉,导致解剖学上不合理的分割结果。由于人工校正难以规模化,需依赖自动质量控制(QC)技术。尽管已有方法表现良好,但其多为器官特定,泛化能力受限。为此,本文提出无新训练质量控制(nnQC),一种基于扩散生成范式的稳健框架,可自适应任意输入器官数据集。nnQC的核心是新型专家团队(ToE)架构:两名专家分别独立编码三维空间感知(通过轴向切片相对位置)与源自原始图像的视觉特征所表征的解剖信息。加权条件模块动态融合这两类独立嵌入(即“意见”),以条件化扩散过程中的采样机制,从而生成具空间感知的伪真实标签,用于预测质量评分。该框架还集成指纹自适应机制,确保在不同器官、数据集和成像模态间的适应性。我们在十二个公开数据集上对七个器官进行了评估。结果表明,nnQC在所有实验中持续超越现有最先进方法,包括分割掩码高度退化或完全缺失的情况,验证了其跨器官的通用性与有效性。
原文摘要 · Abstract (English)
Medical image segmentation using deep learning (DL) has enabled the development of automated analysis pipelines for large-scale population studies. However, state-of-the-art DL methods are prone to hallucinations, which can result in anatomically implausible segmentations. With manual correction impractical at scale, automated quality control (QC) techniques have to address the challenge. While promising, existing QC methods are organ-specific, limiting their generalizability and usability beyond their original intended task. To overcome this limitation, we propose no-new Quality Control (nnQC), a robust QC framework based on a diffusion-generative paradigm that self-adapts to any input organ dataset. Central to nnQC is a novel Team of Experts (ToE) architecture, where two specialized experts independently encode 3D spatial awareness, represented by the relative spatial position of an axial slice, and anatomical information derived from visual features from the original image. A weighted conditional module dynamically combines the pair of independent embeddings, or opinions to condition the sampling mechanism within a diffusion process, enabling the generation of a spatially aware pseudo-ground truth for predicting QC scores. Within its framework, nnQC integrates fingerprint adaptation to ensure adaptability across organs, datasets, and imaging modalities. We evaluated nnQC on seven organs using twelve publicly available datasets. Our results demonstrate that nnQC consistently outperforms state-of-the-art methods across all experiments, including cases where segmentation masks are highly degraded or completely missing, confirming its versatility and effectiveness across different organs.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。