中文

基于域注意力的基础模型精调方法

计算机视觉与模式识别 2024-07-11 v2 人工智能 机器学习

摘要

基础模型(Foundation Models,FMs)已在计算机视觉领域取得突破,使其能够有效地跨越不同域实现学习。然而,这些模型在域迁移情况下的表现仍有待探索。本文通过比较不同骨干网络架构,引入新颖的域注意力组件,利用与域相关的文本嵌入,研究了基础模型在零样态域适应中的潜力。我们提出了一种称为Domino(Domino)的域自适应归一化方法,显式地在精调过程中利用域嵌入,从而使模型具备域注意力。最终,Domino能够实现对各种未见域更具鲁棒性的计算机视觉模型。

关键词

引用

@article{arxiv.2407.03482,
  title  = {Domain-Aware Fine-Tuning of Foundation Models},
  author = {Ugur Ali Kaplan and Margret Keuper and Anna Khoreva and Dan Zhang and Yumeng Li},
  journal= {arXiv preprint arXiv:2407.03482},
  year   = {2024}
}

备注

Accepted at ICML 2024 Workshop on Foundation Models in the Wild