swinv2-base-patch4-window12to16-192to256-22kto1k-ft-galaxy10-decals-head-v2
microsoft/swinv2-base-patch4-window12to16-192to256-22kto1k-ft adapted to galaxy morphology classification on Galaxy10 DECaLS (10 classes) by head-only fine-tuning (backbone frozen, linear head trained), under the revised protocol of the paper When foundation models meet astronomical data (Astronomy and Computing, revised version, 2026). One of the 80 configurations (40 pretrained backbones x two adaptation regimes) of the collection Vision Foundation Models Benchmark.
This repository contains the complete model: the backbone weights are those of the base checkpoint (frozen during training, including the statistics of its normalisation layers) and the classification head is the trained one.
Results (held-out test set, 1,774 images)
| Accuracy (%) | Weighted F1 (%) | |
|---|---|---|
| This checkpoint (seed 42, epoch 29 of 30) | 59.24 | 56.87 |
| Mean +/- s.d. over three seeds (42, 1, 2) | 59.49 +/- 0.47 | 57.16 +/- 0.63 |
Scores are computed at the epoch of minimum validation loss. The per-epoch metrics of all runs, the tables of the paper and the data split are in the dataset vision-foundation-models-benchmark-runs.
Training protocol
- Data: Galaxy10 DECaLS (17,736 images). The public test split (1,774 images) is the held-out test set; the public train split is divided, with stratification by class, into 14,188 training and 1,774 validation images (indices in the dataset repository above). The validation set is used only to select the epoch.
- Recipe (identical for the 40 backbones): AdamW (betas 0.9/0.999, eps 1e-8, weight decay 0.1, gradient clipping 1.0), learning rate 1e-4, linear schedule with 10% warm-up, batches of 32 images with gradient accumulation over 8 steps (effective batch 256), 30 epochs, single precision (fp32, TF32 disabled), one GPU. Augmentation: random resized crop to the input resolution and random horizontal flip. Evaluation: resize and centre crop to 256x256 px, normalisation of the checkpoint's image processor. Seed 42 (seeds 1 and 2 give the mean and s.d. above).
- Regime: head-only fine-tuning (backbone frozen, linear head trained). The head is a single linear layer with ten outputs.
- Selection: epoch of minimum validation loss (epoch 29 here); the weights of this repository are those of that epoch.
- Software: PyTorch 2.3.0, transformers 4.37.2, datasets 2.19.1, torchvision 0.18.0.
Per-epoch metrics (seed 42)
| Epoch | Val. loss | Val. acc. | Test acc. | Test weighted F1 |
|---|---|---|---|---|
| 1 | 2.2610 | 11.89 | 13.30 | 9.50 |
| 2 | 2.0196 | 31.34 | 32.24 | 20.70 |
| 3 | 1.8060 | 37.77 | 41.38 | 34.44 |
| 4 | 1.6680 | 42.78 | 45.94 | 41.14 |
| 5 | 1.5864 | 45.72 | 49.15 | 44.84 |
| 6 | 1.5253 | 48.87 | 51.47 | 47.95 |
| 7 | 1.4849 | 48.82 | 51.24 | 47.56 |
| 8 | 1.4516 | 49.77 | 52.87 | 49.55 |
| 9 | 1.4203 | 51.47 | 54.06 | 50.92 |
| 10 | 1.3960 | 51.86 | 55.13 | 52.14 |
| 11 | 1.3782 | 53.27 | 56.09 | 53.28 |
| 12 | 1.3582 | 53.95 | 56.31 | 53.55 |
| 13 | 1.3458 | 54.17 | 56.03 | 53.25 |
| 14 | 1.3341 | 54.23 | 57.22 | 54.41 |
| 15 | 1.3265 | 53.95 | 56.65 | 54.01 |
| 16 | 1.3133 | 54.51 | 57.61 | 54.93 |
| 17 | 1.3023 | 55.13 | 57.67 | 54.91 |
| 18 | 1.2955 | 55.30 | 57.95 | 55.28 |
| 19 | 1.2892 | 55.58 | 58.57 | 55.87 |
| 20 | 1.2859 | 56.14 | 57.84 | 55.37 |
| 21 | 1.2813 | 56.26 | 58.23 | 55.73 |
| 22 | 1.2745 | 56.20 | 58.17 | 55.45 |
| 23 | 1.2707 | 56.31 | 58.51 | 56.06 |
| 24 | 1.2671 | 56.54 | 58.91 | 56.51 |
| 25 | 1.2638 | 56.65 | 59.19 | 56.85 |
| 26 | 1.2631 | 56.60 | 59.41 | 57.10 |
| 27 | 1.2612 | 56.71 | 59.13 | 56.77 |
| 28 | 1.2606 | 56.76 | 59.24 | 56.86 |
| 29 (selected) | 1.2594 | 56.76 | 59.24 | 56.87 |
| 30 | 1.2594 | 56.71 | 59.19 | 56.81 |
How to use
from transformers import AutoImageProcessor, AutoModelForImageClassification
from torchvision.transforms import Compose, Resize, CenterCrop, ToTensor, Normalize
model = AutoModelForImageClassification.from_pretrained("matthieulel/swinv2-base-patch4-window12to16-192to256-22kto1k-ft-galaxy10-decals-head-v2").eval()
processor = AutoImageProcessor.from_pretrained("matthieulel/swinv2-base-patch4-window12to16-192to256-22kto1k-ft-galaxy10-decals-head-v2")
# Pre-processing used in the paper: resize and centre-crop to 256x256 px, then the processor's normalisation
transform = Compose([Resize(256), CenterCrop((256, 256)), ToTensor(), Normalize(processor.image_mean, processor.image_std)])
# logits = model(pixel_values=transform(image).unsqueeze(0)).logits
Integrity check
The model stored in this repository was reloaded and run on the first 96 test images: predictions identical to the saved logits of the selected epoch for 100.0% of them (maximum absolute logit difference 2.38e-06).
Citation
Paper under revision for Astronomy and Computing; citation to be completed upon publication.