Jay Shah
I'm a Machine Learning Engineer III at PathAI, developing AI-powered pathology solutions to improve patient outcomes. I earned my Ph.D. in Computer Science from the Wu Lab at Arizona State University, co-advised by Dr. Teresa Wu and Dr. Baoxin Li.
My research spans Generative AI, Deep Learning, and Medical Imaging. During my Ph.D., I developed AI methods for the early detection of brain disorders — Alzheimer's Disease and headache — including:
- Capturing brain-aging signatures in Alzheimer's Disease and headache disorders
- Medical image super-resolution to enhance quantitative accuracy
- Harmonization and quantification of medical imaging
This work was conducted in collaboration with Mayo Clinic, Banner Alzheimer's Institute, and Barrow Neurological Institute in Arizona.
I also host the Jay Shah Podcast, where I interview AI engineers, researchers, and practitioners about their journeys and advice for newcomers to the field.
Research
Publications
Selected work; see Google Scholar for the full and most current list. denotes my name.
Ordinal Classification with Distance Regularization for Robust Brain Age Prediction
, M. M. R. Siddiquee, Y. Su, T. Wu, B. Li. Proc. IEEE/CVF WACV, 2024. paper arXiv pdf code
AUCp: Pseudo-AUC for Inference Model Selection with Unlabeled Validation Data in Abnormality Detection
M. M. R. Siddiquee, F. Rafsani, , T. Wu, C. D. Chong, T. J. Schwedt, B. Li. IEEE Trans. Medical Imaging, 2026. paper pdf code
Brainomaly: Unsupervised Neurologic Disease Detection Utilizing Unannotated T1-weighted Brain MR Images
M. M. R. Siddiquee, , T. Wu, C. D. Chong, T. J. Schwedt, G. Dumkrieger, S. Nikolova, B. Li. Proc. IEEE/CVF WACV, 2024. paper arXiv pdf code
Leveraging Multi-modal Foundation Model Image Encoders to Enhance Brain MRI-based Headache Classification
F. Rafsani, D. Sheth, Y. Che, , M. M. R. Siddiquee, C. D. Chong, S. Nikolova, K. Ross, G. Dumkrieger, B. Li, T. Wu, T. J. Schwedt. Scientific Reports, 2025. paper pdf
Enhancing Amyloid PET Quantification: MRI-Guided Super-Resolution Using Latent Diffusion Models
, Y. Che, J. Sohankar, B. Li, Y. Su, T. Wu. Life, 2024. paper preprint pdf code
AnoFPDM: Anomaly Detection with Forward Process of Diffusion Models for Brain MRI
Y. Che, F. Rafsani, , M. M. R. Siddiquee, T. Wu. Proc. IEEE/CVF WACV Workshops, 2025. paper arXiv pdf code
Traumatic Brain Injury Recovery Prediction by Harmonizing Real Brain CT and Synthetic Brain MRI: A Pilot Study
Y. Che, A. Joshi, , M. M. R. Siddiquee, C. D. Chong, S. Nikolova, G. Dumkrieger, B. Li, T. Wu, T. J. Schwedt. Brain Communications, 2026. paper pdf
DinoAtten3D: Slice-Level Attention Aggregation of DinoV2 for 3D Brain MRI Anomaly Classification
F. Rafsani, , C. D. Chong, T. J. Schwedt, T. Wu. Proc. IEEE/CVF ICCV Workshops, 2025. arXiv pdf
HealthyGAN: Learning from Unannotated Medical Images to Detect Anomalies Associated with Human Disease
M. M. R. Siddiquee, , T. Wu, C. D. Chong, T. J. Schwedt, B. Li. SASHIMI Workshop, MICCAI, 2022. paper arXiv pdf code
Headache Classification and Automatic Biomarker Extraction from Structural MRIs Using Deep Learning
M. M. R. Siddiquee, , C. D. Chong, S. Nikolova, G. Dumkrieger, B. Li, T. Wu, T. J. Schwedt. Brain Communications, 2023. paper pdf
Interpretable Deep Learning Framework for Understanding Molecular Changes in Human Brains with Alzheimer's Disease: Implications for Microglia Activation and Sex Differences
M. R. Trivedi, A. Joshi, , B. P. Readhead, M. A. Wilson, Y. Su, E. M. Reiman, T. Wu, Q. Wang. npj Aging, 2025. paper bioRxiv pdf
Predicting Cognitive Decline from Neuropsychiatric Symptoms and Alzheimer's Disease Biomarkers: A Machine Learning Approach to Population-Based Data
, J. Krell-Roesch, E. Forzani, D. S. Knopman, C. R. Jack Jr., R. C. Petersen, Y. Che, T. Wu, Y. E. Geda. J. Alzheimer's Disease, 2025. paper pdf
Neuropsychiatric Symptoms and Commonly Used Biomarkers of Alzheimer's Disease: A Literature Review from a Machine Learning Perspective
, M. M. R. Siddiquee, J. Krell-Roesch, J. A. Syrjanen, W. Kremers, M. Vassilaki, E. Forzani, T. Wu, Y. E. Geda. J. Alzheimer's Disease, 2023. paper pdf
Physical Activity and the Outcome of Cognitive Trajectory: A Machine Learning Approach
B. Barisch-Fritz, , J. Krafft, Y. E. Geda, T. Wu, A. Woll, J. Krell-Roesch. Eur. Review of Aging and Physical Activity, 2025. paper pdf code
Deep Residual Inception Encoder-Decoder Network for Amyloid PET Harmonization
, F. Gao, V. Ghisays, J. Luo, Y. Chen, W. Lee, Y. Zhou, T. Benzinger, E. M. Reiman, K. Chen, Y. Su, T. Wu. Alzheimer's & Dementia, 2022. paper pdf code
Show all publications
Conference Abstracts
Using Large-scale Contrastive Language-Image Pre-training to Maximize Brain MRI-based Headache Classification
F. Rafsani, D. Sheth, Y. Che, , M. M. R. Siddiquee, C. D. Chong, S. Nikolova, G. Dumkrieger, B. Li, T. Wu, T. J. Schwedt. American Academy of Neurology Annual Meeting, 2025. paper
Capturing MRI Signatures of Brain Age as a Potential Biomarker to Predict Persistence of Post-traumatic Headache
, M. M. R. Siddiquee, C. D. Chong, T. J. Schwedt, J. Li, V. Berisha, K. Ross, T. Wu. American Academy of Neurology Annual Meeting, 2024. paper
Applying Generative Adversarial Networks on Structural Brain MRI for Unsupervised Classification of Headache
M. M. R. Siddiquee, , T. J. Schwedt, C. D. Chong, B. Li, T. Wu. American Academy of Neurology Annual Meeting, 2024. paper
Prediction of Headache Improvement Using Multimodal Machine Learning in Patients with Acute Post-traumatic Headache
A. Joshi, M. M. R. Siddiquee, , T. J. Schwedt, C. D. Chong, B. Li, T. Wu. American Academy of Neurology Annual Meeting, 2024. paper
A Multi-class Deep Learning Model to Estimate Brain Age While Addressing Systematic Bias of Regression to the Mean
, J. Luo, J. Sohankar, E. M. Reiman, K. Chen, Y. Su, B. Li, T. Wu. Alzheimer's Association International Conference, 2023. paper pdf
Interpretable Deep Learning Framework Towards Understanding Molecular Changes Associated with Neuropathology in Human Brains with Alzheimer's Disease
A. Joshi, , B. P. Readhead, Y. Su, T. Wu, Q. Wang. Alzheimer's Association International Conference, 2023. paper pdf
A 2.5D Residual U-Net for Improved Amyloid Harmonization Preserving Spatial Information
, J. Sohankar, J. Luo, Y. Chen, S. Li, H. D. Protas, K. Chen, E. M. Reiman, B. Li, T. Wu, Y. Su. Alzheimer's Association International Conference, 2023. paper pdf
End-to-End 3D CycleGAN Model for Amyloid PET Harmonization
X. Dong, Y. Wang, , V. Ghisays, J. Luo, Y. Chen, W. Lee, B. Li, K. Chen, E. M. Reiman, T. Wu, Y. Su. Alzheimer's Association International Conference, 2024. paper pdf
Classification and Biomarker Discovery of Persistent Post-traumatic Headache (PPTH) Using Deep Learning on Structural Brain MRI Data
M. M. R. Siddiquee, , T. J. Schwedt, C. D. Chong, S. Nikolova, G. Dumkrieger, K. Ross, V. Berisha, J. Li, T. Wu. INFORMS Annual Meeting, 2022. paper
Participant-specific Interrogation of Population-based Data to Predict Cognitive Decline from Neuropsychiatric Symptoms and Neuroimaging Biomarkers: A Machine Learning Approach
, J. A. Syrjanen, J. Krell-Roesch, W. Kremers, P. Vemuri, M. Vassilaki, R. C. Petersen, E. Forzani, T. Wu, Y. E. Geda. American Academy of Neurology Annual Meeting, 2023. paper pdf
MRI Signatures of Brain Age in the Alzheimer's Disease Continuum
, V. Ghisays, Y. Chen, J. Luo, B. Li, E. M. Reiman, K. Chen, T. Wu, Y. Su. Alzheimer's Association International Conference, 2022. paper pdf
Transfer Learning Based Deep Encoder-Decoder Network for Amyloid PET Harmonization with Small Datasets
, K. Chen, E. M. Reiman, B. Li, T. Wu, Y. Su. Alzheimer's Association International Conference, 2022. paper pdf
Classification of Post-Traumatic Headache (PTH) Using Deep Learning on Structural Brain MRI Data
M. M. R. Siddiquee, , T. J. Schwedt, C. D. Chong, S. Nikolova, G. Dumkrieger, K. Ross, V. Berisha, J. Li, T. Wu. American Headache Society Annual Meeting, 2022. paper pdf
Migraine Classification Using Deep Learning on Structural Brain MRI Data
M. M. R. Siddiquee, , T. J. Schwedt, C. D. Chong, S. Nikolova, G. Dumkrieger, K. Ross, V. Berisha, J. Li, T. Wu. American Headache Society Annual Meeting, 2022. paper pdf
Interpreting Deep Learning Model Predictions Using Shapley Values
, C. D. Chong, T. J. Schwedt, V. Berisha, J. Li, K. Ross, G. Dumkrieger, J. Zhang, N. Gaw, S. Nikolova, T. Wu. INFORMS Annual Meeting, 2021. pdf
Deep Residual Inception Encoder-Decoder Network for Amyloid PET Harmonization
, V. Ghisays, J. Luo, Y. Chen, W. Lee, B. Li, T. Benzinger, E. M. Reiman, K. Chen, Y. Su, T. Wu. Alzheimer's Association International Conference, 2021. paper pdf
Show all abstracts
Patents
User-guided context-aware music recommendations
Inventors: Jay Shah, Shanti Stewart, Gauri Jagatap, Gouthaman KV, Andrea Fanelli
Dolby Laboratories, 2024Deep Residual Inception Encoder-Decoder Network for Amyloid PET Harmonization
Inventors: Fei Gao, Yi Su, Jay Shah, Teresa Wu
US20240285244A1 WO2023101959A1 pdf
Work Experience
Machine Learning Engineer III
PathAI, Boston
08.2025 - Present
Research Assistant, Ph.D. Student
Arizona State University, Tempe
05.2020 - 07.2025
Ph.D. Research Intern
Dolby Laboratories, San Francisco [Machine Perception and Reasoning]
05.2024 - 08.2024
Research Scientist Intern
Amazon, Seattle [Health Halo Computer Vision]
05.2022 - 08.2022
Graduate Teaching Assistant
Arizona State University, Tempe
10.2019 -05.2020
Research Intern - Computer Vision
Philips Research Labs, Cambridge
06.2019 -08.2019
Graduate Research Assistant
Arizona State University, Tempe
11.2018 -06.2019
Machine Learning Engineer Intern
HackerRank, Bengaluru
01.2018 -05.2018
Visiting Research Assistant
Nanyang Technological University, Singapore
05.2017 -08.2017
News and Highlights
- I will be joining PathAI as a Machine Learning Engineer III
- Successfully defended my Ph.D. thesis on "Novel Deep Learning techniques for Early Detection of Neurological Disorders" slides thesis
-
Invited lecture on "Novel Deep Learning techniques for Early Detection of Neurological Disorders"
- Stephen and Denise Adams Center for Parkinson's Disease, Yale University
-
Invited talk on "AI for Early Detection of Alzheimer’s Disease" link
- AI Club, DAIICT
-
AI-powered medicine article full magazine
- Thrive magazine-summer 2024, Arizona State University
- College Enrollment, Jobs, Medical Research, AGI and Consciousness with Dr. Jay Shah link
-
Capturing MRI Signatures of Brain Age as a Potential Biomarker to Predict Persistence of Post-traumatic Headache slides link
- Oral presentation at American Academy of Neurology Annual Meeting, 2024
-
Invited speaker on PhD student Panel
- SUmmer Research Initiative (SURI) 2023, Arizona State University
-
Invited Young Professionals (YP) speaker at CMD Workshop link
- IEEE IAS Annual Meeting, 2022
-
Fulton Schools CS Doctoral student & researcher explores the quickly evolving world of AI and related smart tech advances on popular podcast link
- FullCircle, Arizona State University Newsletter
-
Using AI to battle Alzheimer’s link asu news
- FullCircle, Arizona State University Newsletter
-
Invited speaker at Emerging Research Topics in Engineering(ERTE) link
- IEEE Gujarat Section
-
Invited talk on "Landscape of Explainable AI, Interpreting Deep Learning predictions and my observations from hosting an ML Podcast" link
- 4th OnCV&AI workshop arranged by the Nordling Lab, National Cheng Kung University in Taiwan
-
From DA-IICT to Arizona State University and working with Nobel Laureate Frank Wilczek: Journey of Jay Shah link
- DA-IICT Blog
-
Interview on growing a technical podcast link link
- IEEE Spectrum and IEEE TV
-
Behind the scenes with a Machine Learning Expert : Jay Shah link
- Curryup Leadership Podcast
-
Python Workshop 2020 Convolutional Neural Networks 2020 2021
- AI Club, Arizona State University
Podcast mentions
- Best 100 Machine Learning Podcasts, Million Podcasts link
- Best 100 Research Podcasts, Million Podcasts link
- A hand-curated list of the best AI Podcasts, AI Depot link
- 8 of the best machine learning podcasts to listen to in 2022, Qwak MLOps link
- 5 Best Machine Learning & AI Podcasts, Unite[dot]AI, Futurist series link
- 20 best Machine Learning Podcasts of 2021, Welp Magazine link
In the media