4 个月前

OFA:通过简单的序列到序列学习框架统一架构、任务和模态

OFA:通过简单的序列到序列学习框架统一架构、任务和模态

摘要

在这项工作中,我们追求一种统一的多模态预训练范式,以打破复杂任务/模态特定定制的框架。我们提出了OFA,这是一种任务无关和模态无关的框架,支持任务全面性。OFA在一个简单的序列到序列学习框架中统一了多种跨模态和单模态任务,包括图像生成、视觉定位、图像描述、图像分类、语言建模等。在预训练和微调阶段,OFA均采用基于指令的学习方法,无需为下游任务添加额外的任务特定层。与依赖于极其大规模跨模态数据集的最新视觉与语言模型相比,OFA仅在2000万公开可用的图像-文本对上进行预训练。尽管其简单且训练数据规模相对较小,OFA在一系列跨模态任务中取得了新的最先进水平(SOTA),同时在单模态任务上也达到了极具竞争力的表现。我们的进一步分析表明,OFA还能有效迁移到未见过的任务和未见过的领域。我们的代码和模型已公开发布于https://github.com/OFA-Sys/OFA。

代码仓库

JHKim-snu/GVCCI
pytorch
GitHub 中提及
ofa-sys/ofa
官方
pytorch
GitHub 中提及
JHKim-snu/PGA
pytorch
GitHub 中提及

基准测试

基准方法指标
image-captioning-on-coco-captionsOFA
BLEU-4: 44.9
CIDER: 154.9
METEOR: 32.5
SPICE: 26.6
object-categorization-on-gritOFA_Large
Categorization (ablation): 22.6
self-supervised-image-classification-on-1OFA (Large)
Number of Params: 473M
Top 1 Accuracy: 85.6%
text-summarization-on-gigawordOFA
ROUGE-1: 39.81
ROUGE-2: 20.66
ROUGE-L: 37.11
visual-entailment-on-snli-ve-testOFA
Accuracy: 91.2
visual-entailment-on-snli-ve-valOFA
Accuracy: 91.0
visual-question-answering-on-grit-1OFA
VQA (ablation): 72.4
visual-question-answering-on-vqa-v2-test-dev-1OFA
Accuracy: 82.0
visual-question-answering-on-vqa-v2-test-std-1OFA
number: 71.44
other: 73.35
overall: 81.98
yes/no: 94.66

用 AI 构建 AI

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

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

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
OFA:通过简单的序列到序列学习框架统一架构、任务和模态 | 论文 | HyperAI超神经