Extrapolation and explainability in hadron mass predictions with machine learning
Abstract
In this work, we study the predictive performance of four machine learning (ML) models for hadron masses, with particular attention to extrapolation beyond the conventional hadrons represented in the training data. Calculations are performed using a multilayer perceptron (MLP), eXtreme Gradient Boosting (XGBoost), Light Gradient Boosting Machine (LightGBM), and Gaussian process regression (GPR). ML models are trained on conventional hadrons, using features that describe valence quark content and quantum numbers. Then, we evaluate their performance on test sets and on a separate extrapolation set of 39 states, including exotic candidates and heavy baryons, none of which appear in the training or test sets. We find that the MLP achieves the lowest extrapolation Root Mean Square Error (RMSE) on the $30$ states with measured masses at 294.2 ± 57.9~MeV, roughly one-third that of the other models. We also show that the MLP reproduces the overall mass scale of nine unobserved doubly heavy baryons predicted by theoretical calculations, although it underestimates the doubly bottom masses. We use SHapley Additive exPlanations (SHAP) analysis and show that heavy quark content is the most significant factor in the MLP's mass predictions, with the remaining features contributing differently across hadron types. Our results quantify the extrapolation accuracy of ML models for hadron masses and identify the features that drive the MLP predictions.
Related
- Precise QCD Predictions for Photon-Associated Hadron Production at the LHC
- A novel machine learning framework based on proper orthogonal decomposition for turbine wake flow predictions in natural waterways
- Learning Chaos Without Seeing Chaos: Extrapolation of Global Dynamics in Autoregressive Transformers
- Separating charm and bottom decay electrons with point-cloud deep learning on the charged-hadron environment
- Opacity predictions in plasmas under stellar conditions using deep learning
BibTeX
@misc{yuksel2026extrapolation,
title = {{Extrapolation and explainability in hadron mass predictions with machine learning}},
author = {Esra Yuksel and Huseyin Bahtiyar and Derya Soydaner},
howpublished = {arXiv},
year = {2026},
month = nov,
doi = {10.48550/arxiv.2611.02185},
eprint = {2611.02185},
archivePrefix = {arXiv},
url = {https://arxiv.org/abs/2611.02185},
note = {CC BY 4.0},
}
RIS
TY - UNPB
TI - Extrapolation and explainability in hadron mass predictions with machine learning
AU - Esra Yuksel
AU - Huseyin Bahtiyar
AU - Derya Soydaner
PY - 2026
DA - 2026/11/01
PB - arXiv
DO - 10.48550/arxiv.2611.02185
UR - https://arxiv.org/abs/2611.02185
AB - In this work, we study the predictive performance of four machine learning (ML) models for hadron masses, with particular attention to extrapolation beyond the conventional hadrons represented in the training data. Calculations are performed using a multilayer perceptron (MLP), eXtreme Gradient Boosting (XGBoost), Light Gradient Boosting Machine (LightGBM), and Gaussian process regression (GPR). ML models are trained on conventional hadrons, using features that describe valence quark content and quantum numbers. Then, we evaluate their performance on test sets and on a separate extrapolation set of 39 states, including exotic candidates and heavy baryons, none of which appear in the training or test sets. We find that the MLP achieves the lowest extrapolation Root Mean Square Error (RMSE) on the $30$ states with measured masses at 294.2 ± 57.9~MeV, roughly one-third that of the other models. We also show that the MLP reproduces the overall mass scale of nine unobserved doubly heavy baryons predicted by theoretical calculations, although it underestimates the doubly bottom masses. We use SHapley Additive exPlanations (SHAP) analysis and show that heavy quark content is the most significant factor in the MLP's mass predictions, with the remaining features contributing differently across hadron types. Our results quantify the extrapolation accuracy of ML models for hadron masses and identify the features that drive the MLP predictions.
ER -