GUSL用无反向传播的模型实现前列腺精准分割,兼顾高精度与可解释性。
GUSL: A Novel and Efficient Machine Learning Model for Prostate Segmentation on MRI
- 采用多层回归的粗到精分割策略,线性特征提取确保透明可解释。
- 在三个数据集上均实现>0.9的骰子系数,且模型尺寸仅为其他模型的几分之一。
- 适合临床部署,特别适合关注可解释性与低资源消耗的医学影像场景。
前列腺及分区分割是前列腺癌临床诊断的关键步骤。现有计算机辅助诊断工具多基于深度学习,但深层神经网络被医生视为“黑箱”,难以在临床落地。本文提出一种前馈式机器学习模型Green U-shaped Learning(GUSL),适用于无需反向传播的医学图像分割。GUSL采用多层回归实现粗到精分割,其特征提取基于线性模型,保障了特征提取过程的可解释性。同时,通过回归对前列腺边界区域进行残差修正,提升易错区域的分割精度。此外,采用两步流程缓解医学图像中固有的类别不平衡问题。在两个公开数据集和一个私有数据集上的实验表明,GUSL在前列腺腺体与分区分割任务中均达到当前最优性能。值得注意的是,该模型极为节能,模型规模为其他方案的数倍小,复杂度显著降低。所有数据集上,腺体分割的骰子相似系数(DSC)均超过0.9。结合其轻量化与可解释特性,GUSL为医学影像应用提供了兼具竞争力与实用性的解决方案。
原文摘要 · Abstract (English)
Prostate and zonal segmentation is a crucial step for clinical diagnosis of prostate cancer (PCa). Computer-aided diagnosis tools for prostate segmentation are based on the deep learning (DL) paradigm. However, deep neural networks are perceived as "black-box" solutions by physicians, thus making them less practical for deployment in the clinical setting. In this paper, we introduce a feed-forward machine learning model, named Green U-shaped Learning (GUSL), suitable for medical image segmentation without backpropagation. GUSL introduces a multi-layer regression scheme for coarse-to-fine segmentation. Its feature extraction is based on a linear model, which enables seamless interpretability during feature extraction. Also, GUSL introduces a mechanism for attention on the prostate boundaries, which is an error-prone region, by employing regression to refine the predictions through residue correction. In addition, a two-step pipeline approach is used to mitigate the class imbalance, an issue inherent in medical imaging problems. After conducting experiments on two publicly available datasets and one private dataset, in both prostate gland and zonal segmentation tasks, GUSL achieves state-of-the-art performance among other DL-based models. Notably, GUSL features a very energy-efficient pipeline, since it has a model size several times smaller and less complexity than the rest of the solutions. In all datasets, GUSL achieved a Dice Similarity Coefficient (DSC) performance greater than $0.9$ for gland segmentation. Considering also its lightweight model size and transparency in feature extraction, it offers a competitive and practical package for medical imaging applications.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。