I am a fourth year Ph.D. candidate in the Department of Computer Science at Virginia Tech, working in the Algorithms Lab under the supervision of Dr. Sharath Raghvendra. My research interests lie in the areas of computational geometry, combinatorial optimization, optimal transport and their applications to machine learning.
Research Work
"A New Robust Partial-Wasserstein-Based Metric for Comparing Distributions" - Submitted to ICML 2024 [arxiv]
Sharath Raghvendra, Pouyan Shirzadian, Kaiyi Zhang
"Computing all optimal partial transports" - Published at ICLR 2023 [openreview]
Abhijeet Phatak, Sharath Raghvendra, Chittaranjan Tripathy, Kaiyi Zhang
"A Combinatorial Algorithm for Approximating the Optimal Transport in the Parallel and MPC Settings" - Published at NeurIPS 2023 [openreview]
Nathaniel Lahn, Sharath Raghvendra, Kaiyi Zhang
Industry Experience
Senior Data Scientist at Walmart Global Tech
Jan 2024 - present, Sunnyvale, CA
Data Science Intern at Walmart Global Tech
Jun 2022 - Aug 2022, Sunnyvale, CA
Software Engineer Intern at Pometic Technology
Jun 2018 - Aug 2018, Houston, TX
Teaching
CS4104 Data and Algorithm Analysis
CS3214 Computer Systems
CS 5526 Data Analytics II
