Sensitivity of machine learning regression models to data structure and quality in crop yield prediction
- 1Laboratoire de Biomathématiques et d’Estimations Forestières, Faculté des Sciences Agronomiques, Université d’Abomey-Calavi, Abomey-Calavi, Atlantique, Bénin
Abstract
Accurate crop yield prediction is essential for agricultural planning, yet machine learning (ML) models remain highly sensitive to the quality and structure of input data. This study uses simulated datasets to systematically investigate how data structure (sample size and number of predictors), data imperfections (missing values and multicollinearity), and pre-processing methods (imputation techniques and principal component analysis) influence ML regression performance. The performance of ML-based models (RF, SVM, MLR, XGBoost, LightGBM, NNet, and kNN) on the pre-processed data was evaluated. In total, each algorithm was tested on 1,728 datasets and pre-processing scenarios, yielding 12,096 model-scenario evaluations across the seven ML algorithms. Results show that missing data decreases performance (R2 drops by up to 20.63%; MAE increases by 12.67%), while multicollinearity may inflate R2 values despite poorer MAE performance. Larger sample sizes consistently improve prediction accuracy (R2 = +18.87%; MAE = −8.16%), whereas more predictors generally reduce it (R2 = −8.47%; MAE = +44.74%). Regression-based imputation improved both R2 and MAE the most, while RF demonstrated greater robustness across varying conditions. These simulation findings were further validated on five real-world crop datasets (Maize, Yam, Cassava, Sorghum, and Peanuts), confirming that despite RF remains the safest and most robust default, no single pre-processing strategy is universally optimal and that the selection of imputation method and dimensionality reduction must be strictly contingent upon the dataset’s specific missingness rate and correlation structure. Overall, this study highlights the complex interplay between data characteristics and pre-processing, urging the development of clearer guidelines for applying ML to agricultural datasets.
Related
- An automated ranking platform for machine learning regression models for meat spoilage prediction using multi-spectral imaging and metabolic profiling
- Neopep: Reinforcement Learning From Machine-Learned Force Field Energy for Non-Canonical Peptide Structure Prediction
- Machine learning and spatial regression approaches to estimating willingness to pay for ecosystem services in Jakarta and Taoyuan
- Regional climate variability and rice yields prediction in Bangladesh using machine learning and deep learning models
- Integrating Multisource Target Evidence with Machine-Learning Potency Prediction for Alzheimer's Disease Drug Discovery
BibTeX
@article{zinzinhedo2026sensitivity,
title = {{Sensitivity of machine learning regression models to data structure and quality in crop yield prediction}},
author = {Mahoukpégo Luc Zinzinhédo and Mèvognon Firmin Mitchozounnou and Kolawolé Valère Salako and Romain Glèlè Kakaï},
journal = {PLOS One},
year = {2026},
month = sep,
doi = {10.1371/journal.pone.0353938},
url = {https://doi.org/10.1371/journal.pone.0353938},
note = {CC BY 4.0},
}
RIS
TY - JOUR
TI - Sensitivity of machine learning regression models to data structure and quality in crop yield prediction
AU - Mahoukpégo Luc Zinzinhédo
AU - Mèvognon Firmin Mitchozounnou
AU - Kolawolé Valère Salako
AU - Romain Glèlè Kakaï
PY - 2026
DA - 2026/09/01
JO - PLOS One
DO - 10.1371/journal.pone.0353938
UR - https://doi.org/10.1371/journal.pone.0353938
AB - Accurate crop yield prediction is essential for agricultural planning, yet machine learning (ML) models remain highly sensitive to the quality and structure of input data. This study uses simulated datasets to systematically investigate how data structure (sample size and number of predictors), data imperfections (missing values and multicollinearity), and pre-processing methods (imputation techniques and principal component analysis) influence ML regression performance. The performance of ML-based models (RF, SVM, MLR, XGBoost, LightGBM, NNet, and kNN) on the pre-processed data was evaluated. In total, each algorithm was tested on 1,728 datasets and pre-processing scenarios, yielding 12,096 model-scenario evaluations across the seven ML algorithms. Results show that missing data decreases performance (R2 drops by up to 20.63%; MAE increases by 12.67%), while multicollinearity may inflate R2 values despite poorer MAE performance. Larger sample sizes consistently improve prediction accuracy (R2 = +18.87%; MAE = −8.16%), whereas more predictors generally reduce it (R2 = −8.47%; MAE = +44.74%). Regression-based imputation improved both R2 and MAE the most, while RF demonstrated greater robustness across varying conditions. These simulation findings were further validated on five real-world crop datasets (Maize, Yam, Cassava, Sorghum, and Peanuts), confirming that despite RF remains the safest and most robust default, no single pre-processing strategy is universally optimal and that the selection of imputation method and dimensionality reduction must be strictly contingent upon the dataset’s specific missingness rate and correlation structure. Overall, this study highlights the complex interplay between data characteristics and pre-processing, urging the development of clearer guidelines for applying ML to agricultural datasets.
SN - 1932-6203
ER -