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

被引:3
|
作者
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
相关论文
共 50 条
  • [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
    Hussain, Mohammed
    Thaher, Thaer
    Almourad, Mohamed Basel
    Mafarja, Majdi
    SCIENTIFIC REPORTS, 2024, 14 (01):
  • [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] Cotton Leaf Disease Classification Using Fine-tuned VGG16 Deep Learning Model
    Kaur, Arshleen
    Sharma, Rishabh
    Chattopadhyay, Saumitra
    Joshi, Kireet
    2024 2ND WORLD CONFERENCE ON COMMUNICATION & COMPUTING, WCONF 2024, 2024,
  • [8] Flame Image Processing and Classification Using a Pre-Trained VGG16 Model in Combustion Diagnosis
    Omiotek, Zbigniew
    Kotyra, Andrzej
    SENSORS, 2021, 21 (02) : 1 - 15
  • [9] Forensic sex classification by convolutional neural network approach by VGG16 model: accuracy, precision and sensitivity
    Pereira, Cristiana Palmela
    Correia, Mariana
    Augusto, Diana
    Coutinho, Francisco
    Silva, Francisco Salvado
    Santos, Rui
    INTERNATIONAL JOURNAL OF LEGAL MEDICINE, 2025, : 1381 - 1393
  • [10] Hybrid Feature Extraction for Breast Cancer Classification Using the Ensemble Residual VGG16 Deep Learning Model
    Wang, Zhenfei
    Ali, Muhammad Mumtaz
    Sahibzada, Kashif Iqbal
    Maqsood, Faiqa
    Rehman, Naveed Urr
    Aftab, Muhammad
    Zia, Qasim
    Weiyan, Hou
    Wei, Dong-Qing
    CURRENT BIOINFORMATICS, 2025, 20 (02) : 149 - 163