94 citations · 229 across the 34 of their papers we have counts for
5 papers · 1 filter
MatrixFormer: A Foundation Model for Matrix Completion
Dwaipayan Saha, Jacob Feitelberg, Kyuseong Choi +2
Matrix completion underlies problems from tabular imputation to causal inference, yet existing tabular foundation models treat it as entry-by-entry prediction, repeating context fo…
ORCA-bench: How Ready Are Language Model Agents for Oncall?
Albert Gong, Kyuseong Choi, Abhineet Agarwal +5
Large language models can write, patch, and search code, but oncall root cause analysis (RCA) demands something different: reasoning over noisy metrics, logs, traces, and source co…
Express Language Modeling
Albert Gong, Annabelle Michael Carrell, Raaz Dwivedi +1
We introduce a new tool, Express, for converting a non-causal attention approximation into a causal approximation with matching approximation guarantees. When combined with the sta…
GAAVI: Global Asymptotic Anytime Valid Inference for the Conditional Mean Function
Brian M Cho, Raaz Dwivedi, Nathan Kallus
Inference on the conditional mean function (CMF) is central to tasks from adaptive experimentation to optimal treatment assignment and algorithmic fairness auditing. In this work,…
GOPO: Policy Optimization using Ranked Rewards
Kyuseong Choi, Dwaipayan Saha, Woojeong Kim +2
Standard reinforcement learning from human feedback (RLHF) trains a reward model on pairwise preference data and then uses it for policy optimization. However, while reward models…