cs.CVMar 23, 2026

Tiny-ViT: A Compact Vision Transformer for Efficient and Explainable Potato Leaf Disease Classification

Authors: Shakil Mia, Umme Habiba, Urmi Akter, SK Rezwana Quadir Raisa, Jeba Maliha, Md. Iqbal Hossain, Md. Shakhauat Hossan Sumon

Organizations: Dept. of CSE, Daffodil International University Dhaka, Bangladesh · Dept. of EEE, American International University-Bangladesh (AIUB), Dhaka, Bangladesh · Dept. of Horticulture, Sher-e-Bangla Agricultural University, Dhaka, Bangladesh · Dept. of CSE, Ahsanullah University of Science and Technology (AUST), Dhaka, Bangladesh · Dept. of CSE, Uttara University, Dhaka, Bangladesh · Dept. of ECE, North South University, Dhaka, Bangladesh

Abstract

Early and precise identification of plant diseases, especially in potato crops is important to ensure the health of the crops and ensure the maximum yield . Potato leaf diseases, such as Early Blight and Late Blight, pose significant challenges to farmers, often resulting in yield losses and increased pesticide use. Traditional methods of detection are not only time-consuming, but are also subject to human error, which is why automated and efficient methods are required. The paper introduces a new method of potato leaf disease classification Tiny-ViT model, which is a small and effective Vision Transformer (ViT) developed to be used in resource-limited systems. The model is tested on a dataset of three classes, namely Early Blight, Late Blight, and Healthy leaves, and the preprocessing procedures include resizing, CLAHE, and Gaussian blur to improve the quality of the image. Tiny-ViT model has an impressive test accuracy of 99.85% and a mean CV accuracy of 99.82% which is better than baseline models such as DEIT Small, SWIN Tiny, and MobileViT XS. In addition to this, the model has a Matthews Correlation Coefficient (MCC) of 0.9990 and narrow confidence intervals (CI) of [0.9980, 0.9995], which indicates high reliability and generalization. The training and testing inference time is competitive, and the model exhibits low computational expenses, thereby, making it applicable in real-time applications. Moreover, interpretability of the model is improved with the help of GRAD-CAM, which identifies diseased areas. Altogether, the proposed Tiny-ViT is a solution with a high level of robustness, efficiency, and explainability to the problem of plant disease classification.

Explore similar work

CardsList
  1. Detection of Potato Plant Leaf Diseases using Convolutional Neural Networks

    Oct 14, 2026Haimanot Bitew Alemayehu, Simachew Melaku Ayalew, Chilotie Desalew Mengesha +3Plant Disease ClassificationConvolutional Neural Networks

  2. Enhancing Precision Agriculture with a Hybrid Deep Learning Framework for Multi-Class Plant Disease Classification and Interpretability

    Jun 13, 2026Hasibul Islam Sufi, Ridam Roy, Shayla Alam Setu +1Plant Disease ClassificationHybrid CNN-Transformer Architectures

  3. AgriMind: An Ensemble Deep Learning Framework for Multi-Class Plant Disease Classification

    May 15, 2026Salma Hoque Talukdar Koli, Fahima Haque Talukder JelyPlant Disease ClassificationEnsemble Learning