Saved in:
Bibliographic Details
Main Authors: Feng, Yu, Zhou, Ben, Lin, Weidong, Roth, Dan
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2404.12494
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917028362715136
author Feng, Yu
Zhou, Ben
Lin, Weidong
Roth, Dan
author_facet Feng, Yu
Zhou, Ben
Lin, Weidong
Roth, Dan
contents Predictive models often need to work with incomplete information in real-world tasks. Consequently, they must provide reliable probability or confidence estimation, especially in large-scale decision-making and planning tasks. Current large language models (LLMs) are insufficient for accurate estimations, but they can generate relevant factors that may affect the probabilities, produce coarse-grained probabilities when the information is more complete, and help determine which factors are relevant to specific downstream contexts. In this paper, we make use of these capabilities of LLMs to provide a significantly more accurate probabilistic estimation. We propose BIRD, a novel probabilistic inference framework that aligns a Bayesian network with LLM abductions and then estimates more accurate probabilities in a deduction step. We show BIRD provides reliable probability estimations that are 30% better than those provided directly by LLM baselines. These estimates further contribute to better and more trustworthy decision making.
format Preprint
id arxiv_https___arxiv_org_abs_2404_12494
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle BIRD: A Trustworthy Bayesian Inference Framework for Large Language Models
Feng, Yu
Zhou, Ben
Lin, Weidong
Roth, Dan
Computation and Language
Predictive models often need to work with incomplete information in real-world tasks. Consequently, they must provide reliable probability or confidence estimation, especially in large-scale decision-making and planning tasks. Current large language models (LLMs) are insufficient for accurate estimations, but they can generate relevant factors that may affect the probabilities, produce coarse-grained probabilities when the information is more complete, and help determine which factors are relevant to specific downstream contexts. In this paper, we make use of these capabilities of LLMs to provide a significantly more accurate probabilistic estimation. We propose BIRD, a novel probabilistic inference framework that aligns a Bayesian network with LLM abductions and then estimates more accurate probabilities in a deduction step. We show BIRD provides reliable probability estimations that are 30% better than those provided directly by LLM baselines. These estimates further contribute to better and more trustworthy decision making.
title BIRD: A Trustworthy Bayesian Inference Framework for Large Language Models
topic Computation and Language
url https://arxiv.org/abs/2404.12494