AI Ensemble Learns to Predict Rice Yields With Stunning Accuracy

Rice feeds nearly half of humanity, and predicting how much of it a field will produce has long been one of agriculture’s hardest problems. A new study published in Discover Artificial Intelligence by researchers at the University of Gondar in Ethiopia now reports that a carefully constructed ensemble of machine learning models, paired with a statistical technique for identifying the most informative variables, can explain more than 97.8 percent of the variance in rice yield from a handful of environmental and management inputs. The work, led by Fikadu Berie Adugna with colleagues Yonatan Motbaynor Tebabal and Esubalew Asmara Desta, is framed as a benchmark study, and its authors are unusually candid about both the promise and the limits of their results.

The research team built their framework around four of the most widely used regression algorithms in modern machine learning: Random Forest, XGBoost, LightGBM, and Gradient Boosting. Each of these belongs to the family of tree-based learners, which make predictions by combining hundreds or thousands of decision trees, each trained on slightly different slices of the data. Random Forest averages the outputs of independently grown trees to suppress variance, while the boosting methods build trees sequentially, with each new tree trained to correct the residual errors left by its predecessors. These approaches excel at capturing the tangled, nonlinear relationships that characterize agricultural systems, where the effect of rainfall on yield may depend on temperature, soil type, and fertilizer application all at once.

Individually, all four models performed remarkably well on the study’s dataset, achieving coefficients of determination, or R-squared values, above 0.975. Gradient Boosting edged out the others with an R-squared of 0.9775, a root mean squared error of 5.60, and a mean absolute error of 4.41. XGBoost followed closely with an R-squared of 0.9759, and LightGBM and Random Forest landed within a few thousandths of a point of the same mark. The near-identical performance across such different algorithms was itself a signal: the data contained a strong, structured relationship between the input variables and the target yield, a point that would later prove central to interpreting the study’s conclusions.

To push beyond the performance of any single model, the researchers constructed a hybrid voting ensemble, a Voting Regressor that aggregates the predictions of all four base learners into a single output. The intuition behind ensembling is straightforward: different algorithms make different kinds of mistakes, and averaging their predictions can cancel out individual errors while preserving their collective strengths. The ensemble without any feature selection achieved an R-squared of 0.9768, marginally below the best individual model. This modest result turned out to be instructive. Ensembling alone did not deliver the headline gains; the real breakthrough came when the researchers combined the ensemble with an aggressive pruning of the input variables.

That pruning was accomplished through two advanced feature selection methods. The first, Boruta, is a wrapper algorithm that judges each variable’s importance by comparing it against randomly shuffled copies of itself, so-called shadow features. Any real variable that consistently outperforms its shadow counterpart is deemed genuinely informative, a design that guards against selecting variables that merely appear important by chance. The second method, permutation importance, takes a more direct approach: it shuffles the values of one variable at a time and measures how much the model’s accuracy degrades. If scrambling a variable wrecks the predictions, that variable matters; if the model barely notices, it does not. Both methods were applied strictly within each training fold of a five-fold cross-validation scheme, a discipline that prevents information from the validation data from leaking into the selection process.

The combination of permutation-based feature selection with the voting ensemble produced the study’s best result: an R-squared of 0.9787, a root mean squared error of 5.4421, and a mean absolute error of 4.3243, the lowest error figures recorded in any configuration. Boruta-based selection also improved over the baseline ensemble, reaching an R-squared of 0.9775, but fell short of the permutation approach. The authors attribute this difference to how the two methods handle correlated predictors. Boruta can retain variables whose contributions overlap with those of related features, whereas permutation importance directly quantifies each variable’s actual impact on prediction error, making it better suited to identifying the truly influential factors in data where environmental and management variables carry redundant information.

Statistical testing reinforced the ranking. Paired t-tests on repeated cross-validation error values, conducted at a significance level of 0.05, showed that the Voting-plus-Permutation framework’s improvements over the plain ensemble, the Boruta variant, and the best individual model were unlikely to be artifacts of random variation. Five-fold cross-validation confirmed the robustness of the winning configuration, yielding a mean cross-validated R-squared of 0.9789 with a standard deviation of just 0.0018, alongside a mean cross-validated RMSE of 5.496. The tight agreement between cross-validation scores and independent test scores suggested the models were not overfitting, and the researchers further guarded against that risk by limiting tree depth, moderating learning rates, and tuning hyperparameters through grid search on validation performance.

Perhaps the most practically interesting finding concerns which variables mattered most. Feature importance analysis crowned fertilizer application rate, measured in kilograms, as the single strongest predictor of rice yield, followed by rainfall in millimeters, temperature in degrees Celsius, and relative humidity. Irrigation method, soil pH, planting density, and soil type rounded out the influential set. The prominence of both management factors and climatic variables underscores a central tension in agricultural AI: some levers, like fertilizer, are within a farmer’s control, while others, like rainfall and temperature, are not. A model that quantifies the relative weight of each can, in principle, help farmers and policymakers decide where intervention will pay off most, particularly as climate variability makes rain-fed rice systems increasingly unpredictable.

Yet the study’s most important caveat is one the authors state repeatedly and plainly: the entire evaluation was performed on a synthetic dataset. The team drew on a Kaggle benchmark of roughly 10,000 simulated agricultural records, filtered down to 1,534 valid rice observations after quality checks, and split 80:20 into training and test sets. Synthetic data of this kind is clean, structured, and relatively free of the noise, gaps, and spatial-temporal messiness that plague real field measurements. The authors explicitly warn that the strikingly high accuracy may partly reflect the dataset’s design rather than genuine agricultural complexity, and that the results should be read as a methodological benchmark rather than evidence of a field-ready forecasting system. Error analysis reinforced this caution: predictions were most reliable for mid-range yields, where data was dense, and less accurate at extreme low and high values and under rare conditions such as extreme rainfall or unusual soil-fertilizer combinations.

The ethical stakes of agricultural AI also receive serious attention. Inaccurate yield forecasts could lead smallholder farmers, who have the least financial cushion, to misallocate fertilizer, irrigation water, labor, or investment. The authors argue that such models should serve strictly as decision-support tools that inform, but never replace, expert agronomic judgment, and that deployment should follow validation across multiple regions, seasons, and farming systems, with periodic recalibration, bias evaluation, and explainable interfaces so that end users understand the confidence behind each prediction. Future work, they suggest, should incorporate real climate station records, soil surveys, remote sensing data, and national crop statistics, and explore deep learning architectures capable of capturing temporal patterns. For now, the study stands as a rigorous demonstration that when ensemble learning is paired with intelligent feature selection, machine learning can extract near-perfect predictive signal from structured agricultural data, provided we remember that the field it modeled, for the moment, exists only in silicon.

Subject of Research: Ensemble machine learning with feature selection for rice yield prediction

Article Title: Benchmark study of ensemble machine learning with feature selection for rice yield prediction

Article References: Adugna, F. B., Tebabal, Y. M., & Desta, E. A. (2026). Benchmark study of ensemble machine learning with feature selection for rice yield prediction. Discover Artificial Intelligence, 6(1), Article 1215. https://doi.org/10.1007/s44163-026-02201-3

Image Credits: AI Generated

DOI: 10.1007/s44163-026-02201-3

Keywords: rice yield prediction, machine learning, ensemble learning, feature selection, XGBoost, Random Forest, LightGBM, permutation importance, Boruta, precision agriculture, food security, synthetic dataset