DES Vision Transformer Strong Lens Finder
This repository contains the trained Vision Transformer (ViT) models used in
González et al. (2025) (Discovering Strong Gravitational Lenses in the Dark
Energy Survey with Interactive Machine Learning and Crowd-sourced Inspection
with Space Warps).
Models
- Main ViT: trained to identify galaxy–galaxy strong lenses in general
- DSPL ViT: trained to identify double-source-plane lens systems
Framework
- PyTorch
- Vision Transformer architecture
Usage
These models are intended for inference and research use.
See the accompanying paper for training details.
Citation
If you use this model, please cite:
González et al. (2025), ApJ, submitted
