Trained on Tokens, Calibrated on Concepts: The Emergence of Semantic Calibration in LLMs

· Source: Apple Machine Learning Research · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Advanced, quick

Summary

Large Language Models (LLMs) often struggle with providing reliable confidence estimates for their outputs. This research investigates "semantic calibration," a sampling-based method for LLMs to assess confidence in the meaning of their responses, rather than just next-token prediction. The study finds that base LLMs are surprisingly well-calibrated semantically for open-domain question-answering tasks, even without explicit training for this capability. The authors propose a theoretical mechanism explaining this emergence as a byproduct of next-token prediction, linking it to local loss optimality and introducing "B-calibration" for parameterized equivalence classes. Experimental validation supports three implications: base LLMs exhibit semantic calibration in Q&A, while RL instruction-tuning and chain-of-thought reasoning systematically degrade this calibration.

Key takeaway

For research scientists developing or deploying LLMs, understanding semantic calibration is crucial. Your base LLMs likely possess an inherent ability to assess confidence in their output's meaning, but applying techniques like RL instruction-tuning or chain-of-thought reasoning will systematically diminish this valuable property. You should evaluate the calibration of your models post-tuning to ensure reliable confidence estimates for downstream applications.

Key insights

Base LLMs exhibit emergent semantic calibration in Q&A, which instruction-tuning and chain-of-thought reasoning degrade.

Principles

Method

The research introduces "B-calibration," a general definition of calibration parameterized by equivalence classes, to theoretically explain and experimentally validate semantic calibration in LLMs.

In practice

Topics

Best for: Research Scientist, AI Researcher, AI Scientist, AI Student

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Apple Machine Learning Research.