Essential parameters needed for a U-Net-based segmentation of individual bones on planning CT images in the head & neck region using limited datasets for radiotherapy application

Ama Katseena Yawson,Alexander Walter,Nora Wolf,Sebastian Klüter,Philipp Hoegen,Sebastian Adeberg,Juergen Debus,Martin Frank,Oliver Jaekel,Kristina Giske
DOI: https://doi.org/10.1088/1361-6560/ad1996
IF: 3.5
2024-01-03
Physics in Medicine and Biology
Abstract:Objective: The field of radiotherapy is highly marked by the lack of datasets even with the availability of public datasets. Our study uses a very limited dataset to provide insights on essential parameters needed to automatically and accurately segment individual bones on planning CT images of head & neck cancer patients. Approach: The study was conducted using 30 planning CT images of real patients acquired from 5 different cohorts. 15 cases from 4 cohorts were randomly selected as training and validation datasets while the remaining were used as test datasets. Four experimental sets were formulated to explore parameters such as background patch reduction, class-dependent augmentation and incorporation of a weight map on the loss function. Main Results: Our best experimental scenario resulted in a mean Dice score of 0.93 ± 0.06 for other bones (skull, mandible, scapulae, clavicles, humeri and hyoid), 0.93 ± 0.02 for ribs and 0.88 ± 0.03 for vertebrae on 7 test cases from the same cohorts as the training datasets. We compared our proposed solution approach to a retrained nnU-Net and obtained comparable results for vertebral bones while outperforming in the correct identification of the left and right instances of ribs, scapulae, humeri and clavicles. Furthermore, we evaluated the generalization capability of our proposed model on a new cohort and the mean Dice score yielded 0.96 ± 0.10 for other bones, 0.95 ± 0.07 for ribs and 0.81 ± 0.19 for vertebrae on 8 test cases. Significance: With these insights, we are challenging the utilization of an automatic and accurate bone segmentation tool into the clinical routine of radiotherapy despite the limited training datasets.
engineering, biomedical,radiology, nuclear medicine & medical imaging
What problem does this paper attempt to address?
The problem that this paper attempts to solve is to perform automatic and accurate segmentation of individual bones in the planning CT images of the head - and - neck region using a limited data set in radiotherapy applications. Specifically, the research aims to explore the necessary parameters required to achieve accurate prediction of all individual bones in the planning CT images of head - and - neck cancer patients under a very limited training data set. ### Problem Background Radiotherapy is an important part of anti - cancer treatment and usually involves many tasks that can be automated or at least partially supported by machine - learning techniques. Among them, bone delineation is an important medical imaging tool used to help clinicians assess the metastatic state of cancer, assist in formulating radiotherapy plans, etc. However, manual bone delineation is a labor - intensive, time - consuming process and is prone to inter - observer variability. Therefore, developing a tool that can automatically and accurately segment individual bones is of great significance. ### Main Challenges 1. **Limited Data Set**: There is a lack of sufficient data sets in the field of radiotherapy, and even public data sets can hardly meet the demand. 2. **Large Bone Variations**: Human bones have a wide range of variations in shape, size, and composition, especially in parts such as the spine and skull. 3. **Image Quality Limitations**: Problems inherent in CT images, such as low signal - to - noise ratio, poor spatial resolution, and metal artifacts, make it difficult for traditional image processing methods to effectively segment individual bones. ### Research Objectives This paper explores the key parameters for achieving accurate segmentation of individual bones in the head - and - neck region under the condition of a limited data set by using a U - Net - based deep - learning model. The research also compares the performance of the proposed method with benchmark segmentation tools (such as nnU - Net and Totalsegmentator) and evaluates the generalization ability of the model under different scanners and acquisition parameters. ### Key Contributions 1. **Parameter Optimization**: Optimize the model training process by means of reducing background patches, introducing class - dependent data augmentation and weight maps, etc. 2. **High - Precision Segmentation**: Achieve a relatively high Dice coefficient on the test data set, especially showing outstanding performance on complex structures such as ribs and vertebrae. 3. **Generalization Ability Verification**: Verify the generalization ability of the model through new cohort data, proving that it can maintain a good segmentation effect under different devices and acquisition parameters. In conclusion, this research provides valuable insights into how to achieve automatic and accurate segmentation of individual bones in the head - and - neck region under the condition of a limited data set, and is expected to promote the development of automated tools in radiotherapy.