UPDF AI

Interpretable Machine Learning for Multi-Class Crop Yield Prediction

Aldian Nurcahyo,Yaya Heryadi,2 Authors,I. Sonata

2023 · DOI: 10.1109/ICICyTA60173.2023.10428914
4 Citations

Abstract

Crop Yield Analysis and Prediction is a fast-expanding discipline that is critical for optimizing agricultural methods. A lack of trustworthy data is one of the challenges in estimating crop yields. We develop predictive models for 22 different fruits and vegetables data. The goals of this study are to create accurate and interpretable crop recommendation models. We used multiple machine learning (ML) models for multi-class crop production prediction to fulfill our research goal. We thoroughly examined the influence of climate and nutrient factors on crop yield, considering their complex interactions. To improve the dataset, augmented data techniques were applied. Configuring the parameters and fine-tuning the hyperparameters is our technique to increase the model performance. Furthermore, we employ explainable artificial intelligence (XAI) techniques and interpretability tools like Shapley Additive exPlanations (SHAP) to improve the interpretability of our prediction model. Our findings reveal that the XGBoost model has the best performance model with 99.86% accuracy, followed by SVM Poly Kernel with 99.32% and Random Forest with 98.82%. Feature selection and analysis are emphasized, particularly in regional agricultural contexts. This study contributes to the creation of accurate and interpretable crop recommendation models while also addressing the issue of untrustworthy data, providing useful insights for optimizing agricultural practices.