Zhuang, Yuchen

27 publications

TMLR 2026 Exploration-Driven Optimization for Test-Time Large Language Model Reasoning ChangHao Li, Yuchen Zhuang, Chenxiao Gao, Haotian Sun, Rushi Qiang, Chao Zhang, Bo Dai

ICLR 2026 MLE-Smith: Scaling MLE Tasks with Automated Multi-Agent Pipeline Rushi Qiang, Yuchen Zhuang, Anikait Singh, Percy Liang, Chao Zhang, Sherry Yang, Bo Dai

ICLR 2026 MedAgentGym: A Scalable Agentic Training Environment for Code-Centric Reasoning in Biomedical Data Science Ran Xu, Yuchen Zhuang, Yishan Zhong, Yue Yu, Zifeng Wang, Xiangru Tang, Hang Wu, May Dongmei Wang, Peifeng Ruan, Donghan Yang, Tao Wang, Guanghua Xiao, Xin Liu, Carl Yang, Yang Xie, Wenqi Shi

NeurIPS 2025 AceSearcher: Bootstrapping Reasoning and Search for LLMs via Reinforced Self-Play Ran Xu, Yuchen Zhuang, Zihan Dong, Ruiyu Wang, Yue Yu, Joyce C. Ho, Linjun Zhang, Haoyu Wang, Wenqi Shi, Carl Yang

NeurIPS 2025 AmorLIP: Efficient Language-Image Pretraining via Amortization Haotian Sun, Yitong Li, Yuchen Zhuang, Niao He, Hanjun Dai, Bo Dai

UAI 2025 DF$^2$: Distribution-Free Decision-Focused Learning Lingkai Kong, Wenhao Mu, Jiaming Cui, Yuchen Zhuang, B. Aditya Prakash, Bo Dai, Chao Zhang

ICLR 2025 Efficient Evolutionary Search over Chemical Space with Large Language Models Haorui Wang, Marta Skreta, Cher Tian Ser, Wenhao Gao, Lingkai Kong, Felix Strieth-Kalthoff, Chenru Duan, Yuchen Zhuang, Yue Yu, Yanqiao Zhu, Yuanqi Du, Alan Aspuru-Guzik, Kirill Neklyudov, Chao Zhang

NeurIPS 2025 MLE-Dojo: Interactive Environments for Empowering LLM Agents in Machine Learning Engineering Rushi Qiang, Yuchen Zhuang, Yinghao Li, V K Dingu Sagar, Rongzhi Zhang, ChangHao Li, Ian Shu-Hei Wong, Sherry Yang, Percy Liang, Chao Zhang, Bo Dai

NeurIPS 2025 Matryoshka Pilot: Learning to Drive Black-Box LLMs with LLMs ChangHao Li, Yuchen Zhuang, Rushi Qiang, Haotian Sun, Hanjun Dai, Chao Zhang, Bo Dai

NeurIPS 2024 Aligning Large Language Models with Representation Editing: A Control Perspective Lingkai Kong, Haorui Wang, Wenhao Mu, Yuanqi Du, Yuchen Zhuang, Yifei Zhou, Yue Song, Rongzhi Zhang, Kai Wang, Chao Zhang

ICMLW 2024 Aligning Large Language Models with Representation Editing: A Control Perspective Lingkai Kong, Haorui Wang, Wenhao Mu, Yuanqi Du, Yuchen Zhuang, Yifei Zhou, Yue Song, Rongzhi Zhang, Kai Wang, Chao Zhang

ICML 2024 BBox-Adapter: Lightweight Adapting for Black-Box Large Language Models Haotian Sun, Yuchen Zhuang, Wei Wei, Chao Zhang, Bo Dai

ICLRW 2024 EHRAgent: Code Empowers Large Language Models for Few-Shot Complex Tabular Reasoning on Electronic Health Records Wenqi Shi, Ran Xu, Yuchen Zhuang, Yue Yu, Jieyu Zhang, Hang Wu, Yuanda Zhu, Joyce C. Ho, Carl Yang, May Dongmei Wang

