Out-of-Distribution Generalization of Risk Aversion in Language Models

📅 2026-07-02
📈 Citations: 0
Influential: 0
📄 PDF
🤖 AI Summary
This study addresses a central challenge in AI alignment: enabling language models to generalize risk-averse behaviors learned in low-stakes settings to out-of-distribution (OOD) scenarios with extremely high stakes. The work presents the first systematic investigation of risk generalization across 98 orders of magnitude in stakes, introducing the RiskAverseOOD benchmark and a multi-model evaluation framework. It evaluates prevalent alignment techniques—including supervised fine-tuning (SFT), direct preference optimization (DPO), tie training, activation steering, and reward model fine-tuning—on models such as Qwen3, Gemma-3, and Llama-3.1. Results demonstrate that these methods substantially improve safety, increasing the proportion of models selecting the safe “cooperate” option from a baseline of 2% to 39%–70% under extreme stakes. Moreover, fine-tuned reward models achieve 99.6% accuracy in discriminating risk-averse reasoning, with consistent performance across diverse model scales and architectures.
📝 Abstract
Training AIs to be risk-averse in resources could offer a failsafe in the event that AIs turn out misaligned. Misaligned but risk-averse AIs would tend to prefer low-risk, low-reward strategies like cooperation over high-risk, high-reward strategies like rebellion, limiting the downsides of any misalignment. But we can only feasibly train AIs to be risk-averse on low-stakes gambles, and we will only be safe if their risk aversion generalizes to astronomically-high-stakes gambles. Will it? To shed light on this question, we introduce RiskAverseOOD: a benchmark for measuring how well risk aversion generalizes out of distribution. We then offer some initial results. Using a variety of methods to make Qwen3-8B choose risk-aversely when the stakes are low, we find that we can induce substantial risk aversion when the stakes are astronomically high. Our models' learned risk aversion generalizes at least partially across 98 orders of magnitude. From a baseline 2% rate of choosing a safe `Cooperate' option, we see rates around 70% (SFT and tie training), 52% (DPO), and 39% (activation steering). In another experiment, our fine-tuned reward model reliably scores risk-averse reasoning above risk-neutral or excessively risk-averse alternatives (99.6% pairwise accuracy). We replicate these effects at different scales (Qwen3-1.7B and Qwen3-14B) and across model families (Gemma-3-12B-IT and Llama-3.1-8B-Instruct). Overall, we find that risk aversion learned at low stakes can generalize OOD to astronomically high stakes, though not yet consistently enough to serve as a reliable failsafe. Achieving that level of consistency is an open problem.
Problem

Research questions and friction points this paper is trying to address.

Out-of-Distribution Generalization
Risk Aversion
Language Models
AI Alignment
Generalization
Innovation

Methods, ideas, or system contributions that make the work stand out.

Out-of-Distribution Generalization
Risk Aversion
Language Model Alignment
Reward Modeling
AI Safety
🔎 Similar Papers
No similar papers found.
K
Kristina Zhang
Massachusetts Institute of Technology, Cambridge, MA, USA
J
Junior Chinomso Okoroafor
Massachusetts Institute of Technology, Cambridge, MA, USA
B
Benjamin Maltbie
Massachusetts Institute of Technology, Cambridge, MA, USA
Andrew Lin
Andrew Lin
Princeton University
CombinatoricsTheoretical Computer Science
A
Abhitej Bokka
Massachusetts Institute of Technology, Cambridge, MA, USA
Elliott Thornley
Elliott Thornley
MIT
AI safetyAI alignmentethicsdecision theory