Skip to content

Survival Prediction of GlioBlastoma Patients using Ensemble architecture of random forest, xgboost and logistic regression classifiers. Uses Optuna for tuning, SMOTE for imbalances, CNN for feature extraction, LDA for feature pre-processing, MPL and Seaborn for visualizations and concordance index as the performance metrics.

License

Notifications You must be signed in to change notification settings

asp616848/Glioblastoma-ML-model

Repository files navigation

Multi-View Learning for Addressing Unbalanced/Partial Data Modalities in Glioblastoma Survival Prediction

Overview

Glioblastoma (GBM) is an aggressive form of brain tumor with a poor prognosis. Accurate survival prediction is crucial for personalized treatment planning. Our study leverages multi-modal learning techniques to handle missing or unbalanced data modalities in Glioblastoma survival prediction.

Motivation

Survival prediction in Glioblastoma patients is challenging due to:

  • Heterogeneous data: Imaging, pathology, molecular, and clinical features vary across patients.
  • Missing modalities: Not all patients have complete data, making traditional deep learning models unreliable.
  • Interpretability issues: Black-box models lack transparency, reducing their adoption in clinical settings.

Approach

We propose a multi-view learning framework that integrates incomplete data modalities to enhance survival prediction. Our methodology includes:

  • Data preprocessing and feature extraction across MRI, histopathology, and clinical data.
  • Multi-modal deep learning models trained on incomplete data.
  • Interpretability techniques for model transparency and clinical trust.

Data Sources

We utilized publicly available datasets from The Cancer Imaging Archive (TCIA):

Methods

Data Preprocessing

  • Converted DICOM to NIfTI for MRI images.
  • Extracted tumor-specific features using radiomics principles.
  • Tiled histopathology images into smaller patches to improve computational efficiency.
  • Performed Exploratory Data Analysis (EDA) on clinical data, handling missing values and feature correlations.

Results & Findings

  • The multi-view model performed better than single-modality models in survival prediction.
  • The interpretability approach helped identify key biomarkers for prognosis.
  • Future work includes expanding datasets and improving generalization.

Research References

Our work is supported by recent research papers:

  1. Glioblastoma: An Update in Pathology, Molecular Mechanisms and Biomarkers
  2. Epidemiology of Glioblastoma Multiforme
  3. Survival Prediction using Machine Learning & Deep Learning
  4. MRI-Based Survival Analysis

About

Survival Prediction of GlioBlastoma Patients using Ensemble architecture of random forest, xgboost and logistic regression classifiers. Uses Optuna for tuning, SMOTE for imbalances, CNN for feature extraction, LDA for feature pre-processing, MPL and Seaborn for visualizations and concordance index as the performance metrics.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 3

  •  
  •  
  •