arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2307.00534cs.LG
收藏

共享增长:基于提示的免方向知识蒸馏的图神经网络

Shared Growth of Graph Neural Networks via Prompted Free-direction Knowledge Distillation

Beijing Institute of Technology(北京理工大学)

Kaituo Feng, Yikun Miao, Changsheng Li, Ye Yuan, Guoren Wang

AI总结:

本文提出免方向知识蒸馏框架FreeKD,通过强化学习在多个浅层GNN间动态交换知识,并利用提示学习增强多样性,在五个基准数据集上大幅超越基础GNN,性能媲美甚至优于传统深层教师蒸馏。

摘要:

知识蒸馏(KD)已被证明能有效提升图神经网络(GNNs)的性能,其典型目标是将知识从较深的教师GNN蒸馏到较浅的学生GNN中。然而,由于众所周知的过参数化和过平滑问题,训练一个令人满意的较深GNN往往颇具挑战性,导致在实际应用中知识迁移无效。在本文中,我们提出了首个基于强化学习的GNN免方向知识蒸馏框架,称为FreeKD,该框架不再需要提供一个较深的、优化良好的教师GNN。我们的核心思想是协作学习两个较浅的GNN,在它们之间交换知识。由于我们观察到,在训练过程中,一个典型的GNN模型通常在不同节点上表现出更好和更差的性能,我们设计了一种动态的、免方向的知识迁移策略,该策略涉及两个层次的动作:1)节点级动作决定两个网络对应节点之间知识迁移的方向;然后2)结构级动作决定由节点级动作生成的局部结构中哪些应被传播。此外,考虑到不同的增强图可能捕获图数据的不同视角,我们提出了FreeKD-Prompt,它基于提示学习学习无失真且多样化的增强,以交换多样化的知识。更进一步,我们不局限于在两个GNN之间进行知识交换,而是开发了FreeKD++,以实现在多个GNN之间的免方向知识迁移。在五个基准数据集上的大量实验表明,我们的方法大幅超越了基础GNN。更令人惊讶的是,我们的FreeKD与从更深更强的教师GNN蒸馏知识的传统KD算法相比,具有相当甚至更好的性能。

英文摘要:

Knowledge distillation (KD) has shown to be effective to boost the performance of graph neural networks (GNNs), where the typical objective is to distill knowledge from a deeper teacher GNN into a shallower student GNN. However, it is often quite challenging to train a satisfactory deeper GNN due to the well-known over-parametrized and over-smoothing issues, leading to invalid knowledge transfer in practical applications. In this paper, we propose the first Free-direction Knowledge Distillation framework via reinforcement learning for GNNs, called FreeKD, which is no longer required to provide a deeper well-optimized teacher GNN. Our core idea is to collaboratively learn two shallower GNNs to exchange knowledge between them. As we observe that one typical GNN model often exhibits better and worse performances at different nodes during training, we devise a dynamic and free-direction knowledge transfer strategy that involves two levels of actions: 1) node-level action determines the directions of knowledge transfer between the corresponding nodes of two networks; and then 2) structure-level action determines which of the local structures generated by the node-level actions to be propagated. Additionally, considering that different augmented graphs can potentially capture distinct perspectives of the graph data, we propose FreeKD-Prompt that learns undistorted and diverse augmentations based on prompt learning for exchanging varied knowledge. Furthermore, instead of confining knowledge exchange within two GNNs, we develop FreeKD++ to enable free-direction knowledge transfer among multiple GNNs. Extensive experiments on five benchmark datasets demonstrate our approaches outperform the base GNNs in a large margin. More surprisingly, our FreeKD has comparable or even better performance than traditional KD algorithms that distill knowledge from a deeper and stronger teacher GNN.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