arrow
Return

A pest image recognition method for long-tail distribution problem

delete2024-07-29
delete0
delete
OA
AI
S
Shengbo Chen
Q
Quan Gao
Y
Yun He *
DOI:10.3389/fenvs.2024.1391770delete
deleteOriginal
deleteOriginal request for help
deleteShare
deleteSave
Abstract

Abstract

En 中文
Deep learning has revolutionized numerous fields, notably image classification. However, conventional methods in agricultural pest recognition struggle with the long-tail distribution of pest image data, characterized by limited samples in rare pest categories, thereby impeding overall model performance. This study proposes two state-of-the-art techniques: Instance-based Data Augmentation (IDA) and Constraint-based Feature Tuning (CFT). IDA collaboratively applies resampling and mixup methods to notably enhance feature extraction for rare class images. This approach addresses the long-tail distribution challenge through resampling, ensuring adequate representation for scarce categories. Additionally, by introducing data augmentation, we further refined the recognition of tail-end categories without compromising performance on common samples. CFT, a refinement built upon pre-trained models using IDA, facilitated the precise classification of image features through fine-tuning. Our experimental findings validate that our proposed method outperformed previous approaches on the CIFAR-10-LT, CIFAR-100-LT, and IP102 datasets, demonstrating its effectiveness. Using IDA and CFT to optimize the ViT model, we observed significant improvements over the baseline, with accuracy rates reaching 98.21%, 88.62%, and 64.26%, representing increases of 0.74%, 3.55%, and 5.73% respectively. Our evaluation of the CIFAR-10-LT and CIFAR-100-LT datasets also demonstrated state-of-the-art performance.
Keywords:
insect pest recognition
long-tail distribution data
data augmentation
deep machine learning
classification

Journal

Frontiers in Environmental Science cover
Frontiers in Environmental Science
IF:
3.7
Papers:
8.0K
Citations:
2.3W

Organization

Y
Yunnan Agricultural University
Scholars:
6.2K
Papers: 2.7K
Citations: 6