MTGCN: A multi-task approach for node classification and link prediction in graph data |
| |
Institution: | 1. School of Information Management, Wuhan University, Wuhan, Hubei, China;2. Center for Studies of Information Resources, Wuhan University, Wuhan, Hubei, China;3. School of Library and Information Science, Simmons University, Boston, USA;1. School of Management Engineering, Shandong Jianzhu University, Jinan, China;2. School of Business, University of Shanghai for Science and Technology, Shanghai, China |
| |
Abstract: | Both node classification and link prediction are popular topics of supervised learning on the graph data, but previous works seldom integrate them together to capture their complementary information. In this paper, we propose a Multi-Task and Multi-Graph Convolutional Network (MTGCN) to jointly conduct node classification and link prediction in a unified framework. Specifically, MTGCN consists of multiple multi-task learning so that each multi-task learning learns the complementary information between node classification and link prediction. In particular, each multi-task learning uses different inputs to output representations of the graph data. Moreover, the parameters of one multi-task learning initialize the parameters of the other multi-task learning, so that the useful information in the former multi-task learning can be propagated to the other multi-task learning. As a result, the information is augmented to guarantee the quality of representations by exploring the complex constructure inherent in the graph data. Experimental results on six datasets show that our MTGCN outperforms the comparison methods in terms of both node classification and link prediction. |
| |
Keywords: | Graph convolutional network Node classification Link prediction Multi-task learning |
本文献已被 ScienceDirect 等数据库收录! |
|