Feb, 2024

通过组合特征对齐增强组合泛化能力

TL;DR在实际的机器学习模型应用中,数据分布的变化常常导致训练数据和测试数据之间存在差异。在常见的多领域多类别设置中,随着类别和领域数量的增加,无法为每个领域 - 类别组合收集足够的训练数据。为了解决这一挑战,我们提出了一个名为 CG-Bench 的测试套件,该测试套件基于现有的真实图像数据集,发现了一种名为 Compositional Feature Alignment (CFA) 的两阶段模型微调技术,它可以促进预训练模型的组合特征学习。实验结果表明,CFA 在组合泛化方面优于常见的微调技术,证实了 CFA 在组合特征学习中的有效性。