Powered by OpenAIRE graph
Found an issue? Give us feedback
image/svg+xml art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos Open Access logo, converted into svg, designed by PLoS. This version with transparent background. http://commons.wikimedia.org/wiki/File:Open_Access_logo_PLoS_white.svg art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos http://www.plos.org/ ZENODOarrow_drop_down
image/svg+xml art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos Open Access logo, converted into svg, designed by PLoS. This version with transparent background. http://commons.wikimedia.org/wiki/File:Open_Access_logo_PLoS_white.svg art designer at PLoS, modified by Wikipedia users Nina, Beao, JakobVoss, and AnonMoos http://www.plos.org/
versions View all 2 versions
addClaim

GraViT: Transfer Learning with Vision Transformers and MLP-Mixer for Strong Gravitational Lens Discovery

Authors: Parlange, René; Cuevas-Tello, Juan C.; Valenzuela, Octavio; Cabrera Rosas, Omar de Jesús; Verdugo, Tomas; More, Anupreeta; Jaelani, Anton Timur;

GraViT: Transfer Learning with Vision Transformers and MLP-Mixer for Strong Gravitational Lens Discovery

Abstract

Gravitational lensing offers a powerful probe into the properties of dark matter and is crucial to infer cosmological parameters.The Legacy Survey of Space and Time (LSST) is predicted to find O(10^5) gravitational lenses over the next decade, demandingautomated classifiers. In this work, we introduce GraViT, a PyTorch pipeline for gravitational lens detection that leveragesextensive pretraining of state-of-the-art Vision Transformer (ViT) models and MLP-Mixer. We assess the impact of transferlearning on classification performance by examining data quality (source and sample size), model architecture (selectionand fine-tuning), training strategies (augmentation, normalization, and optimization), and ensemble predictions. This studyreproduces the experiments in a previous systematic comparison of neural networks and provides insights into the detectabilityof strong gravitational lenses on that common test sample. We fine-tune ten architectures using datasets from HOLISMOKESVI and SuGOHI X, and benchmark them against convolutional baselines, discussing complexity and inference-time analysis.Our publicly available fine-tuned models provide a scalable transfer learning solution for gravitational lens finding in LSST.

The .zip file contains the fine-tuned models for all the experiments in GraViT. There are 12 transfer learning settings and 10 architectures, resulting in 120 models.

Keywords

surveys, software: machine learning, gravitational lensing: strong

  • BIP!
    Impact byBIP!
    selected citations
    These citations are derived from selected sources.
    This is an alternative to the "Influence" indicator, which also reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically).
    0
    popularity
    This indicator reflects the "current" impact/attention (the "hype") of an article in the research community at large, based on the underlying citation network.
    Average
    influence
    This indicator reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically).
    Average
    impulse
    This indicator reflects the initial momentum of an article directly after its publication, based on the underlying citation network.
    Average
Powered by OpenAIRE graph
Found an issue? Give us feedback
selected citations
These citations are derived from selected sources.
This is an alternative to the "Influence" indicator, which also reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically).
BIP!Citations provided by BIP!
popularity
This indicator reflects the "current" impact/attention (the "hype") of an article in the research community at large, based on the underlying citation network.
BIP!Popularity provided by BIP!
influence
This indicator reflects the overall/total impact of an article in the research community at large, based on the underlying citation network (diachronically).
BIP!Influence provided by BIP!
impulse
This indicator reflects the initial momentum of an article directly after its publication, based on the underlying citation network.
BIP!Impulse provided by BIP!
0
Average
Average
Average