Zheng, Kai

33 publications

ICLR 2026 AgentMath: Empowering Mathematical Reasoning for Large Language Models via Tool-Augmented Agent Haipeng Luo, Huawen Feng, Qingfeng Sun, Can Xu, Kai Zheng, Yufei Wang, Tao Yang, Han Hu, Yansong Tang

NeurIPS 2025 Graph-Theoretic Insights into Bayesian Personalized Ranking for Recommendation Kai Zheng, Jianxin Wang, Jinhui Xu

ICLR 2025 Making Transformer Decoders Better Differentiable Indexers Wuchao Li, Kai Zheng, Defu Lian, Qi Liu, Wentian Bao, Enyun Yu, Yang Song, Han Li, Kun Gai

WACV 2025 MaskVD: Region Masking for Efficient Video Object Detection Sreetama Sarkar, Gourav Datta, Souvik Kundu, Kai Zheng, Chirayata Bhattacharyya, Peter A. Beerel

ICML 2025 NEAR: Neural Electromagnetic Array Response Yinyan Bu, Jiajie Yu, Kai Zheng, Xinyu Zhang, Piya Pal

ICLR 2025 ReDeEP: Detecting Hallucination in Retrieval-Augmented Generation via Mechanistic Interpretability ZhongXiang Sun, Xiaoxue Zang, Kai Zheng, Jun Xu, Xiao Zhang, Weijie Yu, Yang Song, Han Li

ICLR 2025 RecFlow: An Industrial Full Flow Recommendation Dataset Qi Liu, Kai Zheng, Rui Huang, Wuchao Li, Kuo Cai, Yuan Chai, Yanan Niu, Yiqun Hui, Bing Han, Na Mou, Hongning Wang, Wentian Bao, Enyun Yu, Guorui Zhou, Han Li, Yang Song, Defu Lian, Kun Gai

AAAI 2025 Trigger3: Refining Query Correction via Adaptive Model Selector Kepu Zhang, Zhongxiang Sun, Xiao Zhang, Xiaoxue Zang, Kai Zheng, Yang Song, Jun Xu

CVPRW 2024 Block Selective Reprogramming for On-Device Training of Vision Transformers Sreetama Sarkar, Souvik Kundu, Kai Zheng, Peter A. Beerel

ICLR 2024 WizardLM: Empowering Large Pre-Trained Language Models to Follow Complex Instructions Can Xu, Qingfeng Sun, Kai Zheng, Xiubo Geng, Pu Zhao, Jiazhan Feng, Chongyang Tao, Qingwei Lin, Daxin Jiang

AAAI 2022 Boosting Contrastive Learning with Relation Knowledge Distillation Kai Zheng, Yuanjiang Wang, Ye Yuan

NeurIPS 2022 Cache-Augmented Inbatch Importance Resampling for Training Recommender Retriever Jin Chen, Defu Lian, Yucheng Li, Baoyun Wang, Kai Zheng, Enhong Chen

IJCAI 2022 MetaER-TTE: An Adaptive Meta-Learning Model for en Route Travel Time Estimation Yu Fan, Jiajie Xu, Rui Zhou, Jianxin Li, Kai Zheng, Lu Chen, Chengfei Liu

UAI 2021 Combinatorial Semi-Bandit in the Non-Stationary Environment Wei Chen, Liwei Wang, Haoyu Zhao, Kai Zheng

AAAI 2021 Efficient Optimal Selection for Composited Advertising Creatives with Tree Structure Jin Chen, Tiezheng Ge, Gangwei Jiang, Zhiqiang Zhang, Defu Lian, Kai Zheng

IJCAI 2021 MFNP: A Meta-Optimized Model for Few-Shot Next POI Recommendation Huimin Sun, Jiajie Xu, Kai Zheng, Pengpeng Zhao, Pingfu Chao, Xiaofang Zhou

ICML 2020 (Locally) Differentially Private Combinatorial Semi-Bandits Xiaoyu Chen, Kai Zheng, Zixin Zhou, Yunchang Yang, Wei Chen, Liwei Wang

IJCAI 2020 Bilinear Graph Neural Network with Neighbor Interactions Hongmin Zhu, Fuli Feng, Xiangnan He, Xiang Wang, Yan Li, Kai Zheng, Yongdong Zhang

IJCAI 2020 Discovering Subsequence Patterns for Next POI Recommendation Kangzhi Zhao, Yong Zhang, Hongzhi Yin, Jin Wang, Kai Zheng, Xiaofang Zhou, Chunxiao Xing

NeurIPS 2020 Locally Differentially Private (Contextual) Bandits Learning Kai Zheng, Tianle Cai, Weiran Huang, Zhenguo Li, Liwei Wang

ICML 2020 On Layer Normalization in the Transformer Architecture Ruibin Xiong, Yunchang Yang, Di He, Kai Zheng, Shuxin Zheng, Chen Xing, Huishuai Zhang, Yanyan Lan, Liwei Wang, Tieyan Liu

IJCAI 2019 DMRAN: A Hierarchical Fine-Grained Attention-Based Network for Recommendation Huizhao Wang, Guanfeng Liu, An Liu, Zhixu Li, Kai Zheng

NeurIPS 2019 Equipping Experts/Bandits with Long-Term Memory Kai Zheng, Haipeng Luo, Ilias Diakonikolas, Liwei Wang

AAAI 2019 Improving One-Class Collaborative Filtering via Ranking-Based Implicit Regularizer Jin Chen, Defu Lian, Kai Zheng

AAAI 2019 Learning Transferable Self-Attentive Representations for Action Recognition in Untrimmed Videos with Weak Supervision Xiaoyu Zhang, Haichao Shi, Changsheng Li, Kai Zheng, Xiaobin Zhu, Lixin Duan

AAAI 2019 Preference-Aware Task Assignment in Spatial Crowdsourcing Yan Zhao, Jinfu Xia, Guanfeng Liu, Han Su, Defu Lian, Shuo Shang, Kai Zheng

IJCAI 2019 Profit-Driven Task Assignment in Spatial Crowdsourcing Jinfu Xia, Yan Zhao, Guanfeng Liu, Jiajie Xu, Min Zhang, Kai Zheng

ACML 2018 A Scalable Heterogeneous Parallel SOM Based on MPI/CUDA Yao Liu, Jun Sun, Qing Yao, Su Wang, Kai Zheng, Yan Liu

NeurIPS 2018 Efficient Online Portfolio with Logarithmic Regret Haipeng Luo, Chen-Yu Wei, Kai Zheng

COLT 2018 Generalization Bounds of SGLD for Non-Convex Learning: Two Theoretical Viewpoints Wenlong Mou, Liwei Wang, Xiyu Zhai, Kai Zheng

IJCAI 2018 LC-RNN: A Deep Learning Model for Traffic Speed Prediction Zhongjian Lv, Jiajie Xu, Kai Zheng, Hongzhi Yin, Pengpeng Zhao, Xiaofang Zhou

ICML 2017 Collect at Once, Use Effectively: Making Non-Interactive Locally Private Learning Possible Kai Zheng, Wenlong Mou, Liwei Wang

IJCAI 2017 Efficient Private ERM for Smooth Objectives Jiaqi Zhang, Kai Zheng, Wenlong Mou, Liwei Wang