ICMLW 2024 Efficient Evolutionary Search over Chemical Space with Large Language Models Haorui Wang, Marta Skreta, Yuanqi Du, Wenhao Gao, Lingkai Kong, Cher Tian Ser, Felix Strieth-Kalthoff, Chenru Duan, Yuchen Zhuang, Yue Yu, Yanqiao Zhu, Alan Aspuru-Guzik, Kirill Neklyudov, Chao Zhang

ICMLW 2024 Efficient Evolutionary Search over Chemical Space with Large Language Models Haorui Wang, Marta Skreta, Yuanqi Du, Wenhao Gao, Lingkai Kong, Cher Tian Ser, Felix Strieth-Kalthoff, Chenru Duan, Yuchen Zhuang, Yue Yu, Yanqiao Zhu, Alan Aspuru-Guzik, Kirill Neklyudov, Chao Zhang

NeurIPS 2024 HYDRA: Model Factorization Framework for Black-Box LLM Personalization Yuchen Zhuang, Haotian Sun, Yue Yu, Rushi Qiang, Qifan Wang, Chao Zhang, Bo Dai

TMLR 2024 MUBen: Benchmarking the Uncertainty of Molecular Representation Models Yinghao Li, Lingkai Kong, Yuanqi Du, Yue Yu, Yuchen Zhuang, Wenhao Mu, Chao Zhang

ICLR 2024 ToolChain*: Efficient Action Space Navigation in Large Language Models with A* Search Yuchen Zhuang, Xiang Chen, Tong Yu, Saayan Mitra, Victor Bursztyn, Ryan A. Rossi, Somdeb Sarkhel, Chao Zhang

AISTATS 2024 Two Birds with One Stone: Enhancing Uncertainty Quantification and Interpretability with Graph Functional Neural Process Lingkai Kong, Haotian Sun, Yuchen Zhuang, Haorui Wang, Wenhao Mu, Chao Zhang

NeurIPS 2023 AdaPlanner: Adaptive Planning from Feedback with Language Models Haotian Sun, Yuchen Zhuang, Lingkai Kong, Bo Dai, Chao Zhang

NeurIPSW 2023 AdaPlanner: Adaptive Planning from Feedback with Language Models Haotian Sun, Yuchen Zhuang, Lingkai Kong, Bo Dai, Chao Zhang

ICML 2023 Autoregressive Diffusion Model for Graph Generation Lingkai Kong, Jiaming Cui, Haotian Sun, Yuchen Zhuang, B. Aditya Prakash, Chao Zhang

NeurIPSW 2023 Knowledge-Infused Prompting Improves Clinical Text Generation with Large Language Models Ran Xu, Hejie Cui, Yue Yu, Xuan Kan, Wenqi Shi, Yuchen Zhuang, Wei Jin, Joyce Ho, Carl Yang

NeurIPS 2023 Large Language Model as Attributed Training Data Generator: A Tale of Diversity and Bias Yue Yu, Yuchen Zhuang, Jieyu Zhang, Yu Meng, Alexander J Ratner, Ranjay Krishna, Jiaming Shen, Chao Zhang

NeurIPSW 2023 MUBen: Benchmarking the Uncertainty of Molecular Representation Models Yinghao Li, Lingkai Kong, Yuanqi Du, Yue Yu, Yuchen Zhuang, Wenhao Mu, Chao Zhang

NeurIPS 2023 ToolQA: A Dataset for LLM Question Answering with External Tools Yuchen Zhuang, Yue Yu, Kuan Wang, Haotian Sun, Chao Zhang

NeurIPS 2022 End-to-End Stochastic Optimization with Energy-Based Model Lingkai Kong, Jiaming Cui, Yuchen Zhuang, Rui Feng, B. Aditya Prakash, Chao Zhang