EpiCaR: Knowing What You Don't Know Matters for Better Reasoning in LLMs

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Yeom, Jewon, Sok, Jaewon, Park, Seonghyeon, Park, Jeongjae, Kim, Taesup
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911366342770688
author Yeom, Jewon
Sok, Jaewon
Park, Seonghyeon
Park, Jeongjae
Kim, Taesup
author_facet Yeom, Jewon
Sok, Jaewon
Park, Seonghyeon
Park, Jeongjae
Kim, Taesup
contents Improving the reasoning abilities of large language models (LLMs) has largely relied on iterative self-training with model-generated data. While effective at boosting accuracy, existing approaches primarily reinforce successful reasoning paths, incurring a substantial calibration cost: models become overconfident and lose the ability to represent uncertainty. This failure has been characterized as a form of model collapse in alignment, where predictive distributions degenerate toward low-variance point estimates. We address this issue by reframing reasoning training as an epistemic learning problem, in which models must learn not only how to reason, but also when their reasoning should be trusted. We propose epistemically-calibrated reasoning (EpiCaR) as a training objective that jointly optimizes reasoning performance and calibration, and instantiate it within an iterative supervised fine-tuning framework using explicit self-evaluation signals. Experiments on Llama-3 and Qwen-3 families demonstrate that our approach achieves Pareto-superiority over standard baselines in both accuracy and calibration, particularly in models with sufficient reasoning capacity (e.g., 3B+). This framework generalizes effectively to OOD mathematical reasoning (GSM8K) and code generation (MBPP). Ultimately, our approach enables a 3X reduction in inference compute, matching the K=30 performance of STaR with only K=10 samples in capable models.
format Preprint
id arxiv_https___arxiv_org_abs_2601_06786
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle EpiCaR: Knowing What You Don't Know Matters for Better Reasoning in LLMs
Yeom, Jewon
Sok, Jaewon
Park, Seonghyeon
Park, Jeongjae
Kim, Taesup
Computation and Language
Improving the reasoning abilities of large language models (LLMs) has largely relied on iterative self-training with model-generated data. While effective at boosting accuracy, existing approaches primarily reinforce successful reasoning paths, incurring a substantial calibration cost: models become overconfident and lose the ability to represent uncertainty. This failure has been characterized as a form of model collapse in alignment, where predictive distributions degenerate toward low-variance point estimates. We address this issue by reframing reasoning training as an epistemic learning problem, in which models must learn not only how to reason, but also when their reasoning should be trusted. We propose epistemically-calibrated reasoning (EpiCaR) as a training objective that jointly optimizes reasoning performance and calibration, and instantiate it within an iterative supervised fine-tuning framework using explicit self-evaluation signals. Experiments on Llama-3 and Qwen-3 families demonstrate that our approach achieves Pareto-superiority over standard baselines in both accuracy and calibration, particularly in models with sufficient reasoning capacity (e.g., 3B+). This framework generalizes effectively to OOD mathematical reasoning (GSM8K) and code generation (MBPP). Ultimately, our approach enables a 3X reduction in inference compute, matching the K=30 performance of STaR with only K=10 samples in capable models.
title EpiCaR: Knowing What You Don't Know Matters for Better Reasoning in LLMs
topic Computation and Language
url https://arxiv.org/abs/2601.06786