Dual center validation of deep learning for automated multi-label segmentation of thoracic anatomy in bedside chest radiographs

Felix Busch*, Lina Xu, Dmitry Sushko, Matthias Weidlich, Daniel Truhn, Gustav Mueller-Franzes, Maurice M. Heimer, Stefan M. Niehues, Marcus R. Makowski, Markus Hinsche, Janis L. Vahldiek, Hugo J. W. L. Aerts, Lisa C. Adams, Keno K. Bressem

*Corresponding author for this work

Research output: Contribution to journalArticleAcademicpeer-review

6 Downloads (Pure)

Abstract

Background and Objectives: Bedside chest radiographs (CXRs) are challenging to interpret but impor-tant for monitoring cardiothoracic disease and invasive therapy devices in critical care and emergency medicine. Taking surrounding anatomy into account is likely to improve the diagnostic accuracy of artifi-cial intelligence and bring its performance closer to that of a radiologist. Therefore, we aimed to develop a deep convolutional neural network for efficient automatic anatomy segmentation of bedside CXRs.Methods: To improve the efficiency of the segmentation process, we introduced a "human-in-the-loop" segmentation workflow with an active learning approach, looking at five major anatomical structures in the chest (heart, lungs, mediastinum, trachea, and clavicles). This allowed us to decrease the time needed for segmentation by 32% and select the most complex cases to utilize human expert annotators effi-ciently. After annotation of 2,0 0 0 CXRs from different Level 1 medical centers at Charite - University Hospital Berlin, there was no relevant improvement in model performance, and the annotation process was stopped. A 5-layer U-ResNet was trained for 150 epochs using a combined soft Dice similarity coeffi-cient (DSC) and cross-entropy as a loss function. DSC, Jaccard index (JI), Hausdorff distance (HD) in mm, and average symmetric surface distance (ASSD) in mm were used to assess model performance. External validation was performed using an independent external test dataset from Aachen University Hospital ( n = 20).Results: The final training, validation, and testing dataset consisted of 1900/50/50 segmentation masks for each anatomical structure. Our model achieved a mean DSC/JI/HD/ASSD of 0.93/0.88/32.1/5.8 for the lung, 0.92/0.86/21.65/4.85 for the mediastinum, 0.91/0.84/11.83/1.35 for the clavicles, 0.9/0.85/9.6/2.19 for the trachea, and 0.88/0.8/31.74/8.73 for the heart. Validation using the external dataset showed an overall robust performance of our algorithm. Conclusions: Using an efficient computer-aided segmentation method with active learning, our anatomy -based model achieves comparable performance to state-of-the-art approaches. Instead of only segmenting the non-overlapping portions of the organs, as previous studies did, a closer approximation to actual anatomy is achieved by segmenting along the natural anatomical borders. This novel anatomy approach could be useful for developing pathology models for accurate and quantifiable diagnosis.(c) 2023 Elsevier B.V. All rights reserved.
Original languageEnglish
Article number107505
Number of pages10
JournalComputer Methods and Programs in Biomedicine
Volume234
Issue number1
DOIs
Publication statusPublished - 1 Jun 2023

Keywords

  • Anatomy
  • Active learning
  • Chest radiograph
  • Artificial intelligence
  • Deep learning
  • Convolutional neural network
  • X-RAY
  • RADIOLOGISTS

Cite this