atlas

Research map / Large language models / Retrieval and knowledge

ChatQA: Surpassing GPT-4 on Conversational QA and RAG

NeurIPS 2024 · Large language models › Retrieval and knowledge · approach: RAG robustness and evaluation · 9 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.

Introduces ChatQA, a two-stage instruction-tuned family of models (up to Llama3-70B) plus a fine-tuned conversational retriever and the ChatRAG Bench, matching or beating GPT-4 on conversational QA and RAG.

Problem

Building open models that match proprietary GPT-4 on conversational question answering and retrieval-augmented generation, including follow-up questions, long-context retrieval, tables, arithmetic, and unanswerable questions.

Approach

A two-stage instruction tuning recipe (SFT followed by context-enhanced instruction tuning with conversational and single-turn QA data plus a small set of unanswerable samples), together with a single-turn retriever fine-tuned on multi-turn query-context pairs that matches LLM-based query rewriting at lower cost.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
apply two-stage instruction tuning (ablation)ChatQA-1.0-70B · ChatRAG Benchaverage score: 43.22 -> 54.14Helped
include stage-1 SFT (ablation)ChatQA-1.0-70B · ChatRAG Benchaverage score: 52.18 -> 54.08Helped
include single-turn QA datasets in stage-2 (ablation)ChatQA-1.0-70B · ChatRAG Benchaverage score: 52.25 -> 54.08Helped
include HumanAnnotatedConvQA in stage-2 (ablation)ChatQA-1.0-70B · ChatRAG Benchaverage score: 48.97 -> 54.08Helped
replace HumanAnnotatedConvQA with SyntheticConvQA (ablation)ChatQA-1.0-70B · ChatRAG Benchaverage score: 54.14 -> 54.08No clear effect
fine-tune Dragon on conversational query-context pairs instead of using query rewritingDragon retriever · Five conversational retrieval datasetsaverage top-1 recall: 54.46 -> 52.72; 80.13 -> 80.67Mixed
fine-tune E5-unsupervised on conversational query-context pairsE5-unsupervised · Five conversational retrieval datasetsaverage top-1 recall: 33.23 -> 47.79Helped
use the human-annotated conversational QA data instead of SyntheticConvQA (ablation)ChatQA-1.0-70B · QuAC and DoQA unanswerable evaluationaverage accuracy: 69.84 -> 77.25Helped
use 1.5k unanswerable training samples instead of 1k (ablation)ChatQA-1.0-70B · QuAC and DoQA unanswerable evaluationaverage accuracy: 76.88 -> 77.25Helped
increase unanswerable training samples from 1.5k to 2.5k (ablation)ChatQA-1.0-70B · QuAC and DoQA unanswerable evaluationaverage accuracy: 77.25 -> 75.87Hurt
use top-5 retrieved chunks for stage-2 training context (ablation)ChatQA-1.0-70B · Retrieval datasets vs. non-retrieval datasetsscore: 42.33 -> 42.91; 65.96 -> 65.16Mixed
replace Dragon + Fine-tune with the original Dragon retriever (ablation)ChatQA-1.0-70B · Retrieval QA datasetsaverage score: 42.31 -> 40.50Hurt

Surprising findings

Limitations

Experimental setup

Datasets: Soda ELI5 FLAN chain-of-thought datasets Self-Instruct Unnatural Instructions OpenAssistant Dolly HumanAnnotatedConvQA SyntheticConvQA DROP NarrativeQA

Benchmarks: ChatRAG Bench Natural Questions TriviaQA HotpotQA

Metrics: F1 exact match top-1 recall top-5 recall accuracy on answerable and unanswerable samples human A/B win rate

Baselines: Llama2-Chat Llama3-Instruct Command R+ GPT-3.5-Turbo-0613 GPT-4-0613 GPT-4-Turbo-2024-04-09 E5-unsupervised Dragon

Builds on: Dragon Llama 2 Llama 3 RA-DIT

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