AION-1: Astronomical Omnimodal Network

AION-xlarge is a 3B 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 (24 blocks each, 2048 dim, 32 heads)
• Parameters: 3B
• Training: Multimodal Masked Modeling (4M) on astronomical survey data
• Modalities: 39 data types including imaging, spectra, catalogs, and photometry

Quick Start

from aion import AION

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

# 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

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/