arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.04434cs.CVcs.LG

基于焦点多样性剪枝的鲁棒少样本集成学习

Robust Few-Shot Ensemble Learning with Focal Diversity-Based Pruning

Georgia Institute of Technology(佐治亚理工学院)

Selim Furkan Tekin, Fatih Ilhan, Tiansheng Huang, Sihao Hu, Ka-Ho Chow, Margaret L. Loper, Ling Liu

AI总结:

提出 FusionShot 少样本集成方法,通过多通道融合、焦点误差多样性剪枝和可学习组合权重,提升模型在新任务、跨域及对抗场景下的鲁棒性与泛化性。

摘要:

本文提出 FusionShot,这是一种焦点多样性优化的少样本集成学习方法,旨在提升预训练少样本模型的鲁棒性和泛化性能。论文作出了三项原创贡献。首先,我们挖掘少样本学习的独特特征,通过创建三种可选融合通道来集成多个少样本(FS)模型。其次,我们引入焦点误差多样性的概念,以学习最高效的集成组队策略,而不是假定由更多基模型组成的集成会优于规模较小的子集成。我们开发了一种焦点多样性集成剪枝方法,以有效剔除集成误差多样性较低的候选集成,并推荐具有最高焦点误差多样性的 top-K 个 FS 集成。最后,我们通过设计 learn-to-combine 算法来捕获集成少样本预测中的复杂非线性模式,该算法能够学习不同的权重分配,从而在不同成员模型上实现鲁棒的集成融合。在具有代表性的少样本基准上进行的大量实验表明,FusionShot 推荐的 top-K 集成能够在新任务(分布不同且训练时未知)上优于具有代表性的 SOTA 少样本模型,并且在跨域设置和对抗设置中均胜过现有的少样本学习器。为便于复现,FusionShot 训练后的模型、结果和代码可在 https://github.com/sftekin/fusionshot 获取。

英文摘要:

This paper presents FusionShot, a focal diversity optimized few-shot ensemble learning approach for boosting the robustness and generalization performance of pre-trained few-shot models. The paper makes three original contributions. First, we explore the unique characteristics of few-shot learning to ensemble multiple few-shot (FS) models by creating three alternative fusion channels. Second, we introduce the concept of focal error diversity to learn the most efficient ensemble teaming strategy, rather than assuming that an ensemble of a larger number of base models will outperform those sub-ensembles of smaller size. We develop a focal-diversity ensemble pruning method to effectively prune out the candidate ensembles with low ensemble error diversity and recommend top-$K$ FS ensembles with the highest focal error diversity. Finally, we capture the complex non-linear patterns of ensemble few-shot predictions by designing the learn-to-combine algorithm, which can learn the diverse weight assignments for robust ensemble fusion over different member models. Extensive experiments on representative few-shot benchmarks show that the top-K ensembles recommended by FusionShot can outperform the representative SOTA few-shot models on novel tasks (different distributions and unknown at training), and can prevail over existing few-shot learners in both cross-domain settings and adversarial settings. For reproducibility purposes, FusionShot trained models, results, and code are made available at https://github.com/sftekin/fusionshot

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