Aug, 2023

建模深度贝叶斯图网络中的边特征

TL;DR我们扩展了上下文图马尔可夫模型,采用深度且概率的机器学习模型来建模边特征的分布。我们在架构上进行了改进,将边特征映射成离散状态的贝叶斯网络,并用于原始模型。通过这样做,即使在没有边特征的情况下,我们也能构建更丰富的图表示,通过在标准图分类基准测试中的性能改进得到了证实。此外,在边特征至关重要的图回归场景中成功测试了我们的提议,我们展示了所学到的边表示在三个链接预测任务上相对于原始模型的重大性能改进。由于计算复杂度与边数量线性相关,该提议的模型适用于大规模图处理。