Risk-conditioned Fine-tuning of Large Language Models

Zixuan Liu, Fangzheng WU, Brian Summa, Zizhan Zheng


Abstract
Large Language Models (LLMs) are increasingly deployed in settings where rare but severe harmful generations can have significant consequences. Existing Risk-Averse RLHF addresses this issue by optimizing Conditional Value-at-Risk (CVaR), but it trains policies for fixed risk levels and therefore cannot adjust the desired degree of risk aversion at inference time. In this paper, we propose *risk-conditioned RLHF*, a framework that trains a single policy that provides a continuous risk-control interface, enabling users to select different degrees of risk aversion without retraining or deploying multiple risk-specific models. Experiments across multiple benchmarks demonstrate that a single risk-conditioned policy can adapt to different risk levels at inference time, enabling more flexible and risk-aware LLM deployment. This paper contains example data that may be offensive or harmful.
Anthology ID:
2026.emnlp-main.956
Volume:
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
EMNLP
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
22036–22076
Language:
URL:
https://aclanthology.org/2026.emnlp-main.956/
DOI:
Bibkey:
Cite (ACL):
Zixuan Liu, Fangzheng WU, Brian Summa, and Zizhan Zheng. 2026. Risk-conditioned Fine-tuning of Large Language Models. In Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing, pages 22036–22076, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
Risk-conditioned Fine-tuning of Large Language Models (Liu et al., EMNLP 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.emnlp-main.956.pdf