metadata

tags:
  - model_hub_mixin
  - pytorch_model_hub_mixin
extra_gated_fields:
  GitHub username: text
  Affiliation: text
  I want to use this model on the following data:
    type: select
    options:
      - Legacy Survey
      - HSC
      - SDSS
      - DESI
      - Gaia
  I want to use this model for the following science case: text
  I agree to provide feedback on the model and API through GitHub issues during the alpha-testing phase: checkbox
  I agree to use this model for non-commercial use ONLY: checkbox
license: mit
datasets:
  - MultimodalUniverse/legacysurvey
  - MultimodalUniverse/hsc
  - MultimodalUniverse/gaia
  - MultimodalUniverse/sdss
  - MultimodalUniverse/desi

AION-1: Astronomical Omnimodal Network

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

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

pip install polymathic-aion

Resources

GitHub: https://github.com/PolymathicAI/AION
Documentation: https://polymathic-ai.github.io/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/