Supervised Machine Learning Methods with Uncertainty Quantification for Exoplanet Atmospheric Retrievals from Transmission Spectroscopy
2508.04982v1
astro-ph.EP, astro-ph.IM, cs.LG, physics.data-an
2025-08-09
Авторы:
Roy T. Forestano, Konstantin T. Matchev, Katia Matcheva, Eyup B. Unlu
Резюме на русском
**Резюме**
Поиск и изучение экзопланет становится возможным благодаря развитию технологий, таких как JWST, которые позволяют анализировать транзитные спектры экзопланет. Однако стандартные байесовские методы для восстановления параметров экзопланетных атмосфер требуют больших вычислительных ресурсов. В статье предлагается исследовать возможность замены этих методов машинным обучением, которое обеспечивает эффективность без потери точности. Авторы проанализировали различные методы регрессии машинного обучения, включая PLS, SVM, KNN, DT, RF, VOTE, STACK и XGB, и оценили их точность, прецизию и скорость на реальных данных. Также исследовались различные методы предобработки данных и их влияние на модели. Наконец, авторы провели студию случая на примере WASP-39b, подтвердив высокую эффективность выбранной модели. Эти результаты открывают путь к более быстрому и точному восстановлению параметров экзопланетных атмосфер с меньшими вычислительными затратами.
Abstract
Standard Bayesian retrievals for exoplanet atmospheric parameters from
transmission spectroscopy, while well understood and widely used, are generally
computationally expensive. In the era of the JWST and other upcoming
observatories, machine learning approaches have emerged as viable alternatives
that are both efficient and robust. In this paper we present a systematic study
of several existing machine learning regression techniques and compare their
performance for retrieving exoplanet atmospheric parameters from transmission
spectra. We benchmark the performance of the different algorithms on the
accuracy, precision, and speed. The regression methods tested here include
partial least squares (PLS), support vector machines (SVM), k nearest neighbors
(KNN), decision trees (DT), random forests (RF), voting (VOTE), stacking
(STACK), and extreme gradient boosting (XGB). We also investigate the impact of
different preprocessing methods of the training data on the model performance.
We quantify the model uncertainties across the entire dynamical range of
planetary parameters. The best performing combination of ML model and
preprocessing scheme is validated on a the case study of JWST observation of
WASP-39b.