This paper presents a computer vision-based system for the early detection of plant leaf diseases, developed using TinyMLOps principles to enable deployment on resourceconstrained embedded devices. The solution leverages the MobileNet architecture with a custom classification head that is properly trained using the PlantVillage dataset to distinguish between healthy, unhealthy, and no leaves images. Given the class imbalance of the relabeled dataset, we augmented the underrepresented classes (No leaves) using basic image transformations to improve model generalization. To ensure both high accuracy and computational feasibility, considering that the target device has limited resources, we apply a multi-objective hyperparameter optimization strategy using the Optuna framework, incorporating the NSGA-II algorithm to identify optimal trade-offs between model accuracy (measured by F1-score) and computational constraints (number of model parameters). The optimization process identifies a set of Pareto-optimal models, with the most promising configurations residing in the “knee region”, offering strong classification performance with significantly reduced parameter counts. From the Pareto front, we selected a few models that were evaluated on a Raspberry Pi Zero 2W to assess their realworld performance in terms of inference latency, memory usage, and overall suitability for deployment in resource-constrained environments. This poses the basis for building a low-cost solution with commodity devices for early disease detection in agriculture, including rural areas where computational resources and connectivity are limited. Finally, the dataset we used for this work has been made public as an asset of the AgrifoodTEF Data Space.
A TinyMLOps-Based Edge AI Approach for Early Detection of Emerging Plant Diseases
Hossein Aqasizade;Mattia Antonini
;Massimo Vecchio;Fabio Antonelli
2025-01-01
Abstract
This paper presents a computer vision-based system for the early detection of plant leaf diseases, developed using TinyMLOps principles to enable deployment on resourceconstrained embedded devices. The solution leverages the MobileNet architecture with a custom classification head that is properly trained using the PlantVillage dataset to distinguish between healthy, unhealthy, and no leaves images. Given the class imbalance of the relabeled dataset, we augmented the underrepresented classes (No leaves) using basic image transformations to improve model generalization. To ensure both high accuracy and computational feasibility, considering that the target device has limited resources, we apply a multi-objective hyperparameter optimization strategy using the Optuna framework, incorporating the NSGA-II algorithm to identify optimal trade-offs between model accuracy (measured by F1-score) and computational constraints (number of model parameters). The optimization process identifies a set of Pareto-optimal models, with the most promising configurations residing in the “knee region”, offering strong classification performance with significantly reduced parameter counts. From the Pareto front, we selected a few models that were evaluated on a Raspberry Pi Zero 2W to assess their realworld performance in terms of inference latency, memory usage, and overall suitability for deployment in resource-constrained environments. This poses the basis for building a low-cost solution with commodity devices for early disease detection in agriculture, including rural areas where computational resources and connectivity are limited. Finally, the dataset we used for this work has been made public as an asset of the AgrifoodTEF Data Space.I documenti in IRIS sono protetti da copyright e tutti i diritti sono riservati, salvo diversa indicazione.
