错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Two Heads Are Better Than One: Teaching MLPs with Multiple Graph Neural Networks via Knowledge Distillation

  • Bo-Wei Yang,
  • Ming-Yi Chang,
  • Chia-Hsun Lu,
  • Chih-Ya Shen

摘要

Knowledge Distillation (KD) has been successfully applied to compress and accelerate Graph Neural Networks (GNNs) in recent years. Recently, KD has been adeptly applied to boost Multi-Layer Perceptrons (MLPs), enabling them to parallel GNNs in performance. This is notable as MLPs depend solely on node features and require extensive knowledge from multiple teachers for enhancement. However, since existing multi-teacher KD methods often rely on a graph’s topological data, they are typically limited to GNN-based student models, thus excluding MLPs. In this paper, we introduce an innovative method that amalgamates various GNNs into a super teacher, which is then distilled into an MLP student. Moreover, our experiments suggest the promising number of selected teachers configurations to boost the performance of the student model. Extensive experiments on five benchmark datasets show that our proposed approaches outperform the state-of-the-art methods and achieve even higher accuracy than the teacher models. Also, the inference time of our approach is 30 \(\times \) -60 \(\times \) faster than GNNs and other KD methods.