基于全波形反演和卷积神经网络的地下岩层分布成像获取方法、系统、终端及可读存储介质与流程

1.本发明属于勘探技术领域，具体涉及一种基于全波形反演和卷积神经网络的地下岩层分布成像获取方法、系统、终端及可读存储介质。


背景技术：

2.随着地质勘探的不断发展，待勘测的地质条件越发复杂，对于地下介质的成像精度要求也越来越高。虽然传统成像方法全波形反演(fwi，full waveform inversion)具有数学理论完备，充分利用采集信息和准确率高的优点，但它也存在以下缺点：1)参数众多，要达到可接受的准确率，需要强大的算力支持，为了达到要求的精度往往需要超算级别百万核的运算，普通企业难以支撑运算消耗的巨大花费。2)存在收敛到局部最优，而非全局最优的情况。从而容易导致最终成像与地下介质的真实分布存在差异。主要的影响因素为初始值的选取。
3.深度学习的发展提供了完全不同与传统全波形反演研究方法的全新思路，从而巧妙地避免了上述问题。深度学习不依赖于传统的公式推导，类似于黑匣子。机器通过对海量数据的学习掌握输入与输出之间的非线性函数关系，训练好的网络具备了良好地推理与预测能力。通过大量的训练数据和模型的改进，能够使得模型输出，即地下介质成像达到很高的正确率。并且深度学习的网络模型训练所需算力和传统全波形反演超算级别的计算资源相比，消耗小了许多。深度学习在成像方面的突破是全卷积网络(fcn，fully convolutional networks forsemantic segmentation)的提出。fcn网络是首个实现点到点，端到端的网络。所谓点到点是指对每个像素都进行分类任务，确定其所属类别；所谓端到端是输入一张图像，输出的也是一张图像。fcn的这一特性，使得深度学习运用于地下岩层分布成像成为了可能。地下岩层分布成像的任务就是将实际生产中通过检波器收集到的二维地震数据，通过一系列的运算得到对应的速度模型(地下岩层分布成像)。其中，二维地震数据是网络的输入，一系列运算是具体的网络架构以及训练过程，而对应的速度模型就是输出，也就是地下介质成像。但 fcn在图像分割这一领域存在着结果不够精细的缺点。
4.综上所述，现有的全波形反演方法以及基于fcn网络的深度学习方法在面对复杂地质情况及地表起伏的情况时均存在着一定的问题，因此，如何充分利用深度学习的优势提高地下介质成像的精度是亟需进一步研究的。


技术实现要素：

