需求描述

使用圖神經(jīng)網(wǎng)絡(luò)計(jì)算圖中節(jié)點(diǎn)重要性。首先用graphsage計(jì)算節(jié)點(diǎn)初始嵌入，然后用基于中心度的注意力機(jī)制進(jìn)行改進(jìn)，就得到了節(jié)點(diǎn)的最終嵌入。然后通過(guò)一個(gè)全連接層計(jì)算出節(jié)點(diǎn)的重要性分?jǐn)?shù)，使用均方誤差進(jìn)行訓(xùn)練。數(shù)據(jù)集為常用的引文網(wǎng)絡(luò)Cora、Citeseer、Pubmed，標(biāo)簽為文章被引用的次數(shù)。評(píng)估指標(biāo)為NDCG@K，baseline為pagerank算法，消融實(shí)驗(yàn)為不使用注意力機(jī)制。