Lightweight Vision Transformer Compression for On-Device Plant Disease Detection in Resource-Constrained Agricultural Field Conditions
On the map: Knowledge distillation → Detection and dense prediction distillation
Abstract (arXiv)
Chilli (Capsicum annuum) is one of India's most economically significant crops, yet its productivity is persistently threatened by diseases that are difficult to identify without expert intervention. While Vision Transformers (ViTs) have achieved high classification accuracy, their large computational footprint makes deployment on resource constrained devices challenging. Existing compression approaches typically address pruning, quantization, and knowledge distillation in isolation, leaving the potential benefits and interactions of their combined application insufficiently explored. We propose a unified Vision Transformer compression framework that combines Hessian-Balanced Adaptive Block Pruning (H-BAC), guided by second-order sensitivity estimation, with quantization and attention-based knowledge distillation. To systematically identify the most effective configuration within each compression family, each technique is first evaluated independently through controlled ablation studies, after which the best-performing components are integrated into a sequential deployment pipeline tailored to real-world agricultural constraints. On a chilli 3-class village-split dataset with a genuine cross-village, cross-device out-of-distribution test split, the resulting compressed models match or exceed the 95.13% FP32 baseline's accuracy, alongside 74-98% model size reduction, and the fully integrated compression pipeline achieves a 54.5x size reduction (327.42 MB to 6.01 MB) at 95.13 +/- 2.32% accuracy across four tested configurations. A direct comparison further reveals that, on this dataset, a directly-trained student of the same final size, without pruning or distillation, reaches comparable accuracy of 94.87%, at the same 6.01 MB INT8 size, indicating where H-BAC and knowledge distillation are, and are not yet shown to be, worth their computational cost.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in Detection and dense prediction distillation, one of 12 directions in Knowledge distillation. Builds on 10 of its 38 references that are in the map. First posted on arXiv 4 Sep 2026.
Builds on
10 of 38 references.
References · 10
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2015Distilling the Knowledge in a Neural Networkcited by 10,324
- 2015Deep Compression: Compressing Deep Neural Networks with Pruning, Trained Quantization and Huffman Codingcited by 2,329
- 2016Using Deep Learning for Image-Based Plant Disease Detectioncited by 219
- 2021Global Vision Transformer Pruning with Hessian-Aware Saliencycited by 70
- 2022ViTKD: Practical Guidelines for ViT feature knowledge distillationcited by 34
- 2021PQK: Model Compression via Pruning, Quantization, and Knowledge Distillationcited by 21
- 2023Constraint-aware and Ranking-distilled Token Pruning for Efficient Transformer Inferencecited by 6
- 2024Edge-AI for Agriculture: Lightweight Vision Models for Disease Detection in Resource-Limited Settingscited by 1
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026TinyCNN: A 193K-Parameter Network for On-Device Plant Disease Detection, with a Cross-Dataset Robustness Diagnosis
- 2025Hybrid Knowledge Transfer through Attention and Logit Distillation for On-Device Vision Systems in Agricultural IoT
- 2025Involution-Infused DenseNet with Two-Step Compression for Resource-Efficient Plant Disease Classification
- 2026AgriKD: Cross-Architecture Knowledge Distillation for Efficient Leaf Disease Classification
- 2026Cross-Architecture Knowledge Distillation from a Vision Foundation Model to a Lightweight Visual State Space Model for Tea Leaf Disease Classification
- 2026Energy-Efficient Plant Monitoring via Knowledge Distillation
- 2025Hardware-Aware YOLO Compression for Low-Power Edge AI on STM32U5 for Weeds Detection in Digital Agriculture
- 2025Self-Consistency in Vision-Language Models for Precision Agriculture: Multi-Response Consensus for Crop Disease Management