核心概念
그래프 신경망 네트워크와 대형 언어 모델을 결합하여 그래프 학습 작업을 수행할 수 있는 새로운 프레임워크인 GraphPrompter를 제안한다.
要約
이 논문은 대형 언어 모델(LLM)을 그래프 학습 작업에 적용하는 방법을 소개한다. 제안하는 GraphPrompter 프레임워크는 다음과 같은 구성으로 이루어져 있다:
-
그래프 섹션: 그래프 신경망 네트워크(GNN)를 사용하여 각 노드의 3-hop 부그래프를 인코딩하고, 이를 프로젝션 레이어를 통해 LLM의 벡터 공간에 정렬한다.
-
LLM 섹션: 각 노드의 텍스트 속성을 LLM의 토크나이저와 임베딩 레이어를 통해 처리한다. 이렇게 얻은 텍스트 임베딩을 GNN 임베딩과 연결하여 LLM에 입력으로 제공한다.
이를 통해 GNN의 구조적 정보와 LLM의 텍스트 이해 능력을 결합하여, 노드 분류와 링크 예측 등의 그래프 학습 작업에서 우수한 성능을 달성할 수 있다. 실험 결과, GraphPrompter는 다양한 벤치마크 데이터셋에서 기존 방법들을 능가하는 성과를 보였다.
統計
그래프 신경망 네트워크(GNN)를 사용하여 각 노드의 3-hop 부그래프를 인코딩한다.
프로젝션 레이어를 통해 GNN 임베딩을 LLM의 벡터 공간에 정렬한다.
각 노드의 텍스트 속성을 LLM의 토크나이저와 임베딩 레이어를 통해 처리한다.
GNN 임베딩과 텍스트 임베딩을 연결하여 LLM에 입력으로 제공한다.
引用
"To the best of our knowledge, this is the very first work investigating whether LLMs can understand graph learning tasks via soft prompting."
"Comprehensive experiments on various benchmark datasets under node classification and link prediction tasks demonstrate the effectiveness of our proposed method."