arXiv:2503.11408cs.LGcs.AI2025-03被引 1

用注意力门控网络加速三维磁测正演模拟,精度超传统方法。

A Neural Network Architecture Based on Attention Gate Mechanism for 3D Magnetotelluric Forward Modeling

  • 设计双路径注意力门控模块,融合浅层异常信息增强深层特征提取。
  • 预测结果结构相似度(SSIM)超0.98,收敛更稳定,泛化能力好。
  • 适合地质建模与电磁法正演仿真研究者使用,提升计算效率。

传统三维磁测(MT)数值正演方法如有限元法(FEM)和有限体积法(FVM)因网格细化与计算资源限制,存在计算成本高、效率低的问题。本文提出一种新型神经网络架构MTAGU-Net,融合注意力门控机制用于3D MT正演建模。通过在编码器与解码器间的跳跃连接中嵌入基于正演响应图像的双路径注意力门控模块,实现浅层特征图中关键异常信息向深层特征图的融合,显著提升网络对异常区域的特征提取能力。此外,引入基于3D高斯随机场(GRF)的合成模型生成方法,高保真复现真实地质电性结构。数值实验表明,MTAGU-Net在收敛稳定性与预测精度上优于传统3D U-Net,正演响应数据的结构相似性指数(SSIM)持续超过0.98。该网络可在未见过的数据集模型上准确预测正演响应,验证了其强泛化能力与实际应用可行性。

原文摘要 · Abstract (English)

Traditional three-dimensional magnetotelluric (MT) numerical forward modeling methods, such as the finite element method (FEM) and finite volume method (FVM), suffer from high computational costs and low efficiency due to limitations in mesh refinement and computational resources. We propose a novel neural network architecture named MTAGU-Net, which integrates an attention gating mechanism for 3D MT forward modeling. Specifically, a dual-path attention gating module is designed based on forward response data images and embedded in the skip connections between the encoder and decoder. This module enables the fusion of critical anomaly information from shallow feature maps during the decoding of deep feature maps, significantly enhancing the network's capability to extract features from anomalous regions. Furthermore, we introduce a synthetic model generation method utilizing 3D Gaussian random field (GRF), which accurately replicates the electrical structures of real-world geological scenarios with high fidelity. Numerical experiments demonstrate that MTAGU-Net outperforms conventional 3D U-Net in terms of convergence stability and prediction accuracy, with the structural similarity index (SSIM) of the forward response data consistently exceeding 0.98. Moreover, the network can accurately predict forward response data on previously unseen datasets models, demonstrating its strong generalization ability and validating the feasibility and effectiveness of this method in practical applications.

神经网络电磁建模注意力机制正演模拟

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。