Early severity prediction of BPD for premature infants from chest X-ray images using deep learning: A study at the 28th day of oxygen inhalation

Abstract

Background and objective: Bronchopulmonary dysplasia is a common respiratory disease in premature infants. The severity is diagnosed at the 56th day after birth or discharge by analyzing the clinical indicators, which may cause the delay of the best treatment opportunity. Thus, we proposed a deep learning-based method using chest X-ray images of the 28th day of oxygen inhalation for the early severity prediction of bronchopulmonary dysplasia in clinic.

Methods: We first adopted a two-step lung field extraction method by combining digital image processing and human-computer interaction to form the one-to-one corresponding image and label. The designed XSEG-Net model was then trained for segmenting the chest X-ray images, with the results being used for the analysis of heart development and clinical severity. Therein, Six-Point cardiothoracic ratio measurement algorithm based on corner detection was designed for the analysis of heart development; and the transfer learning of deep convolutional neural network models were used for the early prediction of clinical severities.

Results: The dice and cross-entropy loss value of the training of XSEG-Net network reached 0.9794 and 0.0146. The dice, volumetric overlap error, relative volume difference, precision, and recall were used to evaluate the trained model in testing set with the result being 98.43 ± 0.39%, 0.49 ± 0.35%, 0.49 ± 0.35%, 98.67 ± 0.40%, and 98.20 ± 0.47%, respectively. The errors between the Six-Point cardiothoracic ratio measurement method and the gold standard were 0.0122 ± 0.0084. The deep convolutional neural network model based on VGGNet had the promising prediction performance, with the accuracy, precision, sensitivity, specificity, and F1 score reaching 95.58 ± 0.48%, 95.61 ± 0.55%, 95.67 ± 0.44%, 96.98 ± 0.42%, and 95.61±0.48%, respectively.

Conclusions: These experimental results of the proposed methods in lung field segmentation, cardiothoracic ratio measurement and clinic severity prediction were better than previous methods, which proved that this method had great potential for clinical application.

Keywords: Bronchopulmonary dysplasia; Cardiothoracic ratio measurement; Deep learning; Early prediction; Image segmentation.