A Deep Multimodal Multi--Head Neural Network for Joint Estimation of Stellar Age, Lifetime, and Evolutionary Stage

Accurate estimation of stellar parameters -- stellar age, lifetime, and evolutionary stage -- remains a fundamental challenge in astrophysics. We introduce a hybrid deep learning architecture combining multimodal spectroscopic and photometric data from SDSS DR17. The model comprises a Multi-Layer Perceptron for numerical features and a CNN with a Vision Transformer for spectra, with three output heads for age, lifetime, and evolutionary stage prediction. Training labels are derived from MIST v1.2 isochrones, with evolutionary stage binned into five classes (Hot, Medium, Cool, Subgiant, Red Giant). We conduct multi-phase evaluation: Phase I explores model architectures and data balancing strategies, Phase II tunes architectural complexity, and Phase III optimizes the multi-task loss composition. The final model achieves a balance between precision (Age RMSE 0.093 in $\log(\mathrm{yrs})$) and physical realism. Monte Carlo Dropout confirms well-calibrated uncertainties, enabling meaningful astrophysical interpretation and establishing a new benchmark for multimodal stellar parameter estimation.

Paper

References (4)

02iii) Denoising using a Savitzky–Golay filter1964
032022, Hands-on machine learning with Scikit-Learn, Keras, and TensorFlow
04ii) Resampling via linear interpolation onto a common wavelength grid (3,600 ˚A to 10,400 ˚A, 1.0 ˚A step), creating a uniform 6800-point vector

Similar papers

© 2026 NYSGPT2525 LLC