The bibliographic entry was discovered and compiled by a machine, and published without a person checking it against the publication. The publication itself is the authors' own work — this badge says nothing about it.

UniT: Multimodal Multitask Learning with a Unified Transformer

The work

AuthorsRonghang Hu; Amanpreet Singh
Editors
Typepreprint
Year2021
Citekeyhu2021unit

Identifiers

arXiv2102.10772 from its doi
DOI10.48550/arxiv.2102.10772

From arxiv.

Abstract

We propose UniT, a Unified Transformer model to simultaneously learn the most prominent tasks across different domains, ranging from object detection to natural language understanding and multimodal reasoning. Based on the transformer encoder-decoder architecture, our UniT model encodes each input modality with an encoder and makes predictions on each task with a shared decoder over the encoded input representations, followed by task-specific output heads. The entire model is jointly trained end-to-end with losses from each task. Compared to previous efforts on multi-task learning with transformers, we share the same model parameters across all tasks instead of separately fine-tuning task-specific models and handle a much higher variety of tasks across different domains. In our experiments, we learn 7 tasks jointly over 8 datasets, achieving strong performance on each task with significantly fewer parameters. Our code is available in MMF at https://mmf.sh.

How it got here

How it got hereagent via arxiv
Added2026-10-23 14:05 UTC
Not denied bya person 2026-10-29 17:12 UTC

Cite it as

@unpublished{hu2021unit,
  title        = {UniT: Multimodal Multitask Learning with a Unified Transformer},
  author       = {Ronghang Hu and Amanpreet Singh},
  year         = {2021},
  doi          = {10.48550/arxiv.2102.10772},
}

This record lives at https://refs.drheap.org/hu2021unit/ and will keep doing so.