融合全局信息和对比学习的图神经网络推荐模型

王彦乐,张瑞峰,李锵

Graph neural network recommendation model integrating global information and contrastive learning

Yanle WANG,Ruifeng ZHANG,Qiang LI

表 2 不同推荐模型在4个公开数据集上的性能参数对比

Tab.2 Performance comparison of different recommendation models on four public datasets

数据集 模型 Recall@10 NDCG@10 Recall@20 NDCG@20 Recall@50 NDCG@50 ML-1M NGCF 0.184 6 0.252 8 0.274 1 0.261 4 0.434 1 0.305 5 LightGCN 0.187 6 0.251 4 0.279 6 0.262 0 0.446 9 0.309 1 SGL 0.188 8 0.252 6 0.284 8 0.264 9 0.448 7 0.311 1 SimGCL 0.205 2 0.272 6 0.298 8 0.281 5 0.451 5 0.328 2 NCL 0.204 8 0.272 2 0.302 5 0.283 2 0.462 8 0.329 2 AdaGCL 0.207 2 0.274 5 0.301 2 0.286 8 0.466 2 0.329 5 GraphAug 0.206 0 0.273 2 0.303 4 0.284 3 0.468 6 0.330 0 GICL 0.210 3 0.277 3 0.304 4 0.287 9 0.470 6 0.333 9 η/% +1.50 +1.02 +0.33 +0.38 +0.43 +1.18 Yelp NGCF 0.063 0 0.044 6 0.102 6 0.056 7 0.186 4 0.078 4 LightGCN 0.073 0 0.052 0 0.116 3 0.065 2 0.201 6 0.087 5 SGL 0.083 3 0.060 1 0.128 8 0.073 9 0.214 0 0.096 4 SimGCL 0.089 6 0.065 8 0.130 4 0.081 0 0.209 8 0.101 8 NCL 0.091 2 0.066 2 0.132 6 0.081 3 0.224 7 0.104 8 AdaGCL 0.090 2 0.067 3 0.131 2 0.082 2 0.221 3 0.102 5 GraphAug 0.092 2 0.068 2 0.137 7 0.081 7 0.224 0 0.104 2 GICL 0.095 8 0.070 4 0.143 4 0.084 2 0.233 2 0.107 4 η/% +3.90 +3.23 +4.14 +2.43 +3.78 +2.48 Books NGCF 0.061 7 0.042 7 0.097 8 0.053 7 0.169 9 0.072 5 LightGCN 0.079 7 0.056 5 0.120 6 0.068 9 0.201 2 0.089 9 SGL 0.089 8 0.064 5 0.133 1 0.077 7 0.215 7 0.099 2 SimGCL 0.092 4 0.065 6 0.133 8 0.082 0 0.215 8 0.100 9 NCL 0.093 0 0.066 4 0.137 7 0.081 2 0.216 4 0.101 7 AdaGCL 0.094 2 0.067 2 0.134 6 0.082 8 0.217 7 0.102 3 GraphAug 0.093 7 0.068 2 0.138 1 0.081 5 0.217 0 0.103 2 GICL 0.098 9 0.071 5 0.146 4 0.085 9 0.230 7 0.108 2 η/% +4.99 +4.84 +6.01 +3.74 +5.97 +4.84 Gowalla NGCF 0.119 2 0.085 2 0.175 5 0.101 3 0.281 1 0.127 0 LightGCN 0.136 2 0.087 6 0.197 6 0.115 2 0.304 4 0.141 4 SGL 0.146 5 0.104 8 0.208 4 0.122 5 0.319 7 0.149 7 SimGCL 0.147 2 0.105 2 0.200 5 0.123 2 0.319 8 0.150 8 NCL 0.148 2 0.106 1 0.211 8 0.125 3 0.322 9 0.153 1 AdaGCL 0.149 1 0.108 2 0.212 7 0.127 5 0.323 8 0.152 2 GraphAug 0.151 2 0.107 3 0.213 2 0.126 5 0.325 4 0.154 1 GICL 0.156 7 0.113 9 0.220 5 0.131 8 0.332 7 0.159 6 η/% +3.64 +5.27 +3.42 +3.37 +2.24 +3.57