README.md

---
tags:
- model_hub_mixin
- pytorch_model_hub_mixin
license: mit
datasets:
- MultimodalUniverse/legacysurvey
- MultimodalUniverse/hsc
- MultimodalUniverse/gaia
- MultimodalUniverse/sdss
- MultimodalUniverse/desi
---
# AION-1: Astronomical Omnimodal Network

[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
[![GitHub Repo](https://img.shields.io/badge/GitHub-Repo-blue?logo=github)](https://github.com/PolymathicAI/AION)
[![arXiv](https://img.shields.io/badge/arXiv-2510.17960-b31b1b.svg)](https://arxiv.org/abs/2510.17960)
[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/PolymathicAI/AION/blob/main/notebooks/Tutorial.ipynb)

**AION-base** is a 300M parameter large omnimodal model specifically designed for astronomical surveys. It
integrates 39 distinct astronomical data types and enables adaptation to a wide range of astronomical tasks
through multimodal masked modeling.

## Model Details

- **Architecture**: Encoder-Decoder Transformer (12 blocks each, 768 dim, 12 heads)
- **Parameters**: 300M
- **Training**: Multimodal Masked Modeling (4M) on astronomical survey data
- **Modalities**: 39 data types including imaging, spectra, catalogs, and photometry

## Quick Start

```python
from aion import AION

# Load the pretrained model
model = AION.from_pretrained('polymathic-ai/aion-base')

# Your astronomical analysis begins here!
```

## Supported Data Types

AION-Base processes data from major astronomical surveys:

- Imaging: Legacy Survey, HSC Wide
- Spectra: SDSS, DESI
- Catalog: Legacy Survey entries
- Gaia: BP/RP spectra, parallax, coordinates, photometry
- Photometry: Legacy Survey (g,r,i,z + WISE), HSC (g,r,i,z,y)
- Shape: Ellipticity and morphological parameters

## Installation

```bash
pip install polymathic-aion
```


## Resources

- GitHub: https://github.com/PolymathicAI/AION
- Tutorial: https://colab.research.google.com/github/PolymathicAI/AION/blob/main/notebooks/Tutorial.ipynb

## License

MIT License - see https://github.com/PolymathicAI/AION/blob/main/LICENSE for details.

---
Built with ❤ for the astronomical community by https://polymathic-ai.org/