融合多头注意力机制的探地雷达反演网络

    Ground Penetrating Radar Inversion Network Combined with Multi-head Attention Mechanism

    • 摘要: 针对已有的基于神经网络的探地雷达反演方法特征提取不全、准确度不高、空间对应性差等局限,基于编码器—解码器结构,融合多头注意力机制和残差结构,提出一种新的探地雷达反演网络——注意力反演网络(AINet)。该网络首先利用单维度卷积和残差结构提取并压缩有效特征;接着使用多头注意力机制提取B-scan的全局语义信息,以改善数据对的空间对应关系并优化相关特征;最后通过介电常数解码器完成介电常数图的重建。实验采用基于时域有限差分法构建的探地雷达反演数据集对AINet训练、验证、测试,成功重建了介电常数分布图。在多种泛化场景测试下,与已有的深度学习反演方法进行对比分析,AINet的均方误差、平均绝对值误差和结构相异指数等指标相比其他网络均有显著降低。使用实测数据对AINet进行反演测试,结果表明文中方法能够根据探地雷达实测数据高效准确地反演出地下介电常数分布。

       

      Abstract: Aiming at the limitations of the existing ground penetrating radar inversion methods based on neural network, such as incomplete feature extraction, low accuracy and poor spatial correspondence, based on the encoder-decoder structure a new ground penetrating radar inversion network—attention inversion network (AINet) is proposed, which integrates multi-head attention mechanism and residual structure. Firstly, single dimensional convolution and residual structure are used to extract and compress effective features; then multi-head attention mechanism is used to extract global semantic information of B-scan, which improves the spatial correspondences between data pairs and optimizes relevant features; finally, the reconstruction of the dielectric constant graph is completed through a dielectric constant decoder. In the experiment, the AINet is trained, verified and tested by the ground penetrating radar inversion data set based on the finite difference time domain method, and the dielectric constant distribution map is successfully reconstructed. Compared with the existing deep learning inversion methods, the mean square error, mean absolute value error and structural dissimilarity index of AINet are significantly lower than those of other networks in a variety of generalization scenarios. The inversion test of AINet is carried out by using measured data, and the results show that the proposed method can efficiently and accurately inverse the distribution of underground dielectric constant according to the measured data of ground penetrating radar.

       

    /

    返回文章
    返回