Deep learning-based prediction of nodal metastasis in lung cancer using endobronchial ultrasound.

OBJECTIVE

Endobronchial ultrasound-guided transbronchial needle aspiration is a vital tool for mediastinal and hilar lymph node staging in patients with lung cancer. Despite its high diagnostic performance and safety, it has a limited negative predictive value. Our objective was to evaluate the diagnostic performance of deep learning-based prediction of lung cancer lymph node metastases using convolutional neural networks developed from automatically extracted images of endobronchial ultrasound videos without supervision of the lymph node location.

METHODS

Patient and lymph node data were collected from a single-center database. The diagnosis of metastasis was confirmed with endobronchial ultrasound-guided transbronchial needle aspiration and/or surgically resected specimens; the diagnosis of normal lymph node was confirmed with surgically resected specimens only. An annotation system facilitated automated image extraction from endobronchial ultrasound videos. Image frames were randomly selected and split into training and validation datasets on a per-patient basis. A deep learning model with convolutional neural networks, SqueezeNet, was used for image classification via transfer learning based on pretraining from ImageNet. Adaptive moment estimation and stochastic gradient descent were applied as optimizers.

RESULTS

SqueezeNet, with adaptive moment estimation, achieved a sensitivity, specificity, accuracy, positive predictive value, and negative predictive value of 96.7% each after 300 epochs, whereas SqueezeNet with stochastic gradient descent achieved 91.1% each. However, SqueezeNet with stochastic gradient descent demonstrated more stable performance than with adaptive moment estimation.

CONCLUSIONS

Deep learning-based image classification using convolutional neural networks showed promising diagnostic accuracy for lung cancer nodal metastasis. Future clinical trials are warranted to validate the algorithm's efficacy in a prospective, large-cohort study.