arXiv:2509.23830cs.LGmath.ST2025-09被引 1

让大模型知道自己的不确定,提升可靠性。

Bayesian Mixture-of-Experts: Towards Making LLMs Know What They Don't Know

  • 用贝叶斯框架在路由中引入概率分布,替代固定选专家。
  • 在30亿参数模型上显著改善校准度和外部数据检测能力。
  • 适合关注模型可信度与自我认知的AI安全研究者。

混合专家(MoE)架构使大型语言模型在保持高效的同时规模巨大。然而,标准的确定性路由机制存在严重局限:其固有的脆弱性是导致模型校准偏差和过度自信的关键原因,使得系统往往无法识别自身知识盲区。本文提出一种结构化的贝叶斯MoE路由框架,不再强制单一专家选择,而是对路由决策本身建模为概率分布。我们系统研究了三类在路由流程不同阶段引入此原则性不确定性的方法:权重空间、逻辑值空间和最终选择空间。在30亿参数的MoE模型上进行一系列受控实验表明,该框架显著提升了路由稳定性、分布内校准能力和分布外(OoD)检测性能。结果表明,通过优化这一核心架构组件,可构建更可靠的内部不确定性信号。本工作为打造更鲁棒、具备自知能力的大型语言模型提供了实用且计算可行的路径,迈出让模型真正‘知道自己不知道’的关键一步。

原文摘要 · Abstract (English)

The Mixture-of-Experts (MoE) architecture has enabled the creation of massive yet efficient Large Language Models (LLMs). However, the standard deterministic routing mechanism presents a significant limitation: its inherent brittleness is a key contributor to model miscalibration and overconfidence, resulting in systems that often do not know what they don't know. This thesis confronts this challenge by proposing a structured \textbf{Bayesian MoE routing framework}. Instead of forcing a single, deterministic expert selection, our approach models a probability distribution over the routing decision itself. We systematically investigate three families of methods that introduce this principled uncertainty at different stages of the routing pipeline: in the \textbf{weight-space}, the \textbf{logit-space}, and the final \textbf{selection-space}. Through a series of controlled experiments on a 3-billion parameter MoE model, we demonstrate that this framework significantly improves routing stability, in-distribution calibration, and out-of-distribution (OoD) detection. The results show that by targeting this core architectural component, we can create a more reliable internal uncertainty signal. This work provides a practical and computationally tractable pathway towards building more robust and self-aware LLMs, taking a crucial step towards making them know what they don't know.

大模型不确定性MoE自知

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