| { |
| "version": "1.0.0", |
| "description": "Stellar Foundation Models - Parameter Estimation and Cross-Modal Retrieval", |
| "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 Å)", |
| "architecture": "SpecFormer" |
| } |
| }, |
| "specclip_models": { |
| "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" |
| } |
| }, |
| "parameter_estimation": { |
| "lamost_lrs": { |
| "sbi": { |
| "age": { |
| "path": "downstream_models/lrs/sbi/age.pkl", |
| "parameters": ["age"], |
| "method": "SBI", |
| "stats": null |
| }, |
| "nu_max": { |
| "path": "downstream_models/lrs/sbi/numax.pkl", |
| "parameters": ["nu_max"], |
| "method": "SBI", |
| "stats": null |
| }, |
| "mass": { |
| "path": "downstream_models/lrs/sbi/mass.pkl", |
| "parameters": ["mass"], |
| "method": "SBI", |
| "stats": null |
| }, |
| "rad": { |
| "path": "downstream_models/lrs/sbi/rad.pkl", |
| "parameters": ["rad"], |
| "method": "SBI", |
| "stats": null |
| }, |
| "dnu": { |
| "path": "downstream_models/lrs/sbi/dnu.pkl", |
| "parameters": ["dnu"], |
| "method": "SBI", |
| "stats": null |
| }, |
| "rv": { |
| "path": "downstream_models/lrs/sbi/rv.pkl", |
| "parameters": ["rv"], |
| "method": "SBI", |
| "stats": null |
| }, |
| "dpi1": { |
| "path": "downstream_models/lrs/sbi/DP.pkl", |
| "parameters": ["dpi"], |
| "method": "SBI", |
| "stats": null |
| }, |
| "teff": { |
| "path": "downstream_models/lrs/sbi/teff.pkl", |
| "parameters": ["teff"], |
| "method": "SBI", |
| "stats": null |
| }, |
| "logg": { |
| "path": "downstream_models/lrs/sbi/logg.pkl", |
| "parameters": ["logg"], |
| "method": "SBI", |
| "stats": null |
| } |
| }, |
| "mlp": { |
| "feh": { |
| "path": "downstream_models/lrs/mlp/feh.pt", |
| "parameters": ["fe_h"], |
| "method": "MLP", |
| "stats": "stats/lrs/feh_mean_std.json" |
| }, |
| "ebprp": { |
| "path": "downstream_models/lrs/mlp/ebprp.pt", |
| "parameters": ["e_bp_rp"], |
| "method": "MLP", |
| "stats": "stats/lrs/ebprp_mean_std.json" |
| }, |
| "afe": { |
| "path": "downstream_models/lrs/mlp/afe.pt", |
| "parameters": ["a_fe"], |
| "method": "MLP", |
| "stats": "stats/lrs/afe_mean_std.json" |
| }, |
| "cfe": { |
| "path": "downstream_models/lrs/mlp/cfe.pt", |
| "parameters": ["c_fe"], |
| "method": "MLP", |
| "stats": "stats/lrs/cfe_mean_std.json" |
| }, |
| "nfe": { |
| "path": "downstream_models/lrs/mlp/nfe.pt", |
| "parameters": ["n_fe"], |
| "method": "MLP", |
| "stats": "stats/lrs/nfe_mean_std.json" |
| }, |
| "alfe": { |
| "path": "downstream_models/lrs/mlp/alfe.pt", |
| "parameters": ["al_fe"], |
| "method": "MLP", |
| "stats": "stats/lrs/alfe_mean_std.json" |
| }, |
| "mgfe": { |
| "path": "downstream_models/lrs/mlp/mgfe.pt", |
| "parameters": ["mg_fe"], |
| "method": "MLP", |
| "stats": "stats/lrs/mgfe_mean_std.json" |
| }, |
| "mnfe": { |
| "path": "downstream_models/lrs/mlp/mnfe.pt", |
| "parameters": ["mn_fe"], |
| "method": "MLP", |
| "stats": "stats/lrs/mnfe_mean_std.json" |
| }, |
| "nife": { |
| "path": "downstream_models/lrs/mlp/nife.pt", |
| "parameters": ["ni_fe"], |
| "method": "MLP", |
| "stats": "stats/lrs/nife_mean_std.json" |
| }, |
| "ofe": { |
| "path": "downstream_models/lrs/mlp/ofe.pt", |
| "parameters": ["o_fe"], |
| "method": "MLP", |
| "stats": "stats/lrs/ofe_mean_std.json" |
| }, |
| "sife": { |
| "path": "downstream_models/lrs/mlp/sife.pt", |
| "parameters": ["si_fe"], |
| "method": "MLP", |
| "stats": "stats/lrs/sife_mean_std.json" |
| }, |
| "tife": { |
| "path": "downstream_models/lrs/mlp/tife.pt", |
| "parameters": ["ti_fe"], |
| "method": "MLP", |
| "stats": "stats/lrs/tife_mean_std.json" |
| }, |
| "crfe": { |
| "path": "downstream_models/lrs/mlp/crfe.pt", |
| "parameters": ["cr_fe"], |
| "method": "MLP", |
| "stats": "stats/lrs/crfe_mean_std.json" |
| }, |
| "cafe": { |
| "path": "downstream_models/lrs/mlp/cafe.pt", |
| "parameters": ["ca_fe"], |
| "method": "MLP", |
| "stats": "stats/lrs/cafe_mean_std.json" |
| } |
| } |
| }, |
| "gaia_xp": { |
| "mlp": { |
| "afe": { |
| "path": "downstream_models/xp/mlp/afe.pt", |
| "parameters": ["a_fe"], |
| "method": "MLP", |
| "stats": "stats/xp/afe_mean_std.json" |
| }, |
| "cfe": { |
| "path": "downstream_models/xp/mlp/cfe.pt", |
| "parameters": ["c_fe"], |
| "method": "MLP", |
| "stats": "stats/xp/cfe_mean_std.json" |
| }, |
| "nfe": { |
| "path": "downstream_models/xp/mlp/nfe.pt", |
| "parameters": ["n_fe"], |
| "method": "MLP", |
| "stats": "stats/xp/nfe_mean_std.json" |
| }, |
| "feh": { |
| "path": "downstream_models/xp/mlp/feh.pt", |
| "parameters": ["fe_h"], |
| "method": "MLP", |
| "stats": "stats/xp/FeH_mean_std.json" |
| }, |
| "ebprp": { |
| "path": "downstream_models/xp/mlp/ebprp.pt", |
| "parameters": ["e_bp_rp"], |
| "method": "MLP", |
| "stats": "stats/xp/ebprp_mean_std.json" |
| } |
| } |
| }, |
| "sbi": { |
| "teff": { |
| "path": "downstream_models/xp/sbi/teff.pkl", |
| "parameters": ["teff"], |
| "method": "SBI", |
| "stats": null |
| }, |
| "logg": { |
| "path": "downstream_models/xp/sbi/logg.pkl", |
| "parameters": ["logg"], |
| "method": "SBI", |
| "stats": null |
| } |
| } |
| }, |
| "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": "~500 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 Å", |
| "gaia_xp": "336-1020 nm" |
| }, |
| "preprocessing": { |
| "lamost_lrs": "GASPP continuum normalization + log-wavelength interpolation", |
| "gaia_xp": "Linear interpolation + normalize at 550nm" |
| } |
| } |
