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

Fast and Accurate Gamma-Ray/Hadronic Particle Shower Classification Using Machine Learning

This research applies dimensionality reduction techniques, such as Pearson Correlation and Principal Component Analysis, to machine learning models like Random Forests and Support Vector Classifiers to simplify the prediction of atmospheric gamma-ray particle showers, achieving a modest accuracy increase while reducing the number of features used.
Leonardo V.
Mentored by Pablo Bonilla
An Investigation Into Applications of Machine Learning Algorithms on Solar Flare Data and Distance Prediction

The current method that NASA uses for flare prediction involves studying various solar cycles that range from 11 days to 80 years. However, there are far too many factors to consider using this method of prediction and the forecasts are often wrong.
Isaac A.
Mentored by Aidan Donaghey
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