InsectMamba: Insect Pest Classification with State Space Model

Qianning Wang,Chenglin Wang,Zhixin Lai,Yucheng Zhou
2024-04-05
Abstract:The classification of insect pests is a critical task in agricultural technology, vital for ensuring food security and environmental sustainability. However, the complexity of pest identification, due to factors like high camouflage and species diversity, poses significant obstacles. Existing methods struggle with the fine-grained feature extraction needed to distinguish between closely related pest species. Although recent advancements have utilized modified network structures and combined deep learning approaches to improve accuracy, challenges persist due to the similarity between pests and their surroundings. To address this problem, we introduce InsectMamba, a novel approach that integrates State Space Models (SSMs), Convolutional Neural Networks (CNNs), Multi-Head Self-Attention mechanism (MSA), and Multilayer Perceptrons (MLPs) within Mix-SSM blocks. This integration facilitates the extraction of comprehensive visual features by leveraging the strengths of each encoding strategy. A selective module is also proposed to adaptively aggregate these features, enhancing the model's ability to discern pest characteristics. InsectMamba was evaluated against strong competitors across five insect pest classification datasets. The results demonstrate its superior performance and verify the significance of each model component by an ablation study.
Computer Vision and Pattern Recognition,Artificial Intelligence
What problem does this paper attempt to address?
### Problems the Paper Aims to Solve The paper aims to address key issues in insect pest classification to ensure food safety and environmental sustainability. Specifically, the paper targets the following challenges: 1. **High Camouflage**: Insect pests have a high level of camouflage in natural environments, making visual recognition difficult. 2. **Species Diversity**: There is a wide variety of insect pests, with subtle differences in characteristics between different species, increasing the difficulty of classification. 3. **Background Complexity**: The similarity between insect pests and their surrounding environment makes it difficult for traditional image processing algorithms to effectively distinguish pests. 4. **Fine-Grained Feature Extraction**: It is necessary to extract information from subtle features to distinguish closely related pest species. Although existing methods have improved accuracy by enhancing network structures and incorporating deep learning techniques, they still fall short in addressing the above challenges. To this end, the paper proposes the **InsectMamba** model, which integrates State Space Models (SSMs), Convolutional Neural Networks (CNNs), Multi-Head Self-Attention (MSA), and Multi-Layer Perceptrons (MLPs). By using Mix-SSM blocks and a Selective Module to extract and integrate comprehensive visual features, the model aims to improve pest classification performance.