ViT-KAN模型驱动下的砂岩热储地热潜力智能评估与可解释归因分析

Intelligent evaluation and interpretable attribution analysis of sandstone geothermal potential using ViT-KAN model

  • 摘要: 砂岩热储地热资源具有清洁、稳定及可再生优势,在“双碳”目标下对区域能源结构转型意义重大,但地质条件复杂、热储非均质性强及勘探不确定性等因素使区域地热潜力精准评估面临挑战. 本文以鲁西北平原为研究区,综合10项地热地质指标,采用皮尔逊相关性分析与方差膨胀因子开展特征筛选;采用KAN(Kolmogorov–Arnold networks)网络替换ViT(Vision transformer)模型Transformer编码器中的多层感知机模块,构建ViT-KAN砂岩热储地热资源潜力评估模型,结合SHAP(Shapley additive explanations)与反事实分析方法,揭示影响地热潜力的关键因素对模型预测的贡献及其作用方向. 结果表明:底板埋深与热储厚度呈显著正相关,相关系数为0.88,经特征筛选后剔除底板埋深特征,以降低输入特征冗余;ViT-KAN模型测试集的受试者工作特征曲线下面积、准确率和召回率分别为0.936、0.840和0.907,均优于ViT模型,预测误差更低;高地热潜力区主要分布于研究区北部德城区、陵城区与平原县,模型预测结果与地热井点空间分布较为一致,ViT-KAN模型在低潜力区识别和勘探风险控制方面具有一定优势;SHAP与反事实分析结果表明,热储温度、孔隙度和热储厚度是影响地热潜力预测的关键因素. 研究成果可为区域地热资源评价与勘探靶区优选提供技术支撑.

     

    Abstract: Geothermal resources hosted in sandstone reservoirs offer a clean, reliable, and renewable energy source that can support regional energy transitions under China’s carbon peaking and carbon neutrality targets. However, complex geology, pronounced reservoir heterogeneity, and exploration uncertainty hinder the accurate regional assessment of geothermal-resource potential. In the Northwestern Shandong Plain, 10 geological variables relevant to geothermal potential were compiled, and feature selection was performed using Pearson correlation analysis and variance inflation factors. Subsequently, a vision transformer Kolmogorov–Arnold network (ViT-KAN) model was developed to assess the geothermal-resource potential of sandstone reservoirs by replacing the multilayer perceptron module in the transformer encoder of the ViT with a KAN. Model performance was evaluated from the complementary perspectives of classification discrimination and probabilistic prediction accuracy using six metrics: area under the receiver operating characteristic curve (AUC), accuracy (ACC), recall, mean squared error (MSE), root mean squared error (RMSE), and mean absolute error (MAE). Shapley additive explanations (SHAP) and counterfactual analysis were integrated to identify key factors governing the model predictions. The depth to the base of the geothermal reservoir was strongly and positively correlated with reservoir thickness, with a Pearson correlation coefficient of 0.88; therefore, it was excluded during feature selection to reduce redundancy among the input variables. On the test set, the ViT-KAN model achieved an AUC of 0.936, an accuracy of 0.840, and a recall of 0.907, thus exceeding corresponding ViT values of 0.914, 0.802, and 0.887, respectively. The test set MSE, RMSE, and MAE of the ViT-KAN model were 0.161, 0.401, and 0.311, respectively, compared with 0.201, 0.448, and 0.344 for the ViT. These results indicate that the ViT-KAN model provides better classification discrimination and higher probabilistic prediction accuracy compared with the ViT alone. The predicted probabilities were classified into five geothermal potential levels using the natural-breaks method. Although the two models yielded broadly consistent spatial patterns, the ViT-KAN model provided a more detailed delineation of areas with high geothermal potential. Under the ViT-KAN model, areas with very low and low geothermal potential constituted 60.9% of the study area, whereas areas with medium, high, and very high geothermal potential constituted 18.1%, 12.2%, and 8.8%, respectively. Areas with high geothermal potential were concentrated primarily in Decheng District, Lingcheng District, and Pingyuan County in the northern region of the study area. The predicted potential classes were generally consistent with the spatial distribution of known wells, and 87.61% of the geothermal wells were located in areas classified by the ViT-KAN model as exhibiting very high geothermal potential. The reduced proportion of non-geothermal wells in areas with medium and high geothermal potential suggests that the ViT-KAN model can identify areas with low geothermal potential and mitigate exploration risk more effectively. SHAP and counterfactual analyses consistently identified reservoir temperature, porosity, and reservoir thickness as the most influential variables. Higher values of all three variables generally increased the predicted geothermal potential, while reducing them individually or jointly shifted a sample from high to low potential class. The agreement between the two interpretation methods enhances the credibility of the model predictions. These findings provide a scientific basis for regional geothermal-resource assessment and the prioritization of exploration targets.

     

/

返回文章
返回