Shallow Neural Networks beat Deep Neural Networks trained with transfer learning: A Use Case based on training Neural Networks to identify Covid-19 in chest X-ray images

被引:1
作者
Manolakis, Dimitrios [1 ]
Spanos, Georgios [1 ]
Refanidis, Ioannis [1 ]
机构
[1] Univ Macedonia, Dept Appl Informat, Thessaloniki, Greece
来源
25TH PAN-HELLENIC CONFERENCE ON INFORMATICS WITH INTERNATIONAL PARTICIPATION (PCI2021) | 2021年
关键词
Deep Learning; Transfer Learning; Convolutional Neural Networks;
D O I
10.1145/3503823.3503834
中图分类号
TP301 [理论、方法];
学科分类号
081202 ;
摘要
Since the start of the covid-19 health crisis, there have been many studies on the application of deep learning models in order to detect the virus on chest X-ray images. Training large neural networks on big data sets is a computationally intensive task, consuming a lot of power and needing a lot of time. Thus, usually only researchers in large institutions or companies have the necessary resources to bring the task to fruition. Other researchers employ transfer learning, a technique that is based on using pre-trained deep neural networks that have been trained on a similar dataset and retrain only their last neuron layers. However, using deep neural networks with transfer learning is not always the best option; in some cases, training a shallow neural network from scratch achieves better results. In this paper we compare training from scratch, shallow neural networks to transfer learning from deep neural models. Our experiments have been conducted on a publicly available dataset containing chest X-ray images concerning covid-19 patients, as well as non-covid-19 ones. Surprisingly enough, training from scratch shallow neural networks produced significantly better results in terms of both specificity and sensitivity. The results of the models' evaluation showed that the three shallow neural networks achieved specificity rates higher than 98%, while having a sensitivity rate of 98%, exceeding the best performing pre-trained model, the DenseNet121, which achieved a specificity rate of 91.3%, while having a sensitivity rate of 98%.
引用
收藏
页码:58 / 62
页数:5
相关论文
共 22 条
  • [1] A Novel Transfer Learning Based Approach for Pneumonia Detection in Chest X-ray Images
    Chouhan, Vikash
    Singh, Sanjay Kumar
    Khamparia, Aditya
    Gupta, Deepak
    Tiwari, Prayag
    Moreira, Catarina
    Damasevicius, Robertas
    de Albuquerque, Victor Hugo C.
    [J]. APPLIED SCIENCES-BASEL, 2020, 10 (02):
  • [2] Glickman Cody, 2020, DATA AUGMENTATION ME
  • [3] Goadrich M, 2006, ACM INT C PROCEEDING, P233, DOI 10.1145/1143844.1143874
  • [4] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 770 - 778
  • [5] Densely Connected Convolutional Networks
    Huang, Gao
    Liu, Zhuang
    van der Maaten, Laurens
    Weinberger, Kilian Q.
    [J]. 30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, : 2261 - 2269
  • [6] Irvin J, 2019, AAAI CONF ARTIF INTE, P590
  • [7] Efficient backprop
    LeCun, Y
    Bottou, L
    Orr, GB
    Müller, KR
    [J]. NEURAL NETWORKS: TRICKS OF THE TRADE, 1998, 1524 : 9 - 50
  • [8] Maguolo Gianluca, 2020, ARTIFICIAL NEURAL NE, P270, DOI [10.1007/978-3-030-01424-7_27, DOI 10.1007/978-3-030-01424-7_27]
  • [9] Deep-COVID: Predicting COVID-19 from chest X-ray images using deep transfer learning
    Minaee, Shervin
    Kafieh, Rahele
    Sonka, Milan
    Yazdani, Shakib
    Soufi, Ghazaleh Jamalipour
    [J]. MEDICAL IMAGE ANALYSIS, 2020, 65
  • [10] Shallow Convolutional Neural Network for COVID-19 Outbreak Screening Using Chest X-rays
    Mukherjee, Himadri
    Ghosh, Subhankar
    Dhar, Ankita
    Obaidullah, Sk Md
    Santosh, K. C.
    Roy, Kaushik
    [J]. COGNITIVE COMPUTATION, 2024, 16 (04) : 1695 - 1708