4 个月前

使用图神经网络对结构化对象进行排序

使用图神经网络对结构化对象进行排序

摘要

图神经网络(GNNs)已在许多结构化数据领域成功应用,其应用范围从分子性质预测到社交网络分析。受GNN广泛适用性的启发,我们提出了一类所谓的RankGNNs,这是神经学习排序(Learning to Rank, LtR)方法与GNN的结合体。RankGNNs通过一组图之间的成对偏好进行训练,表明其中一个图比另一个更受青睐。这一问题的一个实际应用是药物筛选,专家希望在大量候选药物中找到最有潜力的分子。我们通过实验证明,所提出的成对RankGNN方法要么显著优于或至少与基于点的基线方法的排名性能相当,后者通过基于GNN的图回归解决LtR问题。

代码仓库

Cortys/rankgnn
官方
tf
GitHub 中提及

基准测试

基准方法指标
graph-ranking-on-ogbg-molesol2-WL-GNN + DirectRanker
Kendall's Tau: 0.745
graph-ranking-on-ogbg-molesol2-WL-GNN + Utility Regression
Kendall's Tau: 0.747
graph-ranking-on-ogbg-molesol2-WL-GNN + Rank Regression
Kendall's Tau: 0.720
graph-ranking-on-ogbg-molesol2-WL-GNN + CmpNN
Kendall's Tau: 0.718
graph-ranking-on-ogbg-molfreesolv2-WL-GNN + Utility Regression
Kendall's Tau: 0.379
graph-ranking-on-ogbg-molfreesolv2-WL-GNN + DirectRanker
Kendall's Tau: 0.525
graph-ranking-on-ogbg-molfreesolv2-WL-GNN + Rank Regression
Kendall's Tau: 0.524
graph-ranking-on-ogbg-molfreesolv2-WL-GNN + CmpNN
Kendall's Tau: 0.527
graph-ranking-on-ogbg-mollipo2-WL-GNN + Rank Regression
Kendall's Tau: 0.332
graph-ranking-on-ogbg-mollipo2-WL-GNN + CmpNN
Kendall's Tau: 0.503
graph-ranking-on-ogbg-mollipo2-WL-GNN + Utility Regression
Kendall's Tau: 0.318
graph-ranking-on-ogbg-mollipo2-WL-GNN + DirectRanker
Kendall's Tau: 0.505
graph-ranking-on-zinc2-WL-GNN + DirectRanker
Kendall's Tau: 0.894
graph-ranking-on-zinc2-WL-GNN + Rank Regression
Kendall's Tau: 0.810
graph-ranking-on-zinc2-WL-GNN + Utility Regression
Kendall's Tau: 0.803
graph-ranking-on-zinc2-WL-GNN + CmpNN
Kendall's Tau: 0.873

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
使用图神经网络对结构化对象进行排序 | 论文 | HyperAI超神经