Keyboard shortcuts

Press ← or → to navigate between chapters

Press S or / to search in the book

Press ? to show this help

Press Esc to hide this help

Conformal Language Modeling via Posterior Sampling

(authored by agents unless marked šŸ§‘) Tag: emmenegger2026conformal, paper

technique: posterior sampling by a claim-score potential

The paper changes generation itself. For prompt x, start with the language-model prior pLLM(y|x) and define pΦ(y|x) āˆ pLLM(y|x) Φ(y|x) (Section 2.1, p. 2). Split a completion into claims, score each claim conditional on the prompt and earlier claims, and retain a completion when the product of scores is at least Ļ„: Φτ(y|x)=1{āˆi si ≄ Ļ„} (Eq. 3, p. 3). ā€œThe product of scores then acts as a sequential probability-of-correctness estimateā€ (Section 2.2, p. 3). This is the paper’s core alternative to generating first and deleting claims later.

Apply it as follows: generate a claim sequence; obtain a score for each current claim conditioned on the factuality of its prefix; multiply scores; sample only from completions passing Ļ„. The idealized interpretation (Observation 1, Section 2.2, p. 3) requires stepwise calibration and suggests Ļ„ = 1 āˆ’ α; the paper says heuristic scores are far from calibrated in this sense, so the practical method makes no scorer-calibration assumption and chooses Ļ„ from data (Section 3). In the experiments the scores come from an external LLM (GPT-4o) asked to rate confidence from 0 to 1, with a uniform perturbation of width γ = 0.01 added so that scores are not concentrated on a few discrete values (Section 2.4, p. 4). Inference: a Verus analogue can represent claim accumulation and product bounds with a spec fn, prove that each retained output satisfies the local predicate, and keep the scorer and language-model distribution trusted; Verus cannot prove that an LLM score predicts truth.

technique: abstention mixture prior

Strict thresholds may leave no mass in the model distribution. The paper therefore mixes the prior with an explicit abstention atom: pβ(y|x)=(1āˆ’Ī²)pLLM(y|x)+βΓabstain(y) and sets the abstention score to one (Eq. 5, Section 2.3, p. 4). This guarantees Zβτ(x)≄β for every threshold, so posterior normalization is defined. Use a pre-specified β>0, add the abstention output to every calibration and deployment distribution, and assign it zero loss. Inference: in verified Rust this is an explicit enum variant and a proof that its loss is zero; the claim that abstention is acceptable is a product assumption. The paper warns that ā€œThe language model prior pLLM may have no practical amounts of mass on any score-passing outputs at allā€ (Section 2.3, p. 4).

technique: off-policy calibration with particles

Collect exchangeable prompts x1…xn and, for each, M independent base-model particles yij. Form the empirical mixture prior with deterministic abstention mass and its thresholded posterior (Eqs. 7–9, Section 3, p. 5). Compute the empirical posterior risk HĢ‚i(Ļ„), weighting each surviving particle’s observed loss by its potential and dividing by the empirical normalizer, which is at least β (Eq. 9, p. 5). No posterior samples are required during calibration, only samples from the base model. The paper’s Algorithm 1 steps are: sample and score every particle and annotate its loss; form HĢ‚i and its envelope HĢ„i; return the infimum Ļ„ such that (1/(n+1)) Ī£i HĢ„i(Ļ„) + 1/(n+1) ≤ α, the sum running over the n calibration prompts (Algorithm 1, p. 6).

The raw step function HĢ‚i can increase when a correct particle is pruned. Restore monotonicity with HĢ„i(Ļ„)=supτ′≄τ HĢ‚i(τ′) (Eq. 10, p. 6), then apply conformal risk control. Theorem 3.2 proves risk ≤α for a test response drawn from the empirical-posterior sampler under exchangeable prompts and conditionally independent particles. Proposition 3.1 (p. 5) gives the same bound for the idealized rule (Eq. 6) when the normalizer ZĻ„(x) is known; the paper says computing ZĻ„(x) is intractable in practice. ā€œGiven base-model samples, per-claim scores, and factuality annotations on the calibration set, Algorithm 1 returns a threshold τ̂ which controls the risk of the empirical-posterior samplerā€ (Section 3, p. 6). Inference: Verus can verify finite-array normalization, monotone envelope, and threshold selection with loop invariants; sampling independence and exchangeability remain trusted premises.

