跳到正文
arXiv · ML· Jincheng Huang, Jie Xu, Xiaoshuang Shi, Ping Hu, Lei Feng, Xiaofeng Zhu·· 17 小时前AI 评分20

GNNs 最终层置信度校准新方法解析

Understanding and Mitigating Under-Confidence in GNNs from the Final Layer

AI 导读

该研究提出一种简单的图校准方法,揭示最终层模型置信度由类质心级和节点级校准共同控制。降低最终层参数权重衰减可缓解 GNN 的低置信度,节点级校准作为补充鼓励测试节点靠近预测类原型。

来源:arXiv · ML · arxiv.org