人工知能学会全国大会論文集
Online ISSN : 2758-7347
第37回 (2023)
セッションID: 2A4-GS-2-01
会議情報

構造情報を用いたグラフ対照学習の改善
*後藤 輝一武山 泰豊井本 和範
著者情報
キーワード: グラフ, 対照学習
会議録・要旨集 フリー

詳細
抄録

近年、ノード表現学習としてグラフ対照学習手法が注目されている。ノードレベルのグラフ対照学習では、異なる変換を加え水増しした二つのグラフについてグラフ間の同一ノードを正例、それ以外の全てのノードの組み合わせを負例として損失を計算する。一方で、グラフは似た属性をもつノードが周辺に集まりやすいホモフィリー性等の性質を持っている。既存の対照損失ではノードの周辺構造に寄らず正例や負例を生成している為、それらグラフの性質を活用出来ていない。本研究では、既存の対照損失に加え、ノードの周辺構造を考慮した二種類の損失を導入する。1) エッジ再構成損失。あるノードとその一次隣接ノードの個々の組み合わせを正例とする。2) 平均エッジ再構成損失。あるノードとその一次隣接ノードの平均値を正例とする。エッジ再構成損失は隣接する個々のノード同士の表現を近付けるように学習され、平均エッジ再構成損失は同じ組み合わせの隣接ノードを持つノード同士の表現を近付けるように学習される。引用関係や併売関係等の様々な性質を持つグラフデータでの実験において、それぞれの損失を加えた手法が既存手法と比較し高い性能を示した。

著者関連情報
© 2023 人工知能学会
前の記事 次の記事
feedback
Top