Frameless Graph Knowledge Distillation

Dai Shi, Zhiqi Shao, Junbin Gao, Zhiyong Wang, Yi Guo

    Research output: Contribution to journalArticlepeer-review

    Abstract

    Knowledge distillation (KD) has shown great potential for transferring knowledge from a complex teacher model to a simple student model in which the heavy learning task can be accomplished efficiently and without losing too much prediction accuracy. Recently, many attempts have been made by applying the KD mechanism to graph representation learning models such as graph neural networks (GNNs) to accelerate the model's inference speed via student models. However, many existing KD-based GNNs utilize multilayer perceptron (MLP) as a universal approximator in the student model to imitate the teacher model's process without considering the graph knowledge from the teacher model. In this work, we provide a KD-based framework on multiscaled GNNs, known as graph framelet, and prove that by adequately utilizing the graph knowledge in a multiscaled manner provided by graph framelet decomposition, the student model is capable of adapting both homophilic and heterophilic graphs and has the potential of alleviating the oversquashing issue with a simple yet effective graph surgery. Furthermore, we show how the graph knowledge supplied by the teacher is learned and digested by the student model via both algebra and geometry. Comprehensive experiments show that our proposed model can generate learning accuracy identical to or even surpass the teacher model while maintaining the high speed of inference.
    Original languageEnglish
    Pages (from-to)8125-8139
    Number of pages15
    JournalIEEE Transactions on Neural Networks and Learning Systems
    Volume36
    Issue number5
    DOIs
    Publication statusPublished - 2025

    Bibliographical note

    Publisher Copyright:
    © 2012 IEEE.

    Keywords

    • Computational modeling
    • Adaptation models
    • Analytical models
    • Data models
    • Graph framelets
    • Graph neural networks
    • graph neural networks (GNNs)
    • knowledge distillation (KD)
    • Knowledge engineering
    • Spectral analysis

    Fingerprint

    Dive into the research topics of 'Frameless Graph Knowledge Distillation'. Together they form a unique fingerprint.

    Cite this