Prismer: A Vision-Language Model with Multi-Task Experts

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Liu, Shikun, Fan, Linxi, Johns, Edward, Yu, Zhiding, Xiao, Chaowei, Anandkumar, Anima
Formato: Preprint
Publicado: 2023
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911760590569472
author Liu, Shikun
Fan, Linxi
Johns, Edward
Yu, Zhiding
Xiao, Chaowei
Anandkumar, Anima
author_facet Liu, Shikun
Fan, Linxi
Johns, Edward
Yu, Zhiding
Xiao, Chaowei
Anandkumar, Anima
contents Recent vision-language models have shown impressive multi-modal generation capabilities. However, typically they require training huge models on massive datasets. As a more scalable alternative, we introduce Prismer, a data- and parameter-efficient vision-language model that leverages an ensemble of task-specific experts. Prismer only requires training of a small number of components, with the majority of network weights inherited from multiple readily-available, pre-trained experts, and kept frozen during training. By leveraging experts from a wide range of domains, we show Prismer can efficiently pool this expert knowledge and adapt it to various vision-language reasoning tasks. In our experiments, we show that Prismer achieves fine-tuned and few-shot learning performance which is competitive with current state-of-the-arts, whilst requiring up to two orders of magnitude less training data. Code is available at https://github.com/NVlabs/prismer.
format Preprint
id arxiv_https___arxiv_org_abs_2303_02506
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle Prismer: A Vision-Language Model with Multi-Task Experts
Liu, Shikun
Fan, Linxi
Johns, Edward
Yu, Zhiding
Xiao, Chaowei
Anandkumar, Anima
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
Recent vision-language models have shown impressive multi-modal generation capabilities. However, typically they require training huge models on massive datasets. As a more scalable alternative, we introduce Prismer, a data- and parameter-efficient vision-language model that leverages an ensemble of task-specific experts. Prismer only requires training of a small number of components, with the majority of network weights inherited from multiple readily-available, pre-trained experts, and kept frozen during training. By leveraging experts from a wide range of domains, we show Prismer can efficiently pool this expert knowledge and adapt it to various vision-language reasoning tasks. In our experiments, we show that Prismer achieves fine-tuned and few-shot learning performance which is competitive with current state-of-the-arts, whilst requiring up to two orders of magnitude less training data. Code is available at https://github.com/NVlabs/prismer.
title Prismer: A Vision-Language Model with Multi-Task Experts
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2303.02506