מושגי ליבה
대형 언어 모델(LLM)과 그래프 신경망(GNN)을 결합하여 그래프 학습 작업을 수행할 수 있는 새로운 프레임워크인 GraphPrompter를 제안한다.
תקציר
이 논문은 대형 언어 모델(LLM)과 그래프 신경망(GNN)을 결합하여 그래프 학습 작업을 수행할 수 있는 새로운 프레임워크인 GraphPrompter를 소개한다.
-
그래프 섹션:
- 각 노드에 대해 3홉 서브그래프를 추출하고 GNN을 사용하여 노드 임베딩을 계산한다.
- 노드 임베딩을 LLM 벡터 공간에 맞추기 위해 MLP 프로젝션 레이어를 사용한다.
-
LLM 섹션:
- 각 노드의 텍스트 속성(예: 논문 제목 및 초록)을 LLM의 토크나이저와 임베딩 레이어를 사용하여 처리한다.
- 노드 임베딩과 텍스트 임베딩을 연결하여 LLM의 자기 주의 레이어에 입력한다.
-
실험 결과:
- 노드 분류 및 링크 예측 작업에서 제안된 GraphPrompter가 다양한 벤치마크 데이터셋에서 우수한 성능을 보였다.
- 특히 PubMed와 Citeseer 데이터셋에서 각각 94.80%와 73.61%의 최고 정확도를 달성했다.
- 제안 방법은 기존의 순수 GNN, 제로 샷 LLM, 소프트 프롬프트 튜닝, 파인튜닝 등의 기술보다 우수한 성능을 보였다.
סטטיסטיקה
노드 분류 실험에서 PubMed 데이터셋의 GraphPrompter + LoRA 모델이 94.80%의 최고 정확도를 달성했다.
링크 예측 실험에서 Citeseer 데이터셋의 서브그래프 프롬프트 튜닝 모델이 93.49%의 최고 정확도를 달성했다.
ציטוטים
"To the best of our knowledge, this is the very first work investigating whether LLMs can understand graph learning tasks via soft prompting."
"Extensive experiments demonstrate the effectiveness of our proposed framework under both node classification and link prediction tasks across various graph benchmarks."