atlas

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

Cross-Care: Assessing the Healthcare Implications of Pre-training Data on Language Model Bias

NeurIPS 2024 · Large language models › Language, safety and interpretability · approach: LLM unlearning and benchmarks · 18 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 Cross-Care, a benchmark that compares LLM logit-based disease-demographic associations with The Pile co-occurrence counts and real US prevalence data across four languages.

Problem

LLMs may encode demographic biases from pre-training data about disease prevalence, and it is unclear whether their representations are grounded in real-world epidemiological prevalence across demographic subgroups and languages.

Approach

Measure model logits for ten templated statements linking diseases and demographic subgroups, then compare the resulting rankings via Kendall's tau against co-occurrence counts in The Pile and CDC NHIS real-world prevalence, across models, alignment stages, and languages.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use model logits to rank demographic disease prevalenceModels in the wild · Real-world disease prevalence, gender and race/ethnicityKendall's tau: Min = -0.73, Max = 0.33, Median = -0.05, Avg = -0.06, Var = 0.03Hurt
Increase Pythia or Mamba model sizePythia and Mamba · The Pile subgroup rankings across 89 diseasesAgreement of top demographic ranks with co-occurrenceHurt
Compare Pythia and Mamba logits to The Pile co-occurrence rankingsPythia and Mamba · Real-world disease prevalence, gender and race/ethnicityKendall's tauMixed
Increase disease co-occurrence frequency quartilePythia and Mamba · The Pile disease co-occurrence quartilesLogit correlation with co-occurrenceNo clear effect
Change the co-occurrence token window among 50, 100, and 250 tokens (ablation)Cross-Care co-occurrence rankings · Disease-demographic pairsTop disease rank: 50, 100, and 250No clear effect
Apply alignment methods or in-domain continued pre-trainingLlama2 70b series · Real-world disease prevalenceAccuracy of demographic prevalence representationNo clear effect
Apply alignment methods to Llama2 70bLlama2 70b · English gender and race/ethnicity rankingsSubgroup preferenceMixed
Apply SFT or DPO alignmentLlama2 70b · Race/ethnicity subgroup rankingsKendall's tau rank-order stability: δ>=0.8No clear effect
Use RLHF chat tuning or medical continued pre-trainingLlama2 70b · Race/ethnicity subgroup rankingsRank-order stabilityMixed
Use PPO+DPO chat alignment instead of the base modelQwen1.5-7b · English race/ethnicity subgroup preferenceTop subgroup preferenceMixed
Evaluate Mistral and Llama models in Chinese rather than FrenchMistral and Llama series · Gender subgroup preferenceTop subgroup preferenceMixed
Evaluate Qwen1.5 in different languagesQwen1.5 · Gender subgroup preferenceTop subgroup preferenceMixed

Surprising findings

Limitations

Experimental setup

Datasets: The Pile RedPajama CDC National Health Interview Survey

Benchmarks: Cross-Care

Metrics: Kendall's tau average logit top-rank agreement

Baselines: Infini-gram

Builds on: The Pile Pythia Mamba HolisticBias Infini-gram

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