arXiv:2601.22068cs.LG2026-01中稿 · ICML

用奇异值调节知识方向,低成本实现大模型不确定性量化

Quantifying the Uncertainty of Foundation Models with Singular Value Ensembles

  • 通过调节权重矩阵的奇异值构建隐式集成,不增加新参数
  • 仅增加<1%参数量,性能媲美显式集成,提升预测校准度
  • 适用于资源受限场景,支持多模态骨干网络

基础模型在大规模预训练后表现出色,但常产生过度自信的预测。传统不确定性量化依赖多个独立训练模型的集成,计算开销随集成规模线性增长,难以应用于大型基础模型。本文提出奇异值集成(SVE),一种参数高效的隐式集成方法。核心假设是权重矩阵的奇异向量对应于表示空间中的有意义“知识方向”。若奇异向量确实代表正交的知识方向,则可通过调节各方向的贡献强度来构建模型集成。SVE冻结奇异向量,仅训练每成员的奇异值以缩放这些共享知识方向的输出。集成多样性在联合训练中自然涌现:随机初始化与批次采样导致不同成员收敛到同一知识基底的不同组合。SVE在多种骨干网络的自然语言处理与视觉任务上验证有效,性能接近显式集成,仅使基础模型参数量增加<1%,使可解释的不确定性估计在资源受限环境中成为可能。

原文摘要 · Abstract (English)

Foundation models have become a dominant paradigm in machine learning, achieving remarkable performance across diverse tasks through large-scale pretraining. However, they often yield overconfident, uncalibrated predictions. The standard approach to quantifying epistemic uncertainty are ensembles of multiple independently trained models. But their computational cost scales linearly with ensemble size, making them impractical for large foundation models. We propose Singular Value Ensemble (SVE), a parameter-efficient implicit ensembling method. SVE builds on a simple, but powerful core assumption: namely, that the singular vectors of the weight matrices correspond to meaningful directions in the representation space. If the singular vectors are indeed meaningful (orthogonal) "knowledge directions", then a model ensemble can be obtained by modulating only how strongly each direction contributes to the output. Rather than learning new parameters for each ensemble member, we freeze the singular vectors and only train per-member singular values that rescale the contribution of each direction in that shared knowledge basis. Ensemble diversity emerges naturally during joint training as stochastic initialization and random batch sampling cause different members to converge to different combinations of the same underlying knowledge. SVE performs comparable to an explicit ensemble, while increasing the parameter count of the base model by <1%, making principled uncertainty estimation accessible in resource-constrained settings. We validate SVE on NLP and vision tasks with various different backbones and show that it improves calibration while maintaining predictive accuracy.

不确定性量化基础模型奇异值集成学习

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