摘要
本发明公开了一种基于图神经网络的潜在滑坡识别方法及系统,方法包括:构建包括节点层、子网层和全局层的多层级特征融合的图注意力网络模型,其中,图注意力网络模型采用双支路编码器‑译码器结构;采用大样本的数据集1和小样本的数据集2的合并数据集,对双支路编码器‑译码器结构模型进行预训练至收敛,形成预训练模型;基于将数据集1和数据集2输入预训练模型得到的辅助集节点特征,对图注意力网络模型进行训练和验证,直至模型收敛训练完成;将数据集2中需要进行识别的图片输入训练完成的图注意力网络模型,得到潜在滑坡的预测结果。通过本发明的技术方案,实现了小数据集2中目标特征的有效提取,提升了模型的分类性能和泛化能力。
技术关键词
节点特征
注意力
预训练模型
译码器结构
滑坡识别方法
多层感知器
样本
编码器
数据
输出特征
识别系统
支路
模型预训练
源节点
网络
潜在滑坡识别
模型训练模块
层级
系统为您推荐了相关专利信息
三维建模数据
无序点云
内部支撑结构
生成特征
边界特征
电力工控系统
恶意软件识别
融合特征
软件测试环境
融合启发式算法
实体关系抽取
工业
生成训练样本
BERT模型
自然语言
轴承故障诊断方法
铁路转向架
转向架轴箱轴承
故障诊断模型
特征提取网络