atlas

Research map / Large language models / Language, safety and interpretability

Curated LLM: Synergy of LLMs and Data Curation for tabular augmentation in low-data regimes

ICML 2024 · Large language models › Language, safety and interpretability · approach: LLM unlearning and benchmarks · 1 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 CLLM, which prompts a frozen LLM (GPT-4/GPT-3.5) with dataset context and examples to generate synthetic tabular data for low-data regimes, then curates samples using XGBoost learning dynamics (confidence and aleatoric uncertainty).

Problem

Tabular data augmentation when the labeled training set is very small (n<100), where conventional tabular generators (CTGAN, TVAE, TabDDPM, SMOTE) and LLM fine-tuning struggle to produce diverse, accurate synthetic data.

Approach

Use in-context generation from a frozen black-box LLM with feature descriptions, then filter synthetic samples via learning dynamics (average confidence and aleatoric uncertainty across checkpoints of a classifier trained on the small real set), discarding low-confidence, low-aleatoric samples likely to be mislabeled.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Add curation to GPT-4-generated data (ablation)CLLM on Covid (n=20) · Covid (n=20)AUC: 73.78 -> 73.87Helped
Add curation to GPT-4-generated data (ablation)CLLM on CUTRACT (n=20) · CUTRACT (n=20)AUC: 71.15 -> 72.50Helped
Add curation to GPT-4-generated data (ablation)CLLM on MAGGIC (n=20) · MAGGIC (n=20)AUC: 60.70 -> 61.48Helped
Add curation to GPT-4-generated data (ablation)CLLM on SEER (n=20) · SEER (n=20)AUC: 84.53 -> 84.82Helped
Add curation to GPT-4-generated data (ablation)CLLM on Compas (n=20) · Compas (n=20)AUC: 68.01 -> 67.91Hurt
Add curation to GPT-4-generated data (ablation)CLLM on Adult (n=20) · Adult (n=20)AUC: 50.39 -> 71.48Helped
Add curation to GPT-4-generated data (ablation)CLLM on Drug (n=20) · Drug (n=20)AUC: 75.08 -> 75.29Helped
Include contextual information in the prompt (ablation)GPT-4 generation on Covid · Covid (n=20)utility: 0.66 -> 0.74Helped
Include contextual information in the prompt (ablation)GPT-4 generation on Covid · Covid (n=20)precision: 0.13 -> 0.41Helped
Include contextual information in the prompt (ablation)GPT-4 generation on Covid · Covid (n=20)recall: 0.82 -> 0.87Helped
Increase in-context examples from one to 20 (ablation)GPT-4 generation on Covid · Coviddownstream accuracy: $70.20\pm1.60$ -> $73.87\pm 0.50$Helped
Increase in-context examples from 20 to 100 (ablation)GPT-4 generation on Covid · Coviddownstream accuracy: $73.87\pm 0.50$ -> $74.71\pm0.34$Helped

Surprising findings

Limitations

Experimental setup

Datasets: Adult Income Compas Covid-19 (private) CUTRACT Prostate (private) Drug MAGGIC (private) SEER Prostate (private)

Benchmarks: Covid-19 (private) CUTRACT (private) MAGGIC (private) SEER (private) Compas Adult Drug

Metrics: AUC average rank precision recall

Baselines: CTGAN TVAE TabDDPM NFLOW SMOTE GReaT GPT-3.5

Builds on: GReaT TabDDPM CTGAN SMOTE TabPFN

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