• ISSN 0258-2724
  • CN 51-1277/U
  • EI Compendex
  • Scopus 收录
  • 全国中文核心期刊
  • 中国科技论文统计源期刊
  • 中国科学引文数据库来源期刊

基于轻量级模型的隧道岩性快速识别方法

夏毅敏 李清友 邓朝辉 龙斌 姚捷

杨柳, 白朝元, 范平志. 一种压缩感知测量矩阵的联合优化算法[J]. 西南交通大学学报, 2025, 60(2): 454-461. doi: 10.3969/j.issn.0258-2724.20230032
引用本文: 夏毅敏, 李清友, 邓朝辉, 龙斌, 姚捷. 基于轻量级模型的隧道岩性快速识别方法[J]. 西南交通大学学报, 2021, 56(2): 420-427. doi: 10.3969/j.issn.0258-2724.20191057
YANG Liu, BAI Chaoyuan, FAN Pingzhi. Co-optimization Algorithm for Measurement Matrix of Compressive Sensing[J]. Journal of Southwest Jiaotong University, 2025, 60(2): 454-461. doi: 10.3969/j.issn.0258-2724.20230032
Citation: XIA Yimin, LI Qingyou, DENG Chaohui, LONG Bin, YAO Jie. Rapid Identification Method for Lithology of Tunnel Based on Lightweight Model[J]. Journal of Southwest Jiaotong University, 2021, 56(2): 420-427. doi: 10.3969/j.issn.0258-2724.20191057

基于轻量级模型的隧道岩性快速识别方法

doi: 10.3969/j.issn.0258-2724.20191057
基金项目: 国家重点研发计划资助项目(2017YFB1302600);湖南省科技重大专项(2019GK1010)
详细信息
    通讯作者:

    夏毅敏(1987—),男,教授,研究方向为地下工程与大型掘进装备,E-mail:xiamj@csu.edu.cn

  • 中图分类号: V221.3

