Generalizable deep learning framework for 3D medical image segmentation using limited training data

被引:0
作者
Ekman, Tobias [1 ,2 ]
Barakat, Arthur [1 ,2 ]
Heiberg, Einar [1 ,2 ,3 ]
机构
[1] Lund Univ, Dept Med Imaging & Physiol, Lund, Sweden
[2] Skane Univ Hosp, Dept Med Imaging & Physiol, Lund, Sweden
[3] Lund Univ, Wallenberg Ctr Mol Med, Lund, Sweden
基金
瑞典研究理事会;
关键词
Segmentation; Machine learning; Artificial intelligence; Deep learning; 3D printing;
D O I
10.1186/s41205-025-00254-1
中图分类号
R8 [特种医学]; R445 [影像诊断学];
学科分类号
1002 ; 100207 ; 1009 ;
摘要
Medical image segmentation is a critical component in a wide range of clinical applications, enabling the identification and delineation of anatomical structures. This study focuses on segmentation of anatomical structures for 3D printing, virtual surgery planning, and advanced visualization such as virtual or augmented reality. Manual segmentation methods are labor-intensive and can be subjective, leading to inter-observer variability. Machine learning algorithms, particularly deep learning models, have gained traction for automating the process and are now considered state-of-the-art. However, deep-learning methods typically demand large datasets for fine-tuning and powerful graphics cards, limiting their applicability in resource-constrained settings. In this paper we introduce a robust deep learning framework for 3D medical segmentation that achieves high performance across a range of medical segmentation tasks, even when trained on a small number of subjects. This approach overcomes the need for extensive data and heavy GPU resources, facilitating adoption within healthcare systems. The potential is exemplified through six different clinical applications involving orthopedics, orbital segmentation, mandible CT, cardiac CT, fetal MRI and lung CT. Notably, a small set of hyper-parameters and augmentation settings produced segmentations with an average Dice score of 92% (SD = +/- 0.06) across a diverse range of organs and tissues.
引用
收藏
页数:18
相关论文
共 15 条
[1]   Deep learning can yield clinically useful right ventricular segmentations faster than fully manual analysis [J].
Akesson, Julius ;
Ostenfeld, Ellen ;
Carlsson, Marcus ;
Arheden, Hakan ;
Heiberg, Einar .
SCIENTIFIC REPORTS, 2023, 13 (01)
[2]   Semi-automated segmentation of pre-operative low grade gliomas in magnetic resonance imaging [J].
Akkus, Zeynettin ;
Sedlar, Jiri ;
Coufalova, Lucie ;
Korfiatis, Panagiotis ;
Kline, Timothy L. ;
Warner, Joshua D. ;
Agrawal, Jay ;
Erickson, Bradley J. .
CANCER IMAGING, 2015, 15 :1-10
[3]   Review of deep learning: concepts, CNN architectures, challenges, applications, future directions [J].
Alzubaidi, Laith ;
Zhang, Jinglan ;
Humaidi, Amjad J. ;
Al-Dujaili, Ayad ;
Duan, Ye ;
Al-Shamma, Omran ;
Santamaria, J. ;
Fadhel, Mohammed A. ;
Al-Amidie, Muthana ;
Farhan, Laith .
JOURNAL OF BIG DATA, 2021, 8 (01)
[4]   The role of hyperparameters in machine learning models and how to tune them [J].
Arnold, Christian ;
Biedebach, Luka ;
Kuepfer, Andreas ;
Neunhoeffer, Marcel .
POLITICAL SCIENCE RESEARCH AND METHODS, 2024, 12 (04) :841-848
[5]  
Cicek Ozgun, 2016, 3D U-Net: Learning Dense Volumetric Segmentation from Sparse Annotation, P424
[6]   The Cancer Imaging Archive (TCIA): Maintaining and Operating a Public Information Repository [J].
Clark, Kenneth ;
Vendt, Bruce ;
Smith, Kirk ;
Freymann, John ;
Kirby, Justin ;
Koppel, Paul ;
Moore, Stephen ;
Phillips, Stanley ;
Maffitt, David ;
Pringle, Michael ;
Tarbox, Lawrence ;
Prior, Fred .
JOURNAL OF DIGITAL IMAGING, 2013, 26 (06) :1045-1057
[7]   Algorithms used in medical image segmentation for 3D printing and how to understand and quantify their performance [J].
Fogarasi, Magdalene ;
Coburn, James C. ;
Ripley, Beth .
3D PRINTING IN MEDICINE, 2022, 8 (01)
[8]  
Isensee F., 2018, arXiv: arXiv:1809.10486, parXiv.1809.10486, DOI [DOI 10.48550/ARXIV.1809.10486, 10.48550/arxiv.1809.10486]
[9]   Redefining Radiology: A Review of Artificial Intelligence Integration in Medical Imaging [J].
Najjar, Reabal .
DIAGNOSTICS, 2023, 13 (17)
[10]  
Nogus I., 2019, Insights Imaging, V10, P1