Abstract:Remarkable progress has been made in image recognition, primarily due to the availability of large-scale annotated datasets and deep convolutional neural networks (CNNs). CNNs enable learning data-driven, highly representative, hierarchical image features from sufficient training data. However, obtaining datasets as comprehensively annotated as ImageNet in the medical imaging domain remains a challenge. There are currently three major techniques that successfully employ CNNs to medical image classification: training the CNN from scratch, using off-the-shelf pre-trained CNN features, and conducting unsupervised CNN pre-training with supervised fine-tuning. Another effective method is transfer learning, i.e., fine-tuning CNN models pre-trained from natural image dataset to medical image tasks. In this paper, we exploit three important, but previously understudied factors of employing deep convolutional neural networks to computer-aided detection problems. We first explore and evaluate different CNN architectures. The studied models contain 5 thousand to 160 million parameters, and vary in numbers of layers. We then evaluate the influence of dataset scale and spatial image context on performance. Finally, we examine when and why transfer learning from pre-trained ImageNet (via fine-tuning) can be useful. We study two specific computer-aided detection (CADe) problems, namely thoraco-abdominal lymph node (LN) detection and interstitial lung disease (ILD) classification. We achieve the state-of-the-art performance on the mediastinal LN detection, and report the first five-fold cross-validation classification results on predicting axial CT slices with ILD categories. Our extensive empirical evaluation, CNN model analysis and valuable insights can be extended to the design of high performance CAD systems for other medical imaging tasks.

Medical image classification with convolutional neural network

Residual Convolutional Neural Networks With Global And Local Pathways For Classification Of Focal Liver Lesions

A deep convolutional neural network architecture for interstitial lung disease pattern classification

Medical Image Classification Using Deep Learning

Improved Convolutional Neural Network for Lung Cancer Detection

A Lung Disease Classification Based on Feature Fusion Convolutional Neural Network with X-ray Image Enhancement.

Deep convolutional neural network based medical image classification for disease diagnosis

An Efficient End-to-End Deep Neural Network for Interstitial Lung Disease Recognition and Classification

Deep Learning for the Classification of Lung Nodules

Multi-class deep learning architecture for classifying lung diseases from chest X-Ray and CT images

CNN-based Classification Framework for Lung Tissues with Auxiliary Information

Deep Learning for Lung Disease Classification Using Transfer Learning and a Customized CNN Architecture with Attention

Automatic classification of medical image modality and anatomical location using convolutional neural network

Using Deep Learning for Classification of Lung Nodules on Computed Tomography Images

Image Classification Toward Lung Cancer Recognition by Learning Deep Quality Model

Self-paced Convolutional Neural Network for Computer Aided Detection in Medical Imaging Analysis

Lung Nodule Diagnosis on 3D Computed Tomography Images Using Deep Convolutional Neural Networks

Deep Convolutional Neural Networks for Computer-Aided Detection: CNN Architectures, Dataset Characteristics and Transfer Learning

Deep CNN models for pulmonary nodule classification: Model modification, model integration, and transfer learning

Computer-Aided Diagnosis (CAD) of Pulmonary Nodule of Thoracic CT Image Using Transfer Learning