Welcome!

Currently, I am a Statistics PhD candidate at Cornell Tech. My current research interests include post-training of language models, foundational models for matrix completion. I have also worked on statistical algorithms for Markov chains, matrix completion and compression. I am co-advised by Raaz Dwivedi and Kengo Kato. I received MS in Biostatistics at University of Michigan, Ann Arbor and BBA Business Administration at Korea University.

Employment

Member of Technical Staff Intern, Traversal, Research (2026)
Applied Scientist intern, Amazon, Pricing and Promotion Science (2025)

Selected Works

ORCA-Bench: Are code agents ready for Oncall?
Albert Gong, Kyuseong Choi, Abhineet Agarwal, Anish Agarwal, Raaz Dwivedi
COLM 2026,CBW Workshop (Oral)
GOPO: Policy Optimization using Ranked Rewards
Kyuseong Choi, Dwaipayan Saha, Woojeong Kim, Anish Agarwal, Raaz Dwivedi
COLM 2026,SciFM Workshop
Wild regenerative block bootstrap for Harris recurrent Markov chains
Kyuseong Choi, Gabriella Ciolek
Submitted to Electric Journal of Statistics
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
RnR at Annals of Statistics

Preprints and Publications

Gaussian approximation in Reproducing kernel Hilbert space, applications to kernel ridge regression
Kyuseong Choi, Kengo Kato
Distributional matrix completion via nearest neighbors in the Wasserstein space
Jacob Feitelberg, Kyuseong Choi, Anish Agarwal, Raaz Dwivedi
Submitted to Operations Research
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
ICML 2025, CODEML Workshop
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