Wheat disease recognition method based on the SC-ConvNeXt network model
Abstract When utilizing convolutional neural networks for wheat disease identification, the training phase typically requires a substantial amount of labeled data. However, labeling data is both complex and costly. Additionally, the model’s recognition performance is often disrupted by complex facto...
Saved in:
| Main Authors: | , , , , , , |
|---|---|
| Format: | Article |
| Language: | English |
| Published: |
Nature Portfolio
2024-12-01
|
| Series: | Scientific Reports |
| Online Access: | https://doi.org/10.1038/s41598-024-83636-5 |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1841559409691459584 |
|---|---|
| author | Tianliang Dong Xiao Ma Bin Huang Wenyu Zhong Qingan Han Qinghai Wu You Tang |
| author_facet | Tianliang Dong Xiao Ma Bin Huang Wenyu Zhong Qingan Han Qinghai Wu You Tang |
| author_sort | Tianliang Dong |
| collection | DOAJ |
| description | Abstract When utilizing convolutional neural networks for wheat disease identification, the training phase typically requires a substantial amount of labeled data. However, labeling data is both complex and costly. Additionally, the model’s recognition performance is often disrupted by complex factors in natural environments. To address these issues, this paper proposes a wheat disease identification model, SC-ConvNeXt, which integrates the SimCLR pre-training framework and an improved CBAM attention mechanism. The model employs ConvNeXt-T as the feature extraction network. Initially, it uses the self-supervised SimCLR pre-training framework to learn inter-class similarities, reducing the reliance on labeled data during training. Subsequently, the CBAM attention module is integrated into ConvNeXt-T to enhance the model’s feature extraction and generalization capabilities in complex backgrounds, and each attention module’s loss function is improved with a LeakyReLU activation function to prevent neuron deactivation when inputs are negative. Furthermore, by introducing the Focal Loss function, the model addresses the imbalance in the quantity of easy and difficult-to-classify samples. The dataset used in this study comes from the ‘Smart Agriculture’ platform of Jilin Agricultural Science and Technology College, including images of three wheat diseases and healthy wheat. After expanding the dataset with various data augmentation methods, the effectiveness of adding SimCLR and the attention mechanism was sequentially verified. Comparative experiments were also conducted against four classic classification models. The experimental results show that the proposed SC-ConvNeXt model achieves an average classification accuracy of 88.05% on the test set, the highest among all comparative models. The model does not require additional labeled data during training, demonstrating its effectiveness in enhancing wheat disease recognition performance under natural environmental conditions without the need for extra labeled data training. |
| format | Article |
| id | doaj-art-0ce3712c879443ceab4e4703525376a4 |
| institution | Kabale University |
| issn | 2045-2322 |
| language | English |
| publishDate | 2024-12-01 |
| publisher | Nature Portfolio |
| record_format | Article |
| series | Scientific Reports |
| spelling | doaj-art-0ce3712c879443ceab4e4703525376a42025-01-05T12:30:00ZengNature PortfolioScientific Reports2045-23222024-12-0114111310.1038/s41598-024-83636-5Wheat disease recognition method based on the SC-ConvNeXt network modelTianliang Dong0Xiao Ma1Bin Huang2Wenyu Zhong3Qingan Han4Qinghai Wu5You Tang6School of Information and Control Engineering, Jilin University of Chemical TechnologySchool of Computer Science and Information Engineering, Qilu Institute of TechnologySchool of Electrical and Information Engineering, Jilin Agricultural Science and Technology UniversitySchool of Electrical and Information Engineering, Jilin Agricultural Science and Technology UniversitySchool of Electrical and Information Engineering, Jilin Agricultural Science and Technology UniversitySchool of Information and Control Engineering, Jilin University of Chemical TechnologySchool of Information and Control Engineering, Jilin University of Chemical TechnologyAbstract When utilizing convolutional neural networks for wheat disease identification, the training phase typically requires a substantial amount of labeled data. However, labeling data is both complex and costly. Additionally, the model’s recognition performance is often disrupted by complex factors in natural environments. To address these issues, this paper proposes a wheat disease identification model, SC-ConvNeXt, which integrates the SimCLR pre-training framework and an improved CBAM attention mechanism. The model employs ConvNeXt-T as the feature extraction network. Initially, it uses the self-supervised SimCLR pre-training framework to learn inter-class similarities, reducing the reliance on labeled data during training. Subsequently, the CBAM attention module is integrated into ConvNeXt-T to enhance the model’s feature extraction and generalization capabilities in complex backgrounds, and each attention module’s loss function is improved with a LeakyReLU activation function to prevent neuron deactivation when inputs are negative. Furthermore, by introducing the Focal Loss function, the model addresses the imbalance in the quantity of easy and difficult-to-classify samples. The dataset used in this study comes from the ‘Smart Agriculture’ platform of Jilin Agricultural Science and Technology College, including images of three wheat diseases and healthy wheat. After expanding the dataset with various data augmentation methods, the effectiveness of adding SimCLR and the attention mechanism was sequentially verified. Comparative experiments were also conducted against four classic classification models. The experimental results show that the proposed SC-ConvNeXt model achieves an average classification accuracy of 88.05% on the test set, the highest among all comparative models. The model does not require additional labeled data during training, demonstrating its effectiveness in enhancing wheat disease recognition performance under natural environmental conditions without the need for extra labeled data training.https://doi.org/10.1038/s41598-024-83636-5 |
| spellingShingle | Tianliang Dong Xiao Ma Bin Huang Wenyu Zhong Qingan Han Qinghai Wu You Tang Wheat disease recognition method based on the SC-ConvNeXt network model Scientific Reports |
| title | Wheat disease recognition method based on the SC-ConvNeXt network model |
| title_full | Wheat disease recognition method based on the SC-ConvNeXt network model |
| title_fullStr | Wheat disease recognition method based on the SC-ConvNeXt network model |
| title_full_unstemmed | Wheat disease recognition method based on the SC-ConvNeXt network model |
| title_short | Wheat disease recognition method based on the SC-ConvNeXt network model |
| title_sort | wheat disease recognition method based on the sc convnext network model |
| url | https://doi.org/10.1038/s41598-024-83636-5 |
| work_keys_str_mv | AT tianliangdong wheatdiseaserecognitionmethodbasedonthescconvnextnetworkmodel AT xiaoma wheatdiseaserecognitionmethodbasedonthescconvnextnetworkmodel AT binhuang wheatdiseaserecognitionmethodbasedonthescconvnextnetworkmodel AT wenyuzhong wheatdiseaserecognitionmethodbasedonthescconvnextnetworkmodel AT qinganhan wheatdiseaserecognitionmethodbasedonthescconvnextnetworkmodel AT qinghaiwu wheatdiseaserecognitionmethodbasedonthescconvnextnetworkmodel AT youtang wheatdiseaserecognitionmethodbasedonthescconvnextnetworkmodel |