一种融合地质规则的多头注意力模型训练方法

本发明涉及地质分析技术，其公开了一种融合地质规则的多头注意力模型训练方法，提高地质体模型预测的科学性和准确性，并提升地质分析过程的可解释性。本发明方案中，首先将地质属性数据与地质坐标数据的特征向量进行拼接，获得融合特征向量，接着同时考虑空间局部性和方向性，对融合特征向量进行地构多头注意力计算，获得注意力特征序列；然后对输入地质图提取多尺度特征，并基于断层约束构建跨断层权重掩码矩阵，通过融合获得综合特征；采用包含预测损失与断层不可穿透性约束项的联合损失函数计算损失值，通过反向传播算法更新模型参数，完成模型的训练；最后绘制注意力热力图进行可视化展示，并与三维地质模型叠加。