3 个月前

Prismer:一种具备多任务专家的视觉-语言模型

Prismer:一种具备多任务专家的视觉-语言模型

摘要

近期的视觉-语言模型展现出令人瞩目的多模态生成能力。然而,这类模型通常需要在海量数据上训练庞大的参数量模型。作为一种更具可扩展性的替代方案,我们提出了Prismer——一种数据与参数高效的视觉-语言模型,其核心思想是采用任务特定专家的集成架构。Prismer仅需训练少量组件,模型的大部分网络权重直接继承自多个现成可用的预训练专家模型,并在训练过程中保持冻结状态。通过整合来自多种领域的专家知识,我们证明Prismer能够高效汇聚并适配这些专业知识,以应对多样化的视觉-语言推理任务。实验结果表明,Prismer在微调和少样本学习场景下的性能可与当前最先进的模型相媲美,同时所需训练数据量最多可减少两个数量级。代码已开源,地址为:https://github.com/NVlabs/prismer。

代码仓库

nvlabs/prismer
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
image-captioning-on-coco-captionsPrismer
BLEU-4: 40.4
CIDER: 136.5
METEOR: 31.4
SPICE: 24.4
image-captioning-on-nocaps-entirePrismer
B1: 84.87
B2: 69.99
B3: 52.48
B4: 33.66
CIDEr: 110.84
METEOR: 31.13
ROUGE-L: 60.55
SPICE: 14.91
image-captioning-on-nocaps-valPrismer
CIDEr: 107.9
SPICE: 14.8
visual-question-answering-on-vqa-v2-test-devPrismer
Accuracy: 78.43
visual-question-answering-on-vqa-v2-test-stdPrismer
number: 61.39
other: 69.70
overall: 78.49
yes/no: 93.09

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供