JoPA:Explaining Large Language Model's Generation via Joint Prompt Attribution

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chang, Yurui, Cao, Bochuan, Wang, Yujia, Chen, Jinghui, Lin, Lu
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915496214921216
author Chang, Yurui
Cao, Bochuan
Wang, Yujia
Chen, Jinghui
Lin, Lu
author_facet Chang, Yurui
Cao, Bochuan
Wang, Yujia
Chen, Jinghui
Lin, Lu
contents Large Language Models (LLMs) have demonstrated impressive performances in complex text generation tasks. However, the contribution of the input prompt to the generated content still remains obscure to humans, underscoring the necessity of understanding the causality between input and output pairs. Existing works for providing prompt-specific explanation often confine model output to be classification or next-word prediction. Few initial attempts aiming to explain the entire language generation often treat input prompt texts independently, ignoring their combinatorial effects on the follow-up generation. In this study, we introduce a counterfactual explanation framework based on Joint Prompt Attribution, JoPA, which aims to explain how a few prompt texts collaboratively influences the LLM's complete generation. Particularly, we formulate the task of prompt attribution for generation interpretation as a combinatorial optimization problem, and introduce a probabilistic algorithm to search for the casual input combination in the discrete space. We define and utilize multiple metrics to evaluate the produced explanations, demonstrating both the faithfulness and efficiency of our framework.
format Preprint
id arxiv_https___arxiv_org_abs_2405_20404
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle JoPA:Explaining Large Language Model's Generation via Joint Prompt Attribution
Chang, Yurui
Cao, Bochuan
Wang, Yujia
Chen, Jinghui
Lin, Lu
Computation and Language
Machine Learning
Large Language Models (LLMs) have demonstrated impressive performances in complex text generation tasks. However, the contribution of the input prompt to the generated content still remains obscure to humans, underscoring the necessity of understanding the causality between input and output pairs. Existing works for providing prompt-specific explanation often confine model output to be classification or next-word prediction. Few initial attempts aiming to explain the entire language generation often treat input prompt texts independently, ignoring their combinatorial effects on the follow-up generation. In this study, we introduce a counterfactual explanation framework based on Joint Prompt Attribution, JoPA, which aims to explain how a few prompt texts collaboratively influences the LLM's complete generation. Particularly, we formulate the task of prompt attribution for generation interpretation as a combinatorial optimization problem, and introduce a probabilistic algorithm to search for the casual input combination in the discrete space. We define and utilize multiple metrics to evaluate the produced explanations, demonstrating both the faithfulness and efficiency of our framework.
title JoPA:Explaining Large Language Model's Generation via Joint Prompt Attribution
topic Computation and Language
Machine Learning
url https://arxiv.org/abs/2405.20404