Skip to main content

Alex Baker · Research Documentation

Interpretable machine learning
for observational astrophysics

Three open-source research projects at the intersection of stellar spectroscopy, galaxy morphology, and Bayesian cosmology.

350k+spectra processed
99selected features
89.2%AstroSpectro balanced acc.*
83.9%AstroVision balanced acc. (DINOv2 fine-tuned)

* AstroSpectro: XGBoost v6 (349,998 spectra) includes 4 Gaia/photometric features, so its metrics are not purely spectroscopic; see the Data Leakage Audit.

⭐ Key Scientific Finding · AstroSpectro · SHAP Interpretability · v6 · G2 · 349,998 spectra

"SHAP analysis of the XGBoost classifier (350k spectra, ROC-AUC 0.986) reveals that Mg b equivalent width ranks first globally, Ca II K fourth — metallicity-sensitive features dominate over classical Balmer temperature indicators in the learned classification structure. 96.8% of feature importance derives from physically interpretable spectroscopic descriptors."

Caveat: v6 includes 4 Gaia/photometric features (M_BP, M_G, M_RP, ag_gspphot), which hold 3.2% of the total importance — see the Data Leakage Audit.

Read the full SHAP analysis →