MethodsX (Jan 2023)
A protocol for developing a classification system of mosquitoes using transfer learning
Abstract
Mosquito identification and classification are the most important steps in a surveillance program of mosquito-borne diseases. With conventional approach of data collection, the process of sorting and classification are laborious and time-consuming. The advancement of computer vision with transfer learning provides excellent alternative to the challenge. Transfer learning is a type of machine learning that is viable and durable in image classification with limited training images. This protocol aims to develop step-by-step procedure in developing a classification system with transfer learning algorithm for mosquito, we demonstrate the protocol to classify two species of Aedes mosquito - Aedes aegypti L. and Aedes albopitus L, but user can adopt the protocol for higher number of species classification. We demonstrated the way of start from the scratch, fine-tuning two pre-trained model performance by using different combination of hyperparameters – batch size and learning rate, and explain the terminology in the Appendix. This protocol target on the domain expert such as entomologist and public health practices to develop their own model to solve the task of mosquito/insect classification.