Prediction of cadmium contents in rice grains from Quaternary sediment-distributed farmland using field investigations and machine learning

The Quaternary sediment-distributed regions of South China are suitable for rice cultivation, which is crucial for ensuring food security. Spatial correlations between soil cadmium (Cd) and rice Cd contents are generally poor, making the evaluation of rice quality and associated health risks challenging. In this study, we developed machine learning algorithms for predicting rice Cd contents using 654 data pairs of soil-rice samples collected in Guangxi province, China. After a comprehensive comparison, our results showed that the random forest (RF) had the better performance than artificial neural network (ANN) based on all the data (RMSEtesting 0.066 vs. 0.099 and R2testing 0.860 vs. 0.688). The feature importance analysis showed that soil CaO, Cd, elevation, and rainfall were the four most important features affecting the rice Cd contents in the study area. Using the established RF-predicated model, the rice Cd contents were predicted at the provincial level with an additional dataset of 1176 paddy soil samples. The prediction result revealed about 23 % of farmland cultivated rice with Cd content over 0.2 mg kg−1 in the study area. Therefore, it is recommended to implement strict measures by local agricultural departments to reduce rice Cd contents and ensure food safety in these areas. Our study provides valuable insights into the prediction of rice Cd contents, thus contributing to ensuring food safety and preventing Cd exposure-associated health risks.