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. The show has 6,000+ subscribers and over 300,000 downloads.
Research
Publications
Selected work; see Google Scholar for the full and most current list. denotes my name.
Advanced Brain Aging Associated With Migraine and Posttraumatic Headache: A Deep Learning Study
, G. Dumkrieger, C. D. Chong, T. J. Schwedt, T. Wu. Neurology Open Access, 2026. paper pdf editorial editorial pdf
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
Show all publications
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
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
Show all abstracts
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
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
Developing AI-powered pathology solutions to improve patient outcomes.Research Assistant, Ph.D. Student
Arizona State University, Tempe
05.2020 - 07.2025
Developed generative AI methods for early detection of neurological disorders, in collaboration with Mayo Clinic and Banner Alzheimer's Institute. Wu LabPh.D. Research Intern
Dolby Laboratories, San Francisco [Machine Perception and Reasoning]
05.2024 - 08.2024
Built context-aware music recommendations using multimodal data and large vision-language models.Research Scientist Intern
Amazon, Seattle [Health Halo Computer Vision]
05.2022 - 08.2022
Prototyped markerless biomechanical analysis of workouts using human pose estimation.Graduate Teaching Assistant
Arizona State University, Tempe
10.2019 -05.2020
Research Intern - Computer Vision
Philips Research Labs, Cambridge
06.2019 -08.2019
Built a real-time prototype for contactless patient monitoring and vitals measurement.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
Researched significance-based large-scale 3D point cloud compression and representation.
Education
Ph.D. in Computer Science
Arizona State University, Tempe
2025
Wu Lab, co-advised by Dr. Teresa Wu and Dr. Baoxin Li. Alumni listingM.S. in Computer Science
Arizona State University, Tempe
2020 B.Tech. in Information and Communication Technology
Dhirubhai Ambani Institute of Information and Communication Technology, Gandhinagar
2018
News and Highlights
- Paper on advanced brain aging in migraine and posttraumatic headache published in Neurology Open Access, with an accompanying editorial paper editorial
- Joined PathAI as a Machine Learning Engineer III (August 2025)
- Successfully defended my Ph.D. thesis on "Novel Deep Learning techniques for Early Detection of Neurological Disorders" slides thesis DOI
-
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
-
IEEE Transmitter Impact Creator profile
- How to Launch a Data Science Career: Entry Jobs and Evolving Paths link
- AI is Big. Some Want to Make It Small link
- Three Ways Deep Learning Yields New Insights for Medical Researchers link
- IEEE Twitter Chat: The Impact of Technology in 2022 link
- 8 Reasons to Become An Engineer link
- 5 Reasons to Publish Open Access with IEEE link
-
Heard on the Street – 2/15/2024 link
- InsideBigData
-
Chip industry strains to meet AI-fueled demands-will smaller LLMs help? link
- ComputerWorld
-
How AI could revolutionize biology — and vice versa link
- Axios