本文提出了一種新的圖神經網絡解釋方法GraphGI。GraphGI利用博弈理論中的互動值來衡量圖數據結構之間的相互影響,並以此為依據生成解釋子圖。具體來說:
GraphGI從空集開始,逐步添加邊,並計算添加新邊後子圖的互動強度。每次選擇能使子圖互動強度最大化的邊加入。
互動強度的計算基於Shapley值,但GraphGI進一步考慮了正負向互動,以更全面地反映數據特徵之間的相互影響。
為提高計算效率,GraphGI採用了Monte Carlo采樣的方法近似計算Shapley值和互動強度。
實驗結果表明,GraphGI在合成數據集和真實數據集上均取得了較好的解釋效果,在保持高解釋度的同時,也能達到較高的簡潔性。這表明GraphGI能有效捕捉圖神經網絡中數據特徵的相互作用,為模型解釋提供了一種新的思路。
다른 언어로
소스 콘텐츠 기반
arxiv.org
더 깊은 질문