Advancing plant disease classification: A robust and generalized approach with transformer-fused convolution and Wasserstein domain adaptation
Plant diseases pose significant threats to agricultural productivity and food security. Owing to a scarcity of field environment datasets, the prevailing plant disease classification approaches, trained on laboratory-controlled datasets, often grapple with achieving optimal performance in real-world environments. We proposed a novel and robust framework for Unsupervised Domain Adaptation (UDA), employing an adversarial learning approach with a Wasserstein distance-informed algorithm to learn domain invariant feature representations capable of generalizing more diverse features. This approach incorporates insights from a labeled source domain and adopts an unlabeled target domain by minimizing the distribution discrepancies between domains. Recently, mobile vision transformer (MViT)-based methods have been applied to UDA due to their ability to capture long-distance feature dependencies. However, these methods overlook the fact that MViT lacks effectiveness in extracting local feature details. The proposed framework combines the advantages of convolutional neural networks (CNNs) and MViTs, integrating local features extracted by CNNs with global features captured by MViTs. This fusion of local and global representations enhances transferability and feature discriminability within the domains. Furthermore, we incorporate a feature-fusing method to align channel dimensions and enhance the local details of the global representation. Extensive experiments using three plant disease datasets demonstrate the effectiveness and efficiency of our approach, yielding significant improvements in classification performance with 13.67%, compared to state-of-the-art (SOTA) and baseline methods. Our framework offers a promising solution for robust and efficient plant disease classification, providing valuable insights for sustainable agriculture and crop management. • The novel framework combines MViTs and CNNs to enhance transferability and feature discriminability for plant disease classification in field environments. • Domain adaptation with Wasserstein distance learns domain invariant feature representations and improves generalization. • Fusing local CNN features with global MViT features enhances the model’s transferability and feature discriminability within domains. • Significant performance improvements: outperforming state-of-the-art (SOTA) and baseline methods with a 13.67% increase. • Valuable insights for sustainable agriculture: robust and efficient plant disease classification towards ensuring food security.
Authors
- Xiaoyang Zeng (ORCID: https://orcid.org/0009-0001-9928-8782)
- Iwan Yahya (ORCID: https://orcid.org/0000-0003-4443-0486)
- Syed Attique Shah (ORCID: https://orcid.org/0000-0003-2949-7391)
- Ghulam Ali Mallah (ORCID: https://orcid.org/0000-0003-3428-9802)
- Muhammad Hanif Tunio (ORCID: https://orcid.org/0009-0007-4130-0201)
- Awais Ahmed (ORCID: https://orcid.org/0000-0003-4410-2028)
- Hira Ghazal Shaikh (ORCID: https://orcid.org/0000-0001-5604-4166)
- Jianping Li (ORCID: https://orcid.org/0000-0003-2192-1450)
Institutions
- Birmingham City University (GB)
- University of Electronic Science and Technology of China (CN)
- Shah Abdul Latif University (PK)
- Bayero University Kano (NG)
Publication Details
- Journal
- Computers and Electronics in Agriculture
- Published
- 2024-11-21
- DOI
- https://doi.org/10.1016/j.compag.2024.109574
- Citations
- 12
- Primary Topic
- Smart Agriculture and AI
- Type
- article
- Field-Weighted Citation Impact
- 7.20