Skip to main content

ORIGINAL RESEARCH article

Front. Environ. Sci.
Sec. Big Data, AI, and the Environment
Volume 12 - 2024 | doi: 10.3389/fenvs.2024.1391770

A pest image recognition method for long-tail distribution problem

Provisionally accepted
  • 1 College of Big Data, Yunnan Agricultural University, Kunming, Yunnan Province, China
  • 2 The Laboratory for Crop Production and Intelligent, Yunnan Agricultural University, Kunming, Yunnan Province, China

The final, formatted version of the article will be published soon.

    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

    Received: 26 Feb 2024; Accepted: 01 Jul 2024.

    Copyright: © 2024 Chen, He and Gao. This is an open-access article distributed under the terms of the Creative Commons Attribution License (CC BY). The use, distribution or reproduction in other forums is permitted, provided the original author(s) or licensor are credited and that the original publication in this journal is cited, in accordance with accepted academic practice. No use, distribution or reproduction is permitted which does not comply with these terms.

    * Correspondence:
    Shengbo Chen, College of Big Data, Yunnan Agricultural University, Kunming, Yunnan Province, China
    Yun He, College of Big Data, Yunnan Agricultural University, Kunming, Yunnan Province, China

    Disclaimer: All claims expressed in this article are solely those of the authors and do not necessarily represent those of their affiliated organizations, or those of the publisher, the editors and the reviewers. Any product that may be evaluated in this article or claim that may be made by its manufacturer is not guaranteed or endorsed by the publisher.