论文标题

图形策略网络用于图形上可转移的主动学习

Graph Policy Network for Transferable Active Learning on Graphs

论文作者

Hu, Shengding, Xiong, Zheng, Qu, Meng, Yuan, Xingdi, Côté, Marc-Alexandre, Liu, Zhiyuan, Tang, Jian

论文摘要

图形神经网络(GNN)由于其在各种领域的简单性和有效性而吸引了日益普及。但是,训练这些网络通常需要大量标记的数据,这可能非常昂贵。在本文中,我们研究了GNN的主动学习,即如何有效地将节点标记在图上,以降低训练GNN的注释成本。我们将问题作为图表上的顺序决策过程提出,并通过加强学习来学习最佳查询策略。通过在具有完整标签的几个源图上共同培训,我们学习了可转移的主动学习策略,可以直接推广到未标记的目标图。来自不同领域的多个数据集上的实验结果证明了学术策略在促进同一域和不同域之间图形之间转移的两个设置中的主动学习绩效方面的有效性。

Graph neural networks (GNNs) have been attracting increasing popularity due to their simplicity and effectiveness in a variety of fields. However, a large number of labeled data is generally required to train these networks, which could be very expensive to obtain in some domains. In this paper, we study active learning for GNNs, i.e., how to efficiently label the nodes on a graph to reduce the annotation cost of training GNNs. We formulate the problem as a sequential decision process on graphs and train a GNN-based policy network with reinforcement learning to learn the optimal query strategy. By jointly training on several source graphs with full labels, we learn a transferable active learning policy which can directly generalize to unlabeled target graphs. Experimental results on multiple datasets from different domains prove the effectiveness of the learned policy in promoting active learning performance in both settings of transferring between graphs in the same domain and across different domains.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源