arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2501.01630cs.LGcs.SI
收藏

有向图中节点分类的概率模型

A Probabilistic Model for Node Classification in Directed Graphs

Diego Huerta, Gerardo Arizmendi

AI总结:

本文提出一种面向有向图的概率生成式分类模型,利用最大似然或最大后验估计预测节点标签,兼具高可解释性,并在两个数据集上取得优于或媲美最先进方法的性能。

摘要:

在这项工作中,我们提出了一个针对有向图的概率模型,其中节点具有属性和标签。该模型作为一个生成式分类器,能够使用最大似然估计或最大后验估计来预测未见节点的标签。与一些常见的节点分类方法(如图神经网络)相比,该模型做出的预测具有高度可解释性。我们将该模型应用于两个数据集,展示了与最先进方法相比具有竞争力甚至更优的预测性能。所考虑的数据集之一改编自数学谱系计划(Math Genealogy Project),此前尚未被用于此目的。因此,我们在该数据集上评估了若干分类算法,以比较我们模型的性能,并为这一新资源提供基准。

英文摘要:

In this work, we present a probabilistic model for directed graphs where nodes have attributes and labels. This model serves as a generative classifier capable of predicting the labels of unseen nodes using either maximum likelihood or maximum a posteriori estimations. The predictions made by this model are highly interpretable, contrasting with some common methods for node classification, such as graph neural networks. We applied the model to two datasets, demonstrating predictive performance that is competitive with, and even superior to, state-of-the-art methods. One of the datasets considered is adapted from the Math Genealogy Project, which has not previously been utilized for this purpose. Consequently, we evaluated several classification algorithms on this dataset to compare the performance of our model and provide benchmarks for this new resource.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