Rapid Identification Method for Lithology of Tunnel Based on Lightweight Model

  • 摘要: 为了解决隧道岩性现有识别方法中识别时间长、安全性低、主观性大等问题,结合不同岩性表面具有不同的成分特征,提出了一种基于轻量级模型与岩石图像的隧道岩性快速识别方法. 首先,通过相机采集隧道常见的片麻岩、花岗岩、石灰岩、大理岩、凝灰岩、砂岩等6类主要岩石,建立了岩石图像数据集并划分训练集、验证集与测试集;然后,基于轻量级模型MobileNet V2在ImageNet数据集上进行预训练,改进模型分类器结构以适应岩石数据集,并采用模型迁移学习训练方法对1170张训练集图像进行训练,获取了岩石岩性识别模型;最后,选取共计300张测试集图像在离线条件下进行了模型测试,并与VGG16模型与SVM (support vector machine)模型进行了对比. 实验结果表明:模型在测试数据集上的各项总体评估指标均在85%以上,其中凝灰岩各项评价指标达到94%以上,模型大小仅28.3 MB,平均识别时间为2880 ms,表明该识别模型体积小,识别准确率高,识别时间快,在精确率与识别速度上均优于传统方法.

     

  • 2004年,Donoho[1]提出了关于信号采样、压缩和重构的全新理论——压缩感知理论(compressive sensing,CS). 在大规模机器类型通信(massive machine type communication,mMTC)下,即使是在高峰期,活跃用户的数量也不超过10%[2],这使得压缩感知技术应用在大规模通信场景下成为可能. 压缩感知理论的研究主要集中在3个领域:信号的稀疏表示、高性能测量矩阵的构造以及信号的精准重构[3]. 其中,测量矩阵的构造将直接影响信号重构的误差大小,测量矩阵需要符合一定的条件,如有限等距性质[4](restricted isometry property,RIP),这是保证原始信号不失真重建的必要条件.

    学者们尝试通过相关性理论衡量并优化测量矩阵的性能[5]. 研究表明,减小稀疏基与测量矩阵的相关系数能提高压缩感知算法的重构性能. 目前,针对测量矩阵的研究成果不胜枚举. 曾祥洲[6]提出一种基于交替最小化的优化算法,最小化Gram矩阵与目标矩阵差值的傅里叶范数,建立最优测量矩阵,解决最小化单个指标易导致其他指标恶化的问题. Xu等[7]提出一种基于等角紧框架(equiangular tight frame,ETF)的迭代最小化算法,在不改变测量矩阵对应Gram矩阵属性的情况下修改奇异值,以减小Gram矩阵列向量之间的相关性. 宋儒瑛等[8]基于分块矩阵的思想构造新的测量矩阵,验证了测量矩阵的行列数相差不大时所构造的测量矩阵可以代替等角紧框架矩阵. 魏从静[9]对Gram矩阵进行Schmidt正交化,以此增强矩阵列向量之间的独立性,计算过程中的规范正交基能有效提高数值计算过程的稳定性. He等[10]提出一种基于伪随机序列的测量矩阵优化构造方法,将随机高斯矩阵与伪随机序列和Hadamard矩阵相结合,使得测量矩阵拥有更好的性能、较低复杂度和较好的应用价值.

    本文对测量矩阵的性能优化进行研究,基于Gram矩阵,通过特征值分解降低测量矩阵与稀疏基之间的平均相关性. 再基于ETF理论与梯度缩减迭代算法,使得Gram矩阵向等角紧框架矩阵逼近,降低其非主对角线元素最大值,从而降低最大相关性,提高重构质量.

    假设一个长度为N的信号xCN×1经过M×NM<N)的测量矩阵Φ线性投影后,得到长度为M的测量值yCM×1,如式(1)所示.

    y=Φx.
    (1)

    压缩感知技术使用的前提是信号在某种变换基下具有稀疏性. 若x本身不稀疏,但在某个变换域上是稀疏的,式(1)可改写为

    {y=Φx=Aθ,A=ΦΨ,
    (2)

    式中:ΨN×N的稀疏变换矩阵,称为稀疏基或稀疏字典;θxΨ上的投影;A为感知矩阵.

    由式(2)可以看出,通过接收信号y来重构原始信号x,本质上是求解线性方程组. 由于M<N,线性方程组有无穷多解. 但θ是稀疏的,未知数的个数较少,使得重构信号成为可能.

    在RIP准则基础上所构造的测量矩阵都能在一定程度上保留信号原本信息,保证重构精度[11]. 当矩阵A满足RIP准则时,测量矩阵的有限等距常数δK满足式(3).

    (1δK)||x||22||Ax||22(1+δK)||x||22,
    (3)

    式中:K为信号的稀疏度.

    RIP准则对测量矩阵进行了限制,使得经过投影后的压缩信号能够保留尽可能多的有用信息,用以在重构过程中更加精确地重构出信号.

    当矩阵A满足RIP准则时,压缩感知算法就能通过求解如式(4)所示的L0范数最小问题,对原始信号进行重构.

    {ˆθ=arg min||θ||0,s.t.ΦΨθ=y,
    (4)

    式中:ˆθ为原始信号在Ψ下的最稀疏系数向量.

    得到ˆθ后,即可利用稀疏反变换重构出原始信号,如式(5)所示.

    ˆx=Ψˆθ
    (5)

    式中:ˆx为重构所得的信号.

    求解L0范数最小问题属于NP-hard问题. Donoho等[12]提出,在一定条件下,对L0范数问题进行求解可等价于对L1范数问题进行求解,所求得的解是相同的,即式(4)等价于式(6).

    {ˆθ=argmin||θ||1,s.t.ΦΨθ = y.
    (6)

    式(6)可以转化为线性规划问题进行求解. 常用的求解方法有贪婪类算法[13]、凸松弛算法[14]、贝叶斯类算法[15]等.

    在大规模通信场景中,基站端无法保证每个用户都能单独利用资源块,故不可避免地会产生冲突问题,而基站侧也很难得知哪些用户是处于活跃状态,因此活跃用户检测的问题需要得到解决.

    假设社区用户数为N的蜂窝上行链路,在大规模随机接入的场景下,同一时间点中的设备大多为非活跃状态[16]. 以an{0,1}表示用户的活跃状态,0为非活跃状态,1为活跃状态,且1nN,则由an构成的向量a=(a1,a2,,aN)中,大部分元素都为0,即a具备稀疏性,这使得压缩感知技术应用于大规模通信场景成为可能.

    假设每个用户都由基站分配了长度为M的导频序列φn,则这N个用户的导频序列就能组成M×N的测量矩阵Φ,当Φ满足有限等距性质准则时,就可以应用压缩感知理论,基于接收信号y,通过重构算法对原始信号进行估计,进而对活跃用户的身份进行估计.

    基站接收到的信号可以表示为

    y=Nn=1φnanhn=Φx
    (7)

    式中:hn为第n个用户与基站之间的信道参数.

    定义xn=anhnx=(x1,x2,,xN)=(a1h1,a2h2,,aNhN). 显然,x呈现为稀疏状态. 若第n个用户为活跃状态,则xn等于真实的信道参数;若用户为非活跃状态,则xn为0. 综上可知,在mMTC场景中,活跃用户估计与信道估计可以联合实现.

    Φ的列相关性μ可以表示为列向量之间内积的最大值,如式(8)所示[17].

    μ=max|φi,φj|
    (8)

    式中:φiφj分别为矩阵Φ的第ij列,1i,jNij.

    在矩阵Φ列向量单位化后,若NM(M+1)/2,则称Φ的列相关性存在下界,这个下界即Welch界,如式(9)所示.

    μNM(N1)M.
    (9)

    当式(9)取等号时,称该矩阵具备等角紧框架结构[18]. 特别地,若MN,则下界将收敛至1/M. 通过使测量矩阵与稀疏基矩阵的互相关系数逼近Welch界,可以保证测量矩阵具有更好的压缩观测性能.

    测量矩阵Φ与稀疏基Ψ之间的相关性是影响测量矩阵性能的因素之一. 2个矩阵之间的相关性定义为

    μ(Φ,Ψ)=max|φi,γj|
    (10)

    式中:γj为矩阵Ψ的第j列.

    但仅依靠此定义很难得出两矩阵之间的具体联系. 为此,Elad[19]引入了Gram矩阵,从另一个角度对测量矩阵Φ和稀疏基Ψ之间的相关性进行了定义,将最大相关系数定义为Gram矩阵中非对角线元素绝对值的最大值.

    定义平均相关系数μavg为Gram矩阵中所有非主对角元素绝对值之和的平均值. 当 Gram 矩阵的非主对角元素中最大值较大而平均值较小时,如果仅依据最大值来判断,可能会误认为该矩阵性能不佳. 然而,由于其平均值相对较小,实际应用中的性能也相对良好. 因此,引入平均相关系数是为了避免因最大值与平均值差距过大而导致性能的误判.

    首先,定义等价字典DD=ΦΨ,假设ϑi是矩阵D的第i列;然后,将等价字典D的每列进行单位化处理,得到单位化字典矩阵˜D=[˜ϑ1˜ϑ2˜ϑN],并以此求出对应的Gram矩阵,如式(11)所示.

    G=˜DT˜D=[1˜ϑ1˜ϑ2˜ϑ1˜ϑN˜ϑ2˜ϑ11˜ϑ2˜ϑN˜ϑN˜ϑ1˜ϑN˜ϑ21].
    (11)

    由上文可知,单位化后的等价字典˜D所对应的Gram矩阵是一个主对角线元素全为1的N维半正定矩阵,其秩为M,假设其M个非零特征值为λqq=1,2,,M,有λq>0,由于矩阵所有特征值的和等于矩阵对角线上的元素之和,因此可以推出式(12).

    Mq=1λq=N.
    (12)

    假设lq是Gram矩阵G的某一特征向量,λq是其对应的特征值,则有Gλq=lqλq. 又因为G为对称矩阵,则GTGlq=GGlq=Glqλq=lqλ2q,所以矩阵GTG的特征值为G特征值的平方. 假设gi,j是Gram矩阵G中的第i行第j列的元素,则矩阵GTG对角线上的元素之和为Ni,j=1g2i,j,于是推导出式(13).

    Mq=1λ2q=Mi=jg2i,j.
    (13)

    考虑到非主对角线元素绝对值之和与其平方的和具有相同的单调性,因此,可以通过最小化非主对角线元素平方和来降低平均互相关系数μavg,实现平均相关性的优化,结合式(12),可将问题转化为式(14)所示的问题.

    {minijg2i,j=Mq=1λ2qMi=1g2i,i,s.t.Mq=1λq = N.
    (14)

    要在矩阵G的特征值之和保持不变的前提下,求出上述问题最优解,就需要尽可能减小矩阵G的特征值平方和. 而矩阵G的特征值都大于0,因此,将所有特征值设置为平均值即可. 对Gram矩阵G进行特征值分解,如式(15)所示.

    G=PΛPT
    (15)

    式中:P为矩阵G的特征向量组成的矩阵,Λ为特征值组成的对角矩阵.

    将对角矩阵Λ中的非零元素修改为M/N,得到新的矩阵ˆΛ,再将对角矩阵ˆΛ进行分解,如式(16)所示.

    ˆΛ=LTL
    (16)

    式中:LM×N的矩阵.

    L主对角线中,对应ˆΛ中非零项的元素设为N/M,其余元素皆为0. 重新计算Gram矩阵,得到

    G=PLTLPT.
    (17)

    根据式(11),可以得到

    ˆD=LPT
    (18)

    式中:ˆD为更新后的字典矩阵.

    2.2.1   构造ETF矩阵

    由于对字典矩阵D进行了列单位化处理,故得到的Gram矩阵中对角线元素全为1,且非对角线元素的值也都在1和 −1之间. 通过Gram矩阵减小测量矩阵Φ和稀疏基Ψ之间的相关性,等价于减小Gram矩阵的非对角线元素. 当两矩阵之间不相关时,其所对应的Gram矩阵的非对角线元素也将全为0,此时,Gram矩阵等价于单位矩阵E. 所以,使Gram矩阵向E逼近,即可降低其最大相关性,如式(19)所示.

    minGE22.
    (19)

    要让2个矩阵相关性几乎为0,这在实际情况中是比较困难的[20]. 由于当矩阵满足等角紧框架理论时,其相关性呈最小状态[18],同时,考虑到矩阵非对角线元素几乎都是一些接近0的小数,因此将条件放宽,即使非主对角线元素小于等于某一阈值μ0,使得矩阵G的非主对角线元素朝μ0逼近,μ0即为所期望的两矩阵之间的相关性. 假设与矩阵G维度相同的单位矩阵HH=(Hi,j),ij,按照式(20)对矩阵H进行修改.

    Hi,j={gi,j,|gi,j|<μ0,μ0sgngi,j,|gi,j|>μ0.
    (20)

    将阈值μ0设置为Welch界,可以让测量矩阵与稀疏基矩阵的相关系数逼近下界[21],使得测量矩阵具有更好的性能.

    对矩阵H进行更新后,就需要使得Gram矩阵G尽可能逼近矩阵H. 将关于矩阵GH的表达式设为f,如式(21)所示.

    f=GH22.
    (21)

    为使矩阵G尽可能逼近矩阵H,将问题转化为求函数f的最小值,将G=˜DT˜D代入后,式(21)转化为式(22).

    f=˜DT˜DH22.
    (22)

    但是矩阵G的秩为M,而修改后的矩阵H的秩为N,因此,考虑对矩阵H降秩. 首先,进行特征值分解H=PΛHPT,并修改其特征值,得到秩为M的矩阵HM,如式(23)所示,其中,ΛH为特征值组成的对角矩阵,特征值由大到小排列.

    HM=PIΛHPT
    (23)

    式中:P为矩阵H特征向量组成的矩阵;I=[EM000]RN×NEM为维度为M的单位矩阵.

    假设λHM=(λHM,1,λHM,2,,λHM,M,0,,0),是矩阵HM的特征值. 将所有特征值设为平均值,降低矩阵非主对角线的平方和,从而减少矩阵的平均相关性. 令ˆλHM=1MMp=1λHM,p,则修改后的特征值矩阵ˆΛHM=diag(ˆλHM,ˆλHM,,ˆλHM,0,,0). 优化后的矩阵为

    Hopt=PˆΛHMPT.
    (24)

    此时,所求问题变为式(25)所示问题.

    min˜DT˜DHopt22.
    (25)
    2.2.2   基于梯度缩减算法求最优解

    最优测量矩阵ˆΦ存在解析解,可通过交替策略与奇异值分解[22]求出. 但其子问题的计算成本偏高,耗时较长. 因此,本文选择梯度缩减的思想对原问题求最优解.

    由于矩阵Hopt为常量,f是一个关于字典矩阵˜D的函数,求f的最小值问题也就是求矩阵f关于字典矩阵˜D的梯度,由此需要找出目标函数关于字典矩阵˜D的下降方向,假设下降方向为k,如式(26)所示.

    k=di,j˜DT˜DHopt22
    (26)

    式中:di,j˜D的组成元素.

    由矩阵分解规则变换为[23]

    k=di,jTr{(˜DT˜DHopt)(˜DT˜DHopt)T}=4˜D(˜DT˜DHopt).
    (27)

    沿着下降方向迭代,逐渐迭代缩减目标值,如式(28)所示.

    ˜Du=˜Du14β˜Du1(˜DTu1˜Du1Hopt)
    (28)

    式中:˜Du为第u次迭代后的字典矩阵;β为每次缩减的步长,且β>0,通常按照经验取较小的值.

    经过多次迭代后,通过逆运算得到最终结果ˆΦ,如式(29)所示.

    ˆΦ=˜DuΨ1.
    (29)

    首先,对Gram矩阵进行特征值分解,优化对角矩阵中的特征值,降低Gram矩阵非对角元素平均值;随后,构造出等角紧框架矩阵,并按照梯度缩减的策略使得Gram矩阵向等角紧框架矩阵逼近,减少其最大相关性,提高重构质量;最后,得到优化后的测量矩阵. 定义迭代终止参数σ,当式(30)成立时,停止迭代,输出优化后的测量矩阵.

    ˜DTu˜DuHopt22<σ.
    (30)

    基于矩阵分解与ETF理论的测量矩阵优化算法如下所示:

    输入:测量矩阵Φ、测量矩阵Φ的行数M与列数N,稀疏基Ψ、最大迭代次数t、缩减步长β、Welch界μ0,迭代终止参数σ

    输出:优化后的测量矩阵ˆΦ

    初始化:生成N×N的单位矩阵H,迭代次数u=1,计算字典矩阵D,并列标准化得到˜D0

    1) 生成Gram矩阵:G=˜DTu1˜Du1

    2) 对矩阵G进行特征值分解:G=PΛPT

    3) 将矩阵Λ的对角线元素中的非零项设置为N/M,得到ˆΛG

    4) 重新计算Gram矩阵G=PˆΛPT

    5) 将矩阵ˆΛ进行分解,ˆΛ=LTL

    6) 代入G,计算G=PLTLPT,令ˆDi=LPT

    7) 若|gi,j|μ0Hi,j=gi,j,若|gi,j|>μ0Hi,j=μ0sgn(gi,j),对矩阵H进行修改;

    8) 对矩阵H特征值分解H=PΛHPT

    9) 利用矩阵I=[EM000]对矩阵H降秩,并重新计算HM=PHIΛHPTH

    10) 对HM特征值分解,HM=PΛHMPTΛHM=diag(λHM,1,λHM,2,,λHM,M,0,,0)

    11) 令ˆλHM=1MMp=1λHM,p,修改特征值矩阵ˆΛHM=diag(ˆλHM,ˆλHM,,ˆλHM,0,,0)

    12) Hopt=PˆΛHMPT

    13) ˜Du=˜Du14β˜Du1(˜DTu1˜Du1Hopt)

    14) 若||˜DTu˜DuHopt||22<σ,进入16),否则进入15);

    15) 若u<t,则u=u+1,返回1),否则进入16);

    16) 输出ˆΦ=˜DuΨ1.

    为验证本文优化算法的有效性,在MATLAB上进行仿真,以正交匹配追踪(orthogonal matching pursuit,OMP)作为重构算法,实验参数如表1所示.

    表  1  参数表
    Table  1.  Parameters
    参数名称 参数值
    Φ 随机导频矩阵
    Ψ 单位矩阵
    t/次 1000
    β 0.01
    信道类型 随机瑞利衰落
    导频长度 100
    信号长度 256
    潜在用户数/人 30
    下载: 导出CSV 
    | 显示表格

    图1比较ΦΨ的相关系数随观测维度M的变化趋势. 由图可见,经过基于矩阵分解与ETF理论的改进后,测量矩阵与稀疏基之间的相关性明显降低.

    图  1  相关系数随观测维度M的变化
    Figure  1.  Variation of correlation coefficient with observed dimension M

    图2比较不同稀疏度下各种算法错误估计活跃用户的数量. 从图中可以看出:信号稀疏度低于30时,各算法都能准确地估计活跃用户,但当稀疏度大于30以后,随着稀疏度不断增加,错误估计的活跃用户数量都在增加;本文提出的优化算法性能较好.

    图  2  不同稀疏度下错误估计用户数比较
    Figure  2.  Comparison of estimation errors of user number under different sparsities

    图3比较不同稀疏度下单次重构所消耗的时间. 文献[6-10]所提及的矩阵优化算法复杂度分别为O(N2)O(N2)O(1)O(N)O(N),本文所提出的算法复杂度为O(N2). 从图中可以看出,本文提出的算法所消耗的时间较多,是因为通过特征值分解优化了平均相关性,构造出ETF矩阵后又对其进行降秩与降低平均相关性的操作,计算流程较多,因此单次重构时间相对较长.

    图  3  不同稀疏度下单次重构时间比较
    Figure  3.  Comparison of single reconstruction time under different sparsities

    图4比较不同信噪比(SNR)下信号稀疏度为30时,通过不同算法进行信道估计所得的均方误差(MSE). 从结果可以看出,随着信噪比的增加,各算法均方误差都在降低,其中,本文提出的测量矩阵优化算法效果最好.

    图  4  不同SNR下的MSE比较
    Figure  4.  Comparison of MSEs under different SNRs

    图5比较在不同信噪比下信号稀疏度为30时不同算法所得的误码率(BER). 由图可知,随着信噪比的增加,各个算法的误码率也逐渐下降,其中,本文提出的测量矩阵改进算法性能最好.

    图  5  不同SNR下的BER比较
    Figure  5.  Comparison of BERs under different SNRs

    通过上述仿真实验可以看出,本文提出的基于矩阵分解和ETF理论的测量矩阵改进算法,虽然在时间复杂度上有所不足,但在重构性能、信道估计与活跃用户检测上有一定的优势.

    在大规模通信场景下,基于测量矩阵与稀疏基所构造出的Gram矩阵,提出一种矩阵分解与ETF理论相结合的优化算法. 主要结论如下:

    1) 所提出的算法有效减小了测量矩阵与稀疏基之间的相关性,提高了信号重构质量,优化了测量矩阵在信道估计与活跃用户检测中的性能.

    2) 未来可对该算法时间复杂度进行优化,以提高重构效率. 该算法单次重构时间相对较高,后续研究中将尝试在保证算法性能的基础上,降低其时间复杂度.

  • 图 1  标准卷积结构

    Figure 1.  Standard convolution structure

    图 2  深度可分离式卷积结构

    Figure 2.  Structure of depthwise separable convolution

    图 3  传统学习与迁移学习

    Figure 3.  Traditional learning and transfer learning

    图 4  模型训练与识别流程

    Figure 4.  Model training and identification process

    图 5  岩石样本图像示例

    Figure 5.  Rock sample image examples

    图 6  MobileNet V2深度可分离式卷积结构

    Figure 6.  MobileNet V2 depthwise separable convolution structure

    图 7  训练中训练精度、验证精度及损失变化

    Figure 7.  Training accuracy,verification accuracy and loss changes during training

    表  1  MobileNet V2基础结构

    Table  1.   MobileNet V2 basic structure

    输入操作tcns
    2242 × 3conv2d3212
    1122 × 32bottleneck11611
    1122 × 16bottleneck62422
    562 × 24bottleneck63232
    282 × 32bottleneck66442
    142 × 64bottleneck69631
    142 × 96bottleneck616032
    72 × 160bottleneck632011
    72 × 320conv2d 1 × 1128011
    72 × 1280avgpool 7 × 71
    1 × 1 × 1280conv2d1000
    注:c 为输出通道数;s 为模块第 1 次重复时的操作步长(之后重复操作步长均为 1).
    下载: 导出CSV

    表  2  改进的模型结构

    Table  2.   Improved model structure

    输入操作输出
    224 × 224 × 3conv2d112 × 112 × 32
    112 × 112 × 32bottleneck1112 × 112 × 16
    7 × 7 × 160bottleneck167 × 7 × 160
    7 × 7 × 160bottleneck177 × 7 × 320
    7 × 7 × 320conv2d 1 × 17 × 7 × 1280
    7 × 7 × 1280avgpool 7 × 71 × 1 × 1280
    1 × 1 × 1280Dense1 (激活函数 Relu)1 × 1 × 1024
    1 × 1 × 1024Dropout1 × 1 × 1024
    1 × 1 × 1024Dense2 (激活函数 Relu)1 × 1 × 256
    1 × 1 × 256Dropout1 × 1 × 256
    1 × 1 × 256Dense3 (激活函数 Relu)1 × 1 × 6
    1 × 1 × 6Softmax 分类函数分类结果
    下载: 导出CSV

    表  3  基于fine-tune的迁移训练

    Table  3.   Transfer training based on fine-tune

    输入操作输出
    224 × 224 × 3conv2d112 × 112 × 32
    112 × 112 × 32bottleneck1112 × 112 × 16
    7 × 7 × 160bottleneck167 × 7 × 160
    7 × 7 × 160bottleneck177 × 7 × 320
    7 × 7 × 320conv2d 1 × 17 × 7 × 1280
    7 × 7 × 1280avgpool 7 × 71 × 1 × 1280
    1 × 1 × 1280Dense (激活函数 Relu)1 × 1 × 1024
    1 × 1 × 1024Dropout1 × 1 × 1024
    1 × 1 × 1024Dense (激活函数 Relu)1 × 1 × 256
    1 × 1 × 256Dropout1 × 1 × 256
    1 × 1 × 256Dense (激活函数 Relu)1 × 1 × 6
    1 × 1 × 6Softmax 分类函数分类结果
    下载: 导出CSV

    表  4  训练集各项评价指标

    Table  4.   Index values for training set evaluation

    岩石精确率召回率综合评价指标
    片麻岩1.0000.9330.965
    花岗岩1.0000.9670.983
    大理岩0.9381.0000.968
    石灰岩0.9350.9670.951
    凝灰岩0.9681.0000.984
    砂岩1.0000.9670.983
    下载: 导出CSV

    表  5  测试集图片识别结果

    Table  5.   Recognition results for testing set images

    测试图片
    片麻岩花岗岩大理岩石灰岩凝灰岩砂岩
    片麻岩362561
    花岗岩4514
    大理岩1463
    石灰岩4451
    凝灰岩248
    砂岩4314236
    合计405157645137
    下载: 导出CSV

    表  6  测试集各项评价指标

    Table  6.   Index values for testing set evaluation

    岩石PRF1
    片麻岩0.9000.7200.800
    花岗岩0.8820.9000.891
    大理岩0.8070.9200.860
    石灰岩0.7030.9000.789
    凝灰岩0.9410.9600.950
    砂岩0.9730.7200.828
    下载: 导出CSV

    表  7  实验结果对比

    Table  7.   Comparison of experimental results

    分类指标平均识别精确率单张识别时间/ms
    SVM0.6654650
    VGG160.8143680
    MobileNet V20.8682880
    下载: 导出CSV
  • 管会生,吴和北,陶伟,等. 深埋斜井隧道双模式盾构刀盘驱动扭矩分析[J]. 现代隧道技术,2015,52(6): 163-169.

    GUAN Huisheng, WU Hebei, TAO Wei, et al. Analysis of the cutterhead driving torque of a dual-mode shield used for a deeply buried inclined shaft[J]. Modern Tunnelling Technology, 2015, 52(6): 163-169.
    刘书斌, 方勇, 周超月, 等. 承压溶腔对隧道开挖稳定性影响的模型试验[J]. 岩石力学与工程学报, 2017, 36(增刊2): 3781-3791.

    LIU Shubin, FANG Yong, ZHOU Chaoyue, et al. Model test of tunnel excavation stability influenced by concealed cave with internal water pressure[J]. Chinese Journal of Rock Mechanics and Engineering, 2017, 36(S2): 3781-3791.
    严鹏,张晨,高启栋,等. 不同损伤程度下岩石力学参数变化的声波测试[J]. 岩土力学,2015,36(12): 3425-3432.

    YAN Peng, ZHANG Chen, GAO Qidong, et al. Acoustic wave test on mechanical properties variation of rocks under different damage degrees[J]. Rock and Soil Mechanics, 2015, 36(12): 3425-3432.
    苏凯,张妍珺,伍鹤皋,等. 隧洞开挖过程中围岩安全系数演化特征与支护时机选择方法研究[J]. 岩石力学与工程学报,2019,38(增刊1): 2964-2975.

    SU Kai, ZHANG Yanjun, WU Hegao, et al. Evolution of surrounding rock safety factor and support installation time during tunnel excavation[J]. Chinese Journal of Rock Mechanics and Engineering, 2019, 38(S1): 2964-2975.
    WEI X S, QIN X H, RONG C L, et al. Image classification recognition for rock micro-thin section based on probabilistic neural networks[J]. Applied Mechanics and Materials, 2014, 602-605: 2147-2152. doi: 10.4028/www.scientific.net/AMM.602-605.2147
    罗佳,刘大刚. 基于自适应阈值和连通域的隧道裂缝提取[J]. 西南交通大学学报,2018,53(6): 1137-1141. doi: 10.3969/j.issn.0258-2724.2018.06.007

    LUO Jia, LIU Dagang. Tunnel crack extraction based on adaptive threshold and connected domain[J]. Journal of Southwest Jiaotong University, 2018, 53(6): 1137-1141. doi: 10.3969/j.issn.0258-2724.2018.06.007
    刘烨,程国建,马微,等. 基于铸体薄片图像颜色空间与形态学梯度的岩石分类[J]. 中南大学学报(自然科学版),2016,47(7): 2375-2382.

    LIU Ye, CHENG Guojian, MA Wei, et al. Rock classification based on features form color space and morphological gradient of rock thin section image[J]. Journal of Central South University (Science and Technology), 2016, 47(7): 2375-2382.
    程国建,郭文惠,范鹏召. 基于卷积神经网络的岩石图像分类[J]. 西安石油大学学报(自然科学版),2017,32(4): 116-122. doi: 10.3969/j.issn.1673-064X.2017.04.020

    CHENG Guojian, GUO Wenhui, FAN Pengshao. Study on rock image classification based on convolution neural network[J]. Journal of Xi’an Shiyou University (Natural Science Edition), 2017, 32(4): 116-122. doi: 10.3969/j.issn.1673-064X.2017.04.020
    LECUN Y, BENGIO Y, HINTON G. Deep learning[J]. Nature, 2015, 521: 436-444. doi: 10.1038/nature14539
    BIANCO S, BUZZELLI M, MAZZINI D, et al. Deep learning for logo recognition[J]. Neurocomputing, 2017, 245: 23-30. doi: 10.1016/j.neucom.2017.03.051
    侯进,吕志良,徐茂,等. 基于深度学习的复合神经网络机场信号检测框架[J]. 西南交通大学学报,2019,54(4): 863-869.

    HOU Jin, LYU Zhiliang, XU Mao, et al. Combined neural networks based on deep learning for signal detection in aeronautical communications[J]. Journal of Southwest Jiaotong University, 2019, 54(4): 863-869.
    白林,姚钰,李双涛,等. 基于深度学习特征提取的岩石图像矿物成分分析[J]. 中国矿业,2018,27(7): 178-182.

    BAI Lin, YAO Yu, LI Shuangtao, et al. Mineral composition analysis of rock image based on deep learning feature extraction[J]. China Mining Magazine, 2018, 27(7): 178-182.
    徐述腾,周永章. 基于深度学习的镜下矿石矿物的智能识别实验研究[J]. 岩石学报,2018,34(11): 3244-3252.

    XU Shuteng, ZHOU Yongzhang. Artificial intelligence identification of ore minerals under microscope based on deep learning algorithm[J]. Acta Petrologica Sinica, 2018, 34(11): 3244-3252.
    IMAMVERDIYEV Y, SUKHOSTAT L. Lithological facies classification using deep convolutional neural network[J]. Journal of Petroleum Science and Engineering, 2019, 174: 216-228. doi: 10.1016/j.petrol.2018.11.023
    ZHANG Y, WANG G, LI M, et al. Automated classification analysis of geological structures based on images data and deep learning model[J]. Applied Sciences, 2018, 8(12): 2493. doi: 10.3390/app8122493
    PAN S J, YANG Q. A survey on transfer learning[J]. IEEE Transactions on Knowledge and Data Engineering, 2010, 22(10): 1345-1359. doi: 10.1109/TKDE.2009.191
    QURESHI A S, KHAN A, ZAMEER A, et al. Wind power prediction using deep neural network based meta regression and transfer learning[J]. Applied Soft Computing, 2017, 58: 742-755. doi: 10.1016/j.asoc.2017.05.031
    张野,李明超,韩帅. 基于岩石图像深度学习的岩性自动识别与分类方法[J]. 岩石学报,2018,34(2): 333-342.

    ZHANG Ye, LI Mingchao, HAN Shuai. Automatic identification and classification in lithology based on deep learning in rock images[J]. Acta Petrologica Sinica, 2018, 34(2): 333-342.
    LI N, HAO H, GU Q, et al. A transfer learning method for automatic identification of sandstone microscopic images[J]. Computers & Geosciences, 2017, 103: 111-121.
    WANG C, LI Y, FAN G, et al. Quick recognition of rock images for mobile applications[J]. Journal of Engineering Science and Technology Review, 2018, 11(4): 111-117. doi: 10.25103/jestr.114.14
    SANDLER M, HOWARD A, ZHU M, et al. MobileNet V2: inverted residuals and linear bottlenecks[C]//IEEE Conference on Computer Vision and Pattern Recognition. Salt Lake City: IEEE, 2018: 4510-4520.
    CHOLLET F. Xception: deep learning with depthwise separable convolutions[C]//IEEE Conference on Computer Vision and Pattern Recognition. Honolulu: IEEE, 2017: 1800-1807.
    PAN S J, TSANG I W, KWOL J T, et al. Domain adaptation via transfer component analysis[J]. IEEE Transactions on Neural Networks, 2011, 22(2): 199-210. doi: 10.1109/TNN.2010.2091281
    HE K, ZHANG X, REN S, et al. Deep residual learning for image recognition[C]//IEEE Conference on Computer Vision and Pattern Recognition. Las Vegas: IEEE, 2016: 770-778.
    杜恩宇,张宁,李艳荻. 基于自适应分块编码SVM的车道导向箭头多分类方法[J]. 光学学报,2018,38(10): 251-258.

    DU Enyu, ZHANG Ning, LI Yandi. Multi classification method of lane arrow markings based on support vector machines with adaptive partitioning coding[J]. Acta Optica Sinica, 2018, 38(10): 251-258.
  • 期刊类型引用(8)

    1. 王华俊,卿翠贵,马永政,赵勇博,李春光. 耐久性弱化边坡模型试验及DIC位移测试研究. 水利与建筑工程学报. 2023(01): 124-130 . 百度学术
    2. 王文才,李俊鹏,王创业,陈世江,王鹏. 边帮煤采动影响下边坡变形演化特征及失稳形态分析. 煤炭科学技术. 2023(07): 321-336 . 百度学术
    3. 龚逸非,姚爱军,谷坤生,李彦霖,田甜. 强震作用下反倾薄层岩质斜坡倾倒破坏机制及动力响应研究. 振动与冲击. 2023(20): 263-274 . 百度学术
    4. 任旭,王鸿榕,刘梓晗,徐桂弘,汪权明. 可再分散乳胶粉对软岩相似材料性能的影响. 地下空间与工程学报. 2023(S2): 649-655 . 百度学术
    5. 任旭,王鸿榕,李先恒,徐桂弘,杨福敏. 岩质边坡不良地质体模型试验中泥岩相似材料试验研究. 有色金属工程. 2022(07): 158-162+171 . 百度学术
    6. 王胜. 安家岭露天矿开采工作面穿越陷落柱方案分析. 煤矿安全. 2021(03): 237-240 . 百度学术
    7. 姜程程,范文,苑伟娜. 基于环剪试验的含钙质结核古土壤剪切特性. 西南交通大学学报. 2021(04): 809-817 . 本站查看
    8. 熊茹雪,许万忠,史丁康. 库水位升降对岸坡稳定性影响研究综述. 地质灾害与环境保护. 2019(02): 86-91 . 百度学术

    其他类型引用(27)

  • 加载中
图(7) / 表(7)
计量
  • 文章访问数:  640
  • HTML全文浏览量:  324
  • PDF下载量:  24
  • 被引次数: 35
出版历程
  • 收稿日期:  2019-11-04
  • 修回日期:  2020-04-27
  • 网络出版日期:  2020-12-15
  • 刊出日期:  2021-04-15

目录

/

返回文章
返回