AM-RADIO: Agglomerative Vision Foundation Model -- Reduce All Domains Into One

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Ranzinger, Mike, Heinrich, Greg, Kautz, Jan, Molchanov, Pavlo
Format: Preprint
Publié: 2023
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866929443488923648
author Ranzinger, Mike
Heinrich, Greg
Kautz, Jan
Molchanov, Pavlo
author_facet Ranzinger, Mike
Heinrich, Greg
Kautz, Jan
Molchanov, Pavlo
contents A handful of visual foundation models (VFMs) have recently emerged as the backbones for numerous downstream tasks. VFMs like CLIP, DINOv2, SAM are trained with distinct objectives, exhibiting unique characteristics for various downstream tasks. We find that despite their conceptual differences, these models can be effectively merged into a unified model through multi-teacher distillation. We name this approach AM-RADIO (Agglomerative Model -- Reduce All Domains Into One). This integrative approach not only surpasses the performance of individual teacher models but also amalgamates their distinctive features, such as zero-shot vision-language comprehension, detailed pixel-level understanding, and open vocabulary segmentation capabilities. In pursuit of the most hardware-efficient backbone, we evaluated numerous architectures in our multi-teacher distillation pipeline using the same training recipe. This led to the development of a novel architecture (E-RADIO) that exceeds the performance of its predecessors and is at least 7x faster than the teacher models. Our comprehensive benchmarking process covers downstream tasks including ImageNet classification, ADE20k semantic segmentation, COCO object detection and LLaVa-1.5 framework. Code: https://github.com/NVlabs/RADIO
format Preprint
id arxiv_https___arxiv_org_abs_2312_06709
institution arXiv
publishDate 2023
record_format arxiv
spellingShingle AM-RADIO: Agglomerative Vision Foundation Model -- Reduce All Domains Into One
Ranzinger, Mike
Heinrich, Greg
Kautz, Jan
Molchanov, Pavlo
Computer Vision and Pattern Recognition
A handful of visual foundation models (VFMs) have recently emerged as the backbones for numerous downstream tasks. VFMs like CLIP, DINOv2, SAM are trained with distinct objectives, exhibiting unique characteristics for various downstream tasks. We find that despite their conceptual differences, these models can be effectively merged into a unified model through multi-teacher distillation. We name this approach AM-RADIO (Agglomerative Model -- Reduce All Domains Into One). This integrative approach not only surpasses the performance of individual teacher models but also amalgamates their distinctive features, such as zero-shot vision-language comprehension, detailed pixel-level understanding, and open vocabulary segmentation capabilities. In pursuit of the most hardware-efficient backbone, we evaluated numerous architectures in our multi-teacher distillation pipeline using the same training recipe. This led to the development of a novel architecture (E-RADIO) that exceeds the performance of its predecessors and is at least 7x faster than the teacher models. Our comprehensive benchmarking process covers downstream tasks including ImageNet classification, ADE20k semantic segmentation, COCO object detection and LLaVa-1.5 framework. Code: https://github.com/NVlabs/RADIO
title AM-RADIO: Agglomerative Vision Foundation Model -- Reduce All Domains Into One
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2312.06709