Convolutional neural network-based applied research on the enrichment of heavy metals in the soil–rice system in China

The enrichment of heavy metals in the soil–rice system is affected by various factors, which hampers the prediction of heavy metal concentrations. In this research, a prediction model (CNN-HM) of heavy metal concentrations in rice was constructed based on convolutional neural network (CNN) technology and 17 environmental factors. For comparison, other machine learning models, such as multiple linear regression, Bayesian ridge regression, support vector machine, and backpropagation neural networks, were applied. Furthermore, the LH-OAT method was used to evaluate the sensitivity of CNN-HM to each environmental factor. The results showed that the R2 values of CNN-HM for Cd, Pb, Cr, As, and Hg were 0.818, 0.709, 0.688, 0.462, and 0.816, respectively, and both the MAE and RMAE values were acceptable. The sensitivity analysis showed that the concentrations of Cd and Pb, mechanical composition, soil pH, and altitude were the main sensitive features for CNN-HM. Compared with CNN-HM based on all input features, the performance of the quick prediction model that was based on the sensitive features did not degrade significantly, thereby indicating that CNN-HM has stronger stability and robustness. The quick prediction model has extensive application value for timely prediction of the enrichment of heavy metals in emergencies. This study demonstrated the effectiveness and practicability of CNNs in predicting heavy metal enrichment in the soil–rice system and provided a new perspective and solution for heavy metal prediction.