Welcome!

Currently, I am a Member of Technical Staff Intern at Traversal and a Statistics PhD candidate at Cornell Tech. My current research interests include post-training of language models, foundational models for matrix completion and kernel methods. 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
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
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

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
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