Attention-driven graph convolutional neural networks for mineral prospectivity mapping
Changjie Cao, X. Wang, Fan Yang, Miao Xie, Bingli Liu, Yunhui Kong, Cheng Li, Zhongli Zhou
Abstract
• ADGCN for MPM achieves better accuracy and AUC than GCN and GAT. • Enhance interpretability by acquiring key nodes through attention mechanisms. • ADGCN captures complex spatial patterns and mitigates feature smoothing. Mineral Prospectivity Mapping (MPM) is a fundamental technique in the field of geosciences for identifying regions with high mineral potential. Graph Neural Networks (GNNs) have been extensively utilized in MPM, particularly excelling in handling non-Euclidean spatial data, effectively addressing the limitation of traditional deep neural networks, which struggle to capture and utilize spatial information. However, they often rely on direct geographic connections, which limit their ability to recognize long-distance geological relationships. This constraint impacts the understanding of mineralization processes and prediction accuracy, while feature smoothing in multi-layer graph convolution operations further weakens the aggregation of distant features. To address these challenges, this study proposes an Attention-Driven Graph Convolutional Network (ADGCN) that leverages an attention mechanism to select highly correlated nodes for connection and aggregates features from distant nodes for representation learning. Building upon the spatial information-capturing capabilities of traditional Graph Neural Networks, ADGCN further optimizes the process by dynamically prioritizing critical nodes, capturing complex spatial patterns and nonlinear relationships while alleviating feature smoothing and enhancing the aggregation of geographically distant but geologically related units. In tests conducted in the Lhasa region, ADGCN achieved an AUC of 91.67%, surpassing GAT by 2.65%, demonstrating superior prediction accuracy in MPM.