Research map / Large language models / Retrieval and knowledge
Cascaded Language Models for Cost-Effective Human–AI Decision-Making
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 a cascaded LLM decision framework (small base model, large model, human expert) with calibrated confidence-based deferral and uncertainty-based abstention, with online threshold learning from human feedback.
Problem
Balancing answer correctness, computational cost, and abstention to human experts in LLM question answering, with policies that adapt over time rather than staying static.
Approach
A two-stage cascade where a base LLM's answer is verified with calibrated confidence (Bayesian logistic regression over surrogate token probabilities) and deferred to a larger LLM or to a human based on posterior uncertainty; deferral and abstention thresholds are optimised by SGD on a differentiable system-risk loss using feedback from abstentions.
Key results
- On ARC2-Easy with Qwen-2.5 1.5B→7B, calibrated STP (n=1) verified by the large model reaches ΔIBC of 242.7 (Easy) and 89.3 (Challenge), indicating cost-benefit gains over the large model alone.
- Online cumulative regret of the cascade is lower than single-model baselines on four of five datasets (ARC2-Easy, ARC2-Challenge, MMLU, MedMCQA); MedQA shows no clear gain.
- Verifying with the larger model generally gives better cost-benefit than self-verification, while on medical datasets no method consistently helps (ΔIBC near 0 or negative).
Measured effects reported in the paper
| What was changed | Where | Measured | Effect |
|---|---|---|---|
| Use Qwen-2.5-7B rather than Qwen-2.5-1.5B for response verification | Qwen-2.5-1.5B → Qwen-2.5-7B cascade · ARC2 Easy | ΔIBC: -2.4 ± 32.6 -> 258.4 ± 39.1 (SV, n=1); -16.9 ± 33.2 -> 242.7 ± 36.3 (STP, n=1) | Helped |
| Use Qwen-2.5-7B rather than Qwen-2.5-1.5B for response verification | Qwen-2.5-1.5B → Qwen-2.5-7B cascade · ARC2 Challenge | ΔIBC: -11.6 ± 20.5 -> 70.5 ± 26.0 (SV, n=1); -19.3 ± 20.9 -> 89.3 ± 26.2 (STP, n=1) | Helped |
| Use Qwen-2.5-7B rather than Qwen-2.5-1.5B for response verification | Qwen-2.5-1.5B → Qwen-2.5-7B cascade · MMLU | ΔIBC: -20.4 ± 4.5 -> 2.5 ± 5.2 (STP, n=1); -18.7 ± 4.9 -> -7.8 ± 4.7 (SV, n=1) | Mixed |
| Use Qwen-2.5-7B rather than Qwen-2.5-1.5B for response verification | Qwen-2.5-1.5B → Qwen-2.5-7B cascade · MedQA | ΔIBC: -23.9 ± 14.7 -> 1.7 ± 16.3 (STP, n=1) | No clear effect |
| Use Qwen-2.5-7B rather than Qwen-2.5-1.5B for response verification | Qwen-2.5-1.5B → Qwen-2.5-7B cascade · MedMCQA | ΔIBC: -22.9 ± 9.7 -> -10.0 ± 10.1 (STP, n=1) | No clear effect |
| Use STP (n=1) instead of other evaluated verification methods | Qwen-2.5-1.5B → Qwen-2.5-7B cascade with large-model verification · ARC2 Easy and ARC2 Challenge | ΔIBC: 242.7 ± 36.3; 89.3 ± 26.2 | Mixed |
| Increase calibration set size from 100 to 50, 200, or 500 samples (ablation) | STP (n=1) calibration for the cascades · ARC2 Easy, ARC2 Challenge, MedQA, and MedMCQA | ΔIBC: 239.5 ± 36.1; 242.7 ± 36.3; 273.5 ± 42.7; 254.9 ± 43.5 (ARC2 Easy, Qwen, sizes 50, 100, 200, 500) | No clear effect |
| Train deferral and abstention thresholds online using abstention feedback | Qwen-2.5-1.5B → Qwen-2.5-7B cascade · ARC2 Easy, ARC2 Challenge, MMLU, and MedMCQA | cumulative regret | Helped |
| Train deferral and abstention thresholds online using abstention feedback | Qwen-2.5-1.5B → Qwen-2.5-7B cascade · MedQA | cumulative regret | No clear effect |
| Replace gradient-based threshold updates with grid search (ablation) | Online learning for the Qwen-2.5-1.5B → Qwen-2.5-7B cascade · ARC2 Easy | cumulative regret | Hurt |
| Increase label corruption in the calibration set (ablation) | Online Qwen-2.5-1.5B → Qwen-2.5-7B cascade · ARC-Easy | cumulative regret | Hurt |
Surprising findings
- Uncalibrated confidence scores sometimes yield higher ΔIBC than calibrated ones, but with much larger standard errors, suggesting instability.
- On the medical datasets (MedQA, MedMCQA), the cascade shows no consistent cost-benefit benefit over the large model alone.
Limitations
- Cascade can collapse to trivial solutions (always using the cheapest or the most accurate model) depending on cost and abstention weights.
- Relies on human feedback that may be sparse or noisy; imperfect expert labels degrade regret.
- Poor confidence estimation on MedQA limits gains; no theoretical guarantees are provided.
Experimental setup
Datasets: ARC-Easy ARC-Challenge MMLU MedQA MedMCQA
Benchmarks: ARC-Easy ARC-Challenge MMLU MedQA MedMCQA
Metrics: Delta IBC (incremental benefit per cost) cumulative regret (system risk) accuracy vs cost per sample
Baselines: Qwen-2.5-1.5B alone Qwen-2.5-7B alone grid search over thresholds
Builds on: Surrogate token probability Platt scaling Monte Carlo dropout Chain-of-thought prompting
Closest papers by meaning
- Large Language Model Cascades with Mixture of Thought Representations for Cost-Efficient Reasoning (ICLR 2024 · similarity 0.88)
- Online Cascade Learning for Efficient Inference over Streams (ICML 2024 · similarity 0.88)
- C3PO: Optimized Large Language Model Cascades with Probabilistic Cost Constraints for Reasoning (NeurIPS 2025 · similarity 0.86)
- Cost-efficient Knowledge-based Question Answering with Large Language Models (NeurIPS 2024 · similarity 0.86)
- IQA-EVAL: Automatic Evaluation of Human-Model Interactive Question Answering (NeurIPS 2024 · similarity 0.86)
- MA-RLHF: Reinforcement Learning from Human Feedback with Macro Actions (ICLR 2025 · similarity 0.85)
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.
