Multi-source Transfer Learning with Convolutional Neural Networks for Lung Pattern Analysis

Early diagnosis of interstitial lung diseases is crucial for their treatment, but even experienced physicians find it difficult, as their clinical manifestations are similar. In order to assist with the diagnosis, computer-aided diagnosis systems have been developed. These commonly rely on a fixed scale classifier that scans CT images, recognizes textural lung patterns, and generates a map of pathologies. In a previous study, we proposed a method for classifying lung tissue patterns using a deep convolutional neural network (CNN), with an architecture designed for the specific problem. In this study, we present an improved method for training the proposed network by transferring knowledge from the similar domain of general texture classification. Six publicly available texture databases are used to pretrain networks with the proposed architecture, which are then fine-tuned on the lung tissue data. The resulting CNNs are combined in an ensemble and their fused knowledge is compressed back to a network with the original architecture. The proposed approach resulted in an absolute increase of about 2% in the performance of the proposed CNN. The results demonstrate the potential of transfer learning in the field of medical image analysis, indicate the textural nature of the problem and show that the method used for training a network can be as important as designing its architecture.

[1]  Andrew Zisserman,et al.  Very Deep Convolutional Networks for Large-Scale Image Recognition , 2014, ICLR.

[2]  B. Society,et al.  The diagnosis, assessment and treatment of diffuse parenchymal lung disease in adults. Introduction. , 1999, Thorax.

[3]  K. Doi,et al.  Quantitative computerized analysis of diffuse lung disease in high-resolution computed tomography. , 2003, Medical physics.

[4]  B. van Ginneken,et al.  Computer-aided diagnosis in high resolution CT of the lungs. , 2003, Medical physics.

[5]  John Salvatier,et al.  Theano: A Python framework for fast computation of mathematical expressions , 2016, ArXiv.

[6]  Arcot Sowmya,et al.  Multiple kernel learning for classification of diffuse lung disease using HRCT lung images , 2010, 2010 Annual International Conference of the IEEE Engineering in Medicine and Biology.

[7]  Ronald M. Summers,et al.  Deep Learning in Medical Imaging: Overview and Future Promise of an Exciting New Technique , 2016 .

[8]  Richard A. Robb,et al.  High resolution multidetector CT aided tissue analysis and quantification of lung fibrosis , 2006, SPIE Medical Imaging.

[9]  Qi Tian,et al.  Good Practice in CNN Feature Transfer , 2016, ArXiv.

[10]  Geoffrey E. Hinton,et al.  Distilling the Knowledge in a Neural Network , 2015, ArXiv.

[11]  Thomas G. Dietterich Multiple Classifier Systems , 2000, Lecture Notes in Computer Science.

[12]  Arcot Sowmya,et al.  Multiscale sparse representation of high-resolution computed tomography (HRCT) lung images for diffuse lung disease classification , 2011, 2011 18th IEEE International Conference on Image Processing.

[13]  E. Hoffman,et al.  Computer recognition of regional lung disease patterns. , 1999, American journal of respiratory and critical care medicine.

[14]  Barbara Caputo,et al.  Class-Specific Material Categorisation , 2005, ICCV.

[15]  Lauge Sørensen,et al.  Quantitative Analysis of Pulmonary Emphysema Using Local Binary Patterns , 2010, IEEE Transactions on Medical Imaging.

[16]  Edward H. Adelson,et al.  Material perception: What can you see in a brief glance? , 2010 .

[17]  Ronald M. Summers,et al.  Deep Convolutional Neural Networks for Computer-Aided Detection: CNN Architectures, Dataset Characteristics and Transfer Learning , 2016, IEEE Transactions on Medical Imaging.

[18]  Marleen de Bruijne,et al.  Combining Generative and Discriminative Representation Learning for Lung CT Analysis With Convolutional Restricted Boltzmann Machines , 2016, IEEE Transactions on Medical Imaging.

[19]  Atsuto Maki,et al.  From generic to specific deep representations for visual recognition , 2014, 2015 IEEE Conference on Computer Vision and Pattern Recognition Workshops (CVPRW).

[20]  SchmidCordelia,et al.  A Sparse Texture Representation Using Local Affine Regions , 2005 .

