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 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
workspace_premium
Stellar Classification based on Numerous Characteristics using Machine Learning

The task of stellar classification can be tedious and lengthy when done manually. One can expedite stellar classification by creating an artificial intelligence model to automate the process. The current stellar classification model serves to effectively categorize stars for research purposes regarding their distribution around the universe, so automating the development of this resource would allow professionals to allocate more time to explore the bounds of our current understanding of space and the universe. After finding and analyzing a dataset containing numerical and categorical features, a supervised learning approach was then used to train and test different models on their ability to classify the stars in the given test set. A Decision Tree Classifier, Random Forest Classifier, Ridge Classifier, and Support Vector Classifier were trained and tested using the data.
Roberto T.
Mentored by Sophia Barton
Findings Provided by a Machine Learning Clustering Model of Stellar Kinematics in Star Clusters Can be Used to Identify Runaway Stars

The need for developing a machine learning model that can detect these runaway stars is established in this research as it attempts to expand the domain of astrophysics to help astrophysicists understand tidal tail formation and the theoretical dark matter subhalos’ effects on celestial bodies.
Roneet D.
Mentored by Tony Rodriguez