跳到正文
arXiv cs.LG· Zhewei Chen, Hao Zhu, Jiaojiao Jiang, Ahad N. Zehmakan·· 6 小时前AI 评分17

G²MLP:用 Ollivier-Ricci 曲率引导的 GNN 到 MLP 图几何蒸馏

Distilling Graph Geometry: Knowledge Gap from GNNs to MLPs

AI 导读

针对 GNN 到 MLP 蒸馏中学生模型未明确保留教师图结构几何的问题,研究者提出 G²MLP 训练时蒸馏框架,用 Ollivier-Ricci 曲率识别稀疏图上的谱欠拟合与稠密图上的谱过拟合,并在预测级与表示级对齐之间分配监督。

来源:arXiv cs.LG · arxiv.org