用知识蒸馏让MLP高效又可解释地学习多层图结构
Learning Accurate, Efficient, and Interpretable MLPs on Multiplex Graphs via Node-wise Multi-View Ensemble Distillation
- 通过节点级多视角集成蒸馏,让MLP学习多层图的丰富语义
- 相比纯MLP提升约10%准确率,推理速度超原模型35倍以上
- 可自适应分配不同视图权重,适合需要可解释性的场景
多层图包含多个边类型(视图),在相同节点间提供更丰富的结构语义和建模能力。多层图神经网络(MGNN)通常由视图专用的GNN和多视图融合层组成,在下游任务中表现优异。然而,其依赖邻域聚合的机制限制了在低延迟场景中的部署。受近期GNN到MLP知识蒸馏框架启发,本文提出多层图无神经网络(MGFNN和MGFNN+),结合MGNN的高性能与MLP的高效推理。MGFNN直接以节点特征为输入,使用教师MGNN的软标签训练学生MLP;MGFNN+进一步采用低秩近似重参数化,学习节点级系数,实现各视图专用GNN的自适应知识集成。该节点级多视图集成蒸馏策略使学生MLP能为不同节点学习更具信息量的多层语义。实验表明,MGFNN相比基础MLP平均提升约10%准确率,性能可媲美甚至超越教师MGNN(准确);推理速度相较MGNN提升35.40×–89.14×(高效);MGFNN+能根据节点差异自适应分配多视图集成系数(可解释)。
原文摘要 · Abstract (English)
Multiplex graphs, with multiple edge types (graph views) among common nodes, provide richer structural semantics and better modeling capabilities. Multiplex Graph Neural Networks (MGNNs), typically comprising view-specific GNNs and a multi-view integration layer, have achieved advanced performance in various downstream tasks. However, their reliance on neighborhood aggregation poses challenges for deployment in latency-sensitive applications. Motivated by recent GNN-to-MLP knowledge distillation frameworks, we propose Multiplex Graph-Free Neural Networks (MGFNN and MGFNN+) to combine MGNNs' superior performance and MLPs' efficient inference via knowledge distillation. MGFNN directly trains student MLPs with node features as input and soft labels from teacher MGNNs as targets. MGFNN+ further employs a low-rank approximation-based reparameterization to learn node-wise coefficients, enabling adaptive knowledge ensemble from each view-specific GNN. This node-wise multi-view ensemble distillation strategy allows student MLPs to learn more informative multiplex semantic knowledge for different nodes. Experiments show that MGFNNs achieve average accuracy improvements of about 10% over vanilla MLPs and perform comparably or even better to teacher MGNNs (accurate); MGFNNs achieve a 35.40$\times$-89.14$\times$ speedup in inference over MGNNs (efficient); MGFNN+ adaptively assigns different coefficients for multi-view ensemble distillation regarding different nodes (interpretable).
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。