arXiv cs.LG· Lintao Yanga, Sirui Lia, Yaqing Wang, Pietro Li\`o, Xu Shen, Baisong Liu, Chengbin Peng·· 6 小时前AI 评分17
N-GNAS:节点级图神经网络架构搜索框架
Node-level Graph Neural Architecture Search Framework
AI 导读
研究者提出节点级图神经网络架构搜索算法 N-GNAS,可在更新节点特征时为每个节点子集自动选择合适网络架构,并引入对比学习损失分离不同类别样本特征。在八个节点与图分类数据集上,N-GNAS 优于现有主流 GNAS 方法和人工设计的 GNN,在 CiteSeer 数据集上准确率达 78.26%。
来源:arXiv cs.LG · arxiv.org