Machine Learning · Neuroimaging · Computational Neuroscience

Bahram
Jafrasteh, Ph.D.

Developing novel machine learning frameworks for medical neuroimaging — from probabilistic Gaussian processes to deep graph neural networks for MRI and DTI analysis.

Postdoctoral Research Associate — Weill Cornell Medicine, New York Department of Radiology baj4003@med.cornell.edu
Bahram Jafrasteh
15+
Publications
~300
Citations
67+
Peer reviews
25+
Open-source repos
4
Countries
About

My research advances clinical artificial intelligence by developing novel machine learning frameworks to analyze complex scientific data, with a primary focus on medical neuroimaging.

My career represents a logical progression from my Ph.D., where I built foundational ML models for geophysical data, to my current focus on medicine. The core challenge remains the same: extracting meaningful information from noisy, high-dimensional, and often incomplete data.

This transition is exemplified by my work evolving Gaussian Processes — originally used for mineral estimation — into a novel framework (MGP) to solve the critical problem of missing data in clinical neuroimaging.

At Weill Cornell Medicine, I focus on enhancing the reliability and clinical application of ML in medical imaging and healthcare data analysis, including deep graph neural networks for MRI/DTI, harmonization, and modeling trajectories from longitudinal EHR data.

Current Position

Postdoctoral Research Associate
Dept. of Radiology, Weill Cornell Medicine
New York, USA — Jul 2024–Present

Research Focus

Probabilistic ML · Gaussian Processes · Deep Graph Neural Networks · Neonatal Neuroimaging · Longitudinal EHR

Education

Ph.D. Mining Engineering — Isfahan University of Technology, 2018
M.Sc. — Isfahan University of Technology, 2012
B.Sc. — University of Tehran, 2009

Accreditation & Service

ANECA accredited (Spain) · Peer reviewer for Neural Networks, MICCAI, Knowledge-Based Systems, and more

Research Experience
Jul 2024 – Present
Postdoctoral Research Associate
Department of Radiology, Weill Cornell Medicine — New York, USA
Machine intelligence in neuroimaging; deep graph neural networks for MRI/DTI. Multimodal preprocessing, harmonization, and outcome prediction. Deep learning models for longitudinal electronic health record trajectory analysis.
Graph Neural Networks DTI/FC Longitudinal EHR MRI harmonization
Feb 2021 – Jul 2024
Postdoctoral Researcher
INiBICA, Hospital Universitario Puerta del Mar — Cádiz, Spain
Deep learning for neonatal MRI/US segmentation (MELAGE software), missing data imputation, and biomarker extraction. Led preprocessing/registration pipelines; contributed to multicenter projects AUTO-NEUS and PARENT.
Neonatal MRI Segmentation MELAGE AUTO-NEUS PARENT (H2020)
Sep 2020 – Feb 2021
Postdoctoral Researcher
Autonomous University of Madrid (UAM) — Madrid, Spain
Deep and sparse Gaussian Processes for regression and classification; open-source implementations.
Gaussian Processes Variational Inference
Jan 2020 – Aug 2020
Postdoctoral Researcher
INRIA & GeoAzur, Université Côte d'Azur — Sophia Antipolis, France
GAN-based fault and fracture extraction from high-resolution satellite imagery; vectorization algorithms.
GANs Remote Sensing Deep Learning
Feb 2016 – Dec 2016
Research Fellow
Autonomous University of Madrid (UAM) — Madrid, Spain
Machine learning for ore grade estimation using Gaussian Processes, neural networks, and geostatistics.
Sep 2012 – Sep 2018
Ph.D. Researcher
Isfahan University of Technology — Isfahan, Iran
Advanced ML for ore grade estimation. Developed Gaussian process models in C++, warped RBF neural networks, and Bayesian optimization for exploration planning. GPA: 18.68/20.
Skills
Programming
Python PyTorch TensorFlow Scikit-learn OpenCV C++ R MATLAB Bash/Shell SQL Docker
Research Methods
Gaussian Processes Variational Inference Graph Neural Networks Bayesian Optimization GANs Diffusion Models CNNs Missing data imputation
Neuroimaging & Tools
FreeSurfer 3D Slicer ANTs MIRTK FSL GDAL QGIS ArcGIS HPC
Languages
PersianNative
EnglishFluent
SpanishFluent
FrenchBasic
Professional
Scientific Communication Student Mentoring Collaborative Research Peer Review (67+ reviews) ANECA Accredited Grant Participation
Selected Publications
2025
WASABI: A metric for evaluating morphometric plausibility of synthetic brain MRIs MICCAI 2025
MICCAI 2025
Jafrasteh, B., Peng, W., Wan, C., Luo, Y., Adeli, E., & Zhao, Q.
2025
Statistical variability in comparing accuracy of neuroimaging-based classification models via cross validation
Nature Scientific Reports, 15(1), 28745
Jafrasteh, B., et al.
2025
Symmetry-Aware Brain MRI Inpainting Using Denoising Diffusion Models MICAD 2025
MICAD 2025
Santorum, A., Suárez, A., & Jafrasteh, B.
2024
MGA-Net: Mask-guided attention neural network for precision neonatal brain imaging
NeuroImage, 120872
Jafrasteh, B., et al.
2023
Gaussian Processes for Missing Value Imputation
Knowledge-Based Systems, 110603
Jafrasteh, B., Hernández-Lobato, D., et al.
2023
Deep SIFT CNN for total brain volume estimation from 3D ultrasound
Computer Methods and Programs in Biomedicine, 107805
Jafrasteh, B., Lubián-López S.P., Benavente-Fernández I.
2022
Input Dependent Sparse Gaussian Processes ICML 2022
International Conference on Machine Learning (PMLR)
Jafrasteh, B., Villacampa-Calvo, C., Hernández-Lobato, D.

Full list available on Google Scholar.

Software & Projects

MELAGE

Registered software for medical image analysis — neonatal MRI/US segmentation and biomarker extraction. Reg. No. 2211222681375, Spain, 2022.

PyPI Downloads

MGP Imputer

A Python package for missing data imputation using Gaussian Processes, designed for clinical neuroimaging datasets with incomplete observations.

PyPI Downloads

Author and maintainer of 25+ public repositories on GitHub covering deep learning, Gaussian processes, and scientific computing.

Research Projects
AUTO-NEUS
ISCIII · Ref. DTS22/00142 · 2023–2025 · Automatic neonatal brain injury detection via ultrasound imaging
PARENT
H2020-MSCA-ITN · 2020–2024 · Early diagnosis of motor/cognitive impairment in preterm infants · Co-supervisor
Brain Development at 8 Years After Preterm Birth
Plan Andaluz de Investigación · PI-0016-2020 · 2020–2023
Contact
📍
Location
Department of Radiology
Weill Cornell Medicine, New York, USA
Academic & Social Profiles