Multi-Scale Distillation from Multiple Graph Neural Networks

نویسندگان

چکیده

Knowledge Distillation (KD), which is an effective model compression and acceleration technique, has been successfully applied to graph neural networks (GNNs) recently. Existing approaches utilize a single GNN as the teacher distill knowledge. However, we notice that models with different number of layers demonstrate classification abilities on nodes degrees. On one hand, for high degrees, their local structures are dense complex, hence more message passing needed. Therefore, perform better. other low whose relatively sparse simple, repeated can easily lead over-smoothing. Thus, less suitable. existing single-teacher knowledge distillation based model, sub-optimal. To this end, propose novel approach multi-scale knowledge, learns from multiple capture topological semantic at scales. Instead learning equally, proposed method automatically assigns proper weights each via attention mechanism enables student select teachers structures. Extensive experiments conducted evaluate four public datasets. The experimental results superiority our over state-of-the-art methods. Our code publicly available https://github.com/NKU-IIPLab/MSKD.

برای دانلود رایگان متن کامل این مقاله و بیش از 32 میلیون مقاله دیگر ابتدا ثبت نام کنید

اگر عضو سایت هستید لطفا وارد حساب کاربری خود شوید

منابع مشابه

ذخیره در منابع من


  با ذخیره ی این منبع در منابع من، دسترسی به آن را برای استفاده های بعدی آسان تر کنید

ژورنال

عنوان ژورنال: Proceedings of the ... AAAI Conference on Artificial Intelligence

سال: 2022

ISSN: ['2159-5399', '2374-3468']

DOI: https://doi.org/10.1609/aaai.v36i4.20354