[21]  Rich Caruana,et al.  Multitask Learning , 1998, Encyclopedia of Machine Learning and Data Mining.

[22]  Yoshua Bengio,et al.  How transferable are features in deep neural networks? , 2014, NIPS.

[23]  Jimmy Ba,et al.  Adam: A Method for Stochastic Optimization , 2014, ICLR.

[24]  Bram van Ginneken,et al.  Computer analysis of computed tomography scans of the lung: a survey , 2006, IEEE Transactions on Medical Imaging.

[25]  David Dagan Feng,et al.  Lung image patch classification with automatic feature learning , 2013, 2013 35th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC).

[26]  Marios Anthimopoulos,et al.  Classification of interstitial lung disease patterns using local DCT features and random forest , 2014, 2014 36th Annual International Conference of the IEEE Engineering in Medicine and Biology Society.

[27]  Rui Xu,et al.  Classification of diffuse lung diseases patterns by a sparse representation based method on HRCT images , 2013, 2013 35th Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC).

[28]  Marios Anthimopoulos,et al.  Lung Pattern Classification for Interstitial Lung Diseases Using a Deep Convolutional Neural Network , 2016, IEEE Transactions on Medical Imaging.

[29]  Michael Unser,et al.  Optimized steerable wavelets for texture analysis of lung tissue in 3-D CT: Classification of usual interstitial pneumonia , 2015, 2015 IEEE 12th International Symposium on Biomedical Imaging (ISBI).

[30]  Iasonas Kokkinos,et al.  Describing Textures in the Wild , 2013, 2014 IEEE Conference on Computer Vision and Pattern Recognition.

[31]  Lauge Sørensen,et al.  A Texton-Based Approach for the Classification of Lung Parenchyma in CT Images , 2010, MICCAI.

[32]  Cordelia Schmid,et al.  A sparse texture representation using local affine regions , 2005, IEEE Transactions on Pattern Analysis and Machine Intelligence.

[33]  Antoine Geissbühler,et al.  Building a reference multimedia database for interstitial lung diseases , 2012, Comput. Medical Imaging Graph..

[34]  Antonio Torralba,et al.  Learning Aligned Cross-Modal Representations from Weakly Aligned Data , 2016, 2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR).

[35]  Stefan Carlsson,et al.  CNN Features Off-the-Shelf: An Astounding Baseline for Recognition , 2014, 2014 IEEE Conference on Computer Vision and Pattern Recognition Workshops.

[36]  Lena Costaridou,et al.  Texture-Based Identification and Characterization of Interstitial Pneumonia Patterns in Lung Multidetector CT , 2010, IEEE Transactions on Information Technology in Biomedicine.

[37]  Rich Caruana,et al.  Ensemble selection from libraries of models , 2004, ICML.

[38]  Henning Müller,et al.  Using Multiscale Visual Words for Lung Texture Classification and Retrieval , 2011, MCBR-CDS.

[39]  Gertjan J. Burghouts,et al.  Material-specific adaptation of color invariant features , 2009, Pattern Recognit. Lett..

[40]  Rich Caruana,et al.  Model compression , 2006, KDD '06.

[41]  Gustaf Kylberg,et al.  Kylberg Texture Dataset v. 1.0 , 2011 .

[42]  Geoffrey E. Hinton,et al.  ImageNet classification with deep convolutional neural networks , 2012, Commun. ACM.

[43]  Yoshua Bengio,et al.  Gradient-based learning applied to document recognition , 1998, Proc. IEEE.

[44]  Dimitri Van De Ville,et al.  Near-Affine-Invariant Texture Learning for Lung Tissue Analysis Using Isotropic Wavelet Frames , 2012, IEEE Transactions on Information Technology in Biomedicine.

[45]  Xiaogang Wang,et al.  Medical image classification with convolutional neural network , 2014, 2014 13th International Conference on Control Automation Robotics & Vision (ICARCV).

[46]  Qiang Yang,et al.  A Survey on Transfer Learning , 2010, IEEE Transactions on Knowledge and Data Engineering.

[47]  Nima Tajbakhsh,et al.  Convolutional Neural Networks for Medical Image Analysis: Full Training or Fine Tuning? , 2016, IEEE Transactions on Medical Imaging.