Selected Publications

For a complete list, see my Google Scholar profile.

* denotes equal contribution.

2026

Firefly: Illuminating Verified Real-world Tool Call Data Generation

Yuxuan Lu, Ziyi Wang, Yingzhou Lu, Yisi Sang, Jiri Gesi, Xianfeng Tang, Yimeng Zhang, Zhenwei Dai, Hui Liu, Hanqing Lu, Chen Luo, Qi He, Benoit Dumoulin, Jing Huang, Dakuo Wang.

NeurIPS 2026, poster

A Reward-Guided Dual-Phase Framework for Adaptive Inference-Time Reasoning

Yingqian Cui, Zhenwei Dai, Pengfei He, Bing He, Hui Liu, Zhan Shi, Xianfeng Tang, Jingying Zeng, Suhang Wang, Yue Xing, Jiliang Tang, Benoit Dumoulin.

Findings of ACL 2026

TRAJECT-Bench: A Trajectory-Aware Benchmark for Evaluating Agentic Tool Use

Pengfei He, Zhenwei Dai, Bing He, Hui Liu, Xianfeng Tang, Hanqing Lu, Juanhui Li, Jiayuan Ding, Subhabrata Mukherjee, Suhang Wang, Yue Xing, Jiliang Tang, Benoit Dumoulin.

ICLR 2026

To Trust or Not to Trust: Attention-based Trust Management for LLM Multi-Agent Systems

Pengfei He, Zhenwei Dai, Xianfeng Tang, Yue Xing, Hui Liu, Jingying Zeng, Qiankun Peng, Shrivats Agrawal, Samarth Varshney, Suhang Wang, Jiliang Tang, Qi He.

ACL 2026, main conference; NeurIPS 2026, poster

2025

Keeping an Eye on LLM Unlearning: The Hidden Risk and Remedy

Jie Ren, Zhenwei Dai, Xianfeng Tang, Yue Xing, Shenglai Zeng, Jingying Zeng, Qiankun Peng, Samarth Varshney, Suhang Wang, Qi He, Charu Aggarwal, Hui Liu.

NeurIPS 2025

Examples as the Prompt: A Scalable Approach for Efficient LLM Adaptation in E-Commerce

Jingying Zeng*, Zhenwei Dai*, Hui Liu*, Samarth Varshney, Zhiji Liu, Chen Luo, Zhen Li, Qi He, Xianfeng Tang.

SIGIR 2025, industry track

Cite Before You Speak: Enhancing Context-Response Grounding in E-commerce Conversational LLM-Agents

Jingying Zeng*, Hui Liu*, Zhenwei Dai*, Xianfeng Tang, Chen Luo, Samarth Varshney, Zhen Li, Qi He.

arXiv preprint, 2025

A General Framework to Enhance Fine-tuning-based LLM Unlearning

Jie Ren*, Zhenwei Dai*, Xianfeng Tang, Hui Liu, Jingying Zeng, Zhen Li, Rahul Goutam, Suhang Wang, Yue Xing, Qi He, Hui Liu.

Findings of ACL 2025

Stepwise Perplexity-Guided Refinement for Efficient Chain-of-Thought Reasoning in Large Language Models

Yingqian Cui, Pengfei He, Jingying Zeng, Hui Liu, Xianfeng Tang, Zhenwei Dai, Yan Han, Chen Luo, Jing Huang, Zhen Li, Suhang Wang, Yue Xing, Jiliang Tang, Qi He.

Findings of ACL 2025

How Far Are LLMs from Real Search? A Comprehensive Study on Efficiency, Completeness, and Inherent Capabilities

Minhua Lin, Hui Liu, Xianfeng Tang, Jingying Zeng, Zhenwei Dai, Chen Luo, Zheng Li, Xiang Zhang, Qi He, Suhang Wang.

arXiv preprint, 2025

2024

RA-NER: Retrieval Augmented NER for Knowledge Intensive Named Entity Recognition

Zhenwei Dai, Chen Luo, Zhen Li, Xianfeng Tang, Hanqing Lu, Rahul Goutam, Haiyang Zhang.

ICLR 2024, Tiny Papers track

2023

Graph Self-supervised Learning via Proximity Distribution Minimization

Tianyi Zhang, Zhenwei Dai, Zhaozhuo Xu, Anshumali Shrivastava.

UAI 2023

2022

ScatterSample: Diversified Label Sampling for Data Efficient Graph Neural Network Learning

Zhenwei Dai, Vasileios Ioannidis, Soji Adeshina, Zak Jost, Christos Faloutsos, George Karypis.

Learning on Graphs Conference 2022

2021

Active Sampling Count Sketch (ASCS) for Online Sparse Estimation of a Trillion Scale Covariance Matrix

Zhenwei Dai, Aditya Desai, Reinhard Heckel, Anshumali Shrivastava.

ACM SIGMOD 2021

2020

Adaptive Learned Bloom Filter (Ada-BF): Efficient Utilization of the Classifier with Application to Real-Time Information Filtering on the Web

Zhenwei Dai, Anshumali Shrivastava.

NeurIPS 2020

2019

Channel Normalization in Convolutional Neural Network Avoids Vanishing Gradients

Zhenwei Dai, Reinhard Heckel.

ICML 2019, Deep Phenomena workshop