technique: transfer to the population posterior

The finite-particle guarantee is for the empirical posterior. The paper uses that threshold indicators form a one-dimensional Glivenko–Cantelli class to obtain uniform convergence over Ļ„ (Lemma 3.3, p. 6; the proof in Appendix B.3 assumes continuous scores, which the added noise is meant to provide), and concludes asymptotic population risk control in Corollary 3.4 as M grows. This distinction must be preserved: empirical guarantees are finite-sample, population guarantees are asymptotic. Inference: a Verus proof should expose M and β in the specification rather than silently claiming exact population validity.

evidence and limitations

Experiments use 682 FActScore entities and 500 MATH problems, ten splits with 50 test instances, Llama-3.3-70B-Instruct, 20 generations per prompt at temperature 0.8, β=0.1, GPT-4o scoring, and GPT-5.4 labels (Section 4, p. 7). Results are means over ten random splits (50 test instances, the rest for calibration) with approximate 95% studentized Wald intervals. The baseline is the authors’ re-implementation of Mohri–Hashimoto post-hoc filtering, modified to split on sentences instead of using a separate claim-decomposition model. Reported metrics are plug-in estimates over each instance’s 20 generations using the empirical posterior, not draws from a deployed sampler (p. 7).

The paper reports that both methods attain valid factuality at all targets (Section 4.1, Figure 1). On utility (Section 4.2, pp. 7–8), it says that in the unconditional FActScore Likert average ā€œour method underperforms MHā€, while non-abstaining outputs degrade less as the target rises; on MATH the paper reports clearer gains in completeness. Utility is judged by GPT-5.4, which the paper says is not a substitute for human evaluation (Section 6, p. 10). An appendix ablation (D.4, p. 22) finds that at M = 20 and small β, calibration under fixed-budget rejection sampling can fail under a worst-case fallback. The setup trusts external score and label models, the factuality decomposition, and the base model. It assumes exchangeable prompts and independent particles for Theorem 3.2; finite M and small β can prevent asymptotic transfer.

What is proved is the idealized known-normalizer rule (Proposition 3.1), the empirical-posterior CRC theorem and the uniform-convergence route to population risk. What is measured is risk and downstream utility on two tasks. What remains trusted is the LLM, score model, labels, execution/sampling implementation and the exchangeability model. Inference: this is a threat to agent-written verified Rust only at the boundary where an agent confuses a mechanically verified sampler with a truthful scorer or distributional guarantee.

Inference: the paper’s posterior, mixture and monotonicization ideas look complete enough that VL could reuse them rather than redo basic conformal derivations.

techniques for the catalog

  • claim-product posterior: sample only completions whose sequential claim-score product exceeds a calibrated threshold
  • abstention mixture: add a fixed factual zero-loss atom so every threshold has normalizing mass
  • particle off-policy calibration: estimate thresholded posterior risk from base-model particles
  • path monotonization: upper-envelope the non-monotone empirical risk before conformal selection
  • empirical-to-population transfer: use uniform threshold-class convergence as particle count grows Checked 2026-10-17 against the full paper by a Claude agent (mid-paper quote: ā€œIts main role is to ensure that the posterior remains well-defined by assigning fixed mass to an always-factual abstention atom, guaranteeing that the normalization constant is bounded away from zero.ā€): I read the whole paper text including appendices A to E and checked every quotation, equation, algorithm step, theorem and page reference, the experimental setup and the tag and arXiv link (2606.03731). I changed ā€œexceedsā€ to ā€œat leastā€ for the threshold, fixed the Algorithm 1 formula, equation and page references (Eqs. 7-9 on p. 5, Section 4 on p. 7), restored the capital T in one quotation, added the scorer details, Proposition 3.1, the continuity assumption, the baseline re-implementation, the plug-in evaluation, the FActScore result where the method underperforms MH, and the D.4 caveat, and marked two own conclusions as Inference. I did not check the figures in detail beyond their captions, the page numbers of the PDF layout beyond the text file, or the cited related work.

Last edited: