Enhancement of VGG16 model with multi-view and spatial dropout for classification of mosquito vectors

被引:1
|
作者
Pora, Wanchalerm [1 ]
Kasamsumran, Natthakorn [1 ]
Tharawatcharasart, Katanyu [1 ]
Ampol, Rinnara [2 ]
Siriyasatien, Padet [2 ]
Jariyapan, Narissara [2 ]
机构
[1] Chulalongkorn Univ, Fac Engn, Dept Elect Engn, Bangkok, Thailand
[2] Chulalongkorn Univ, Fac Med, Ctr Excellence Vector Biol & Vector Borne Dis, Dept Parasitol, Bangkok, Thailand
来源
PLOS ONE | 2023年 / 18卷 / 07期
关键词
CULICIDAE; DIPTERA; VIRUS;
D O I
10.1371/journal.pone.0284330
中图分类号
O [数理科学和化学]; P [天文学、地球科学]; Q [生物科学]; N [自然科学总论];
学科分类号
07 ; 0710 ; 09 ;
摘要
Mosquitoes transmit pathogens that can cause numerous significant infectious diseases in humans and animals such as malaria, dengue fever, chikungunya fever, and encephalitis. Although the VGG16 model is not one of the most advanced CNN networks, it is reported that a fine-tuned VGG16 model achieves accuracy over 90% when applied to the classification of mosquitoes. The present study sets out to improve the accuracy and robustness of the VGG16 network by incorporating spatial dropout layers to regularize the network and by modifying its structure to incorporate multi-view inputs. Herein, four models are implemented: (A) early-combined, (B) middle-combined, (C) late-combined, and (D) ensemble model. Moreover, a structure for combining Models (A), (B), (C), and (D), known as the classifier, is developed. Two image datasets, including a reference dataset of mosquitoes in South Korea and a newly generated dataset of mosquitoes in Thailand, are used to evaluate our models. Regards the reference dataset, the average accuracy of ten runs improved from 83.26% to 99.77%, while the standard deviation decreased from 2.60% to 0.12%. When tested on the new dataset, the classifier's accuracy was also over 99% with a standard deviation of less than 2%. This indicates that the algorithm achieves high accuracy with low variation and is independent of a particular dataset. To evaluate the robustness of the classifier, it was applied to a small dataset consisting of mosquito images captured under various conditions. Its accuracy dropped to 86.14%, but after retraining with the small dataset, it regained its previous level of precision. This demonstrates that the classifier is resilient to variation in the dataset and can be retrained to adapt to the variation. The classifier and the new mosquito dataset could be utilized to develop an application for efficient and rapid entomological surveillance for the prevention and control of mosquito-borne diseases.
引用
收藏
页数:20
相关论文
共 48 条
  • [1] An Improved VGG16 Model for Pneumonia Image Classification
    Jiang, Zhi-Peng
    Liu, Yi-Yang
    Shao, Zhen-En
    Huang, Ko-Wei
    APPLIED SCIENCES-BASEL, 2021, 11 (23):
  • [2] A classification method for breast images based on an improved VGG16 network model
    Dong, Yi
    Liu, Jinjiang
    Lan, Yihua
    ELECTRONIC RESEARCH ARCHIVE, 2023, 31 (04): : 2358 - 2373
  • [3] Multi Label Classification Of Retinal Disease On Fundus Images Using AlexNet And VGG16 Architectures
    Prawira, Reyhansyah
    Bustamam, Alhadi
    Anki, Prasnurzaki
    2021 4TH INTERNATIONAL SEMINAR ON RESEARCH OF INFORMATION TECHNOLOGY AND INTELLIGENT SYSTEMS (ISRITI 2021), 2020,
  • [4] Optimizing VGG16 deep learning model with enhanced hunger games search for logo classification
    Mohammed Hussain
    Thaer Thaher
    Mohamed Basel Almourad
    Majdi Mafarja
    Scientific Reports, 14 (1)
  • [5] Transfer Learning VGG16 Model for Classification of Tomato Plant Leaf Diseases: A Novel Approach for Multi-Level Dimensional Reduction
    Borugadda, Premkumar
    Lakshmi, Ramasami
    Sahoo, Satyasangram
    PERTANIKA JOURNAL OF SCIENCE AND TECHNOLOGY, 2023, 31 (02): : 813 - 841
  • [6] Image Classification Via Multi-View Model
    Cheng, Yanyun
    Zhu, Songhao
    Liang, Zhiwei
    Xu, Guozheng
    PROCEEDINGS OF THE 28TH CHINESE CONTROL AND DECISION CONFERENCE (2016 CCDC), 2016, : 3333 - 3337
  • [7] Flame Image Processing and Classification Using a Pre-Trained VGG16 Model in Combustion Diagnosis
    Omiotek, Zbigniew
    Kotyra, Andrzej
    SENSORS, 2021, 21 (02) : 1 - 15
  • [8] Multi-view Classification Model for Knowledge Graph Completion
    Jiang, Wenbin
    Guo, Mengfei
    Chen, Yufeng
    Li, Ying
    Xu, Jinan
    Lyu, Yajuan
    Zhu, Yong
    1ST CONFERENCE OF THE ASIA-PACIFIC CHAPTER OF THE ASSOCIATION FOR COMPUTATIONAL LINGUISTICS AND THE 10TH INTERNATIONAL JOINT CONFERENCE ON NATURAL LANGUAGE PROCESSING (AACL-IJCNLP 2020), 2020, : 726 - 734
  • [9] Classification of Diagnosis of Alzheimer's Disease Based on Convolutional Layers of VGG16 Model using Speech Data
    Kim, Minwoo
    Kim, Hyungjun
    Lim, Joon S.
    11TH INTERNATIONAL CONFERENCE ON ICT CONVERGENCE: DATA, NETWORK, AND AI IN THE AGE OF UNTACT (ICTC 2020), 2020, : 456 - 459
  • [10] A Probabilistic Hierarchical Model for Multi-View and Multi-Feature Classification
    Li, Jinxing
    Yong, Hongwei
    Zhang, Bob
    Li, Mu
    Zhang, Lei
    Zhang, David
    THIRTY-SECOND AAAI CONFERENCE ON ARTIFICIAL INTELLIGENCE / THIRTIETH INNOVATIVE APPLICATIONS OF ARTIFICIAL INTELLIGENCE CONFERENCE / EIGHTH AAAI SYMPOSIUM ON EDUCATIONAL ADVANCES IN ARTIFICIAL INTELLIGENCE, 2018, : 3498 - 3505