Domain adaptation and weight initialization of neural networks for diagnosing interstitial lung diseases
Onkar Thorat
Computer Engineering Department, Dwarkadas J. Sanghvi College of Engineering, University of Mumbai, Maharashtra, India
Search for more papers by this authorSiddharth Salvi
Computer Engineering Department, Dwarkadas J. Sanghvi College of Engineering, University of Mumbai, Maharashtra, India
Search for more papers by this authorCorresponding Author
Shrey Dedhia
Computer Engineering Department, Dwarkadas J. Sanghvi College of Engineering, University of Mumbai, Maharashtra, India
Correspondence
Shrey Dedhia, Computer Engineering Department, Dwarkadas J. Sanghvi College of Engineering, University of Mumbai, Maharashtra, India.
Email: [email protected]
Search for more papers by this authorChetashri Bhadane
Computer Engineering Department, Dwarkadas J. Sanghvi College of Engineering, University of Mumbai, Maharashtra, India
Search for more papers by this authorDeepika Dongre
Computer Engineering Department, Dwarkadas J. Sanghvi College of Engineering, University of Mumbai, Maharashtra, India
Search for more papers by this authorOnkar Thorat
Computer Engineering Department, Dwarkadas J. Sanghvi College of Engineering, University of Mumbai, Maharashtra, India
Search for more papers by this authorSiddharth Salvi
Computer Engineering Department, Dwarkadas J. Sanghvi College of Engineering, University of Mumbai, Maharashtra, India
Search for more papers by this authorCorresponding Author
Shrey Dedhia
Computer Engineering Department, Dwarkadas J. Sanghvi College of Engineering, University of Mumbai, Maharashtra, India
Correspondence
Shrey Dedhia, Computer Engineering Department, Dwarkadas J. Sanghvi College of Engineering, University of Mumbai, Maharashtra, India.
Email: [email protected]
Search for more papers by this authorChetashri Bhadane
Computer Engineering Department, Dwarkadas J. Sanghvi College of Engineering, University of Mumbai, Maharashtra, India
Search for more papers by this authorDeepika Dongre
Computer Engineering Department, Dwarkadas J. Sanghvi College of Engineering, University of Mumbai, Maharashtra, India
Search for more papers by this authorAbstract
Interstitial lung diseases (ILDs) are adverse disorders, damaging the lung tissues, thus making timely diagnosis imperative. To counter the scarcity of publicly available high-resolution computed tomography (HRCT) data, architecture employing HRCT and x-ray data have been proposed for diagnosing ILDs. A model is first trained to diagnose three ILDs and a healthy lung from x-ray data and then with a small amount of HRCT data of the same four classes. We introduce an EfficientNet+AlexNet model and a custom deep convolutional neural network and compare their performances with different weight initialization at the start of x-ray image training. The EfficientNet+AlexNet model initialized with ImageNet weights performed best on the x-ray dataset and was later used for domain adaptation. We also compare our model's performance to a single pre-trained EfficientNetB0 model (trained on the same HRCT data). Our model gave an accuracy of 97.4% on the testing dataset, a notable increase in the performance compared to the recently proposed methodologies. The experimental results and graphical plots depict the superiority of domain adaptation and the potentials of weights that are a starting point for training the models.
Open Research
DATA AVAILABILITY STATEMENT
Data sharing is not applicable to this article as no new data were created or analyzed in this study.
REFERENCES
- 1Anthimopoulos M, Christodoulidis S, Ebner L, Christe A, Mougiakakou S. Lung pattern classification for interstitial lung diseases using a deep convolutional neural network. IEEE Trans Med Imaging. 2016; 35(5): 1207-1216. doi:10.1109/TMI.2016.2535865
- 2Litjens G, Kooi T, Bejnordi BE, et al. A survey on deep learning in medical image analysis. Med Image Anal. 2017; 42: 60-88. doi:10.1016/j.media.2017.07.005
- 3McBee MP, Awan OA, Colucci AT, et al. Deep learning in radiology. Acad Radiol. 2018; 25(11): 1472-1480. doi:10.1016/j.acra.2018.02.018
- 4Sahiner B, Pezeshk A, Hadjiiski LM, et al. Deep learning in medical imaging and radiation therapy. Med Phys. 2019; 46(1): e1-e36. doi:10.1002/mp.13264
- 5Trusculescu A, Manolescu D, Tudorache E, Oancea C. Deep learning in interstitial lung disease—how long until daily practice. Eur Radiol. 2020; 30: 6285-6292. doi:10.1007/s00330-020-06986-4
- 6Tan C, Sun F, Kong T, Zhang W, Yang C, Liu C. A Survey on Deep Transfer Learning. Artificial Neural Networks and Machine Learning – ICANN 2018. ICANN 2018. Lecture Notes in Computer Science, vol. 11141. Cham: Springer, 2018.
- 7Pan SJ, Yang Q. A survey on transfer learning. IEEE Trans Knowl Data Eng. 2010; 22(10): 1345-1359. doi:10.1109/TKDE.2009.191
- 8Deng J, Dong W, Socher R, Li LJ, Li K, Fei-Fei L. ImageNet: a large-scale hierarchical image database, 2009 IEEE Conference on Computer Vision and Pattern Recognition, 2009, pp. 248-255, doi: 10.1109/CVPR.2009.5206848.
- 9Cheplygina V. Cats or CAT scans: transfer learning from natural or medical image source data sets? Curr Opin Biomed Eng. 2019; 9: 21-27. doi:10.1016/j.cobme.2018.12.005
10.1016/j.cobme.2018.12.005 Google Scholar
- 10Raghu M, Zhang C, Kleinberg J, Bengio S. Transfusion: understanding transfer learning for medical imaging. Proceedings of the 33rd International Conference on Neural Information Processing Systems. Red Hook, NY, USA: Curran Associates Inc., 2019.
- 11Lo SCB, Lin JS, Freedman MT, Mun SK. Computer-assisted diagnosis of lung nodule detection using artificial convolution neural network. Int Soc Opt Photonics. 1993; 1898: 859-869.
- 12Bondfale N, Bhagwat DS. Convolutional neural network for categorization of lung tissue patterns in interstitial lung diseases, 2018 Second International Conference on Inventive Communication and Computational Technologies (ICICCT). 2018; 1150-1154.
- 13Negahdar M, Coy A, Beymer D. An end-to-end deep learning pipeline for emphysema quantification using multi-label learning, 2019 41st Annual International Conference of the IEEE Engineering in Medicine and Biology Society (EMBC). 2019; 929-932.
- 14Sloun vRJG, Demi L. Localizing B-lines in lung ultrasonography by weakly supervised deep learning, in-vivo results. IEEE J Biomed Health Inform. 2020; 24(4): 957-964. doi:10.1109/JBHI.2019.2936151
- 15Roy S, MenapaceW OS, et al. Deep learning for classification and localization of COVID-19 markers in point-of-care lung ultrasound. IEEE Trans Med Imaging. 2020; 39(8): 2676-2687. doi:10.1109/TMI.2020.2994459
- 16Ke Q, Zhang J, Wei W, et al. A neuro-heuristic approach for recognition of lung diseases from X-ray images. Expert Syst Appl. 2019; 126: 218-232. doi:10.1016/j.eswa.2019.01.060
- 17Capizzi G, Sciuto GL, Napoli C, Połap D, Wo1niak M. Small lung nodules detection based on fuzzy-logic and probabilistic neural network with bioinspired reinforcement learning. IEEE Trans Fuzzy Syst. 2020; 28(6): 1178-1189. doi:10.1109/TFUZZ.2019.2952831
- 18Wo1niak M, Siłka J, Wieczorek M. Deep neural network correlation learning mechanism for CT brain tumor detection. Neural Comput Appl. 2021; 1-16. doi:10.1007/s00521-021-05841-x
10.1007/s00521?021?05841?x Google Scholar
- 19Chethan K, Vishwanath S, Patil RV, Vijetha K. Segementation and prediction from CT images for detecting lung cancer, 2020 11th International Conference on Computing, Communication and Networking Technologies (ICCCNT). 2020; 1-6.
- 20Singh U, Totla A, Kumar DP. Deep learning model to predict pneumonia disease based on observed patterns in lung X-rays, 2020 4th International Conference on Electronics, Communication and Aerospace Technology (ICECA), 2020; 1315-1320.
- 21Serener A, Serte S. Deep learning to distinguish COVID-19 from other lung infections, pleural diseases, and lung tumors, 2020 Medical Technologies Congress (TIPTEKNO). 2020; 1-4.
- 22Kumaraguru T, Abirami P, Darshan K, Angeline Kirubha S, Latha S, Muthu P. Smart access development for classifying lung disease with chest x-ray images using deep learning. Mater Today Proc. 2021; 47: 76-79. doi:10.1016/j.matpr.2021.03.650
- 23Chenyang L, Chan SC. A joint detection and recognition approach to lung cancer diagnosis from CT images with label uncertainty. IEEE Access. 2020; 8: 228905-228921. doi:10.1109/ACCESS.2020.3044941
- 24Wu P, Sun X, Zhao Z, Wang H, Pan S, Schuller B. Classification of lung nodules based on deep residual networks and migration learning. Comput Intell Neurosci. 2020; 2020: 1-10. doi:10.1155/2020/8975078
- 25Bharati S, Podder P, Mondal MRH. Hybrid deep learning for detecting lung diseases from x-ray images. Inform Med Unlocked. 2020; 20:100391. doi:10.1016/j.imu.2020.100391
- 26Ibrahim DM, Elshennawy NM, Sarhan AM. Deep-chest: multi-classification deep learning model for diagnosing COVID- 19, pneumonia, and lung cancer chest diseases. Comput Biol Med. 2021; 132:104348. doi:10.1016/j.compbiomed.2021.104348
- 27Jiang H, Tang S, Liu W, Zhang Y. Deep learning for COVID-19 chest CT (computed tomography) image analysis: a lesson from lung cancer. Comput Struct Biotechnol J. 2021; 19: 1391-1399. doi:10.1016/j.csbj.2021.02.016
- 28Ho TKK, Gwak J. Utilizing knowledge distillation in deep learning for classification of chest x-ray abnormalities. IEEE Access. 2020; 8: 160749-160761. doi:10.1109/ACCESS.2020.3020802
- 29Agarwala S, Kale M, Kumar D, et al. Deep learning for screening of interstitial lung disease patterns in high-resolution CT images. Clin Radiol. 2020; 75(6): 481.e1-481.e8. doi:10.1016/j.crad.2020.01.010
- 30Christodoulidis S, Anthimopoulos M, Ebner L, Christe A, Mougiakakou S. Multisource transfer learning with convolutional neural networks for lung pattern analysis. IEEE J Biomed Health Inform. 2017; 21(1): 76-84. doi:10.1109/JBHI.2016.2636929
- 31Motiian S, Piccirilli M, Adjeroh DA, Doretto G. Unified deep supervised domain adaptation and generalization, 2017 IEEE International Conference on Computer Vision (ICCV). 2017:5716–5726.
- 32Koniusz P, Tas Y, Porikli F. Domain adaptation by mixture of alignments of second-or higher-order scatter tensors, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). 2017;7139-7148.
- 33Daumé III H, Kumar A, Saha A. Frustratingly easy semi-supervised domain adaptation, Proceedings of the 2010 Workshop on Domain Adaptation for Natural Language Processing. Uppsala, Sweden: Association for Computational Linguistics. 2010; 53-59.
- 34Chen S, Harandi M, Jin X, Yang X. Semi-supervised domain adaptation via asymmetric joint distribution matching. IEEE Trans Neural Netw Learn Syst. 2020; 32: 1-15. doi:10.1109/TNNLS.2020.3027364
- 35Long M, Zhu H, Wang J, Jordan MI. Unsupervised domain adaptation with residual transfer networks. arXiv preprint arXiv:1602.04433 2016.
- 36Li R, Jiao Q, Cao W, Wong HS, Wu S. Model adaptation: unsupervised domain adaptation without source data, 2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR). 2020;9638-9647.
- 37Wang X, Peng Y, Lu L, Lu Z, Bagheri M, Summers RM. ChestX-ray8: hospital-scale chest x-ray database and benchmarks on weakly-supervised classification and localization of common thorax diseases, 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR). 2017. doi:10.1109/CVPR.2017.369
- 38Depeursinge A, Vargas A, Platon A, Geissbuhler A, Poletti PA, Müller H. Building a reference multimedia database for interstitial lung diseases. Comput Med Imaging Graph. 2012; 36(3): 227-238. doi:10.1016/j.compmedimag.2011.07.003
- 39Tan M, Le QV. EfficientNet: rethinking model scaling for convolutional neural networks, Proceedings of the 36th International Conference on Machine Learning. CoRR. 2019; 97: 6105-6114.
- 40Krizhevsky A, Sutskever I, Hinton GE. ImageNet classification with deep convolutional neural networks. In: F Pereira, CJC Burges, L Bottou, KQ Weinberger, eds. Advances in Neural Information Processing Systems. Vol 25. Curran Associates, Inc; 2012.
- 41Kingma DP, Ba J. Adam: a method for stochastic optimization, 3rd International Conference on Learning Representations (ICLR), San Diego, CA, USA. 2015;7-9.
- 42Chollet F. Keras. https://keras.io; 2015.
- 43Handelman G, Kok HK, Chandra R, et al. Peering into the black box of artificial intelligence: evaluation metrics of machine learning methods. AJR Am J Roentgenol. 2019; 212(1): 38-43.