用投票机制统一上百个预训练模型的知识,提升无监督物体识别性能
UNIFORM: Unifying Knowledge from Large-scale and Diverse Pre-trained Models
- 设计双层投票机制,在logit和特征层面融合多模型共识
- 在无监督物体识别上超越强基线,且支持超百个教师模型
- 无需假设数据分布或网络结构,适合异构模型知识迁移
深度学习时代,海量在线预训练模型蕴含丰富知识。这些模型架构各异、训练数据不同,对世界有独特理解,其集体共识可能具有普遍性与泛化能力。然而,模型异质性使有效整合知识成为根本挑战。现有方法依赖训练数据分布或网络结构的强假设,仅能从特定类型模型学习,导致数据与归纳偏置。本文提出UNIFORM框架,无需此类限制,即可将多样化现成模型的知识迁移到一个学生模型中。我们设计专用投票机制:在logit层面融合可预测目标类别的教师模型输出;在特征层面利用任意标签空间学习的视觉表示。大量实验表明,相较于强基线,UNIFORM显著提升无监督物体识别性能。尤其具备卓越扩展性,可从超过一百个教师模型获益,而现有方法在较小规模即饱和。
原文摘要 · Abstract (English)
In the era of deep learning, the increasing number of pre-trained models available online presents a wealth of knowledge. These models, developed with diverse architectures and trained on varied datasets for different tasks, provide unique interpretations of the real world. Their collective consensus is likely universal and generalizable to unseen data. However, effectively harnessing this collective knowledge poses a fundamental challenge due to the heterogeneity of pre-trained models. Existing knowledge integration solutions typically rely on strong assumptions about training data distributions and network architectures, limiting them to learning only from specific types of models and resulting in data and/or inductive biases. In this work, we introduce a novel framework, namely UNIFORM, for knowledge transfer from a diverse set of off-the-shelf models into one student model without such constraints. Specifically, we propose a dedicated voting mechanism to capture the consensus of knowledge both at the logit level -- incorporating teacher models that are capable of predicting target classes of interest -- and at the feature level, utilizing visual representations learned on arbitrary label spaces. Extensive experiments demonstrate that UNIFORM effectively enhances unsupervised object recognition performance compared to strong knowledge transfer baselines. Notably, it exhibits remarkable scalability by benefiting from over one hundred teachers, while existing methods saturate at a much smaller scale.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。