ML Anthology

Dong, Kefan

15 publications

ICML 2025 STP: Self-Play LLM Theorem Provers with Iterative Conjecturing and Proving Kefan Dong, Tengyu Ma
NeurIPSW 2024 Formal Theorem Proving by Rewarding LLMs to Decompose Proofs Hierarchically Kefan Dong, Arvind V. Mahankali, Tengyu Ma
ICLR 2023 Asymptotic Instance-Optimal Algorithms for Interactive Decision Making Kefan Dong, Tengyu Ma
NeurIPS 2023 Beyond NTK with Vanilla Gradient Descent: A Mean-Field Analysis of Neural Networks with Polynomial Width, Samples, and Time Arvind Mahankali, Haochen Zhang, Kefan Dong, Margalit Glasgow, Tengyu Ma
ICLR 2023 First Steps Toward Understanding the Extrapolation of Nonlinear Models to Unseen Domains Kefan Dong, Tengyu Ma
AAAI 2023 Model-Based Offline Reinforcement Learning with Local Misspecification Kefan Dong, Yannis Flet-Berliac, Allen Nie, Emma Brunskill
COLT 2023 Toward L_∞Recovery of Nonlinear Functions: A Polynomial Sample Complexity Bound for Gaussian Random Fields Kefan Dong, Tengyu Ma
NeurIPSW 2022 First Steps Toward Understanding the Extrapolation of Nonlinear Models to Unseen Domains Kefan Dong, Tengyu Ma
NeurIPS 2021 Design of Experiments for Stochastic Contextual Linear Bandits Andrea Zanette, Kefan Dong, Jonathan N Lee, Emma Brunskill
NeurIPS 2021 Provable Model-Based Nonlinear Bandit and Reinforcement Learning: Shelve Optimism, Embrace Virtual Curvature Kefan Dong, Jiaqi Yang, Tengyu Ma
ICML 2020 Multinomial Logit Bandit with Low Switching Cost Kefan Dong, Yingkai Li, Qin Zhang, Yuan Zhou
ICML 2020 On the Expressivity of Neural Networks for Deep Reinforcement Learning Kefan Dong, Yuping Luo, Tianhe Yu, Chelsea Finn, Tengyu Ma
ICLR 2020 Q-Learning with UCB Exploration Is Sample Efficient for Infinite-Horizon MDP Kefan Dong, Yuanhao Wang, Xiaoyu Chen, Liwei Wang
COLT 2020 Root-N-Regret for Learning in Markov Decision Processes with Function Approximation and Low Bellman Rank Kefan Dong, Jian Peng, Yining Wang, Yuan Zhou
NeurIPS 2019 Exploration via Hindsight Goal Generation Zhizhou Ren, Kefan Dong, Yuan Zhou, Qiang Liu, Jian Peng