Extracting Low-/High- Frequency Knowledge from Graph Neural Networks and Injecting it into MLPs: An Effective GNN-to-MLP Distillation Framework

计算机科学 蒸馏 推论 可扩展性 人工智能 机器学习 图形 领域知识 频域 过程(计算) 数据挖掘 模式识别(心理学) 理论计算机科学 化学 操作系统 有机化学 数据库 计算机视觉
作者
Lirong Wu,Haitao Lin,Yufei Huang,Tianyu Fan,Stan Z. Li
出处
期刊:Cornell University - arXiv [Cornell University]
被引量:3
标识
DOI:10.48550/arxiv.2305.10758
摘要

Recent years have witnessed the great success of Graph Neural Networks (GNNs) in handling graph-related tasks. However, MLPs remain the primary workhorse for practical industrial applications due to their desirable inference efficiency and scalability. To reduce their gaps, one can directly distill knowledge from a well-designed teacher GNN to a student MLP, which is termed as GNN-to-MLP distillation. However, the process of distillation usually entails a loss of information, and ``which knowledge patterns of GNNs are more likely to be left and distilled into MLPs?" becomes an important question. In this paper, we first factorize the knowledge learned by GNNs into low- and high-frequency components in the spectral domain and then derive their correspondence in the spatial domain. Furthermore, we identified a potential information drowning problem for existing GNN-to-MLP distillation, i.e., the high-frequency knowledge of the pre-trained GNNs may be overwhelmed by the low-frequency knowledge during distillation; we have described in detail what it represents, how it arises, what impact it has, and how to deal with it. In this paper, we propose an efficient Full-Frequency GNN-to-MLP (FF-G2M) distillation framework, which extracts both low-frequency and high-frequency knowledge from GNNs and injects it into MLPs. Extensive experiments show that FF-G2M improves over the vanilla MLPs by 12.6% and outperforms its corresponding teacher GNNs by 2.6% averaged over six graph datasets and three common GNN architectures.
最长约 10秒,即可获得该文献文件

科研通智能强力驱动
Strongly Powered by AbleSci AI
科研通是完全免费的文献互助平台,具备全网最快的应助速度,最高的求助完成率。 对每一个文献求助,科研通都将尽心尽力,给求助人一个满意的交代。
实时播报
1秒前
张先生发布了新的文献求助10
2秒前
我是老大的应助被yyds采纳,获得10
2秒前
2秒前
安逸1发布了新的文献求助10
3秒前
Jayson完成签到,获得积分10
4秒前
4秒前
Kenny发布了新的文献求助50
5秒前
兰亭序完成签到 ,获得积分10
6秒前
6秒前
Lipuer发布了新的文献求助10
7秒前
隐形曼青的应助被小白采纳,获得10
8秒前
9秒前
10秒前
羊屎蛋完成签到 ,获得积分10
10秒前
jinmai发布了新的文献求助10
10秒前
11秒前
14秒前
14秒前
安逸1完成签到,获得积分10
14秒前
ss发布了新的文献求助10
15秒前
17秒前
18秒前
yyds发布了新的文献求助10
18秒前
板烧堡堡发布了新的文献求助10
18秒前
张文康完成签到 ,获得积分10
19秒前
追寻微笑完成签到,获得积分10
19秒前
20秒前
Hello的应助被yyds采纳,获得10
21秒前
23秒前
张先生发布了新的文献求助10
23秒前
朴素完成签到,获得积分10
23秒前
祁风发布了新的文献求助10
24秒前
nitin_sixer发布了新的文献求助10
24秒前
mu完成签到,获得积分10
25秒前
光亮雨完成签到 ,获得积分10
26秒前
Forever完成签到,获得积分10
28秒前
ss完成签到,获得积分10
28秒前
29秒前
彭于晏的应助被科研通管家采纳,获得10
29秒前
高分求助中
(应助此贴封号)【重要!!请各用户(尤其是新用户)详细阅读】【科研通的精品贴汇总】 10000
自動車の空力技術 800
Essentials of Carbohydrate Chemistry and Biochemistry, 4th Edition 800
Organizational Behavior 510
Management and the Arts 510
Matrix Methods in Data Mining and Pattern Recognition Second Edition 510
Issues in Task-Based Language Teaching 500
热门求助领域 (近24小时)
化学 材料科学 医学 生物 纳米技术 计算机科学 化学工程 工程类 有机化学 物理 复合材料 生物化学 内科学 细胞生物学 基因 遗传学 免疫学 冶金 光电子学 癌症研究
热门帖子
关注 科研通微信公众号,转发送积分 7783428
求助须知:如何正确求助?哪些是违规求助? 9322741
关于积分的说明 20391410
捐赠科研通 7372095
什么是DOI,文献DOI怎么找? 3320669
关于科研通互助平台的介绍 2468702
邀请新用户注册赠送积分活动 2336902