README.md

---
license: apache-2.0
base_model: mwalmsley/zoobot-encoder-convnext_nano
tags:
- astronomy
- galaxy-morphology
- image-classification
- pytorch
- zoobot
- convnext
- fine-tuned
datasets:
- galaxy10-decals
metrics:
- accuracy
pipeline_tag: image-classification
library_name: pytorch
language:
- en
---

# AstroVision Advanced — Galaxy Morphology Classifier

Fine-tuned Zoobot (ConvNeXT Nano) for galaxy morphology classification on the Galaxy10 DECals dataset.

## Model Details

- **Base model:** [mwalmsley/zoobot-encoder-convnext_nano](https://huggingface.co/mwalmsley/zoobot-encoder-convnext_nano)
- **Architecture:** ConvNeXT Nano encoder + LinearHead classifier
- **Training framework:** PyTorch Lightning
- **Training dataset:** Galaxy10 DECals (17,736 images)
- **Epochs:** 10
- **Validation accuracy:** 92.9%

## Classes

| Label | Class |
|---|---|
| 0 | Spiral Galaxy |
| 1 | Elliptical Galaxy |
| 2 | Edge-on Disk |
| 3 | Irregular Galaxy |
| 4 | Merger |

## Results

| Class | Confidence (sample) |
|---|---|
| Spiral Galaxy | 99.89% |
| Elliptical Galaxy | 81.42% |
| Edge-on Disk | 100.00% |
| Irregular Galaxy | 99.17% |
| Merger | 99.99% |

## Training Data

Galaxy10 DECals — 10 original classes remapped to 5:

| Galaxy10 Original | Mapped To |
|---|---|
| Barred Spiral, Unbarred Tight Spiral, Unbarred Loose Spiral | Spiral Galaxy |
| Round Smooth, In-between Smooth, Cigar Shaped Smooth | Elliptical Galaxy |
| Edge-on without Bulge, Edge-on with Bulge | Edge-on Disk |
| Disturbed Galaxies | Irregular Galaxy |
| Merging Galaxies | Merger |

## Known Limitations

The dataset has a class imbalance — Irregular Galaxy has 1,081 samples vs 6,500 for Spiral. The model is less reliable on ambiguous irregular galaxies as a result.

## Links

- **Live App:** [astrovision-advanced.streamlit.app](https://astrovision-advanced.streamlit.app/)
- **GitHub:** [zev-walker/Astrovision-Advanced](https://github.com/zev-walker/Astrovision-Advanced)
- **Zoobot:** [mwalmsley/zoobot](https://github.com/mwalmsley/zoobot)