|
|
--- |
|
|
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 |
|
|
|
|
|
[](https://opensource.org/licenses/MIT) |
|
|
|
|
|
**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 |
|
|
- 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/ |
