Zhang, Kexun

13 publications

ICLR 2025 DeFT: Decoding with Flash Tree-Attention for Efficient Tree-Structured LLM Inference Jinwei Yao, Kaiqi Chen, Kexun Zhang, Jiaxuan You, Binhang Yuan, Zeke Wang, Tao Lin

ICLR 2025 Diversity Empowers Intelligence: Integrating Expertise of Software Engineering Agents Kexun Zhang, Weiran Yao, Zuxin Liu, Yihao Feng, Zhiwei Liu, R N Rithesh, Tian Lan, Lei Li, Renze Lou, Jiacheng Xu, Bo Pang, Yingbo Zhou, Shelby Heinecke, Silvio Savarese, Huan Wang, Caiming Xiong

ICLR 2025 Generalization V.s. Memorization: Tracing Language Models’ Capabilities Back to Pretraining Data Xinyi Wang, Antonis Antoniades, Yanai Elazar, Alfonso Amayuelas, Alon Albalak, Kexun Zhang, William Yang Wang

ICLR 2025 SWE-Search: Enhancing Software Agents with Monte Carlo Tree Search and Iterative Refinement Antonis Antoniades, Albert Örwall, Kexun Zhang, Yuxi Xie, Anirudh Goyal, William Yang Wang

ICLRW 2024 DeFT: Flash Tree-Attention with IO-Awareness for Efficient Tree-Search-Based LLM Inference Jinwei Yao, Kexun Zhang, Kaiqi Chen, Jiaxuan You, Zeke Wang, Binhang Yuan, Tao Lin

ICMLW 2024 Generalization vs. Memorization: Tracing Language Models' Capabilities Back to Pretraining Data Antonis Antoniades, Xinyi Wang, Yanai Elazar, Alfonso Amayuelas, Alon Albalak, Kexun Zhang, William Yang Wang

NeurIPS 2024 Invisible Image Watermarks Are Provably Removable Using Generative AI Xuandong Zhao, Kexun Zhang, Zihao Su, Saastha Vasan, Ilya Grishchenko, Christopher Kruegel, Giovanni Vigna, Yu-Xiang Wang, Lei Li

ICML 2024 Understanding Reasoning Ability of Language Models from the Perspective of Reasoning Paths Aggregation Xinyi Wang, Alfonso Amayuelas, Kexun Zhang, Liangming Pan, Wenhu Chen, William Yang Wang

ICLRW 2024 Understanding the Reasoning Ability of Language Models from the Perspective of Reasoning Paths Aggregation Xinyi Wang, Alfonso Amayuelas, Kexun Zhang, Liangming Pan, Wenhu Chen, William Yang Wang

NeurIPS 2023 ALGO: Synthesizing Algorithmic Programs with Generated Oracle Verifiers Kexun Zhang, Danqing Wang, Jingtao Xia, William Yang Wang, Lei Li

ICMLW 2023 Generative Autoencoders as Watermark Attackers: Analyses of Vulnerabilities and Threats Xuandong Zhao, Kexun Zhang, Yu-Xiang Wang, Lei Li

ICML 2023 ReDi: Efficient Learning-Free Diffusion Inference via Trajectory Retrieval Kexun Zhang, Xianjun Yang, William Yang Wang, Lei Li

NeurIPSW 2023 ToolDec: Syntax Error-Free and Generalizable Tool Use for LLMs via Finite-State Decoding Hongqiao Chen, Kexun Zhang, Lei Li, William Yang Wang