Model: parlange/cait-gravit-s2
Author: parlange
Pipeline: image-classification
Library: timm
Tags: timm, pytorch, safetensors, vision-transformer, image-classification, cait, gravitational-lensing, strong-lensing, astronomy, astrophysics, dataset:parlange/gravit-c21, arxiv:2509.00226, license:apache-2.0, model-index, region:us
Card metadata: license: apache-2.0
tags: vision-transformer, image-classification, pytorch, timm, cait, gravitational-lensing, strong-lensing, astronomy, astrophysics
datasets: parlange/gravit-c21
metrics: accuracy, auc, f1
paper: title: GraViT: A Gravitational Lens Discovery Toolkit with Vision Transformers
url: https://arxiv.org/abs/2509.00226
authors: Parlange et al.
model-index: name: CaiT-s2
results: task: type: image-classification
name: Strong Gravitational Lens Discovery
dataset: type: common-test-sample
name: Common Test Sample (More et al. 2024)
metrics: type: accuracy
value: 0.8026
name: Average Accuracy
verified: no, type: auc
value: 0.8532
name: Average AUC-ROC
verified: no, type: f1
value: 0.5362
name: Average F1-Score
verified: no
Likes: 0 | All-time downloads: 35
Created: 2025-09-06 22:07:06