Reproduce “Conformal Language Modeling via Posterior Sampling” on OpenResearch

Turn your favorite coding agents into research agents that review literature, develop hypotheses, and run experiments.

Universal app for Apple silicon and Intel. Requires macOS 11 or later.

About this paper

Nicolas Emmenegger, Theo X. Olausson, Armando Solar-Lezama, Chara Podimata

Emmenegger et al. from MIT introduce a method for controlling hallucination in Large Language Models by integrating factuality guarantees directly into the generative sampling process. This approach reweights the LLM's distribution to favor more reliable and useful responses, achieving validated risk control and higher downstream utility, particularly in tasks with strong inter-claim dependencies like mathematical reasoning and biography generation.

Key insights

  • Risk control can be fundamentally integrated into an LLM's generative process by reweighting its intrinsic distribution towards statistically more reliable and practically more useful completions, moving beyond post-generation correction.
  • Calibrating the sampling distribution by conditioning on high-confidence output regions allows for effective statistical risk control while preserving the coherence and utility of generated text.
  • An off-policy calibration procedure with a monotonized objective provides finite-sample risk control guarantees for the empirical posterior, and asymptotically converges to the true population posterior, making the method robust to the specific heuristic scorer's accuracy.

Summary by alphaXiv.

Tip: change arxiv.org to autoarxiv.org in any arXiv abstract link to open the paper here.