arXiv:2605.27747stat.MLcs.LG2026-05

让大模型学会识别不确定性,通过软分工提升推理可靠性

Soft Specialists: $α$-Rényi Ensembles for Uncertainty-Aware LLM Post-Training

论文配图:Soft Specialists: $α$-Rényi Ensembles for Uncertainty-Aware LLM Post-Training
图 1 · 摘自论文原文
  • 用α-Rényi变分框架学习参数分布,替代单一模型参数
  • 在微调中实现多适配器软路由,任务间自动分工并输出置信度
  • 适合需要可解释不确定性的高风险场景如医疗、金融

现有大语言模型训练方法基于海量异构数据学习单一参数集,被迫压缩矛盾目标与内在不确定性。本文提出α-Rényi变分框架,学习后训练参数的分布,提供一种面向不确定性的深度集成替代方案。该框架在经典变分贝叶斯与预测导向后验学习间平衡,兼顾全局合理个体模型与互补专家系统。我们识别局部稳定性条件,证明模型误设时非退化后验扩散更优,使矛盾数据表现为认知不确定性。将该框架应用于大模型后训练,通过共享冻结基模型连接多个LoRA适配器,实现监督微调与偏好优化的可扩展训练。模型可对训练样本进行软路由,促进任务特化,并为不同任务提供可操作的不确定性估计。

原文摘要 · Abstract (English)

Existing training approaches for large language models learn a single set of parameters, based on large volumes of data, which is typically heterogeneous, conflicting and often outright contradictory. As a result, the model is forced to compress conflicting goals, and inherent uncertainties into a single, averaged pattern of behaviour. We propose an $α$-Rényi variational framework for learning distributions over post-training parameters, offering an uncertainty-aware alternative to deep ensemble approaches. The resulting variational objective interpolates between classical variational Bayes and predictively oriented posterior learning, balancing between globally plausible individual models against systems of complementary specialists. We identify local stability criteria, demonstrating how model misspecification can make non-degenerate posterior spread locally favourable, manifesting contradictory or conflicting data as epistemic uncertainty. We apply our framework to LLM post-training, learning an ensemble of LoRA adapters attached to a shared, frozen base model, providing a scalable training procedure for both supervised fine-tuning and preference optimisation. Our approach enables training examples to be softly routed across ensemble members, promoting model specialisation and providing actionable uncertainty estimates across different tasks.

大模型不确定性后训练适配器

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。