atlas

Research map / Large language models / Agents, code and math

Are Large Language Models Good Statisticians?

NeurIPS 2024 · Large language models › Agents, code and math · approach: Mathematical and structural reasoning benchmarks · 5 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 StatQA, a 11,623-example benchmark (plus 1,163-example mini split) for evaluating LLMs on statistical method applicability and relevant-column selection for hypothesis testing and descriptive statistics, with LoRA fine-tuning and a human comparison.

Problem

LLMs' ability to perform statistical analysis, particularly assessing the applicability of statistical methods (prerequisites, data type, normality, sample size) and selecting relevant columns, is under-explored; existing benchmarks focus on computation accuracy.

Approach

Reverse the usual dataset construction: start from target answers (methods and columns) computed on real tables, run automated prerequisite checks, then synthesize questions from hand-crafted templates, refined by GPT-3.5-Turbo, to build a scalable applicability benchmark; also compare LLMs against human statisticians and non-statisticians.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Fine-tune LLaMA-3 8B on the StatQA training setLLaMA-3 8B · mini-StatQAAcc(C, M) overall: 36.11 -> 77.13Helped
Add one example and domain knowledge to the prompt (ablation)GPT-4o · mini-StatQAAcc(C, M) overall: 44.23 -> 64.83Helped
Add one example to the prompt (ablation)LLaMA-3 8B · mini-StatQAAcc(C, M) overall: 23.56 -> 31.90Helped
Add one example and domain knowledge to the prompt (ablation)LLaMA-2 7B · mini-StatQAAcc(C, M) overall: 8.08 -> 16.08Helped
Use chain-of-thought with zero-shot prompting (ablation)LLaMA-2 7B · mini-StatQAAcc(C, M) overall: 8.08 -> 6.36Hurt
Use chain-of-thought with one-shot prompting (ablation)GPT-4o · mini-StatQAAcc(C, M) overall: 49.36 -> 48.67Hurt
Use one-shot prompting instead of zero-shot (ablation)GPT models · mini-StatQAAcc(C, M) overallHelped
Use one-shot prompting instead of zero-shot (ablation)LLaMA-2/3 models · mini-StatQAAcc(C, M) overallHelped
Increase model scale from Qwen2 1.5B Instruct to Qwen2 72B InstructQwen2 Instruct · mini-StatQAAcc(C, M) overall: 20.38 -> 47.12Helped
Provide supplemental applicability information (ablation)Human (Stats) · mini-StatQA subsetAcc(C, M) overall: 23.28 -> 53.45Helped

Surprising findings

Limitations

Experimental setup

Datasets: StatQA training split (D_train, from tables 37-78)

Benchmarks: StatQA mini-StatQA

Metrics: Acc(C,M) accuracy on column and method selection error taxonomy proportions

Baselines: GPT-3.5-Turbo GPT-4 GPT-4o LLaMA-2 7B LLaMA-2 13B LLaMA-3 8B LLaMA-3 8B Instruct

Builds on: LoRA chain-of-thought prompting LLaMA-Factory

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