A novel prediction approach driven by graph representation learning for heavy metal concentrations.

ABSTRACT

The potential risk of heavy metals (HMs) to public health is an issue of great concern. Early prediction is an effective means to reduce the accumulation of HMs. The current prediction methods rarely take internal correlations between environmental factors into consideration, which negatively affects the accuracy of the prediction model and the interpretability of intrinsic mechanisms. Graph representation learning (GraRL) can simultaneously learn the attribute relationships between environmental factors and graph structural information. Herein, we developed the GraRL-HM method to predict the HM concentrations in soil-rice systems. The method consists of two modules, which are PeTPG and GCN-HM. In PeTPG, a graphic structure was generated using graph representation and communitization technology to explore the correlations and transmission paths of different environmental factors. Subsequently, the GCN-HM model based on the graph convolutional neural network (GCN) was used to predict the HM concentrations. The GraRL-HM method was validated by 2295 sets of data covering 21 environmental factors. The results indicated that the PeTPG model simplified correlation paths between factor nodes from 396 to 184, reducing by 53.5 % graph scale by eliminating the invalid paths. The concise and efficient graph structure enhanced the learning efficiency and representation accuracy of downstream prediction models. The GCN-HM model was superior to the four benchmark models in predicting the HM concentration in the crop, improving R2 by 36.1 %. This study develops a novel approach to improve the prediction accuracy of pollutant accumulation and provides valuable insights into intelligent regulation and planting guidance for heavy metal pollution control.