AMoE: Agglomerative Mixture-of-Experts Vision Foundation Model

Sofian Chaybouti, Sanath Narayan, Yasser Dahou, Phúc H. Lê Khac, Ankit Singh +4 more
12/23/2025
cs.CV

No confident code implementation found for this paper.

Abstract

Vision foundation models trained via multi-teacher distillation offer a promising path toward unified visual representations, yet the learning dynamics and data efficiency of such approaches remain underexplored. In this paper, we systematically study multi-teacher distillation for vision foundation models and identify key factors that enable training at lower computational cost. We introduce Agglomerative Mixture-of-Experts Vision Foundation Models (AMoE), which distill knowledge from SigLIP2 and DINOv3 simultaneously into a Mixture-of-Experts student. We show that (1) our Asymmetric Relation-Knowledge Distillation loss preserves the geometric properties of each teacher while enabling effective knowledge transfer, (2) token-balanced batching that packs varying-resolution images into sequences with uniform token budgets stabilizes representation learning across resolutions without sacrificing performance, and (3) hierarchical clustering and sampling of training data--typically reserved for self-supervised learning--substantially improves sample efficiency over random sampling for multi-teacher distillation. By combining these findings, we curate OpenLVD200M, a 200M-image corpus that demonstrates superior efficiency for multi-teacher distillation. Instantiated in a Mixture-of-Experts. We release OpenLVD200M and distilled models.

View on arXivView PDF

Code Implementations

No confident code match yet

We couldn't find an author-owned or strongly-evidenced community implementation for this paper. Any repos shown below are weak matches — verify before relying on them.

No code implementations found yet.

Know of an implementation? Let us know in the comments below!

Cite this paper

@article{chaybouti2025amoe,
  title  = {AMoE: Agglomerative Mixture-of-Experts Vision Foundation Model},
  author = {Sofian Chaybouti and Sanath Narayan and Yasser Dahou and Phúc H. Lê Khac and Ankit Singh and Ngoc Dung Huynh and Wamiq Reyaz Para and Hilde Kuehne and Hakim Hacid},
  year   = {2025},
  eprint = {2512.20157},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2512.20157v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2512.20157.svg)](https://codeofpaper.com/paper/2512.20157)

Discussion