基于边缘特征的图神经网络架构搜索

作     者：周鹏 杨军 ZHOU Peng;YANG Jun

作者机构：兰州交通大学电子与信息工程学院兰州730070 兰州交通大学测绘与地理信息学院兰州730070 

基　　金：国家自然科学基金(42261067,61862039) 甘肃省自然科学基金(21JR7RA286) 甘肃省高等学校创新能力提升项目(2019B-056) 甘肃省高等学校创新基金(2020B-116) 兰州市人才创新创业项目(2020-RC-22) 兰州交通大学天佑创新团队(TY202002) 

出 版 物：《兰州交通大学学报》 (Journal of Lanzhou Jiaotong University)

年 卷 期：2022年第41卷第6期

页      码：44-53页

摘      要：针对图神经网络架构搜索方法存在注重实体特征信息、忽略边缘特征潜在信息的问题,提出一种边缘特征图神经网络架构搜索方法,采用随机梯度下降的搜索策略优化图神经网络架构以提高搜索性能.首先,设计基于边缘特征的搜索空间,用于实体和边缘特征的提取;其次,通过边的权重进行决策,逐步对边进行裁剪,得到一个无需权重共享的网络架构.在三维点云数据集ModelNet上的实验结果表明:所提方法能有效提取图神经网络的边缘信息;与连续贪婪神经网络搜索、基于资源平衡的架构搜索、含注入噪声的可微分神经网络搜索等自动搜索的网络架构和人工设计的网络架构相比,所提方法能搜索出高效的网络架构,在三维点云模型分类中具有较高的准确率.

主 题 词：图神经网络 神经网络架构搜索 边缘特征 点云 

学科分类：081203[081203] 08[工学] 0835[0835] 0812[工学-测绘类] 

D　O　I：10.3969/j.issn.1001-4373.2022.06.007

馆 藏 号：203115684...