atlas

Research map / Large language models / Retrieval and knowledge

LACIE: Listener-Aware Finetuning for Calibration in Large Language Models

NeurIPS 2024 · Large language models › Retrieval and knowledge · approach: Knowledge-augmented question answering · 2 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes LACIE, a listener-aware DPO finetuning method that trains LLM speakers to express calibrated confidence (explicit and implicit) by optimizing against a simulated listener's acceptance decisions.

Problem

LLMs are overconfident when conveying answers, both through explicit confidence markers and implicit cues (tone, detail, backstories), which misleads human listeners into accepting wrong answers.

Approach

Cast calibration as preference optimization with a two-agent speaker-listener game: sampled speaker responses are judged by a listener model for acceptance and compared against gold correctness, with a preference function that rewards true accepts/rejects and penalizes false accepts, so the speaker learns to induce a calibrated listener distribution.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Apply listener-aware preference finetuningMistral-7B base · TriviaQAAUROC: 0.54\interval{0.00} -> 0.74\interval{0.02}Helped
Apply listener-aware preference finetuningMistral-7B chat · TriviaQAECE: 0.23\interval{0.01} -> 0.09\interval{0.01}Helped
Apply listener-aware preference finetuningLlama3-8B base · TriviaQAPrecision: 0.55\interval{0.02} -> 0.70\interval{0.01}Helped
Apply listener-aware preference finetuningLlama3-70B base · TriviaQAAUROC: 0.53\interval{0.02} -> 0.80\interval{0.02}Helped
Apply listener-aware preference finetuningMistral-7B base · TriviaQA human evaluationFalse Accept: 32 -> 17Helped
Apply listener-aware preference finetuning on TriviaQAMistral-7B · TruthfulQA evaluation splitTruthfulness: 0.27 -> 0.55Helped
Apply listener-aware preference finetuning on TriviaQAMistral-7B · TruthfulQA evaluation splitInformativeness: 0.99 -> 0.90Hurt
Use correctness-only preference finetuning instead of listener-aware preference finetuningMistral-7B base · TriviaQAAUROC: 0.58 -> 0.74Hurt
Increase the number of training questionsLACIE with Mistral-7B listener · TriviaQAPrecisionHelped
Increase the number of training questions toward 10000LACIE with Mistral-7B listener · TriviaQAPrecisionHelped

Surprising findings

Limitations

Experimental setup

Datasets: TriviaQA TruthfulQA

Benchmarks: TriviaQA TruthfulQA

Metrics: AUROC ECE precision recall abstention rate truthfulness informativeness

Baselines: base model truthful-only finetuning chat/instruct models

Builds on: DPO QLoRA Rational Speech Acts

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy