An open set model for pest identification

Abstract

Agricultural pest identification is a prerequisite for increasing crop production and meeting global food demands. Generally, numerous phenotypic and genotypic features are widely utilized for species-level pest identification. However, the approaches are time-consuming and require expert knowledge in relevant fields. Numerous image-based machine learning (ML) models also exist to identify insect pests in agricultural fields. The models are significantly rely on a large, manually curated dataset and are close-set in nature. Our study aims to develop an open set pest identification approach by adding the capability of rejecting any irrelevant inputs. Tephritid fruit flies (Diptera:Tephritidae) are considered as an example since they are the most economically important agricultural pests worldwide. Images of the fruit flies were collected from a publicly available database and filtered to exclude uninformative images using a deep learning model (Inception-V3) and an unsupervised k-means clustering method. For the closed-set identification task, our EfficientNet-B2 model classified four major genera of notorious tephritid flies, namely, Anastrepha, Ceratitis, Rhagoletis, and Bactrocera with an accuracy of 89.65%. We further improvise our proposed model for open-set recognition tasks to leverage the identification beyond the trained datasets. The open set model achieved an overall accuracy of 86.48% and a macro F1-score of 94.44% on the four genera and an unknown class. Our proposed model can be a practical and effective pest identification tool for harmful fruit flies. In addition, the model is easy to implement with existing agricultural pest control systems in an open-world scenario.