Welcome!

Currently, I am a Statistics PhD student at Cornell Tech. My current research interests include efficient and interpretable policy optimization, personalization in causal inference (via foundational models) and central limit theorems for high-dimensional data points. I am co-advised by Raaz Dwivedi and Kengo Kato.

Education

Statistics, PhD, Cornell Tech (2021 - Present)
Biostatistics, MS, University of Michigan, Ann Arbor (2021)
BBA Business Administration, Korea University (2019)

Work Experience

Applied Scientist intern, Amazon, Pricing and Promotion Science (2025)

Working Papers

GRPO++: Improved policy optimization using ranked rewards
Kyuseong Choi, Dwaipayan Saha, Woojeong Kim, Anish Agarwal, Raaz Dwivedi
Gaussian approximation in Reproducing kernel Hilbert space, applications to kernel ridge regression
Kyuseong Choi, Kengo Kato

Preprints

Wild regenerative block bootstrap for Harris recurrent Markov chains
Kyuseong Choi, Gabriella Ciolek
TabImpute: Accurate and Fast Zero-Shot Missing-Data Imputation with a Pre-Trained Transformer
Jacob Feitelberg, Dwaipayan Saha, Kyuseong Choi, Zaid, Ahmad, Anish Agarwal, Raaz Dwivedi
Learning counterfactual distribution via kernel nearest neighbors
Kyuseong Choi, Jacob Feitelberg, Anish Agarwal, Raaz Dwivedi
Distributional matrix completion via nearest neighbors in the Wasserstein space
Jacob Feitelberg, Kyuseong Choi, Anish Agarwal, Raaz Dwivedi
N2: A Unified Python Package and Test Bench for Nearest Neighbor-Based Matrix Completion
Caleb Chin, Aashish Khubchandani, Harshvardhan Maskara, Kyuseong Choi, Jacob Feitelberg, Albert Gong, Manit Paul, Tathagata Sadhukhan, Anish Agarwal, Raaz Dwivedi

Publications

Supervised kernel thinning
Albert Gong, Kyuseong Choi, Raaz Dwivedi
NeurIPS 2024
Robust data integration from multiple external sources for generalized linear models with binary outcomes
Kyuseong Choi, Jeremy M.G. Taylor, Peisong Han
Biometrics, Volume 80, Issue 1, March 2024
Data integration: exploiting ratios of parameter estimates from a reduced external model
Jeremy M.G. Taylor, Kyuseong Choi, Peisong Han
Biometrika, Volume 110, Issue 1, March 2023, Pages 119-134