The effect of tuning, feature engineering, and feature selection in data mining applied to rainfed sugarcane yield modelling

Abstract

Crop yield models can assist decision makers within any agro-industrial supply chain, even with regard to decisions that are unrelated to the crop production. Considering the characteristics of the mechanisms and data related to yield, data mining techniques are suitable candidates for modelling. The use of these techniques within a context with feature engineering, feature selection, and proper tuning can further improve performance beyond a simple replacement of multiple linear regression. To evaluate the impact of the different steps in the mentioned context, we evaluated sugarcane (Saccharum spp.) yield modelling with data obtained from a sugarcane mill. For a combination of six techniques, tuning, feature selection, and feature engineering, leading to 66 combinations, we assessed final model performance. Average performance across combinations resulted in a mean absolute error (MAE) of 6.42Mgha−1. Using different techniques led to a range of MAE from 4.57 to 8.80Mgha−1 on average. The best and worst performances for an individual model were MAEs of 4.11 and 9.00Mgha−1. Models with lower performance were close to simply predicting yield from the average yield for each number of cuts (MAE of 9.86Mgha−1). Tuning and feature engineering reduced the MAE on average by 1.17 and 0.64Mgha−1, respectively. Feature selection removed nearly 40% of the features but increased the MAE by 0.19Mgha−1. The performance of models was improved by simple strategies such as decomposing weather attributes and detailing fertilisation. Evaluation of feature importance provided by the RReliefF feature selection algorithm was used to explain the performance gains. If empirical models are needed, they will rely on using advanced techniques, but they will need proper algorithm tuning and feature engineering to extract most of the information from datasets. Based on the results, we recommend following the presented workflow for the development of yield models.