arXiv:2412.07156eess.IVcs.CV2024-12被引 5

提出QCResUNet模型,同时预测脑肿瘤分割整体质量与局部错误区域。

QCResUNet: Joint Subject-level and Voxel-level Segmentation Quality Prediction

  • 多任务网络同时输出患者级和体素级质量评估结果。
  • 在脑肿瘤与心脏MRI数据上均达到高精度的错误定位效果。
  • 适合临床辅助诊断,支持人工修正关键错误区域。

近年来深度学习在磁共振成像(MRI)脑肿瘤自动分割方面取得显著进展。然而,这些工具的可靠性受制于低质量分割异常值的影响,尤其是在分布外样本中,限制了其在临床中的应用。因此,亟需质量控制(QC)机制来筛查分割结果质量。尽管已有多种自动QC方法被提出,但多数针对单一模态、单组织类型的心脏MRI分割,且仅提供患者级质量判断,无法定位需要修正的具体错误区域。为此,本文提出一种新型多任务深度学习架构QCResUNet,可生成患者级分割质量指标及每个组织类别的体素级分割误差图。我们在两个不同分割任务上验证该方法:一是脑肿瘤分割,使用一个内部数据集和两个外部数据集;二是自动化心脏诊断挑战赛中的心脏MRI分割。实验表明,所提方法在预测患者级质量指标及精准识别体素级错误方面表现优异,具有在临床环境中指导人机协作改进分割结果的潜力。

原文摘要 · Abstract (English)

Deep learning has made significant strides in automated brain tumor segmentation from magnetic resonance imaging (MRI) scans in recent years. However, the reliability of these tools is hampered by the presence of poor-quality segmentation outliers, particularly in out-of-distribution samples, making their implementation in clinical practice difficult. Therefore, there is a need for quality control (QC) to screen the quality of the segmentation results. Although numerous automatic QC methods have been developed for segmentation quality screening, most were designed for cardiac MRI segmentation, which involves a single modality and a single tissue type. Furthermore, most prior works only provided subject-level predictions of segmentation quality and did not identify erroneous parts segmentation that may require refinement. To address these limitations, we proposed a novel multi-task deep learning architecture, termed QCResUNet, which produces subject-level segmentation-quality measures as well as voxel-level segmentation error maps for each available tissue class. To validate the effectiveness of the proposed method, we conducted experiments on assessing its performance on evaluating the quality of two distinct segmentation tasks. First, we aimed to assess the quality of brain tumor segmentation results. For this task, we performed experiments on one internal and two external datasets. Second, we aimed to evaluate the segmentation quality of cardiac Magnetic Resonance Imaging (MRI) data from the Automated Cardiac Diagnosis Challenge. The proposed method achieved high performance in predicting subject-level segmentation-quality metrics and accurately identifying segmentation errors on a voxel basis. This has the potential to be used to guide human-in-the-loop feedback to improve segmentations in clinical settings.

脑肿瘤分割质量评估多任务学习

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。