reczoo / RecZoo

A curated model zoo for recommendation tasks
Apache License 2.0
170 stars 43 forks source link

【UltraGCN】论文中的公式16与代码中的实现不一致 #29

Closed Light-lyy closed 10 months ago

Light-lyy commented 11 months ago

https://github.com/reczoo/RecZoo/blob/main/matching/gnn/UltraGCN/main.py#L146

代码中146行到157行,按照代码中的写法,得到的会是

$$ \omega_{i,j} = \frac{1}{g_i} * \sqrt{\frac{g_i+1}{g_j+1}} $$

显然与论文中的公式16不同。事实上代码中的这种写法只用到了矩阵G(代码中的A,即I-I共现矩阵)每一行元素之和(或者说每一列元素之和,一样的,因为是对称矩阵)的信息,根本没用到矩阵中的某个元素。而在论文公式16中是用到了矩阵中的元素的

$$ G{i,i}, G{i,j} $$

是代码中发现这样更好特意这么改写的还是因为疏忽导致的不一致呢?希望能得到解答