kakaobrain/coyo-align-b7-base

Model

0

stars

6

commits

1

linked in READMEs

Nov 11, 2022

updated

align
clip

README

Model Details

This is an unofficial implementation of ALIGN trained on COYO-700M. The official ALIGN is trained on its dataset of 1.8B samples. That dataset is not released to the public. Instead, we trained our implementation of ALIGN model on COYO-700M.

It's developed by Kakao Brain to validate the performance of COYO-700M dataset on a large-scale model.

The training took about 8 days on TPU V3-512.

Model Date

April 2022

Model Type

This is dual encoder model where

  • image encoder is using EfficientNet-B7 architecture
  • text encoder is using BERT-base architecture

Training data

This model is trained on COYO-700M dataset.

Evaluation results

DatasetImageNetFlickr30kMsCOCO
KNNI2T R@1T2I R@1I2T R@1T2I R@1
ALIGN-L2-Large(Google)ALIGN 1.8B76.488.675.758.645.6
ALIGN-B7-Base(Google)ALIGN 1.8B69.3--55.441.7
COYO-ALIGN-B7-Base(Kakao Brain)COYO-700M68.688.173.261.243.1

Contributors

bgyoon

6 commits

kakaobrain/coyo-align-b7-base

Model

0

stars

6

commits

1

linked in READMEs

Nov 11, 2022

updated

align
clip

README

Model Details

This is an unofficial implementation of ALIGN trained on COYO-700M. The official ALIGN is trained on its dataset of 1.8B samples. That dataset is not released to the public. Instead, we trained our implementation of ALIGN model on COYO-700M.

It's developed by Kakao Brain to validate the performance of COYO-700M dataset on a large-scale model.

The training took about 8 days on TPU V3-512.

Model Date

April 2022

Model Type

This is dual encoder model where

  • image encoder is using EfficientNet-B7 architecture
  • text encoder is using BERT-base architecture

Training data

This model is trained on COYO-700M dataset.

Evaluation results

DatasetImageNetFlickr30kMsCOCO
KNNI2T R@1T2I R@1I2T R@1T2I R@1
ALIGN-L2-Large(Google)ALIGN 1.8B76.488.675.758.645.6
ALIGN-B7-Base(Google)ALIGN 1.8B69.3--55.441.7
COYO-ALIGN-B7-Base(Kakao Brain)COYO-700M68.688.173.261.243.1

Contributors

bgyoon

6 commits