跳到正文
arXiv · ML· Alessio Borgi, Mario Severino, Luke Braithwaite, Steve Azzolin, R\'emi Bourgerie, Olga Zaghen, Emanuele Mule, Riccardo Ali, Leo Minh Kustermann, Edoardo Coppola, Antonio Purificato, Stefano Fiorini, Christopher Irwin, Filippo Spiga, Fabrizio Silvestri, Michael Bronstein, Pietro Li\`o·· 3 小时前AI 评分27

Sheaf-MPNN:基于消息传递的Sheaf神经网络库

Sheaf-MPNN: A Library for Sheaf Neural Networks as Message Passing

AI 导读

作者推出PyTorch Geometric库Sheaf-MPNN,将Sheaf扩散重构为局部传输聚合,免去构建全局Laplacian矩阵。 支持15类Sheaf神经网络架构,提供涵盖节点与图任务的基准测试套件,理论证明其等价性。

来源:arXiv · ML · arxiv.org