Optimizing convolutional neural networks for Chronic Obstructive Pulmonary Disease detection in clinical computed tomography imaging
Tina Dorosti, Manuel Schultheiß, Felix Hofmann, Johannes Thalhammer, Luisa Kirchner, Theresa Urban, Franz Pfeiffer, Florian Schaff, Tobias Lasser, Daniela Pfeiffer
Abstract
We aim to optimize the binary detection of Chronic Obstructive Pulmonary Disease (COPD) based on emphysema presence in the lung with convolutional neural networks (CNN) by exploring manually adjusted versus automated window-setting optimization (WSO) on computed tomography (CT) images. 7194 contrast-enhanced CT images (3597 with COPD; 3597 healthy controls) from 78 subjects were selected retrospectively (01.2018–12.2021) and preprocessed. For each image, intensity values were manually clipped to the emphysema window setting and a baseline ‘full-range’ window setting. Class-balanced train, validation, and test sets contained 3392, 1114, and 2688 images. The network backbone was optimized by comparing various CNN architectures. Furthermore, automated WSO was implemented by adding a customized layer to the model. The image-level area under the Receiver Operating Characteristics curve (AUC) [lower, upper limit 95% confidence] was utilized to compare model variations. Repeated inference (n = 7) on the test set showed that the DenseNet was the most efficient backbone and achieved a mean AUC of 0.80 [0.76, 0.85] without WSO. Comparably, with input images manually adjusted to the emphysema window, the DenseNet model predicted COPD with a mean AUC of 0.86 [0.82, 0.89]. By adding a customized WSO layer to the DenseNet, an optimal window in the proximity of the emphysema window setting was learned automatically, and a mean AUC of 0.82 [0.78, 0.86] was achieved. Detection of COPD with DenseNet models was improved by WSO of CT data to the emphysema window setting range. • Manual versus automated window-setting optimization for COPD detection in CT images. • Manual preprocessing to emphysema window setting improved binary detection of COPD. • Window-setting optimization was automated by adding a customized layer to the CNN. • Automated window-setting optimization learned settings near the emphysema window.