Attention mechanism-enhanced graph convolutional neural network for unbalanced lithology identification

Abstract

Abstract In this study, we propose a novel method for identifying lithology using an attention mechanism-enhanced graph convolutional neural network (AGCN). The aim of this method is to address the limitations of traditional approaches that evaluate unbalanced lithology by improving the identification of thin layers and small samples, while providing reliable data support for reservoir evaluation. To achieve this goal, we begin by using Principal Component Analysis (PCA) with maximum and minimum distance clustering (Max-min-distance) to correct the logging curves, which compensates for the low resolution of thin layers and enhances the accuracy of stratigraphic representation. Subsequently, we transform the logging data into graph-structured data by connecting distance similarity points and feature similarity points of the logging samples. We then use the graph convolutional network (GCN) to identify lithology, leveraging both labeled and unlabeled data to enhance the ability to identify lithology in small sample datasets. Additionally, our model incorporates a channel and spatial attention mechanism that assigns weights to the graph structure during lithology identification, improving the model’s capability to discern differences across samples. To evaluate the performance of our model, we constructed a lithology dataset comprising five wells and conducted experiments. The results indicate that our approach achieves a maximum accuracy of 97.67%, surpassing the performance of a singlestructure model in lithology identification. In conclusion, our proposed method provides a promising and effective approach for unbalanced lithology identification, significantly improving accuracy levels.