Skip to content

This project uses machine learning to detect brain tumors early. Different models are trained on metabolite data to classify healthy and carcinoma-afflicted individuals. The top-performing EvoHDTree model is enhanced with the ADASYN up-sampling algorithm to accurately distinguish between malignant and non-malignant tumors.

License

Notifications You must be signed in to change notification settings

mudit2004/Enhancing-Machine-Learning-Models-for-Early-Brain-Tumor-Diagnosis

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

3 Commits
 
 
 
 

Repository files navigation

Project Title: ANALYSING AND IMPROVISING OF MACHINE LEARNING MODELS FOR EARLY DIAGNOSIS OF BRAIN TUMOUR

Author

Mudit Golchha

Name of the Guide

Dr. Brindha GR, Assistant Professor-II, School of Computing , SASTRA University

Abstract

Brain tumors, classified broadly as glioma (malignant) and meningeal tumor (non-malignant), present a grave threat by damaging and compressing vital parts of the brain. Traditional diagnostic methods often prove time-consuming, impeding prompt treatment. This project endeavors to expedite tumor detection using machine learning methodologies. Leveraging a dataset featuring metabolite profiles specific to glioma (categorized by grade) and meningeal tumors, ten diverse machine learning models are trained and rigorously tested. These models include Naive Bayes (NB), Generalized Linear Model (GLM), Logistic Regression (LR), Fast Large Margin (FLM), Deep Learning (DL), Decision Tree (DT), Random Forest (RF), Gradient Boosted Trees (GBT), Support Vector Machine (SVM), and Evolutionary Heterogeneous Decision Tree (EvoHDTree). Through rigorous evaluation based on performance metrics such as Area Under the Curve (AUC), Accuracy (ACC), and F1-score, these models are compared for their efficacy in discriminating between healthy individuals and those afflicted with carcinoma. Furthermore, the top-performing EvoHDTree model (with an ACC of 0.9195) undergoes enhancement through the novel application of the ADASYN up-sampling algorithm during preprocessing. This augmentation yields the ultimate model capable of distinguishing between malignant and non-malignant tumors with heightened accuracy.

Specific Contribution

  • Developing machine learning methods other than EvoHDTree
  • Working on GUI.

Specific Learning

  • Learnt the impact of upsampled data on various ML models
  • Designing GUI.

Technical Limitations & Ethical Challenges Faced

  • Uploading the trained model to GUI
  • Overcoming the overfitting of the model.

License

This project is licensed under the BSD-3 License. See the LICENSE file for details.

About

This project uses machine learning to detect brain tumors early. Different models are trained on metabolite data to classify healthy and carcinoma-afflicted individuals. The top-performing EvoHDTree model is enhanced with the ADASYN up-sampling algorithm to accurately distinguish between malignant and non-malignant tumors.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published