HybridViT for robust wheat leaf disease detection using CLAHE and attention-based feature fusion.

Abstract

Wheat (Triticum aestivum L.) is a staple crop of paramount importance to global food security; however, its productivity is significantly compromised by foliar diseases. Conventional diagnostic approaches, relying on manual observation or laboratory analyses, are often labor-intensive and susceptible to inaccuracies. While recent advancements in deep learning present promising avenues for automated disease detection, persistent challenges such as limited annotated datasets, environmental heterogeneity, and model generalization continue to hinder optimal performance. This study proposes a novel hybrid deep learning model called HybridViT, which combines ConvNeXt and Vision Transformer (ViT) architectures with the Convolutional Block Attention Module (CBAM) to improve the classification of wheat leaf diseases. While ConvNeXt ensures local feature extraction and ViT provides global contextual understanding, CBAM dynamically highlights the most discriminative features. Additionally, the Contrast Limited Adaptive Histogram Equalization (CLAHE) method is employed to enhance the visibility of disease symptoms in low-contrast leaf images. Unlike conventional hybrid CNN-Transformer approaches that rely on static feature concatenation, the proposed model employs an adaptive gated fusion mechanism to dynamically balance local and global feature representations. The fused features are further refined using a lightweight CBAM module to enhance discriminative capability. Additionally, Contrast Limited Adaptive Histogram Equalization (CLAHE) is applied to improve feature visibility under varying illumination conditions. Evaluated on three different datasets obtained under both controlled and field conditions, HybridViT achieved 100% accuracy on balanced datasets and 99.10% accuracy on complex images captured in real-world conditions, surpassing existing methods. Furthermore, a 5-fold cross-validation strategy yielded an average accuracy of 99.04% ± 0.22, demonstrating the model's robustness and stability across different data splits. The results demonstrate the model's robustness against environmental noise, lighting variations, and class imbalance. This approach, which enables early and accurate disease diagnosis, supports sustainable agricultural practices, reduces pesticide use, and contributes to global food security.

Plant phenotyping relevance

小麦葉の病徴を画像から分類する深層学習手法を開発・検証しており、植物病害状態の取得・推定が研究の中心です。

Code and data availability

The paper evaluates HybridViT on three public wheat leaf disease image datasets from Kaggle, cited in the reference list with explicit URLs. These are the paper-specific image inputs used for its disease-classification measurements. No author analysis code, trained model checkpoints, or supplementary code deposit is披露d