arrow
返回

Self-supervised Bipartite Graph Representation Learning: A Dirichlet Max-margin Matrix Factorization Approach

delete2024-05-17
delete0
PRE
AI
S
Shenghai Zhong
S
Shu Guo
J
Jing Liu
黄洪仁 封面图
黄洪仁 (Hongren Huang)
王立红 封面图
王立红 (Lihong Wang) *
J
Jianxin Li
C
Chen Li
Y
Yiming Hei
DOI:10.1145/3645098delete
delete原文链接
delete原文求助
delete分享
delete收藏
摘要

摘要

En 中文
Bipartite graph representation learning aims to obtain node embeddings by compressing sparse vectorized representations of interactions between two types of nodes, e.g., users and items. Incorporating structural attributes among homogeneous nodes, such as user communities, improves the identification of similar interaction preferences, namely, user/item embeddings, for downstream tasks. However, existing methods often fail to proactively discover and fully utilize these latent structural attributes. Moreover, themanual collection and labeling of structural attributes is always costly. In this article, we propose a novel approach called Dirichlet Max-margin Matrix Factorization (DM3F), which adopts a self-supervised strategy to discover latent structural attributes and model discriminative node representations. Specifically, in self-supervised learning, our approach generates pseudo group labels (i.e., structural attributes) as a supervised signal using the Dirichlet process without relying on manual collection and labeling, and employs them in a max-margin classification. Additionally, we introduce a Variational Markov Chain Monte Carlo algorithm (Variational MCMC) to effectively update the parameters. The experimental results on six real datasets demonstrate that, in the majority of cases, the proposed method outperforms existing approaches based on matrix factorization and neural networks. Furthermore, the modularity analysis confirms the effectiveness of our model in capturing structural attributes to produce high-quality user embeddings.
Keyword:
Self-supervised learning
bipartite graph
representation learning
user
embedding
matrix completion
item recommendation

期刊

ACM Transactions on Intelligent Systems and Technology 封面图
ACM Transactions on Intelligent Systems and Technology
IF:
6.6
论文数:
1.5K
被引数:
6.2K

机构

B
Beihang University
学者数:
5.2W
论文数: 4.1W
被引数: 37
引用论文

引用论文

err分享
err收藏
A method for generating uniform size-segregated pyrite particle fractions
err2007-10-10
err0
errOAAI
errAmy L Wolfe; Ran Liu; Brian W Stewart; Rosemary C Capo; David A Dzombak
err分享
err收藏
Safety and tolerability of inhaled hypertonic saline in young children with cystic fibrosis
err2008-09-30
err0
PREAI
errElisabeth P. Dellon; Scott H. Donaldson; Robin Johnson; Stephanie D. Davis
err分享
err收藏
Characteristics and management of acute ST-segment elevation myocardial infarctions occurring in ski resorts in the French Alps: Impact of an acute coronary care network
err2010-08-01
err0
PREAI
errMathieu Chacornac; Gilles Baronne-Rochette; Marie-Hélène Schmidt; Dominique Savary; Daniel Habold; Hélène Bouvaist; Stéphanie Marliere; Loic Belle; Jacques Machecourt; Gérald Vanzetto
err分享
err收藏
Exercise‐Induced Ventricular Arrhythmia after Repair of Tetralogy of Fallot
err2007-10-09
err0
PREAI
errJunichiro Fukushige; Kunihisa Shimomura; Tatsuo Harada; Mitsuru Fukazawa; Kohji Ueda; Kouichi Tokunaga
err分享
err收藏
Classification of Wool Follicles in Sheep
err1957-07-01
err0
PREAI
errA. B. WILDMAN
err分享
err收藏
err分享
err收藏
学者 查看更多内容