Training Language Models To Be Coherent Decision-Makers
cs.AI
Submitted: 2026-10-06
Updated: 2026-10-06
License: http://creativecommons.org/licenses/by/4.0/
The gist: Reliable decision-making requires more than accurate prediction: a model must preserve its beliefs, apply the relevant utilities, and recognize when the information needed to justify an action is
Abstract
Reliable decision-making requires more than accurate prediction: a model must preserve its beliefs, apply the relevant utilities, and recognize when the information needed to justify an action is missing. We study whether language models can learn this decision procedure from supervised fine-tuning and generalize it across domains and differing natural-language expressions of the decision challenge. Across 20 datasets, we explore challenges of belief instability and decision-making errors by first eliciting probabilities of outcomes and then varying only the utilities and the framing of the decision problems, while holding the evidence fixed. We train models to preserve elicited beliefs while selecting the action that maximizes expected utility, and evaluate transfer to unseen application domains, held-out framings, and different classes of payoff structures. We further introduce incomplete-information settings in which required utilities are withheld and replaced with irrelevant text, testing whether models can distinguish missing decision-relevant information from merely additional context. We find that targeted fine-tuning substantially improves coherent decision-making and that in many situations, learning transfers across domains and framings to situations unobserved during training. Further, models trained for decidability learn to identify when action cannot be justified based on missing information. Finally, we show the value of a routed system that considers separately the recognition of decision completeness and utility-sensitive decision execution.
Sources
- Using cognitive psychology to understand GPT-3
- An Expert is Worth One Token: Synergizing Multiple Expert LLMs as Generalist via Expert Token Routing
- Evaluating language models as risk scores
- Cognitive Bias in Decision-Making with LLMs
- Distilling Step-by-Step! Outperforming Larger Language Models with Less Training Data and Smaller Model Sizes
- Language Models (Mostly) Know What They Know
- What Do Learning Dynamics Reveal About Generalization in LLM Reasoning?
- LLMs Get Lost In Multi-Turn Conversation
- QuestBench: Can LLMs ask the right question to acquire information in reasoning tasks?
- Teaching Models to Express Their Uncertainty in Words
- Routing to the Expert: Efficient Reward-guided Ensemble of Large Language Models
- Utility Engineering: Analyzing and Controlling Emergent Value Systems in AIs
- Consistent Estimators for Learning to Defer to an Expert
- RouteLLM: Learning to Route LLMs with Preference Data
- Training language models to follow instructions with human feedback
- Knowing What You Know Is Not Enough: Large Language Model Confidences Don't Align With Their Actions
- Learning Dynamics of LLM Finetuning
- Just Ask for Calibration: Strategies for Eliciting Calibrated Confidence Scores from Language Models Fine-Tuned with Human Feedback
- Finetuned Language Models Are Zero-Shot Learners
- Chain-of-Thought Prompting Elicits Reasoning in Large Language Models
Related papers
- MAVEN-T: Reinforced Heterogeneous Distillation for Real-Time Multi-Agent Trajectory Prediction
- Model Discovery Agent: LLM-assisted Bayesian experiment design for data-efficient discovery of mechanistic world models
- Agents' Last Exam
- The Clinician's Veto: Navigating Trust, Liability, and Uncertainty in Autonomous AI Prescribing
- MindHelper: Closed-Loop Embodied Mental-State Reasoning for Precision Intervention
- Incumbent Advantage: Brand Bias and Cognitive Manipulation Dynamics in LLM Recommendation Systems