arXiv:2409.16797cs.LGcs.AI2024-09被引 2

无需OOD样本即可实现大规模模型集成多样性,提升分布外泛化与检测能力。

Scalable Ensemble Diversification for OOD Generalization and Detection

  • 通过动态识别难样本促使模型间分歧,避免繁琐的成对比较计算。
  • 在ImageNet上验证,多样性强的集成显著提升分布外泛化和检测性能。
  • 适用于大模型场景,适合追求鲁棒性与不确定估计的开发者使用。

训练多样化集成模型在多个实际场景中具有应用价值,例如提升分布外(OOD)泛化性能以供模型选择,或基于贝叶斯原理检测OOD样本。现有方法通过鼓励模型在给定的OOD样本上产生分歧来实现多样性,但该方法计算开销大,且依赖良好的ID与OOD样本分离,仅在小规模设置中验证过。本文提出可扩展的集成多样化方法(SED),适用于大规模场景(如ImageNet),且无需OOS样本。SED在训练过程中动态识别难样本,并促使集成成员在这些样本上产生分歧。为提升可扩展性,我们展示了如何避免现有方法中耗时的全对模型分歧计算。实验在ImageNet上验证了多样化带来的收益:一方面,在输出空间(经典)集成和权重空间集成(model soups)中均观察到显著的分布外泛化提升;另一方面,将集成假设的多样性转化为新型不确定性评分器,在众多基线方法中表现更优。代码已公开:https://github.com/AlexanderRubinstein/diverse-universe-public。

原文摘要 · Abstract (English)

Training a diverse ensemble of models has several practical applications such as providing candidates for model selection with better out-of-distribution (OOD) generalization, and enabling the detection of OOD samples via Bayesian principles. An existing approach to diverse ensemble training encourages the models to disagree on provided OOD samples. However, the approach is computationally expensive and it requires well-separated ID and OOD examples, such that it has only been demonstrated in small-scale settings. $\textbf{Method.}$ This work presents a method for Scalable Ensemble Diversification (SED) applicable to large-scale settings (e.g. ImageNet) that does not require OOD samples. Instead, SED identifies hard training samples on the fly and encourages the ensemble members to disagree on these. To improve scaling, we show how to avoid the expensive computations in existing methods of exhaustive pairwise disagreements across models. $\textbf{Results.}$ We evaluate the benefits of diversification with experiments on ImageNet. First, for OOD generalization, we observe large benefits from the diversification in multiple settings including output-space (classical) ensembles and weight-space ensembles (model soups). Second, for OOD detection, we turn the diversity of ensemble hypotheses into a novel uncertainty score estimator that surpasses a large number of OOD detection baselines. Code is available here: https://github.com/AlexanderRubinstein/diverse-universe-public.

集成学习OOD检测模型多样性可扩展性

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