You can deploy this model and start earning money today!

chaenayo/
id-sim_dinov2_vitb14_cls_patch

Name: id-sim_dinov2_vitb14_cls_patch
Availability: InStock

image-feature-extraction

2604.05039

mit

Views

No views yet

ID-Sim (DINOv2 ViT-B/14, cls_patch)

ID-Sim is a fine-grained identity similarity metric trained to capture the subtle visual differences humans rely on when distinguishing highly similar subjects across varied viewpoints and contexts. This checkpoint uses a DINOv2 ViT-B/14 backbone with LoRA fine-tuning and outputs both CLS and patch-level embeddings (cls_patch).

Paper: ID-Sim: An Identity-Focused Similarity Metric — CVPR 2026
Project page: https://juliachae.github.io/id_sim.github.io/
Code: https://github.com/JuliaChae/id_sim

Usage

DINOv2 weights are public and managed automatically via torch.hub — no manual download needed.

python

import torch
from PIL import Image
from id_sim import id_sim

model, preprocess = id_sim(pretrained=True, model_type="dinov2_vitb14", device="cuda")

img_a = preprocess(Image.open("a.jpg")).cuda()
img_b = preprocess(Image.open("b.jpg")).cuda()

with torch.inference_mode():
    distance = model(img_a, img_b)  # lower = more similar

Install: pip install git+https://github.com/JuliaChae/id_sim.git

Model Details


Base model	DINOv2 ViT-B/14 (`dinov2_vitb14`)
Fine-tuning	LoRA (r=16, α=32, dropout=0.05) + MLP heads (hidden=512)
Feature type	`cls_patch` — joint CLS + Sinkhorn patch loss
License	MIT
Developed by	Julia Chae, Nicholas Kolkin, Jui-Hsien Wang, Richard Zhang, Sara Beery, Cusuh Ham

Training

Trained on ~10k triplets (anchor / positive / negative) spanning ~10k instances across 10 real datasets (landmarks, fashion, objects, art, animals), augmented with generative identity-preserving and identity-altering edits. Key hyperparameters: lr=3e-4, effective batch size 32 (8×4 grad accum), 3 epochs, bf16, InfoNCE + Sinkhorn patch loss.

Evaluation

The model was evaluated across a diverse set of instance-level tasks ranging from concept-preservation evaluation in generative models and perceptual similarity ranking to fine-grained re-identification and retrieval. For full evaluation instructions see the repository; for full evaluation results see the paper.

Citation

bibtex

@InProceedings{Chae_2026_CVPR,
    author    = {Chae, Julia and Kolkin, Nicholas and Wang, Jui-Hsien and Zhang, Richard and Beery, Sara and Ham, Cusuh},
    title     = {ID-Sim: An Identity-Focused Similarity Metric},
    booktitle = {Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)},
    month     = {June},
    year      = {2026},
    pages     = {11250-11262}
}

Framework versions

PEFT 0.16.0

chaenayo/id-sim_dinov2_vitb14_cls_patch