-
题名基于迁移学习及通道先验注意力机制的地质构造识别
- 1
-
-
作者
刘俊杰
马凯
黄泽华
田苗
邱芹军
陶留锋
谢忠
-
机构
湖北省水电工程智能视觉监测重点实验室(三峡大学)
三峡大学计算机与信息学院
地质探测与评估教育部重点实验室(中国地质大学(武汉))
中国地质大学(武汉)计算机学院
地理信息系统国家地方联合工程实验室(中国地质大学(武汉))
-
出处
《广西师范大学学报(自然科学版)》
北大核心
2025年第2期107-120,共14页
-
基金
国家自然科学基金(42301492)
国家重点研发计划项目(2022YFB3904200,2022YFF0711601)
+1 种基金
湖北省自然科学基金(2022CFB640)
地质探测与评估教育部重点实验室主任基金(GLAB2023ZR01)。
-
文摘
针对平面地质图件中地质构造背景复杂、符号表示多样化而导致识别效果不佳的问题,本文提出一种基于迁移学习和通道先验注意力机制的地质构造识别模型MsAttenEfficientNet。该模型以EfficientNet为主干网络架构,并使用通道先验注意力(channel prior convolution attention,CPCA)模块替换EfficientNet特征提取模块MBConv中的压缩和激励网络(squeeze-and-excitation net,SENet),使模型能够动态地分配通道和空间注意力权重,更准确地捕捉到图像中的重要区域和空间结构;其次对顶层预测模块进行改进,引入Swish激活函数和Dropout层,加强模型的泛化性能;最后使用Adam优化算法提高网络的收敛速度,并利用迁移学习实现特征参数共享。通过在地质构造数据集GeoStr18上进行训练及测试,实验结果表明,MsAttenEfficientNet模型对地质构造的识别精准率为96.92%,召回率为96.89%,F 1分数为96.90%,优于ResNet50、ShuffleNetV2和DenseNet121等主流分类识别模型,可有效用于地质构造识别。
-
关键词
图像识别
地质构造
EfficientNet网络
通道先验注意力
迁移学习
-
Keywords
image recognition
geological structure
EfficientNet network
CPCA
transfer learning
-
分类号
TP183
[自动化与计算机技术—控制理论与控制工程]
TP391
[自动化与计算机技术—计算机应用技术]
-
