技术领域
[0001] 本
发明涉及
深度学习技术领域,具体涉及地震
数据处理中的一种曲率驱动扩散全卷积网络地震数据坏道重建与去噪方法。
背景技术
[0002] 在地震数据的采集过程中,一般会出现
地震道缺失损坏和噪声污染,极大的影响了地震资料
质量,地震数据坏道重建主要有以下两种方法:基于变换域的Radon变换,Fourier变换,曲波变换等,基于预测滤波的f-x域坏道重建方法等,但传统坏道重建方法均基于一定的假设前提,计算较为复杂,这些方法在特定的情况下有才能有好的坏道重建效果,泛化性不强;传统去噪需根据
信号和噪声的特征差异,常用的方法有:多项式拟合法、K-L变换法、Curvelet变换、
小波变换法等,多项式拟合法只有在去除相干噪声的后才有好的去噪效果;K-L变换法只有当样本充足时,矩阵的估计才会变得精确;小波变换法去噪时忽略每个
像素的特点,去噪的同时会产生模糊现象;Curvelet变换克服了傅里叶变换和小波变换的缺点,可以保持地震数据边缘和纹理细节信息,但会过度消除部分Curvelet变换系数,去噪同时会产生伪影。
[0003] 为解决传统浅层方法的不足,基于深度学习的去噪模型被提出,其中自编码网络、
卷积神经网络、生成对抗网络广泛应用于去噪领域,但基于深度学习的地震去噪方法较少,基于残差卷积神经网络的随机噪声去除
算法,具有更强的去噪性能;基于卷积神经网络去噪模型,可以去除方差未知的地震随机噪声;将残差卷积神经网络图像去噪方法应用于地震资料去噪,可有效去除随机噪声,但以上深度学习方法仅能去除随机噪声坏道重建效果不佳,效率不高。
发明内容
[0004] 为了克服传统坏道重建和去噪方法的局限性,提出一种基于曲率驱动扩散和深度学习的坏道重建和去噪方法。
[0005] 本发明公开了一种曲率驱动扩散全卷积网络震数据坏道重建与去噪方法,包括以下步骤:(1)制作训练集和测试集;(2)提出32层的曲率驱动扩散全卷积网络模型,此模型由
输入层、第一部分、第二部分、
输出层组成,其中,第一部分由曲率驱动扩散层和多尺度卷积层组成,第二部分主要由三个八卷积结构、一个高低级组合结构和多尺度跳线组成;(3)采用冻结网络的方法训练网络,首先,读入总数据集的1/a部分,冻结网络的第二部分,只进行第一部分的坏道重建和多尺度特征提取,a为正整数,然后,冻结第一部分,将第一部分输出的4通道数据,分批次输入第二部分,训练第二部分;接着,当第二部分达到最优结果时,再次循环(1)和(2),训练剩余的数据集,最后,测试集达到最优结果时,保存网络模型;(5)利用保存的最终网络模型,处理含有坏道和噪声的地震数据,最终输出坏道重建和去噪后的地震资料。本发明公开的地震数据坏道重建与去噪方法,不但可高效的重建坏道,而且经压制噪声后的数据,在不产生假象的同时,还可使地震数据的局部细节得以完整保留。
[0006] 本发明所提供的技术方案为:一种曲率驱动扩散全卷积网络地震数据坏道重建与去噪方法,包括如下步骤:
[0007] 1.一种曲率驱动扩散全卷积网络地震数据重建与去噪方法,其特征在于包括如下步骤:
[0008] 步骤1、制作训练集和测试集,具体步骤如下:
[0009] (1)将原始SEG-Y格式的地震数据处理为CSV格式的幅值数据集;
[0010] (2)将原始不含噪声数据集加入随机噪声,并选取n%作为坏道删除,然后和原始不含噪声的地震资料作为一组数据,制作训练集,n为大于零小于100的正整数;
[0011] (3)选取不同于训练集的地震数据,按照制作训练集的方法制作测试集;
[0012] 步骤2、设计32层曲率驱动扩散全卷积网络模型,此网络模型由输入层、第一部分、第二部分、输出层组成:
[0013] 第一部分由曲率驱动扩散层和多尺度卷积层组成;
[0014] 第二部分主要由3个八卷积结构、1个高低级组合结构和多尺度跳线结构组成,具体组成如下,八卷积结构由7层卷积层和1个上
采样层组成,第一层分别由两个卷积核大小为5×5和3×3的多尺度空洞卷积组成,经此卷积层后数据将被压缩为原始的1/2,接下来6层分别由卷积核大小为1×3,3×1,1×5,5×1,1×3,3×1的卷积层组成,最后上采样层采用双线性插值算法,将前6
层压缩提取出的特征图扩大2倍;
[0015] 高低级组合结构,前三层为了压缩维度由3层卷积核大小均为3×3,步长均为2的卷积层组成,后三层均为采用双线性插值算法的上采样层,每个输入经过上采样层都被扩大2倍后传给下一层,且前三层卷积层的输入均通过跳线结构分别与三层上采样层的输出相加,将浅层和深层特征融合输出;
[0016] 多尺度跳线结构一共含三个加型跳线和一个减型跳线,跳线结构的输入均为第一部分曲率驱动扩散层的输出,第一个加型跳线的输入经过卷积核大小依次为1×3和3×1的两层卷积层,与第一个八卷积结构的输出相加,第二个加型跳线结构同样经过两层卷积层,大小依次为1×5和5×1,并与第一个跳线结构两层卷积后的输出连接,经3×3卷积层调整通道数后与第二个八卷积结构的输出相加,作为第三个八卷积结构的输入,第三个加型跳线结构和第二个加型跳线结构一致,区别在于两层卷积层的卷积核大小依次变为1×7和7×1,经3×3卷积层调整通道数后的输出与第三个八卷积结构的输出相加,将结果输入高低级组合结构;减型跳线结构,由第一部分曲率驱动扩散层的输出与第二部分最后一个3×3卷积层的输出相减组成;
[0017] 步骤3、采用冻结网络方法训练模型,冻结网络方法指只训练
指定的某些层,其它层的权重此时将不更新,训练模型分为4个阶段:
[0018] 阶段1)读入总数据集的1/a部分,冻结模型的第二部分,只进行第一部分的坏道重建和多尺度特征提取,a为大于2小于10的正整数;
[0019] 阶段2)冻结第一部分,将第一部分输出的4通道数据,分批次输入到第二部分,训练第二部分;
[0020] 阶段3)当第二部分达到最优结果时,再次循环阶段1)和阶段2),训练剩余的数据集;
[0021] 阶段4)测试集达到最优结果时,保存网络模型;
[0022] 步骤4、利用保存的网络模型,处理含有坏道和噪声的地震数据,最后输出坏道重建和去噪后的地震数据。
[0023] 2.根据
权利要求1所述的一种曲率驱动扩散全卷积网络地震数据重建与去噪方法,其特征在于所述步骤2中的第一部分,具体包括:
[0024] 曲率驱动扩散层输入为预处理后的含有噪声和坏道的数据集;
[0025] 多尺度卷积层由卷积核大小分别为3×3、5×5、255×255三种卷积核组成,层数为1,输入为经曲率驱动扩散层坏道重建后的数据;
[0026] 其中,第一部分输出为4通道数据,分别由一通道曲率驱动扩散层的输出和3通道多尺度卷积层的输出组成。
[0027] 3.根据权利要求1所述的一种曲率驱动扩散全卷积网络地震数据重建与去噪方法,其特征在于所述步骤3中的阶段1),具体包括:
[0028] 将步骤1预处理后的单通道地震数据,先经过曲率驱动扩散层初次进行坏道重建,其中,
迭代次数为b次,掩膜(mask)由含坏道和噪声数据中值为0的数据组成,
单层的多尺度卷积层采用误差反向传播,通过冻结网络的方法只进行一次权重更新,将输出的4个通道均作为特征数据,输入到第二部分。
[0029] 4.根据权利要求1所述的一种曲率驱动扩散全卷积网络地震数据重建与去噪方法,其特征在于所述步骤3中的阶段2),具体包括:
[0030] (1)将第一部分输出的4通道数据集,输入到第二部分网络中,批处理样本数为m,m为大于1小于10的正整数;
[0031] (2)采用误差反向传播,并以对数双曲余弦损失函数来衡量原始不含噪声和坏道数据与预测数据之间的距离,通过数据集的每次迭代,使用梯度下降来调整神经元之间的权重以降低代价函数,进而优化网络;
[0032] 对数双曲余弦损失函数为:
[0033]
[0034] 其中:
[0035]
[0036] N表示有N对训练样本;yi为第i个真实不含坏道和噪声的数据,zi为模型预测输出的第i个数据,对数双曲余弦损失越小代表zi与yi越接近,网络坏道修复和去噪效果越好;
[0037] (3)以定量的峰值
信噪比、结构相似性和定性的视觉感受判断网络去噪效果,设定
峰值信噪比和结构相似性各自
阈值,当其均等于或大于设定的阈值时,则停止迭代,保存最终网络模型的各个参数,否则继续迭代训练网络;
[0038] 峰值信噪比公式为:
[0039]
[0040] 其中:
[0041]
[0042] N表示有N对训练样本;yi为第i个真实不含坏道和噪声的数据,zi为模型预测输出的第i个数据;max(yi)为yi的最大值;MSE为yi和zi的均方误差,PSNR数值越大表示网络坏道修复和去噪效果越好;
[0043] 结构相似性公式为:
[0044] SSIM(yi,zi)=l(yi,zi)*c(yi,zi)*s(yi,zi)
[0045] 其中:
[0046]
[0047]
[0048]
[0049] 分别为yi和zi的均值, 分别为yi和zi的方差, 为yi和zi的协方差,c1=(k1L)2,c2=(k2L)2、 k1=0.01,k2=0.03,L=max(yi),l(yi,zi)为照明度比较部分,c(yi,zi)为
对比度比较部分,s(yi,zi)为结构比较部分,SSIM(yi,zi)∈[0,1],若yi和zi完全相同时SSIM值为1。
附图说明
[0050] 图1为本发明的
流程图,网络训练分为第一部分和第二部分;
[0051] 图2为本发明的曲率驱动扩散全卷积网络地震数据重建与去噪方法总体结构图,主要由输入层、第一部分、第二部分、输出层组成;
[0052] 图3为本发明的八卷积结构和高低级组合结构,八卷积结构共8层,高低级组合共6层;
[0053] 图4为本发明坏道重建和去噪示例,其中,121、141、151和152道为重建的坏道部分。
[0054] 图中符号说明如下:
[0055] CDD:表示曲率驱动扩散算法;
[0056] CDD-FCN:表示曲率驱动扩散全卷积网络;
[0057] concat:表示Tensorflow中的特征融合函数;
[0058] conv:表示常规卷积;
[0059] Dconv:表示空洞卷积;
[0060] bn:表示批归一化层;
[0061] 8-Conv:表示八卷积结构
[0062] L&H Combine:表示高低级组合结构;
[0063] Unsampled:表示上采样;具体实施方式:
[0064] 为了高效重建坏道和压制随机噪声,以曲率驱动扩散层、八卷积结构、高低级组合结构、多尺度跳线结构为
基础,提出了曲率驱动扩散全卷积网络地震数据重建与去噪模型,如图2所示。参照图1,示出的本发明流程图,包括如下步骤。
[0065] 步骤1、制作训练集和测试集,具体步骤如下:
[0066] 1.一种曲率驱动扩散全卷积网络地震数据重建与去噪方法,其特征在于包括如下步骤:
[0067] 步骤1、制作训练集和测试集,具体步骤如下:
[0068] (1)将原始SEG-Y格式的地震数据处理为CSV格式的幅值数据集;
[0069] (2)将原始不含噪声数据集加入随机噪声,并选取10%作为坏道删除,然后和原始不含噪声的地震资料作为一组数据,制作训练集;
[0070] (3)选取不同于训练集的地震数据,按照制作训练集的方法制作测试集;
[0071] 步骤2、设计32层曲率驱动扩散全卷积网络模型,此网络模型由输入层、第一部分、第二部分、输出层组成:
[0072] 第一部分由曲率驱动扩散层和多尺度卷积层组成;
[0073] 第二部分主要由3个八卷积结构、1个高低级组合结构和多尺度跳线结构组成,具体组成如下,八卷积结构由7层卷积层和1个上采样层组成,第一层分别由两个卷积核大小为5×5和3×3的多尺度空洞卷积组成,经此卷积层后数据将被压缩为原始的1/2,接下来6层分别由卷积核大小为1×3,3×1,1×5,5×1,1×3,3×1的卷积层组成,最后上采样层采用双线性插值算法,将前6层压缩提取出的特征图扩大2倍;
[0074] 高低级组合结构,前三层为了压缩维度由3层卷积核大小均为3×3,步长均为2的卷积层组成,后三层均为采用双线性插值算法的上采样层,每个输入经过上采样层都被扩大2倍后传给下一层,且前三层卷积层的输入均通过跳线结构分别与三层上采样层的输出相加,将浅层和深层特征融合输出;
[0075] 多尺度跳线结构一共含三个加型跳线和一个减型跳线,跳线结构的输入均为第一部分曲率驱动扩散层的输出,第一个加型跳线的输入经过卷积核大小依次为1×3和3×1的两层卷积层,与第一个八卷积结构的输出相加,第二个加型跳线结构同样经过两层卷积层,大小依次为1×5和5×1,并与第一个跳线结构两层卷积后的输出连接,经3×3卷积层调整通道数后与第二个八卷积结构的输出相加,作为第三个八卷积结构的输入,第三个加型跳线结构和第二个加型跳线结构一致,区别在于两层卷积层的卷积核大小依次变为1×7和7×1,经3×3卷积层调整通道数后的输出与第三个八卷积结构的输出相加,将结果输入高低级组合结构;减型跳线结构,由第一部分曲率驱动扩散层的输出与第二部分最后一个3×3卷积层的输出相减组成;
[0076] 步骤3、采用冻结网络方法训练模型,冻结网络方法指只训练指定的某些层,其它层的权重此时将不更新,训练模型分为4个阶段:
[0077] 阶段1)读入总数据集的1/5部分,冻结模型的第二部分,只进行第一部分的坏道重建和多尺度特征提取;
[0078] 阶段2)冻结第一部分,将第一部分输出的4通道数据,分批次输入到第二部分,训练第二部分;
[0079] 阶段3)当第二部分达到最优结果时,再次循环阶段1)和阶段2),训练剩余的数据集;
[0080] 阶段4)测试集达到最优结果时,保存网络模型;
[0081] 步骤4、利用保存的网络模型,处理含有坏道和噪声的地震数据,最后输出坏道重建和去噪后的地震数据。
[0082] 2.根据权利要求1所述的一种曲率驱动扩散全卷积网络地震数据重建与去噪方法,其特征在于所述步骤2中的第一部分,具体包括:
[0083] 曲率驱动扩散层输入为预处理后的含有噪声和坏道的数据集;
[0084] 多尺度卷积层由卷积核大小分别为3×3、5×5、255×255三种卷积核组成,层数为1,输入为经曲率驱动扩散层坏道重建后的数据;
[0085] 其中,第一部分输出为4通道数据,分别由一通道曲率驱动扩散层的输出和3通道多尺度卷积层的输出组成。
[0086] 3.根据权利要求1所述的一种曲率驱动扩散全卷积网络地震数据重建与去噪方法,其特征在于所述步骤3中的阶段1),具体包括:
[0087] 将步骤1预处理后的单通道地震数据,先经过曲率驱动扩散层初次进行坏道重建,其中,迭代次数为6次,掩膜由含坏道和噪声数据中值为0的数据组成,单层的多尺度卷积层采用误差反向传播,通过冻结网络的方法只进行一次权重更新,将输出的4个通道均作为特征数据,输入到第二部分。
[0088] 4.根据权利要求1所述的一种曲率驱动扩散全卷积网络地震数据重建与去噪方法,其特征在于所述步骤3中的阶段2),具体包括:
[0089] (1)将第一部分输出的4通道数据集,输入到第二部分网络中,批处理样本数为6;
[0090] (2)采用误差反向传播,并以对数双曲余弦损失函数来衡量原始不含噪声和坏道数据与预测数据之间的距离,通过数据集的每次迭代,使用梯度下降来调整神经元之间的权重以降低代价函数,进而优化网络;
[0091] 对数双曲余弦损失函数为:
[0092]
[0093] 其中:
[0094]
[0095] N表示有N对训练样本;yi为第i个真实不含坏道和噪声的数据,zi为模型预测输出的第i个数据,对数双曲余弦损失越小代表zi与yi越接近,网络坏道修复和去噪效果越好;
[0096] (3)以定量的峰值信噪比、结构相似性和定性的视觉感受判断网络去噪效果,若峰值信噪比的值等于或大于15dB,且结构相似性的值等于或大于0.80,则停止迭代,保存最终网络模型的各个参数,否则继续迭代训练网络;
[0097] 峰值信噪比公式为:
[0098]
[0099] 其中:
[0100]
[0101] N表示有N对训练样本;yi为第i个真实不含坏道和噪声的数据,zi为模型预测输出的第i个数据;max(yi)为yi的最大值;MSE为yi和zi的均方误差,PSNR数值越大表示网络坏道修复和去噪效果越好;
[0102] 结构相似性公式为:
[0103] SSIM(yi,zi)=l(yi,zi)*c(yi,zi)*s(yi,zi)
[0104] 其中:
[0105]
[0106]
[0107]
[0108] 分别为yi和zi的均值, 分别为yi和zi的方差, 为yi和zi的协方差,c1=(k1L)2,c2=(k2L)2、 k1=0.01,k2=0.03,L=max(yi),l(yi,zi)为照明度比较部分,c(yi,zi)为对比度比较部分,s(yi,zi)为结构比较部分,SSIM(yi,zi)∈[0,1],若yi和zi完全相同时SSIM值为1。
[0109] 图4本发明的地震坏道重建与去噪方法的示例展示,其中,121、141、151和152道为重建的坏道部分。