Model Agnostic Meta-Learning (MAML)-Based Ensemble Model for Accurate Detection of Wheat Diseases Using Vision Transformer and Graph Neural Networks

Yasir Maqsood, Syed Muhammad Usman*, Musaed Alhussein, Khursheed Aurangzeb*, Shehzad Khalid, Muhammad Zubair

*Corresponding author for this work

Research output: Contribution to journalArticlepeer-review

17 Scopus citations

Abstract

Wheat is a critical crop, extensively consumed worldwide, and its production enhancement is essential to meet escalating demand. The presence of diseases like stem rust, leaf rust, yellow rust, and tan spot significantly diminishes wheat yield, making the early and precise identification of these diseases vital for effective disease management. With advancements in deep learning algorithms, researchers have proposed many methods for the automated detection of disease pathogens; however, accurately detecting multiple disease pathogens simultaneously remains a challenge. This challenge arises due to the scarcity of RGB images for multiple diseases, class imbalance in existing public datasets, and the difficulty in extracting features that discriminate between multiple classes of disease pathogens. In this research, a novel method is proposed based on Transfer Generative Adversarial Networks for augmenting existing data, thereby overcoming the problems of class imbalance and data scarcity. This study proposes a customized architecture of Vision Transformers (ViT), where the feature vector is obtained by concatenating features extracted from the custom ViT and Graph Neural Networks. This paper also proposes a Model Agnostic Meta Learning (MAML) based ensemble classifier for accurate classification. The proposed model, validated on public datasets for wheat disease pathogen classification, achieved a test accuracy of 99.20% and an F1-score of 97.95%. Compared with existing state-of-the-art methods, this proposed model outperforms in terms of accuracy, F1-score, and the number of disease pathogens detection. In future, more diseases can be included for detection along with some other modalities like pests and weed.

Original languageEnglish
Pages (from-to)2795-2811
Number of pages17
JournalComputers, Materials and Continua
Volume79
Issue number2
DOIs
StatePublished - 2024
Externally publishedYes

Bibliographical note

Publisher Copyright:
© 2024 Tech Science Press. All rights reserved.

Keywords

  • Wheat disease detection
  • deep learning
  • graph neural network
  • model agnostic meta learning
  • vision transformer

ASJC Scopus subject areas

  • Biomaterials
  • Modeling and Simulation
  • Mechanics of Materials
  • Computer Science Applications
  • Electrical and Electronic Engineering

Fingerprint

Dive into the research topics of 'Model Agnostic Meta-Learning (MAML)-Based Ensemble Model for Accurate Detection of Wheat Diseases Using Vision Transformer and Graph Neural Networks'. Together they form a unique fingerprint.

Cite this