AIDC
← 返回全部动态
arXiv 机器学习AI 评分 70/10009月24日 12:00

已训练图神经网络中的局部证据与几何读出修复

许多用于节点分类的图神经网络(GNN)将线性分类器应用于局部消息的非负混合中。预测错误往往源于混合权重不佳或可达 logit 集合的几何位置不适合该分类器。该研究通过精确质量线性规划和两种后验修复方法区分了这两类原因。研究表明,每个重加权预测都对应一个中心化的 logit 平移。在8个数据集、8个 GNN 骨干网络及10种划分上的实验表明,重加权使冻结模型的平均准确率从62.6%提升至63.8%。

推荐理由提出了一种针对已训练图神经网络的几何读出后验修复方法,有效提升了节点分类准确率。

原标题:Local Evidence and Geometric Readout Repair in Trained GNNs

阅读 arXiv 机器学习 原文 ↗