Back To Projects

Using Machine Learning to Classify Stars, Quasars, and Galaxies

Chinmay R.

In this project, we looked at data from stars, quasars, and galaxies from the sixteenth data release of the Sloan Digital Sky Survey Telescope. The aim of the project was to accurately and quickly classify these three types of objects using machine learning. We used three machine learning algorithms, namely logistic regression, multi-layer perceptron, and decision tree classifier.


In this project, we looked at data from stars, quasars, and galaxies from the sixteenth data release of the Sloan Digital Sky Survey Telescope. The aim of the project was to accurately and quickly classify these three types of objects using machine learning. We used three machine learning algorithms, namely logistic regression, multi-layer perceptron, and decision tree classifier. The classification was done based on measurements of the object’s redshift and its u,g,r,i, and z light emissions. Logistic regression offered the least accurate results, with an overall accuracy of 94.83%, and a runtime of 28.2 seconds. Better in both aspects was the decision tree classifier, with an overall accuracy of 98.90%, and a runtime of 6.36 seconds. Several structures with different kinds of neuron-layer arrangements were tried for the MLP classifier. While all yielded around a 98% total accuracy, the more complex a structure was, or the more nodes and layers it had, the more time it took to run, and none even came close to the low amount of time taken by the decision tree classifier.

Explore More!

Chinmay R.
Amanda Wang
MS in Computer Science, Computer Science and Business Analytics MIT Alum

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
Gamma Ray Classification

This research paper is in this domain of physics, specifically using Artificial intelligence (AI) to classify high-energy Gamma particles as background or signal. I was able to create an AI model to classify these gamma particles with high accuracy. I was able to accomplish this by using a wide range of data, from the MAGIC Gamma Telescope Dataset, which consists of key features, to construct a model.
Eesha S.
Mentored by Victoria Lloyd
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