中文

图神经网络中的熵感知消息传递

机器学习 2024-03-08 v1

摘要

深层图神经网络面临过平滑问题。本文引入了一种新颖的、受物理学启发的 GNN 模型,旨在缓解该问题。我们的方法与现有 GNN 架构相集成,引入了一个熵感知消息传递项。该项在节点聚合过程中对熵执行梯度上升,从而在嵌入中保持一定程度的熵。我们在各种常见数据集上对模型与 state-of-the-art GNN 进行了比较分析。

关键词

引用

@article{arxiv.2403.04636,
  title  = {Entropy Aware Message Passing in Graph Neural Networks},
  author = {Philipp Nazari and Oliver Lemke and Davide Guidobene and Artiom Gesp},
  journal= {arXiv preprint arXiv:2403.04636},
  year   = {2024}
}

备注

4 pages, 3 figures