Graph Neural Prompting with Large Language Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Tian, Yijun, Song, Huan, Wang, Zichen, Wang, Haozhu, Hu, Ziqing, Wang, Fang, Chawla, Nitesh V., Xu, Panpan
Format: Preprint
Published: 2023
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910283037933568
author Tian, Yijun
Song, Huan
Wang, Zichen
Wang, Haozhu
Hu, Ziqing
Wang, Fang
Chawla, Nitesh V.
Xu, Panpan
author_facet Tian, Yijun
Song, Huan
Wang, Zichen
Wang, Haozhu
Hu, Ziqing
Wang, Fang
Chawla, Nitesh V.
Xu, Panpan
contents Large language models (LLMs) have shown remarkable generalization capability with exceptional performance in various language modeling tasks. However, they still exhibit inherent limitations in precisely capturing and returning grounded knowledge. While existing work has explored utilizing knowledge graphs (KGs) to enhance language modeling via joint training and customized model architectures, applying this to LLMs is problematic owing to their large number of parameters and high computational cost. Therefore, how to enhance pre-trained LLMs using grounded knowledge, e.g., retrieval-augmented generation, remains an open question. In this work, we propose Graph Neural Prompting (GNP), a novel plug-and-play method to assist pre-trained LLMs in learning beneficial knowledge from KGs. GNP encompasses various designs, including a standard graph neural network encoder, a cross-modality pooling module, a domain projector, and a self-supervised link prediction objective. Extensive experiments on multiple datasets demonstrate the superiority of GNP on both commonsense and biomedical reasoning tasks across different LLM sizes and settings. Code is available at https://github.com/meettyj/GNP.
format Preprint
id arxiv_https___arxiv_org_abs_2309_15427
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Graph Neural Prompting with Large Language Models
Tian, Yijun
Song, Huan
Wang, Zichen
Wang, Haozhu
Hu, Ziqing
Wang, Fang
Chawla, Nitesh V.
Xu, Panpan
Computation and Language
Artificial Intelligence
Machine Learning
Large language models (LLMs) have shown remarkable generalization capability with exceptional performance in various language modeling tasks. However, they still exhibit inherent limitations in precisely capturing and returning grounded knowledge. While existing work has explored utilizing knowledge graphs (KGs) to enhance language modeling via joint training and customized model architectures, applying this to LLMs is problematic owing to their large number of parameters and high computational cost. Therefore, how to enhance pre-trained LLMs using grounded knowledge, e.g., retrieval-augmented generation, remains an open question. In this work, we propose Graph Neural Prompting (GNP), a novel plug-and-play method to assist pre-trained LLMs in learning beneficial knowledge from KGs. GNP encompasses various designs, including a standard graph neural network encoder, a cross-modality pooling module, a domain projector, and a self-supervised link prediction objective. Extensive experiments on multiple datasets demonstrate the superiority of GNP on both commonsense and biomedical reasoning tasks across different LLM sizes and settings. Code is available at https://github.com/meettyj/GNP.
title Graph Neural Prompting with Large Language Models
topic Computation and Language
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2309.15427