Huang Hong, Song Yu, Wu Yao, Shi Jia, Xie Xia, Jin Hai
IEEE Trans Neural Netw Learn Syst. 2022 Mar;33(3):983-995. doi: 10.1109/TNNLS.2020.3036825. Epub 2022 Feb 28.
Link prediction and node classification are two important downstream tasks of network representation learning. Existing methods have achieved acceptable results but they perform these two tasks separately, which requires a lot of duplication of work and ignores the correlations between tasks. Besides, conventional models suffer from the identical treatment of information of multiple views, thus they fail to learn robust representation for downstream tasks. To this end, we tackle link prediction and node classification problems simultaneously via multitask multiview learning in this article. We first explain the feasibility and advantages of multitask multiview learning for these two tasks. Then we propose a novel model named MT-MVGCN to perform link prediction and node classification tasks simultaneously. More specifically, we design a multiview graph convolutional network to extract abundant information of multiple views in a network, which is shared by different tasks. We further apply two attention mechanisms: view the attention mechanism and task attention mechanism to make views and tasks adjust the view fusion process. Moreover, view reconstruction can be introduced as an auxiliary task to boost the performance of the proposed model. Experiments on real-world network data sets demonstrate that our model is efficient yet effective, and outperforms advanced baselines in these two tasks.
链路预测和节点分类是网络表示学习的两个重要下游任务。现有方法已取得可接受的结果,但它们分别执行这两个任务,这需要大量重复工作且忽略了任务之间的相关性。此外,传统模型对多视图信息进行相同处理,因此无法为下游任务学习到鲁棒的表示。为此,在本文中我们通过多任务多视图学习同时解决链路预测和节点分类问题。我们首先解释多任务多视图学习对于这两个任务的可行性和优势。然后我们提出一种名为MT-MVGCN的新颖模型来同时执行链路预测和节点分类任务。更具体地说,我们设计了一个多视图图卷积网络来提取网络中多视图的丰富信息,这些信息由不同任务共享。我们进一步应用两种注意力机制:视图注意力机制和任务注意力机制,以使视图和任务调整视图融合过程。此外,可以引入视图重建作为辅助任务来提升所提模型的性能。在真实网络数据集上的实验表明,我们的模型高效且有效,并且在这两个任务上优于先进的基线模型。