# MangoLeafViT: Leveraging Lightweight Vision Transformer with Runtime Augmentation for Efficient Mango Leaf Disease Classification

arXiv: https://arxiv.org/abs/2505.23961 · HTML: https://mlsanity.com/paper/2505.23961/

## Key facts

- Cited by 4 papers in the map (data through 14 Oct 2026).
- Builds on 18 of its 32 references that are in the map.
- First posted on arXiv 29 May 2025.

## Builds on (most cited first, 18 of 18)

- [Attention Is All You Need](https://mlsanity.com/paper/1706.03762/) (2017)
- [Adam: A Method for Stochastic Optimization](https://mlsanity.com/paper/1412.6980/) (2014)
- [An Image is Worth 16x16 Words: Transformers for Image Recognition at Scale](https://mlsanity.com/paper/2010.11929/) (2020)
- [Grad-CAM: Visual Explanations from Deep Networks via Gradient-based Localization](https://mlsanity.com/paper/1610.02391/) (2016)
- [Densely Connected Convolutional Networks](https://mlsanity.com/paper/1608.06993/) (2016)
- [MobileNetV2: Inverted Residuals and Linear Bottlenecks](https://mlsanity.com/paper/1801.04381/) (2018)
- [Do Vision Transformers See Like Convolutional Neural Networks?](https://mlsanity.com/paper/2108.08810/) (2021)
- [MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformer](https://mlsanity.com/paper/2110.02178/) (2021)
- [TinyViT: Fast Pretraining Distillation for Small Vision Transformers](https://mlsanity.com/paper/2207.10666/) (2022)
- [Deep learning in agriculture: A survey](https://mlsanity.com/paper/1807.11809/) (2018)
- [An open access repository of images on plant health to enable the development of mobile disease diagnostics](https://mlsanity.com/paper/1511.08060/) (2015)
- [EfficientViT: Multi-Scale Linear Attention for High-Resolution Dense Prediction](https://mlsanity.com/paper/2205.14756/) (2022)
- [Less is More: Lighter and Faster Deep Neural Architecture for Tomato Leaf Disease Classification](https://mlsanity.com/paper/2109.02394/) (2021)
- [Two Decades of Bengali Handwritten Digit Recognition: A Survey](https://mlsanity.com/paper/2206.02234/) (2022)
- [MangoLeafBD: A Comprehensive Image Dataset to Classify Diseased and Healthy Mango Leaves](https://mlsanity.com/paper/2209.02377/) (2022)
- [Fruit Quality Assessment with Densely Connected Convolutional Neural Network](https://mlsanity.com/paper/2212.04255/) (2022)
- [Rethinking Cooking State Recognition with Vision Transformers](https://mlsanity.com/paper/2212.08586/) (2022)
- [An Efficient Transfer Learning-based Approach for Apple Leaf Disease Classification](https://mlsanity.com/paper/2304.06520/) (2023)

## Built on by (most cited first, 4 of 4)

- [AQUA20: A Benchmark Dataset for Underwater Species Classification under Challenging Conditions](https://mlsanity.com/paper/2506.17455/) (2025)
- [DExNet: Combining Observations of Domain Adapted Critics for Leaf Disease Classification with Limited Data](https://mlsanity.com/paper/2506.18173/) (2025)
- [Bi-cLSTM: Residual-Corrected Bidirectional LSTM for Aero-Engine RUL Estimation](https://mlsanity.com/paper/2603.00745/) (2026)
- [SpectroFusion-ViT: A Lightweight Transformer for Speech Emotion Recognition Using Harmonic Mel-Chroma Fusion](https://mlsanity.com/paper/2603.00746/) (2026)

## Abstract (arXiv)

Ensuring food safety is critical due to its profound impact on public health, economic stability, and global supply chains. Cultivation of Mango, a major agricultural product in several South Asian countries, faces high financial losses due to different diseases, affecting various aspects of the entire supply chain. While deep learning-based methods have been explored for mango leaf disease classification, there remains a gap in designing solutions that are computationally efficient and compatible with low-end devices. In this work, we propose a lightweight Vision Transformer-based pipeline with a self-attention mechanism to classify mango leaf diseases, achieving state-of-the-art performance with minimal computational overhead. Our approach leverages global attention to capture intricate patterns among disease types and incorporates runtime augmentation for enhanced performance. Evaluation on the MangoLeafBD dataset demonstrates a 99.43% accuracy, outperforming existing methods in terms of model size, parameter count, and FLOPs count.