# Gated-Dilated Networks for Lung Nodule Classification in CT scans

**Authors:** Mundher Al-Shabi, Hwee Kuan Lee, Maxine Tan

arXiv: 1901.00120 · 2019-12-17

## TL;DR

This paper introduces Gated-Dilated networks, a novel CNN architecture that effectively classifies lung nodules as malignant or benign by capturing size variations through dilated convolutions and a context-aware mechanism, outperforming existing models.

## Contribution

The study proposes a new CNN architecture with dilated convolutions and a context-aware sub-network, improving lung nodule classification accuracy over state-of-the-art models.

## Key findings

- GD network achieves >0.95 AUC on LIDC dataset.
- Outperforms baseline models like ResNet and DenseNet.
- Improves classification of mid-sized nodules.

## Abstract

Different types of Convolutional Neural Networks (CNNs) have been applied to detect cancerous lung nodules from computed tomography (CT) scans. However, the size of a nodule is very diverse and can range anywhere between 3 and 30 millimeters. The high variation of nodule sizes makes classifying them a difficult and challenging task. In this study, we propose a novel CNN architecture called Gated-Dilated (GD) networks to classify nodules as malignant or benign. Unlike previous studies, the GD network uses multiple dilated convolutions instead of max-poolings to capture the scale variations. Moreover, the GD network has a Context-Aware sub-network that analyzes the input features and guides the features to a suitable dilated convolution. We evaluated the proposed network on more than 1,000 CT scans from the LIDC-LDRI dataset. Our proposed network outperforms state-of-the-art baseline models including Multi-Crop, Resnet, and Densenet, with an AUC of >0.95. Compared to the baseline models, the GD network improves the classification accuracies of mid-range sized nodules. Furthermore, we observe a relationship between the size of the nodule and the attention signal generated by the Context-Aware sub-network, which validates our new network architecture.

---
Source: https://tomesphere.com/paper/1901.00120