paper-with-me

Papers

A Teacher-Free Graph Knowledge Distillation Framework with Dual Self-Distillation

2024-03-06 · Lirong Wu, Haitao Lin, Zhangyang Gao, Guojiang Zhao, Stan Z. Li

Recent years have witnessed great success in handling graph-related tasks with Graph Neural Networks (GNNs). Despite their great academic success, Multi-Layer Perceptrons (MLPs) remain the primary workhorse for practical industrial applications. One reason for such an academic-industry gap is the neighborhood-fetching latency incurred by data dependency in GNNs. To reduce their gaps, Graph Knowledge Distillation (GKD) is proposed, usually based on a standard teacher-student architecture, to distill knowledge from a large teacher GNN into a lightweight student GNN or MLP. However, we found in this paper that neither teachers nor GNNs are necessary for graph knowledge distillation. We propose a Teacher-Free Graph Self-Distillation (TGS) framework that does not require any teacher model or GNNs during both training and inference. More importantly, the proposed TGS framework is purely based on MLPs, where structural information is only implicitly used to guide dual knowledge self-distillation between the target node and its neighborhood. As a result, TGS enjoys the benefits of graph topology awareness in training but is free from data dependency in inference. Extensive experiments have shown that the performance of vanilla MLPs can be greatly improved with dual self-distillation, e.g., TGS improves over vanilla MLPs by 15.54% on average and outperforms state-of-the-art GKD algorithms on six real-world datasets. In terms of inference speed, TGS infers 75X-89X faster than existing GNNs and 16X-25X faster than classical inference acceleration methods.

📄 PDF Abstract BibTeX arXiv:2403.03483

Code (1)

lirongwu/tgs 공식 구현 pytorch

Tasks

Knowledge Distillation

Methods 이 논문이 사용한 방법론

Knowledge Distillation A very simple way to improve the performance of almost any machine learning algorithm is to train many different models on the same data and then to average their predictions.…

Similar Papers 제목 키워드 기반

Train Your Own GNN Teacher: Graph-Aware Distillation on Textual Graphs

2023-04-20 · Costas Mavromatis, Vassilis N. Ioannidis, Shen Wang, Da Zheng 외

How can we learn effective node representations on textual graphs? Graph Neural Networks (GNNs) that use Language Models (LMs) to encode textual information of graphs achieve state-of-the-art performance in many node cla…

Knowledge DistillationNode Classification

Adversarial Curriculum Graph-Free Knowledge Distillation for Graph Neural Networks

2025-04-01 · Yuang Jia, Xiaojuan Shan, Jun Xia, Guancheng Wan 외

Data-free Knowledge Distillation (DFKD) is a method that constructs pseudo-samples using a generator without real data, and transfers knowledge from a teacher model to a student by enforcing the student to overcome dimen…

Data-free Knowledge DistillationKnowledge Distillation

Data-Free Adversarial Knowledge Distillation for Graph Neural Networks

2022-05-08 · Yuanxin Zhuang, Lingjuan Lyu, Chuan Shi, Carl Yang 외

Graph neural networks (GNNs) have been widely used in modeling graph structured data, owing to its impressive performance in a wide range of practical applications. Recently, knowledge distillation (KD) for GNNs has enab…

Generative Adversarial NetworkGraph ClassificationKnowledge DistillationModel Compression+1

On Self-Distilling Graph Neural Network

2020-11-04 · Yuzhao Chen, Yatao Bian, Xi Xiao, Yu Rong 외

Recently, the teacher-student knowledge distillation framework has demonstrated its potential in training Graph Neural Networks (GNNs). However, due to the difficulty of training over-parameterized GNN models, one may no…

Graph EmbeddingGraph Neural NetworkKnowledge Distillation

FreeKD: Free-direction Knowledge Distillation for Graph Neural Networks

2022-06-14 · Kaituo Feng, Changsheng Li, Ye Yuan, Guoren Wang

Knowledge distillation (KD) has demonstrated its effectiveness to boost the performance of graph neural networks (GNNs), where its goal is to distill knowledge from a deeper teacher GNN into a shallower student GNN. Howe…

Knowledge Distillationreinforcement-learningReinforcement Learning (RL)Transfer Learning