5.本发明的目的是提供一种基于全波形反演和卷积神经网络的地下岩层分布成像获取方法、系统、终端及可读存储介质，所述方法通过应用u
‑
net网络以及深度学习生成地下岩层分布成像的预测模型，其中，u
‑
net网络在成像方面比fcn显得更加精细，可以克服fcn在图像分割存在的精细问题，此外，所述方法引入can网络，将传统的全波形反演算法作为判别器，提升前述预测模型的成像精度。
6.一方面，本发明提供的一种基于全波形反演和卷积神经网络的地下岩层分布成像获取方法，包括以下步骤：
7.步骤1：获取数据集，所述数据集包括二维地质数据及其速度模型，所述速度模型为地下岩层分布成像，并设为标签；
8.步骤2：搭建u
‑
net网络架构作为生成器以及基于全波形反演算法构建判别器，并利用所述生成器训练生成地下岩层分布成像的预测模型，以及利用所述判别器优化所述生成器的网络权重参数得到精细的地下岩层分布成像的预测模型；
9.其中，将所述数据集的数据输入生成器进行深度学习训练得到地下岩层分布成像的预测模型，所述地下岩层分布成像的预测模型的输入数据为二维地质数据，输出结果为速度模型；
10.基于所述地下岩层分布成像的预测模型的预测结果，并利用所述判别器的全波形反演算法计算出二维地质数据，再与真实的二维地质数据进行比较调整地下岩层分布成像的预测模型的网络权重参数；
11.步骤3：将待预测的二维地质数据输入精细的地下岩层分布成像的预测模型得到速度模型，得到的速度模型为地下岩层分布成像。
12.声波在不同的岩层中传播的速度不一样，在相同岩层中的传播速度是一样的。在代码中就展现的是一块地质是一个速度，另一块地质是另一个速度。所以在实际中，本发明使用速度模型来侧面反映地下岩层的分布。
13.本发明所述方法所使用的u
‑
net网络是对fcn网络的进一步重大改进，它与fcn网络结构大体相似，但采用了与fcn不同的特征融合方式：fcn采用的是逐点相加，对应 tensorflow的tf.add()函数。u
‑
net采用的是channel维度拼接融合，对应tensorflow的tf.concat() 函数。u
‑
net网络在成像方面比fcn显得更加精细，因此，本发明利用u
‑
net网络的优势，提高了地下岩层分布成像精度。其次，本发明还引入了can网络，即生成器的基础上，后续增加了以全波形反演算法为基础的判别器，利用判别器来提升生成器的成像精度。
14.可选地，步骤2中利用所述判别器优化所述生成器的网络权重参数得到精细的地下岩层分布成像的预测模型的过程如下：
15.步骤2
‑
1：将预测模型的网络权重参数映射到预测模型输出的速度模型得到基于网络权重参数的速度域模型；
16.v
k
＝g(w
k
)
17.式中，k表示第k次循环迭代，w
k
代表第k次循环迭代过程中得到的神经网络权重模型参数，v
k
代表生成速度模型，g为映射函数；
18.步骤2
‑
2：将基于网络权重参数的速度域模型并利用全波形反演算法计算出二维地质数据；
19.步骤2
‑
3：以步骤2
‑
2计算出的二维地质数与实际的二维地质数据构建目标函数，并反向计算网络权重参数的损失函数梯度，进而更新网络权重参数，再更新速度域模型，更新后返回步骤2
‑
2进入下一次迭代，直至模型精度满足预设要求或者得到迭代终止条件，得到最终更新的网络权重参数；
20.其中，基于最终更新的网络权重参数的预测模型视为精细的地下岩层分布成像的预测模型。
21.可选地，步骤2
‑
3中采用adam算法的梯度下降法更新网络权重参数，更新公式如下：
[0022][0023]
其中，k代表着迭代次数，p
k
和q
k
是两个波场时刻对应第k次迭代的估计值，∈为常数值， w
k
、w
k 1
为第k次迭代、第k 1次迭代对应的网络权重参数，α
k
为第k次迭代对应的学习率；
[0024][0025][0026]
式中，ε和η是两个超参数，为网络权重参数对应的损失函数梯度。
[0027]
可选地，第k次迭代对应的学习率α
k
的公式如下：
[0028][0029]
其中，μ是步长。
[0030]
可选地，所述生成器的u
‑
net网络中每一层的单节点使用循环残差块结构的节点替换；
[0031]
所述循环残差结构的节点的数据处理过程为：先进行至少两次或两次以上的卷积操作；再将卷积操作后的特征图进入残差模块进行残差计算。
[0032]
可选地，所述生成器的u
‑
net网络的上采样过程中每一层的跳跃连接之前设有控制门。
[0033]
可选地，步骤1中所述数据集由荷兰f3地质块数据集和segsalt数据集混合组成；
[0034]
其中，利用正演算法演算出所述荷兰f3地质块数据集中二维地质数据的速度模型，将其速度模型作为标签。
[0035]
第二方面，本发明提供一种基于上述地下岩层分布成像获取方法的系统，其包括：
[0036]
数据集获取模块，用于获取数据集，所述数据集包括二维地质数据及其速度模型，所述速度模型为地下岩层分布成像，并设为标签；
[0037]
预测模型构建模块，用于利用生成器训练生成地下岩层分布成像的预测模型，以及利用判别器优化所述生成器的网络权重参数得到精细的地下岩层分布成像的预测模型；
[0038]
其中，所述生成器为u
‑
net网络架构，所述判别器基于全波形反演算法实现；
[0039]
其中，预测模型构建模块包括：预测模型生成单元和网络权重参数优化单元，所述预测模型生成单元用于利用生成器训练生成地下岩层分布成像的预测模型，所述网络权重参数优化单元用于利用判别器优化所述生成器的网络权重参数得到精细的地下岩层分布成像的预测模型。
[0040]
预测模块，用于将待预测的二维地质数据输入精细的地下岩层分布成像的预测模型得到速度模型，得到的速度模型为地下岩层分布成像。
[0041]
第三方面，本发明提供一种终端，其包括：
[0042]
一个或多个处理器；
[0043]
存储了一个或多个程序的存储器；
[0044]
所述处理器调用所述存储器存储的程序以实现：
[0045]
一种基于全波形反演和卷积神经网络的地下岩层分布成像获取方法的步骤。
[0046]
第四方面，本发明提供一种可读存储介质，其存储了计算机程序，所述计算机程序被处理器调用以实现：
[0047]
一种基于全波形反演和卷积神经网络的地下岩层分布成像获取方法的步骤。
[0048]
有益效果
[0049]
1.本发明提供的一种基于全波形反演和卷积神经网络的地下岩层分布成像获取方法，引入了u
‑
net网络，充分利用u
‑
net网络的优势，提高地下岩层分布成像的预测模型的成像精度。与此同时，所述方法还引入了基于全波形反演算法构建的判别器，从而利用所述判别器优化所述生成器的网络权重参数得到精细的地下岩层分布成像的预测模型，进一步提升地下岩层分布成像精度。
[0050]
2.本发明进一步的优选方案中，针对u
‑
net网络进行优化，一方面用循环残差块结构的节点取代u
‑
net网络里面的单节点，其中，通过循环次数可以增加整个网络的复杂度，模型的延展性得到了很大的提升，在残差结构和循环结构共同作用下，原本u
‑
net遇到的随着网络层数的增加，损失率不下降的问题也能够得到一定的解决。二方面，在跳跃链接之前加入控制门，从而可以通过计算系数的控制来将特征更好地区分开，其中计算系数的取值可以是根据实验或经验设定。
附图说明
[0051]
图1是本发明实施例提供的attention
‑
r2u
‑
net进行图像预测的流程示意图；
[0052]
图2是本发明改进后得网络节点图；
[0053]
图3是attention
‑
r2u
‑
net网络结构图；
[0054]
图4是控制门流程示意图；
[0055]
图5是r2unet网络中加入控制门后的attention
‑
r2u
‑
net网络流程图；
[0056]
图6是gan网络生成器与判别器流程图；
具体实施方式
[0057]
下面将结合实施例对本发明做进一步的说明。
[0058]
实施例1：
[0059]
本实施例提供一种基于全波形反演和卷积神经网络的地下岩层分布成像获取方法，包括以下步骤：
[0060]
步骤1：获取数据集，所述数据集包括二维地质数据及其速度模型，所述速度模型为地下岩层分布成像，并设为标签。
[0061]
其中，本实施例中选择由荷兰f3地质块数据集和segsalt数据集混合组成数据集。其中，荷兰f3地质块数据集中原本只有真实地质数据，没有对应该地质数据的速度模型。相当于只有模型的原始输入数据，没有对应的标签数据。而segsalt数据集是由地球物理学者协会提出的，既包含地质数据也包含对应的速度模型。但该数据集数量稀少，只依赖该数据集并没有办法完成网络的训练。
[0062]
为此，本实施例中先通过正演算法，将荷兰f3数据集中原始的二维地质数据进行
演算得到该地质数据具体对应的地下介质分布状况(速度模型)。所谓正演过程为：模拟地震波在地底下传播过程。在假设一个地质模型之后，使用正演方程来模拟真实环境中使用放炮的形式来得到地震波的传播过程，然后从具体的传播过程中推得相应地下介质分布状况。本实施例中通过正演方程(核心公式：有限差分法声波方程)可以推理出荷兰f3数据集中的地震数据所具体对应的地下介质分布状况，使得荷兰f3数据集获得了相应的标签。优选采用devito 正演，devito正演在底层封装了openmp实现了内存共享，通过mpi来支持分布式内存并行。其他的可行实施例中，也可以选择其他正演算法，本发明对此不进行具体的限定。
[0063]
从上述可知，针对荷兰f3数据集，通过正演算法得到标签；本发明为了进一步使得实验数据(荷兰f3数据集)能够接近真实数据，将其与真实数据(segsalt盐层速度模型数据集) 进行混合，用于网络的训练测试，最终能够得到高精度的成像效果。
[0064]
步骤2：搭建u
‑
net网络架构作为生成器以及基于全波形反演算法构建判别器，并利用所述生成器训练生成地下岩层分布成像的预测模型，以及利用所述判别器优化所述生成器的网络权重参数得到精细的地下岩层分布成像的预测模型。
[0065]
针对生成器：
[0066]
将所述数据集的数据输入生成器进行深度学习训练得到地下岩层分布成像的预测模型，所述地下岩层分布成像的预测模型的输入数据为二维地质勘探数据(如地震数据)，输出结果为速度模型。
[0067]
如图3所示，u
‑
net是一种基于fcn网络改进的神经网络算法，其主要特点为u型对称结构，左侧是卷积层(编码器)，右侧是上采样层(解码器)。如图3所示的为5层网络架构，其首先进行两次卷积和池化下采样；然后反卷积进行上采样，上采样之前要与之前的低层特征图进行融合，即跳跃链接。再次上采样。重复这个过程，直到获得输出388x388的特征图，最后经过1*1的卷积，输出结果。
[0068]
本实施例中优选对u
‑
net网络架构进行两方面的优化，如下：
[0069]
1.对循环网络和残差网络特点进行融合，引入循环残差块结构，用循环残差块结构来替换u
‑
net网络中的结构块。循环残差结构式先进行多次卷积操作，一次加大网络的复杂度，再进行残差计算。本实施例中，循环块使用了一个3x3大小的卷积核，并搭配relu激活函数进行特征图的提取，以此提高网络的非线性拟合能力。当输入数据进入到该循环残差块之后，经过3*3卷积核的卷积计算之后，其尺寸大小缩小了2；之后与传统单一结构块不同的地方在于，其输出数据不直接输出，而是再次进入循环残差块中，又进行了一次卷积操作，其尺寸大小又缩小了2。其循环次数可人为控制，以此来调整循环深度。经过两个卷积核之后的特征图进入到残差模块中，该模块将经过卷积核卷积之后的输出结果与该层的输入特征图进行加运算输出特征图，同时进行归一化和relu激活函数。整个循环残差结构的主要特点是可以通过调节循环次数来增加整个网络的复杂度。应用的过程是对所有的原网络的卷积块结构进行了替换，而对其他优秀设计不做变动。这样能够使得u
‑
net网络的特点尤其是跳跃连接的能力被保留下来，同时又增加了循环残差结构的特性。并且随着循环残差块的引入，模型的延展性得到了很大的提升，譬如，要想增加网络的深度，只需要修改循环模块中的循环次数。在残差结构和循环结构共同作用下，原本u
‑
net遇到的随着网络层数的增加，损失率不下降的问题也能够得到一定的解决。
[0070]
需要说明的式，残差计算以及卷积计算均是本领域的常规技术手段，因此，针对其具体的计算过程不进行详细的陈述，本实施例引入循环残差结构是对网络结构的一次优化。
[0071]
2.在传统的unet网络的基础上进行了改进：即跳跃链接之前加入控制门，通过计算系数的控制来将特征更好的区分开。其中，计算系数可以是通过实验测试出来的，也是可以根据经验设置的经验值。传统的unet网络在上采样的过程中，为了避免随着深度的增加导致的浅层特征的消失，会将反卷积得到的输出与对应层级下采样过程中输出的特征图直接进行融合然后再进入到下一步的反卷积。这样可以很好地解决浅层特征与深层特征的矛盾问题，但是并没有对浅层特征进行区分。为此，将控制门(attention gate，简称ag)引入该网络，通过计算系数可以将特征更好的区分开。
[0072]
控制门能够将特征点中的主要信息进行捕获并放大，将网络注意力聚焦在岩层的形状和位置当中，简单概括就是想办法使得目标区域的值变大。控制门模块是用在u
‑
net网络的跳跃连接上。
[0073]
由于在传播过程中，u
‑
net网络可能出现重复提取地震资料中的不明显特征，这将导致网络效率的降低，计算资源的过度使用和模型参数的冗余。若在反卷积的过程当中，加入控制门之后，可以改变输入数据的局部重要程度，将网络注意力聚焦在岩层的形状和位置这类主要特征信息上。加入控制门并不需要额外训练模型和添加大量的参数，也就是说，在 r2u
‑
net网络中加入控制门不会造成增加成本负担。由于控制门也是现有技术，本发明的改进在于将其引入在u
‑
net网络中，设置在跳跃链接之前，因此对控制门的具体技术细节不进行阐述。
[0074]
综上，本实施例中改进后的u
‑
net网络架构可以理解为attention
‑
r2u
‑
net网络架构。其他可行的实施例中，在不脱离本发明构思的基础上，对u
‑
net网络架构不进行优化或者除上述之外进一步优化等都属于本发明的保护范围。
[0075]
搭建基于u
‑
net网络架构的生成器后，将数据集中的二维地质数据输入生成器，并基于对应标签进行训练得到地下岩层分布成像的预测模型。为了进一步提升成像精度，还需要对预测模型的模型参数进行优化，即对网络权重参数进行优化。
[0076]
为此，本发明利用全波形反演算法构建判别器，利用判别器计算出二维地质数据，再与真实的二维地质数据进行比较调整地下岩层分布成像的预测模型的网络权重参数。具体实现过程如下：
[0077]
步骤2
‑
1：将预测模型的网络权重参数映射到预测模型输出的速度模型得到基于网络权重参数的速度域模型。本步骤是将attention
‑
r2u
‑
net模型训练得到的网络权重参数转化成为可以被全波形反演算法处理的输入，即基于网络权重参数的速度域模型：
[0078]
v
k
＝g(w
k
)
[0079]
式中，k表示第k次循环迭代，w
k
代表第k次循环迭代过程中得到的神经网络权重模型参数，v
k
代表生成速度模型，g为映射函数。
[0080]
应当理解，基于前一预测模型的预测结果以及网络权重参数进行映射，从而使得网络权重参数可以应用到后续全波形反演算法中，并进行更新，同理，每更新一次网络权重参数，基于网络权重参数的速度域模型可知，其速度模型也是同步更新的，通过不断迭代更新可以得到最终的网络权重参数，即对应得到精细的地下岩层分布成像的预测模型。
[0081]
步骤2
‑
2：将基于网络权重参数的速度域模型并利用全波形反演算法计算出二维地质数据。其中，由全波形反演算法推导出的二维地质数据是个理论上的预测值，如下：
[0082]
d
c
＝f(m
k
)＝f(g(w
k
))
[0083]
其中，d
c
代表计算出的二维地质数据，f代表波动方程式运算表示符，m
k
是由第一步生成的速度模型。
[0084]
全波形反演算法的迭代过程是假设一个速度模型，利用正演得到的到声波传播规律，去模拟声波在假设的速度模型上的传播得到检波器数据(地震数据)。
[0085]
步骤2
‑
3：以步骤2
‑
2计算出的二维地质数与实际的二维地质数据构建目标函数，并反向计算网络权重参数的损失函数梯度，进而更新网络权重参数，再更新速度域模型，更新后返回步骤2
‑
2进入下一次迭代，直至模型精度满足预设要求或者得到迭代终止条件，得到最终更新的网络权重参数。
[0086]
其中，上述计算得到的是理论上的预测值，将其与真实的二维地质数据由一定的误差，以预测值与真实值的最小二乘为目标函数反演计算与权重w
k
有关的损失函数的梯度，其中损失函数e
fwi
(w
k
)定义如下，其中t表示矩阵转置,*表示共轭。根据链式法则，梯度计算公式：
[0087][0088]
根据链式法则，梯度计算公式为：
[0089][0090]
从而在一些可行的方式中设定网络权重参数的更新公式为：
[0091][0092]
其中，学习率α为学习率，w
k 1
为第k 1次迭代的网络权重参数，即更新后的网络权重参数，w
k
为第k次迭代的网络权重参数，即更新前的网络权重参数。
[0093]
本实施例中优选用adam算法来代替传统全波形反演算法的梯度下降法来加速收敛。adam是一种基于一阶梯度来优化随机目标函数的算法，其优势在于能够动态改变学习率α，并且能够谨慎的选择每一次的步长大小，同时还提供更高的计算效率，在较大的数据集上同样能够适应，在处理稀疏矩阵也具有一定优势，应用到全波形反演算法当中能够取得更好的效果。其网络权重更新公式如下：
[0094][0095]
式中，k代表着迭代次数，p
k
和q
k
是两个波场时刻对应第k次迭代的估计值，∈为常数值(设置成一个非常小的数值，其主要作用防止分母为零的情况出现)，w
k
、w
k 1
为第k次迭代、第k 1次迭代对应的网络权重参数，α
k
为第k次迭代对应的学习率；
[0096][0097][0098]
式中，ε和η是两个超参数，其值的范围都在[0，1)之间。为网络权重参数对应的损失函数梯度。
[0099]
第k次迭代对应的学习率α
k
的公式如下：
[0100][0101]
其中，μ是步长。
[0102]
从上述的公式以及算法介绍可知，应用判别器不断迭代更新网络权重参数，直至预测模型达到预设要求，其中，迭代终止条件可以是以模型精度为标准设定，也可以是以最大迭代次数为标准设定，或者以其他标准设定。如本实施例中针对判别器：attentin
‑
r2u
‑
net网络生成的速度模型的误差降到0.001后，停止训练。如本实施例中针对生成器，将原始的地震数据以及对应的速度模型即断层标签数据体作为训练集，以mse(mean squared error，均方误差)作为损失函数，在40个epoch之后，损失梯度不再下降，损失曲线变得平缓，得到训练后的attention
‑
r2u
‑
net网络。
[0103]
还需要说明的是，在一些可行的方式中，生成器与判别器的配合方式为：生成器进行训练得到预测模型，判别器每迭代更新一次或若干次网络权重参数后，基于更新的网络权重参数返回生成器的训练过程进行训练，重复上述过程，直至满足要求；另一些可行的方式，生成器进行训练得到预测模型，判别器迭代更新网络权重参数直至满足要求。
[0104]
步骤3：将待预测的二维地质数据输入精细的地下岩层分布成像的预测模型得到速度模型，得到的速度模型为地下岩层分布成像。
[0105]
具体应用实例：
[0106]
1)利用f3荷兰地质数据，使用devito框架构建572*572尺寸的速度模型，二维地震数据对应的地下岩层结构作为标签数据体，以此作为训练样本，并混合segsalt数据集。
[0107]
2)进入网络输入层，该层输入数据为572*572尺寸的图片；然后进入到第一层网络中，采用3*3的卷积核，设置循环次数为2，在进行第一次循环卷积后，尺寸大小从572*572变成了568*568，再把原尺寸图片经过裁剪变成相同尺寸大小，与输出的特征图片进行残差操作，得到输出图片，尺寸大小为568*568。
[0108]
3)然后进入第二层网络，首先经过池化层，采用2*2的最大值池化，进过池化后，尺寸缩小一半，变为284*284。最大值池之后进行第二次循环卷积，同样是3*3的卷积核，循环次数为2，尺寸从284*284变成了280*280，再把这一层的输入图片裁剪成相同的尺寸大小与输出图片进行残差操作，得到输出图片为280*280。
[0109]
4)再然后进入第三层网络。经过第二层的池化层，将特征图大小缩减为原先的一半，从 280*280下降到140*140；后面的网络结构同上面一样，次数为2的循环卷积后进行残差操作，输出的特征图大小为136*136。
[0110]
5)然后进入了第四层网络。该层网络结构同前四层网络，先是一层池化，将尺寸将为 68*68，然后将特征图进行两次卷积，其尺寸分别缩减到66*66和64*64。
[0111]
6)然后进入第五层网络，池化操作将特征图缩减到32*32，对特征图进行了两次卷积，进一步提取全局特征，其尺寸下降到28*28。至此，整体下采样过程结束，特征提取完毕， u
‑
net网络左侧编码器结束，整体流程是从上往下，由第一层到第五层。从下一步开始进入到右侧解码器过程。也就是上采样的过程。该过程从下往上依次进行。从第五层逐层操作，直到第一层。
[0112]
7)第五层网络结束之后，图片尺寸为28*28。然后进入上采样过程中的第四层，首
先进行一次反卷积操作，反卷积核为2。其特征图尺寸从28*28扩大为56*56；接着将下采样过程中的第四层输出的特征图经过控制门，然后与反卷积得到的56*56的特征图进行跳跃链接。拼接方法可以简单选择矩阵相加操作。然后进过循环次数为2的卷积操作，同样卷积核大小为2，得到上采样过程中第四层的网络输出为52*52的特征图。
[0113]
8)然后进入第三层网络，先进行一步反卷积操作，将其特征图还原到104*104，下采样过程中的第三层输出的特征图经过控制门与该图进行特征融合，然后经过两次卷积核大小为 2的卷积操作，输出大小为100*100的特征图片。
[0114]
9)进入第二层网络后，与前面两层的网络结构一样。进行反卷积操作，将特征图尺寸恢复到200*200，同样将解码器中的特征图与编码器中的第二层特征图进行拼接，之后在进行两次卷积操作，将其尺寸分别缩减到198*198和196*196。
[0115]
10)然后到达第一层网络。特征图进行反卷积操作，将特征图还原到392*392，将解码器中的特征图与编码器中的特征图进行拼接，之后在进行两次卷积操作，尺寸变为388*388。
[0116]
11)最后通过一个1*1大小的卷积核，分别对特征图中的每一个像素进行分类。至此，整个attention
‑
r2u
‑
net网络完成了一次迭代。
[0117]
12)40个epoch的训练后attention
‑
r2u
‑
net网络训练完成。
[0118]
13)通过建立神经网络参数权重与速度模型之间1：1的关系，同时将全波形反演算法作为判别器来修正本发明的速度模型。通过修正后的速度模型，可以更新神经网络的权重。
[0119]
14)循环进行13步，直到生成器，即attentin
‑
r2u
‑
net网络生成的速度模型的误差降到0.001后，停止训练。本发明通过gan网络的进一步训练，attention
‑
r2u
‑
net网络在预测地下岩层结构上拥有了更加优异的表现，更高的准确率以及更强的泛化能力。网络训练结束，最终得到一个有着能从地震资料中，直接反演出高精确度速度模型的网络。
[0120]
应当理解，本应用实例中网络参数的设定为举例说明，本发明并不局限于此。
[0121]
实施例2：
[0122]
本实施例提供一种基于上述方法的系统，其包括：数据集获取模块和预测模型构建模块以及预测模块。
[0123]
数据集获取模块，用于获取数据集，所述数据集包括二维地质数据及其速度模型，所述速度模型为地下岩层分布成像，并设为标签；
[0124]
预测模型构建模块，用于利用生成器训练生成地下岩层分布成像的预测模型，以及利用判别器优化所述生成器的网络权重参数得到精细的地下岩层分布成像的预测模型；
[0125]
其中，所述生成器为u
‑
net网络架构，所述判别器基于全波形反演算法实现；
[0126]
其中，预测模型构建模块包括：预测模型生成单元和网络权重参数优化单元，所述预测模型生成单元用于利用生成器训练生成地下岩层分布成像的预测模型，所述网络权重参数优化单元用于利用判别器优化所述生成器的网络权重参数得到精细的地下岩层分布成像的预测模型。
[0127]
预测模块，用于将待预测的二维地质数据输入精细的地下岩层分布成像的预测模型得到速度模型，得到的速度模型为地下岩层分布成像。
[0128]
其中，各个单元模块的具体实现过程请参照前述方法的对应过程。应当理解，上述
单元模块的具体实现过程参照方法内容，本发明在此不进行具体的赘述，且上述功能模块单元的划分仅仅是一种逻辑功能的划分，实际实现时可以有另外的划分方式，例如多个单元或组件可以结合或者可以集成到另一个系统，或一些特征可以忽略，或不执行。同时，上述集成的单元既可以采用硬件的形式实现，也可以采用软件功能单元的形式实现。
[0129]
实施例3：
[0130]
本实施例提供一种终端，其包括一个或多个处理器和存储了一个或多个程序的存储器，所述处理器调用所述存储器存储的程序以实现：一种基于全波形反演和卷积神经网络的地下岩层分布成像获取方法的步骤，具体如下：
[0131]
步骤1：获取数据集，所述数据集包括二维地质数据及其速度模型，所述速度模型为地下岩层分布成像，并设为标签。
[0132]
步骤2：搭建u
‑
net网络架构作为生成器以及基于全波形反演算法构建判别器，并利用所述生成器训练生成地下岩层分布成像的预测模型，以及利用所述判别器优化所述生成器的网络权重参数得到精细的地下岩层分布成像的预测模型。
[0133]
步骤3：将待预测的二维地质数据输入精细的地下岩层分布成像的预测模型得到速度模型，得到的速度模型为地下岩层分布成像。
[0134]
执行网络权重参数优化时，具体执行：
[0135]
步骤2
‑
1：将预测模型的网络权重参数映射到预测模型输出的速度模型得到基于网络权重参数的速度域模型。
[0136]
步骤2
‑
2：将基于网络权重参数的速度域模型并利用全波形反演算法计算出二维地质数据。
[0137]
步骤2
‑
3：以步骤2
‑
2计算出的二维地质数与实际的二维地质数据构建目标函数，并反向计算网络权重参数的损失函数梯度，进而更新网络权重参数，再更新速度域模型，更新后返回步骤2
‑
2进入下一次迭代，直至模型精度满足预设要求或者得到迭代终止条件，得到最终更新的网络权重参数。
[0138]
该终端还包括：通信接口，用于与外界设备进行通信，进行数据交互传输。
[0139]
其中，存储器可能包含高速ram存储器，也可能还包括非易失性除颤器，例如至少一个磁盘存储器。
[0140]
如果存储器、处理器和通信接口独立实现，则存储器、处理器和通信接口可以通过总线相互连接并完成相互间的通信。所述总线可以是工业标准体系结构总线，外部设备互联总线或扩展工业标准体系结构总线等。所述总线可以分为地址总线、数据总线、控制总线等。
[0141]
可选的，在具体实现上，如果存储器、处理器和通信接口集成在一块芯片上，则存储器、处理器即通信接口可以通过内部接口完成相互之间的通信。
[0142]
各个步骤的具体实现过程请参照前述方法的阐述。
[0143]
应当理解，在本发明实施例中，所称处理器可以是中央处理单元(central processing unit， cpu)，该处理器还可以是其他通用处理器、数字信号处理器(digital signal processor，dsp)、专用集成电路(application specific integrated circuit，asic)、现成可编程门阵列 (field
‑
programmable gatearray，fpga)或者其他可编程逻辑器件、分立门或者晶体管逻辑器件、分立硬件组件等。通用处理器可以是微处理器或者该处
理器也可以是任何常规的处理器等。存储器可以包括只读存储器和随机存取存储器，并向处理器提供指令和数据。存储器的一部分还可以包括非易失性随机存取存储器。例如，存储器还可以存储设备类型的信息。
[0144]
实施例4：
[0145]
本发明提供一种可读存储介质，其存储了计算机程序，所述计算机程序被处理器调用以实现：
[0146]
一种基于全波形反演和卷积神经网络的地下岩层分布成像获取方法的步骤，具体如下：
[0147]
步骤1：获取数据集，所述数据集包括二维地质数据及其速度模型，所述速度模型为地下岩层分布成像，并设为标签。
[0148]
步骤2：搭建u
‑
net网络架构作为生成器以及基于全波形反演算法构建判别器，并利用所述生成器训练生成地下岩层分布成像的预测模型，以及利用所述判别器优化所述生成器的网络权重参数得到精细的地下岩层分布成像的预测模型。
[0149]
步骤3：将待预测的二维地质数据输入精细的地下岩层分布成像的预测模型得到速度模型，得到的速度模型为地下岩层分布成像。
[0150]
执行网络权重参数优化时，具体执行：
[0151]
步骤2
‑
1：将预测模型的网络权重参数映射到预测模型输出的速度模型得到基于网络权重参数的速度域模型。
[0152]
步骤2
‑
2：将基于网络权重参数的速度域模型并利用全波形反演算法计算出二维地质数据。
[0153]
步骤2
‑
3：以步骤2
‑
2计算出的二维地质数与实际的二维地质数据构建目标函数，并反向计算网络权重参数的损失函数梯度，进而更新网络权重参数，再更新速度域模型，更新后返回步骤2
‑
2进入下一次迭代，直至模型精度满足预设要求或者得到迭代终止条件，得到最终更新的网络权重参数。
[0154]
各个步骤的具体实现过程请参照前述方法的阐述。
[0155]
所述可读存储介质为计算机可读存储介质，其可以是前述任一实施例所述的控制器的内部存储单元，例如控制器的硬盘或内存。所述可读存储介质也可以是所述控制器的外部存储设备，例如所述控制器上配备的插接式硬盘，智能存储卡(smart media card,smc)，安全数字 (secure digital,sd)卡，闪存卡(flash card)等。进一步地，所述可读存储介质还可以既包括所述控制器的内部存储单元也包括外部存储设备。所述可读存储介质用于存储所述计算机程序以及所述控制器所需的其他程序和数据。所述可读存储介质还可以用于暂时地存储已经输出或者将要输出的数据。
[0156]
基于这样的理解，本发明的技术方案本质上或者说对现有技术做出贡献的部分，或者该技术方案的全部或部分可以以软件产品的形式体现出来，该计算机软件产品存储在一个存储介质中，包括若干指令用以使得一台计算机设备(可以是个人计算机，服务器，或者网络设备等)执行本发明各个实施例所述方法的全部或部分步骤。而前述的可读存储介质包括：u盘、移动硬盘、只读存储器(rom，read
‑
onlymemory)、随机存取存储器(ram，randomaccessmemory)、磁碟或者光盘等各种可以存储程序代码的介质。
[0157]
需要强调的是，本发明所述的实例是说明性的，而不是限定性的，因此本发明不限
于具体实施方式中所述的实例，凡是由本领域技术人员根据本发明的技术方案得出的其他实施方式，不脱离本发明宗旨和范围的，不论是修改还是替换，同样属于本发明的保护范围。