Papers
arxiv:2606.25610

The Galaxy's Guide to the Tokenizer: A Benchmark for Scientific Foundation Models

Published on Jun 24
· Submitted by
Mike Smith
on Jun 29
Authors:
,
,
,
,

Abstract

Four tokenization methods for astronomical images show distinct strengths in reconstruction quality, physical property prediction, and morphological preservation, with no single approach excelling across all tasks.

Tokenization is central to adapting scientific data for transformer-based foundation models, yet its impact on learned representations remains poorly understood. We compare four tokenization strategies, Affine, AIM, JetFormer, and VQ-VAE, within a unified transformer framework for astronomical imaging. Using 640,000 galaxy images from the DESI Legacy Survey and a shared AstroPT backbone, we evaluate each method on reconstruction fidelity and prediction of physical properties. Our results reveal trade-offs across approaches. The flow-based JetFormer achieves higher reconstruction quality, while VQ-VAE yields strong probe performance for galaxy physical properties. Affine and AIM better preserve localized morphological information. We find that reconstruction and representation quality are decoupled, and no single method consistently performs best across the tasks considered here. By grounding our evaluation in independently measured physical quantities, we hope this study serves to highlight the potential of scientific data as a basis for constructing interpretable benchmarks for foundation models.

Community

Paper author Paper submitter

This paper tested four different scientific imagery tokenization schemes (MLP, Linear, VQ-GAN, and Flow-Based) on AstroPT in a controlled setting and found that reconstruction fidelity is not a good predictor of downstream physical performance.

One big takeaway is that we may need more powerful probing techniques to squeeze the most out of our astronomical foundation models. We found that the Flow-Based tokenization scheme has all the information flowing through the network, but that it is inaccessible by either MLP or Linear probes!

This is an automated message from the Librarian Bot. I found the following papers similar to this paper.

The following papers were recommended by the Semantic Scholar API

Please give a thumbs up to this comment if you found it helpful!

If you want recommendations for any Paper on Hugging Face checkout this Space

You can directly ask Librarian Bot for paper recommendations by tagging it in a comment: @librarian-bot recommend

Sign up or log in to comment

Get this paper in your agent:

hf papers read 2606.25610
Don't have the latest CLI?
curl -LsSf https://hf.co/cli/install.sh | bash

Models citing this paper 0

No model linking this paper

Cite arxiv.org/abs/2606.25610 in a model README.md to link it from this page.

Datasets citing this paper 0

No dataset linking this paper

Cite arxiv.org/abs/2606.25610 in a dataset README.md to link it from this page.

Spaces citing this paper 0

No Space linking this paper

Cite arxiv.org/abs/2606.25610 in a Space README.md to link it from this page.

Collections including this paper 0

No Collection including this paper

Add this paper to a collection to link it from this page.