Hugging Face's logo

ai-forever
/
RUDOLPH-2.7B-FBC2

PyTorch
RUDOLPH
text-image
image-text
decoder
Model card Files Community
RUDOLPH-2.7B-FBC2 / README.md
sberbank-ai
Update README.md
56f0ba6
preview code
|
raw
history blame
4.45 kB
metadata
tags:
  - RUDOLPH
  - text-image
  - image-text
  - decoder
datasets:
  - sberquad

RUDOLPH-2.7B-FBC2 (XL)

RUDOLPH: One Hyper-Tasking Transformer Can be Creative as DALL-E and GPT-3 and Smart as CLIP

This is a fine-tuned version of the pre-trained RUDOLPH 2.7B model. Model was trained by Sber AI and AIRI teams.

Model Description

RUssian Decoder On Language Picture Hyper-tasking (RUDOLPH) 2.7B is the largest text-image-text transformer designed for an easy fine-tuning for a range of tasks: from generating images by text description and image classification to visual question answering and more. This model demonstrates the power of Hyper-tasking Transformers.

Hyper-tasking model is a generalized multi-tasking model, i.e., the model that can solve almost all tasks within supported modalities, mandatory including mutual pairwise translations between modalities (two modalities in case of RUDOLPH: images and Russian texts).

  • Tasks: text2image generation, self reranking, text ranking, image ranking, image2text generation, zero-shot image classification, text2text generation, text qa, math qa, image captioning, image generation, text recognition in the wild, visual qa, and so on
  • Language: Russian
  • Type: decoder
  • Num Parameters: 2.7B
  • Training Data Volume: 119 million text-image pairs, 60 million text paragraphs
  • Fine-tuning Data Volume: 43 334 text question-answer pairs, 100 000 math tasks, 85 000 text-image pairs (for captioning, generation), 85 759 visual question-answer pairs, 140 000 image-text pairs for text recognition

The model was prepared as a baseline for FusionBrain Challenge 2.0 (as a part of AI Journey Contest 2022) and is a fine-tuned version of the pre-trained RuDOLPH 2.7B model using 6 tasks:

  • Text QA: on SberQUaD dataset.
  • Math QA: on DeepMind Mathematics Dataset.
  • Image Captioning: on COCO dataset translated into Russian (MT).
  • Image Generation: on COCO dataset translated into Russian (MT).
  • VQA: on COCO dataset with prepared question set.
  • Text Recognition in the Wild: on START dataset (SynThesized and Annotated dataset for Text Recognition) consisting of synthetic and real-world human-annotated data for text recognition task.

Details of architecture

The maximum sequence length that this model may be used with depends on the modality and stands for 384 - 576 - 128 for the left text tokens, image tokens, and right text tokens, respectively.

RUDOLPH 2.7B is a Transformer-based decoder model with the following parameters:

  • num_layers (32) — Number of hidden layers in the Transformer decoder.
  • hidden_size (2560) — Dimensionality of the hidden layers.
  • num_attention_heads (32) — Number of attention heads for each attention layer.

Sparse Attention Masks

The primary proposed method is to modify the sparse transformer's attention mask to better control modalities. It allows us to calculate the transitions of modalities in both directions, unlike another similar work DALL-E Transformer, which used only one direction, "text to image". The proposed "image to right text" direction is achieved by extension sparse attention mask to the right for auto-repressively text generation with both image and left text condition.

Authors