计算机科学
图形
人工智能
分类器(UML)
先验概率
机器学习
图形模型
二元分类
马尔可夫链
理论计算机科学
模式识别(心理学)
算法
支持向量机
贝叶斯概率
作者
Junehee Yoo,Jung-Hun Kim,Hoyoung Yoon,Geonsoo Kim,Changwon Jang,U Kang
标识
DOI:10.1109/icdm51629.2021.00094
摘要
How can we classify graph-structured data only with positive labels? Graph-based positive-unlabeled (PU) learning is to train a binary classifier given only the positive labels when the relationship between examples is given as a graph. The problem is of great importance for various tasks such as detecting malicious accounts in a social network, which are difficult to be modeled by supervised learning when the true negative labels are absent. Previous works for graph-based PU learning assume that the prior distribution of positive nodes is known in advance, which is not true in many real-world cases. In this work, we propose GRAB (Graph-based Risk minimization with iterAtive Belief propagation), a novel end-to-end approach for graph-based PU learning that requires no class prior. GRAB models a given graph as a Markov network and runs the marginalization and update steps iteratively. The marginalization step estimates the marginals of latent variables, while the update step trains a classifier network utilizing the computed priors in the objective function. Extensive experiments on five datasets show that GRAB achieves state-of-the-art accuracy, even compared with previous methods that are given the true prior.
科研通智能强力驱动
Strongly Powered by AbleSci AI