Back to Search Start Over

Improving plant disease classification by adaptive minimal ensembling

Authors :
Antonio Bruno
Davide Moroni
Riccardo Dainelli
Leandro Rocchi
Silvia Morelli
Emilio Ferrari
Piero Toscano
Massimo Martinelli
Source :
Frontiers in Artificial Intelligence, Vol 5 (2022)
Publication Year :
2022
Publisher :
Frontiers Media S.A., 2022.

Abstract

A novel method for improving plant disease classification, a challenging and time-consuming process, is proposed. First, using as baseline EfficientNet, a recent and advanced family of architectures having an excellent accuracy/complexity trade-off, we have introduced, devised, and applied refined techniques based on transfer learning, regularization, stratification, weighted metrics, and advanced optimizers in order to achieve improved performance. Then, we go further by introducing adaptive minimal ensembling, which is a unique input to the knowledge base of the proposed solution. This represents a leap forward since it allows improving the accuracy with limited complexity using only two EfficientNet-b0 weak models, performing ensembling on feature vectors by a trainable layer instead of classic aggregation on outputs. To the best of our knowledge, such an approach to ensembling has never been used before in literature. Our method was tested on PlantVillage, a public reference dataset used for benchmarking models' performances for crop disease diagnostic, considering both its original and augmented versions. We noticeably improved the state of the art by achieving 100% accuracy in both the original and augmented datasets. Results were obtained using PyTorch to train, test, and validate the models; reproducibility is granted by providing exhaustive details, including hyperparameters used in the experimentation. A Web interface is also made publicly available to test the proposed methods.

Details

Language :
English
ISSN :
26248212
Volume :
5
Database :
Directory of Open Access Journals
Journal :
Frontiers in Artificial Intelligence
Publication Type :
Academic Journal
Accession number :
edsdoj.4aaf011d63824301aca9a5236e70d37b
Document Type :
article
Full Text :
https://doi.org/10.3389/frai.2022.868926