| --- |
| 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) |
