Light-weight MobileNet for Fast Detection of COVID-19

  • Muhammad Hafidh Firmansyah Kyungpook National University, Republic of Korea
  • Seok-Joo Koh Kyungpook National University, Republic of Korea
  • Wahyu Kurnia Dewanto Politeknik Negeri Jember, Jawa Timur, Indonesia
  • Trismayanti Dwi Puspitasari Politeknik Negeri Jember, Jawa Timur, Indonesia

Abstract

The machine learning models based on Convolutional Neural Networks (CNNs) can be effectively used for detection and recognition of objects, such as Corona Virus Disease 19 (COVID-19). In particular, the MobileNet and Single Shot multi-box Detector (SSD) have recently been proposed as the machine learning model for object detection. However, there are still some challenges for deployment of such architectures on the embedded devices, due to the limited computational power. Another problem is that the accuracy of the associated machine learning model may be decreased, depending on the number of concerned parameters and layers. This paper proposes a light-weight MobileNet (LMN) architecture that can be used to improve the accuracy of the machine learning model, with a small number of layers and lower computation time, compared to the existing models. By experimentation, we show that the proposed LMN model can be effectively used for detection of COVID-19 virus. The proposed LMN can achieve the accuracy of 98% with the file size of 27.8 Mbits by replacing the standard CNN layers with separable convolutional layers.

Published
2021-06-30
Section
Articles