Yan, Kai

11 publications

ICLRW 2025 MIR-Bench: Benchmarking LLM's Long-Context Intelligence via Many-Shot In-Context Inductive Reasoning Kai Yan, Zhan Ling, Kang Liu, Yifan Yang, Ting-Han Fan, Lingfeng Shen, Zhengyin Du, Jiecao Chen

NeurIPS 2025 MIR-Bench: Can Your LLM Recognize Complicated Patterns via Many-Shot In-Context Reasoning? Kai Yan, Zhan Ling, Kang Liu, Yifan Yang, Ting-Han Fan, Lingfeng Shen, Zhengyin Du, Jiecao Chen

ICLRW 2024 Language Agent Tree Search Unifies Reasoning Acting and Planning in Language Models Andy Zhou, Kai Yan, Michal Shlapentokh-Rothman, Haohan Wang, Yu-Xiong Wang

ICML 2024 Language Agent Tree Search Unifies Reasoning, Acting, and Planning in Language Models Andy Zhou, Kai Yan, Michal Shlapentokh-Rothman, Haohan Wang, Yu-Xiong Wang

ICML 2024 Offline Imitation from Observation via Primal Wasserstein State Occupancy Matching Kai Yan, Alex Schwing, Yu-Xiong Wang

NeurIPS 2024 Reinforcement Learning Gradients as Vitamin for Online Finetuning Decision Transformers Kai Yan, Alexander G. Schwing, Yu-Xiong Wang

NeurIPS 2023 A Simple Solution for Offline Imitation from Observations and Examples with Possibly Incomplete Trajectories Kai Yan, Alex Schwing, Yu-Xiong Wang

ICCV 2023 Neural-PBIR Reconstruction of Shape, Material, and Illumination Cheng Sun, Guangyan Cai, Zhengqin Li, Kai Yan, Cheng Zhang, Carl Marshall, Jia-Bin Huang, Shuang Zhao, Zhao Dong

NeurIPSW 2023 Offline Imitation from Observation via Primal Wasserstein State Occupancy Matching Kai Yan, Alex Schwing, Yu-Xiong Wang

NeurIPS 2022 CEIP: Combining Explicit and Implicit Priors for Reinforcement Learning with Demonstrations Kai Yan, Alex Schwing, Yu-Xiong Wang

NeurIPS 2021 A Surrogate Objective Framework for Prediction+Programming with Soft Constraints Kai Yan, Jie Yan, Chuan Luo, Liting Chen, Qingwei Lin, Dongmei Zhang