Scaling Laws for Galaxy Images
Abstract
We present the first systematic investigation of supervised scaling laws outside of an ImageNet-like context - on images of galaxies. We use 840k galaxy images and over 100M annotations by Galaxy Zoo volunteers, comparable in scale to Imagenet-1K. We find that adding annotated galaxy images provides a power law improvement in performance across all architectures and all tasks, while adding trainable parameters is effective only for some (typically more subjectively challenging) tasks. We then compare the downstream performance of finetuned models pretrained on either ImageNet-12k alone vs. additionally pretrained on our galaxy images. We achieve an average relative error rate reduction of 31% across 5 downstream tasks of scientific interest. Our finetuned models are more label-efficient and, unlike their ImageNet-12k-pretrained equivalents, often achieve linear transfer performance equal to that of end-to-end finetuning. We find relatively modest additional downstream benefits from scaling model size, implying that scaling alone is not sufficient to address our domain gap, and suggest that practitioners with qualitatively different images might benefit more from in-domain adaption followed by targeted downstream labelling.
Cite
@article{arxiv.2404.02973,
title = {Scaling Laws for Galaxy Images},
author = {Mike Walmsley and Micah Bowles and Anna M. M. Scaife and Jason Shingirai Makechemu and Alexander J. Gordon and Annette M. N. Ferguson and Robert G. Mann and James Pearson and Jürgen J. Popp and Jo Bovy and Josh Speagle and Hugh Dickinson and Lucy Fortson and Tobias Géron and Sandor Kruk and Chris J. Lintott and Kameswara Mantha and Devina Mohan and David O'Ryan and Inigo V. Slijepevic},
journal= {arXiv preprint arXiv:2404.02973},
year = {2024}
}
Comments
10+6 pages, 12 figures. Appendix C2 based on arxiv:2206.11927. Code, demos, documentation at https://github.com/mwalmsley/zoobot