GNNAVI: Navigating the Information Flow in Large Language Models by Graph Neural Network

Publikation: Beitrag in Buch/Konferenzbericht/Sammelband/GutachtenBeitrag in KonferenzbandBeigetragenBegutachtung

Beitragende

Abstract

Large Language Models (LLMs) exhibit strong In-Context Learning (ICL) capabilities when prompts with demonstrations are used. However, fine-tuning still remains crucial to further enhance their adaptability. Prompt-based fine-tuning proves to be an effective fine-tuning method in low-data scenarios, but high demands on computing resources limit its practicality. We address this issue by introducing a prompt-based parameter-efficient finetuning (PEFT) approach. GNNAVI leverages insights into ICL's information flow dynamics, which indicates that label words act in prompts as anchors for information propagation. GNNAVI employs a Graph Neural Network (GNN) layer to precisely guide the aggregation and distribution of information flow during the processing of prompts by hardwiring the desired information flow into the GNN. Our experiments on text classification tasks with GPT-2 and Llama2 show GNNAVI surpasses standard prompt-based fine-tuning methods in few-shot settings by updating just 0.2% to 0.5% of parameters. We compare GNNAVI with prevalent PEFT approaches, such as prefix tuning, LoRA and Adapter in terms of performance and efficiency. Our analysis reveals that GNNAVI enhances information flow and ensures a clear aggregation process.

Details

OriginalspracheEnglisch
Titel62nd Annual Meeting of the Association for Computational Linguistics, ACL 2024 - Proceedings of the Conference
Redakteure/-innenLun-Wei Ku, Andre Martins, Vivek Srikumar
Herausgeber (Verlag)Association for Computational Linguistics (ACL)
Seiten3987-4001
Seitenumfang15
ISBN (elektronisch)979-889176099-8
PublikationsstatusVeröffentlicht - 16 Aug. 2024
Peer-Review-StatusJa

Konferenz

Titel62nd Annual Meeting of the Association for Computational Linguistics
KurztitelACL 2024
Veranstaltungsnummer62
Dauer11 - 16 August 2024
Webseite
OrtCentara Grand and Bangkok Convention Centre
StadtBangkok
LandThailand

Externe IDs

Scopus 85205293526
ORCID /0000-0001-5458-8645/work/193180540

Schlagworte