Lung Nodule Malignancy Prediction From Longitudinal CT Scans With Siamese Convolutional Attention Networks

IEEE Open J Eng Med Biol. 2020 Sep 11:1:257-264. doi: 10.1109/OJEMB.2020.3023614. eCollection 2020.

Abstract

Goal: We propose a convolutional attention-based network that allows for use of pre-trained 2-D convolutional feature extractors and is extendable to multi-time-point classification in a Siamese structure. Methods: Our proposed framework is evaluated for single- and multi-time-point classification to explore the value that temporal information, such as nodule growth, adds to malignancy prediction. Results: Our results show that the proposed method outperforms a comparable 3-D network with less than half the parameters on single-time-point classification and further achieves performance gains on multi-time-point classification. Conclusions: Attention-based, Siamese 2-D pre-trained CNNs lead to fast training times and are effective for malignancy prediction from single-time-point or multiple-time-point imaging data.

Keywords: Lung cancer diagnosis; X-ray CT; deep learning; longitudinal studies; siamese networks.

Grants and funding

This work was partially supported by the University of Louisville Endowment for Bioimaging.