| { | |
| "version": "1.0.0-no-param", | |
| "description": "SpecCLIP: Aligning and Translating Spectroscopic Measurements for Stars", | |
| "founation_models": { | |
| "encoders": { | |
| "xp_encoder": { | |
| "path": "encoders/xp_encoder.ckpt", | |
| "description": "Gaia XP spectrum encoder (epoch=191)", | |
| "input": "Gaia XP spectra (336-1020 nm)", | |
| "architecture": "SpecFormer" | |
| }, | |
| "xp_encoder_ae_768": { | |
| "path": "encoders/xp_encoder_mlp.ckpt", | |
| "description": "Gaia XP spectrum encoder with autoencoders (epoch=191)", | |
| "input": "Gaia XP spectra (336-1020 nm)", | |
| "architecture": "MLP" | |
| }, | |
| "lrs_encoder": { | |
| "path": "encoders/lrs_encoder.ckpt", | |
| "description": "LAMOST LRS spectrum encoder (epoch=128)", | |
| "input": "LAMOST LRS spectra (4000-5600 \u00c5)", | |
| "architecture": "SpecFormer" | |
| } | |
| }, | |
| "specclip_models": { | |
| "specclip_model_base": { | |
| "path": "specclip/specclip_mlp.ckpt", | |
| "description": "SpecClip cross-modal model with unified embeddings, no decoders, xp encoded with mlp (epoch=110)", | |
| "architecture": "SpecClipModel_mlp" | |
| }, | |
| "specclip_model_predrecon_mlp": { | |
| "path": "specclip/specclip_model_predrecon_mlp.ckpt", | |
| "description": "SpecClip cross-modal model with unified embeddings, both decoders, xp encoded with mlp (epoch=110)", | |
| "architecture": "SpecClipModel_reconstruct_embed768_mlp" | |
| }, | |
| "specclip_model_split_mlp": { | |
| "path": "specclip/specclip_model_split_mlp.ckpt", | |
| "description": "SpecClip cross-modal model with split embeddings, both decoders, xp encoded with mlp (epoch=110)", | |
| "architecture": "SpecClipModel_reconstruct_split_5122562_mlp_recordloss" | |
| } | |
| }, | |
| "retrieval": { | |
| "model_path": "specclip/specclip_model_split_mlp.ckpt", | |
| "description": "Used for cross-modal retrieval and prediction" | |
| } | |
| }, | |
| "test_data": { | |
| "gaia_lamost_split": { | |
| "path": "test_data/gaia_lamost_test_only.h5", | |
| "description": "Test set with paired Gaia XP and LAMOST LRS spectra", | |
| "size": "~10 MB", | |
| "contents": { | |
| "test/source_ids": "Gaia source identifiers", | |
| "test/gaia_spectra": "Gaia XP spectra (N, 343)", | |
| "test/lamost_spectra": "LAMOST LRS spectra (N, 1462)" | |
| }, | |
| "required_for": [ | |
| "retrieval", | |
| "embedding_database" | |
| ], | |
| "optional": true | |
| } | |
| }, | |
| "wavelength_ranges": { | |
| "lamost_lrs": "4000-5600 \u00c5", | |
| "gaia_xp": "336-1020 nm" | |
| }, | |
| "preprocessing": { | |
| "lamost_lrs": "GASPP continuum normalization + log-wavelength interpolation", | |
| "gaia_xp": "Linear interpolation + normalize at 550nm" | |
| } | |
| } | |