PRISM: Promoting Sampling Diversity of Large Language Models with Implicit Stochastic Modulation

Qixin Xu, Zihan Zhu, Xiaojiang Liu, Bohan Lyu


Abstract
Large language models (LLMs) exhibit mode collapse, where their output distribution concentrates on familiar semantic modes. When diversity disappears, repeated sampling becomes semantic repetition rather than discovery, making rare but stronger solutions unlikely to surface and leaving self-evolution and continual learning difficult to sustain. Most remedies intervene only at inference time, shaping samples without making diversity a model capability. We propose PRISM, a training-time framework that internalizes diversity within the model. PRISM combines (i) submodular curation that pairs each prompt with diverse-yet-correct multi-teacher solutions, and (ii) stochastic embedding modulation, which turns special-token perturbations into implicit signals for sampling distinct outputs. A theoretical analysis establishes the stability of this mechanism. On WritingPrompts and BookMIA, PRISM moves the quality-diversity frontier outward, achieving the best generation quality in all model-benchmark settings while maintaining the strongest or near-strongest semantic diversity across two open-weight model families. It also improves sampling efficiency on math and code benchmarks. Beyond inference, PRISM yields a stronger initialization for reinforcement learning, sustaining diverse rollouts and mitigating the diversity collapse commonly observed during RL fine-tuning.
Anthology ID:
2026.findings-emnlp.748
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
15748–15772
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.748/
DOI:
Bibkey:
Cite (ACL):
Qixin Xu, Zihan Zhu, Xiaojiang Liu, and Bohan Lyu. 2026. PRISM: Promoting Sampling Diversity of Large Language Models with Implicit Stochastic Modulation. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 15748–15772, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
PRISM: Promoting Sampling Diversity of Large Language Models with Implicit Stochastic Modulation (Xu et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.748.pdf