3D U-Net在肺栓塞分割中表现最佳,但远端栓塞仍难处理。
Rethinking Pulmonary Embolism Segmentation: A Study of Current Approaches and Challenges with an Open Weight Model
- 用490例独立患者数据评估9种模型,验证3D架构优势
- 3D模型平均Dice达0.81,远超2D模型;远端栓塞分割误差高
- 开源最佳模型权重,推动可复现研究
肺栓塞(PE)是危及生命的疾病,准确及时的检测对患者救治至关重要。然而,我们对现有分割算法的系统性研究发现,当前研究存在诸多局限:数据集小且不一致、缺乏可复现基线、模型间对比不足。本研究构建了一个密集标注的数据集,包含490例患者的CTPA扫描(430例用于训练,60例用于测试)。我们评估了九种常用分割架构(包括基于CNN和ViT的模型),在2D与3D配置下,采用平均骰子系数(mDSC)和平均对称表面距离(ASSD)作为评估指标。结果表明:(1) 基于ResNet编码块的3D U-Net仍是高效架构;(2) 3D模型始终优于2D模型;(3) 所有模型在相同数据集上训练与评估时,错误模式高度一致;(4) 远端栓塞因任务复杂性和高质量数据稀缺,仍具挑战性,凸显需更全面覆盖远端栓塞的数据集。为促进可复现性,最优模型的结构与预训练权重已公开发布于https://github.com/mazurowski-lab/PulmonaryEmbolismSegmentation。
原文摘要 · Abstract (English)
Pulmonary Embolism (PE) is a life-threatening condition for which accurate and timely detection is critical to patient care. However, our systematic study of PE segmentation algorithms reveals concerning limitations in the current state of research. Challenges such as small and inconsistent datasets, a lack of reproducible baselines, and limited comparative evaluation across models are hindering progress in the field. In this study, we curated a densely annotated dataset comprising 490 CTPA scans, each from a unique patient (430 for training and 60 for testing). We evaluated nine widely used segmentation architectures, including both CNN- and ViT-based models, in 2D and 3D configurations, using mean Dice Similarity Coefficient (mDSC) and Average Symmetric Surface Distance (ASSD) as evaluation metrics. Furthermore, the highest-performing model was evaluated on a public dataset without fine-tuning and achieved reasonable generalization performance. Our results show that: (1) a 3D U-Net with ResNet encoding blocks remains a highly effective architecture for PE segmentation; (2) 3D models consistently outperform their 2D counterparts; (3) across all architectures, when trained and evaluated on the same datasets, model error patterns are highly consistent; and (4) distal emboli remain particularly challenging due to both task complexity and the scarcity of high-quality datasets, highlighting the need for datasets with more comprehensive and consistent distal PE coverage. To promote research reproducibility, the architecture and pretrained weights of our best-performing model are publicly available at https://github.com/mazurowski-lab/PulmonaryEmbolismSegmentation
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。