Attention-enhanced YOLOv8n-CLS for lightweight and efficient tomato leaf disease classification
12 affiliations
- 1Department of Computer Science and Engineering, Galgotias University
- 2Department of Computer Science, St. Xavier’s College
- 3Department of IT, BVRIT Hyderabad College of Engineering for Women
- 4Department of Computer Science and Engineering, Kalasalingam Academy of Research and Education
- 5Department of Computer Science and Engineering (IoT & Cyber Security including Blockchain Technology), RR Institute of Technology
- 6Department of Computer Science and Engineering, School of Computing, Vel Tech Rangarajan Dr. Sagunthala R&D Institute of Science and Technology
- 7Department of Information Technology, College of Computer and Information Sciences, Princess Nourah bint Abdulrahman University
- 8Department of Computer Science, College of Computing and Information Technology, Shaqra University
- 9Department of Computer Science, Faculty of Computer and Information Sciences, Ain Shams University
- 10Department of Mechatronics, Faculty of Engineering, Ain shams University
- 11Hourani Center for Applied Scientific Research, Al-Ahliyya Amman University
- 12College of Information Sciences and Technology, Penn State University, University Park
Abstract
Introduction: Tomato leaf diseases substantially reduce crop yield and fruit quality, creating a need for accurate, computationally efficient, and potentially deployable image-based disease-screening systems for precision agriculture. This study presents an Attention-Enhanced YOLOv8n-CLS framework for tomato leaf disease classification by integrating a lightweight spatial attention mechanism to enhance the representation of spatially informative image features while maintaining a compact computational structure.
Methods: Unlike conventional object detection approaches that require bounding-box annotations, the proposed framework formulates disease recognition as an image-level classification task, requiring only class labels and thereby simplifying dataset preparation and model training. Experiments were conducted on two publicly available benchmark datasets, namely the PlantVillage Tomato Dataset and the Kaggle Tomato Leaf Disease Dataset. A systematic evaluation comprising binary and multiclass classification, five-fold cross-validation, ablation analysis, cross-dataset evaluation, comparative benchmarking, computational assessment, and Grad-CAM-based visualization was performed to assess classification performance, stability, transferability, and computational efficiency.
Results: The Attention-Enhanced YOLOv8n-CLS achieved a maximum classification Accuracy of 98.90%, an AUC of 0.991, a computational complexity of 7.5 GFLOPs, and an average inference speed of approximately 45 FPS on the evaluated computing platform. In five-fold cross-validation, the proposed model achieved a mean Accuracy of 98.50% ± 0.14% on the PlantVillage Tomato Dataset and 97.82% ± 0.29% on the Kaggle Tomato Leaf Disease Dataset. The model also demonstrated competitive performance relative to representative deep learning architectures, including ResNet50, DenseNet201, ResNet152V2, MobileNetV2, and YOLOv5. The cross-dataset experiments further demonstrated transferability between the evaluated PlantVillage and Kaggle image distributions using the 10 classes shared by the two datasets.
Discussion: The results indicate a favourable balance between classification performance and computational requirements under the evaluated experimental conditions, with potential for further investigation in resource-constrained applications. Grad-CAM visualization provided qualitative information about image regions associated with the model's classification decisions but did not independently establish biological causality or pathogen-specific recognition. Importantly, the proposed framework performs image-based symptom classification and preliminary disease screening using dataset-provided labels; it does not independently confirm causal pathogens. Therefore, the reported classification performance should not be interpreted as definitive pathogen diagnosis or field-level validation. Further evaluation using independently collected field images, laboratory-confirmed disease or pathogen status, diverse symptom presentations, and representative agricultural edge devices is required before practical agricultural deployment can be established.
Related
- Lightweight YOLOv8n marigold corolla detection and picking-point localization method based on ECR progressive channel compression and dual attention mechanisms
- LeafXNet: A Lightweight Explainable Model With Integrated Attention Mechanism for Fruit Leaf Disease Detection
- A lightweight privacy-aware federated vision transformer for tomato leaf disease classification
- HECA-MobileNet: an explainable lightweight attention-based deep learning framework for custard apple disease detection
- MangoSwin-attention: a swin transformer with attention pooling for generalizable mango leaf disease classification
BibTeX
@article{lilhore2026attentionenhanced,
title = {{Attention-enhanced YOLOv8n-CLS for lightweight and efficient tomato leaf disease classification}},
author = {Umesh Kumar Lilhore and J. Rexy and Kayal Padmanandam and Balakrishnan D. and Rasmi A. and Athinarayanan Subramanian and Hanaa A. Abdallah and Abdelaziz A. Abdelhamid and Ehab Seif Ghith and Aseel Smerat and Kaamran Raahemifar and Sarita Simaiya},
journal = {Frontiers in Plant Science},
year = {2026},
month = oct,
doi = {10.3389/fpls.2026.1917550},
url = {https://www.frontiersin.org/articles/10.3389/fpls.2026.1917550/full},
note = {CC BY 4.0},
}
RIS
TY - JOUR
TI - Attention-enhanced YOLOv8n-CLS for lightweight and efficient tomato leaf disease classification
AU - Umesh Kumar Lilhore
AU - J. Rexy
AU - Kayal Padmanandam
AU - Balakrishnan D.
AU - Rasmi A.
AU - Athinarayanan Subramanian
AU - Hanaa A. Abdallah
AU - Abdelaziz A. Abdelhamid
AU - Ehab Seif Ghith
AU - Aseel Smerat
AU - Kaamran Raahemifar
AU - Sarita Simaiya
PY - 2026
DA - 2026/10/14
JO - Frontiers in Plant Science
DO - 10.3389/fpls.2026.1917550
UR - https://www.frontiersin.org/articles/10.3389/fpls.2026.1917550/full
AB - Introduction: Tomato leaf diseases substantially reduce crop yield and fruit quality, creating a need for accurate, computationally efficient, and potentially deployable image-based disease-screening systems for precision agriculture. This study presents an Attention-Enhanced YOLOv8n-CLS framework for tomato leaf disease classification by integrating a lightweight spatial attention mechanism to enhance the representation of spatially informative image features while maintaining a compact computational structure. Methods: Unlike conventional object detection approaches that require bounding-box annotations, the proposed framework formulates disease recognition as an image-level classification task, requiring only class labels and thereby simplifying dataset preparation and model training. Experiments were conducted on two publicly available benchmark datasets, namely the PlantVillage Tomato Dataset and the Kaggle Tomato Leaf Disease Dataset. A systematic evaluation comprising binary and multiclass classification, five-fold cross-validation, ablation analysis, cross-dataset evaluation, comparative benchmarking, computational assessment, and Grad-CAM-based visualization was performed to assess classification performance, stability, transferability, and computational efficiency. Results: The Attention-Enhanced YOLOv8n-CLS achieved a maximum classification Accuracy of 98.90%, an AUC of 0.991, a computational complexity of 7.5 GFLOPs, and an average inference speed of approximately 45 FPS on the evaluated computing platform. In five-fold cross-validation, the proposed model achieved a mean Accuracy of 98.50% ± 0.14% on the PlantVillage Tomato Dataset and 97.82% ± 0.29% on the Kaggle Tomato Leaf Disease Dataset. The model also demonstrated competitive performance relative to representative deep learning architectures, including ResNet50, DenseNet201, ResNet152V2, MobileNetV2, and YOLOv5. The cross-dataset experiments further demonstrated transferability between the evaluated PlantVillage and Kaggle image distributions using the 10 classes shared by the two datasets. Discussion: The results indicate a favourable balance between classification performance and computational requirements under the evaluated experimental conditions, with potential for further investigation in resource-constrained applications. Grad-CAM visualization provided qualitative information about image regions associated with the model's classification decisions but did not independently establish biological causality or pathogen-specific recognition. Importantly, the proposed framework performs image-based symptom classification and preliminary disease screening using dataset-provided labels; it does not independently confirm causal pathogens. Therefore, the reported classification performance should not be interpreted as definitive pathogen diagnosis or field-level validation. Further evaluation using independently collected field images, laboratory-confirmed disease or pathogen status, diverse symptom presentations, and representative agricultural edge devices is required before practical agricultural deployment can be established.
SN - 1664-462X
ER -