Back To Projects

Molecular Analysis of Stellar Clusters Final Paper

James W.

By testing a multitude of models from the python package scikit-learn upon the APOGEE dataset, we’ve managed to produce a resultant product that can predict a star’s iron concentration depending upon its gravitational pull (LOGG) value or effective temperature (TEFF) value.


WIth the diversity of research methods and measurement instruments found in astronomy, it is often difficult to draw direct comparisons between conclusions due to differences in both kind and quality of measurements. By using a machine learning algorithm we have been able to predict measurements that may not have been gathered, notably , the iron content of a star based upon temperature or gravity readings. The problem has not been answered before with this data set, and the scientific community does not have a large amount of research that has pertained to this specific APOGEE data release. By addressing this previously unaddressed problem, we can advance the scientific community with a new contribution that allows scientists of the future to gain additional data in conditions where such data may not be so readily available. We have approached the question by using various machine learning algorithms to discover correlations between the data points we were given, creating a predictive model to produce a numerical result. Gravity and temperature do not present a direct linear relationship to the iron content of a star. However, as their data points tend to cluster together, we can utilize the KNeighborsRegressor to find correlations in the way that these points have, and produce a numerical prediction. Throughout testing we’ve found that the KNeighborsRegressor does better on all accounts in comparison to a large variety of other regression models such as the GaussianProcessRegressor. Considering the results of the Mean Absolute Percentage Error function upon the results of both of these, we can note how the LinearRegressor, mathematically, does marginally better than the KNeighborsRegressor, at a value of 1.13% of average error in comparison to 2.53%.

Explore More!

James W.
Aidan Donaghey
Physics Yale Alumni, Researcher in Bioinformatics and Astrophysics

Related Projects

Using Machine Learning to identify the Habitability of Exoplanets from TESS Transit Data

With so many new tools and methods to detect exoplanets, it is time to start determining which are deserving of more research in the hunt for other habitable planets.
Eve B.
Mentored by Tomer Arnon
Using Machine Learning Models to Analyze the Aerodynamic Properties of Airfoils

This research posits that leveraging artificial intelligence could significantly reduce financial and computational costs while identifying optimal airfoil geometries.
Aaron W.
Mentored by Ronil Synghal
Using Deep Learning to Predict the Half-Lives of Isotopes Given Proton and Neutron Count

This study explores using deep learning regression models to predict superheavy isotopes' half-lives based on proton and neutron counts, finding that additional complexity in input variables and model architecture may be necessary to improve prediction accuracy.
Akilan P.
Mentored by Erick Ruiz