Views
No views yet
1import torch
2from PIL import Image
3from transformers import AutoModel, AutoImageProcessor
4
5# Load model and processor
6model_id = "tiiuae/siglino-moe-0.3-0.6B"
7model = AutoModel.from_pretrained(model_id, trust_remote_code=True).to("cuda", dtype=torch.bfloat16)
8processor = AutoImageProcessor.from_pretrained(model_id, trust_remote_code=True)
9
10# Preprocess image
11image = Image.open("image.jpg").convert("RGB")
12inputs = processor(image, return_tensors="pt").to("cuda")
13inputs["pixel_values"] = inputs["pixel_values"].to(torch.bfloat16)
14
15# Inference
16with torch.no_grad():
17 outputs = model(**inputs)
18
19# Access specialized features
20# Options: 'siglino' (768d), 'siglip2' (1152d), 'dinov3' (1024d)
21patch_features = outputs["patch_features"]["siglino"] # (Batch, Tokens, 768)
22summary_features = outputs["summary_features"]["siglip2"] # (Batch, 1152)| Attribute | Value |
|---|---|
| Architecture | MoE (top-6/28) |
| Parameters (active) | 0.3B |
| Parameters (total) | 0.6B |
| Layers | 18 |
| Hidden Dim | 768 |
| MoE Dim | 384 |
| Patch Size | 16x16 |
| Teachers | DINOv3, SigLIP2 |
| Task | Metric | Score |
|---|---|---|
| kNN (ImageNet) | Acc | 85.9 |
| kNN (6-dataset avg) | Acc | 90.5 |
| Zero-shot cls (ImageNet) | Acc | 79.9 |
| Flickr30K I2T | R@1 | 94.6 |
| MSCOCO I2T | R@1 | 70.8 |
| Pascal VOC (1024) | mIoU | 88.9 |
| Cityscapes (1024) | mIoU | 65.4 |
1@article{chaybouti2025amoe,
2 title={AMoE: Agglomerative Mixture-of-Experts Vision Foundation Models},
3 author={Chaybouti, Sofian and Narayan, Sanath and Dahou, Yasser and Le Khac, Phuc H. and Singh, Ankit and Huynh, Ngoc Dung and Para, Wamiq Reyaz and Kuehne, Hilde and Hacid, Hakim},
4 journal={arXiv preprint arXiv:2512.20157},
5 year={2025}
6}