Eliahu's picture
Add model card
800ef6d verified
metadata
base_model: facebook/dino-vitb16
library_name: transformers
pipeline_tag: image-classification
tags:
  - probex
  - model-j
  - weight-space-learning

Model-J: DINO Model (model_idx_0005)

This model is part of the Model-J dataset, introduced in:

Learning on Model Weights using Tree Experts (CVPR 2025) by Eliahu Horwitz*, Bar Cavia*, Jonathan Kahana*, Yedid Hoshen

🌐 Project | 📃 Paper | 💻 GitHub | 🤗 Dataset

ProbeX

Model Details

Attribute Value
Subset DINO
Split val
Base Model facebook/dino-vitb16
Dataset CIFAR100 (50 classes)

Training Hyperparameters

Parameter Value
Learning Rate 3e-05
LR Scheduler constant_with_warmup
Epochs 4
Max Train Steps 1332
Batch Size 64
Weight Decay 0.03
Seed 5
Random Crop True
Random Flip True

Performance

Metric Value
Train Accuracy 0.9859
Val Accuracy 0.9117
Test Accuracy 0.9002

Training Categories

The model was fine-tuned on the following 50 CIFAR100 classes:

lawn_mower, fox, ray, pickup_truck, lamp, man, leopard, motorcycle, bus, house, bottle, dinosaur, snake, poppy, streetcar, plain, rocket, willow_tree, pine_tree, lizard, forest, porcupine, whale, otter, shrew, skyscraper, clock, apple, couch, crocodile, tank, sunflower, caterpillar, mountain, flatfish, telephone, shark, rabbit, hamster, pear, lobster, rose, possum, lion, worm, castle, bee, snail, skunk, wolf