← Search

Kai Zheng

28 accepted papers

2026

AgentMath: Empowering Mathematical Reasoning for Large Language Models via Tool-Augmented Agent

ICLR 2026poster

Large Reasoning Models (LRMs) like o3 and DeepSeek-R1 have achieved remarkable progress in natural language reasoning with long chain-of-thought. However, they remain computationally inefficient and struggle with accuracy when solving problems requiring complex mathematical operations. In this work,…

Cited by 0SourceScholar
2026

NI-Tex: Non-isometric Image-based Garment Texture Generation

CVPR 2026

Existing industrial 3D garment meshes already cover most real-world clothing geometries, yet their texture diversity remains limited. To acquire more realistic textures, generative methods are often used to extract Physically-based Rendering (PBR) textures and materials from large collections of wil

Cited by 1SourcecodeScholar
2026

ReWeaver: Towards Simulation-Ready and Topology-Accurate Garment Reconstruction

CVPR 2026

High-quality 3D garment reconstruction plays a crucial role in mitigating the sim-to-real gap in applications such as digital avatars, virtual try-on and robotic manipulation. However, existing garment reconstruction methods typically rely on unstructured representations, such as 3D Gaussian Splats,

Cited by 0SourcecodeScholar
2025

Making Transformer Decoders Better Differentiable Indexers

ICLR 2025poster

Retrieval aims to find the top-k items most relevant to a query/user from a large dataset. Traditional retrieval models represent queries/users and items as embedding vectors and use Approximate Nearest Neighbor (ANN) search for retrieval. Recently, researchers have proposed a generative-based retri…

Cited by 0SourcePDFScholar
2025

ReDeEP: Detecting Hallucination in Retrieval-Augmented Generation via Mechanistic Interpretability

ICLR 2025spotlight

Retrieval-Augmented Generation (RAG) models are designed to incorporate external knowledge, reducing hallucinations caused by insufficient parametric (internal) knowledge. However, even with accurate and relevant retrieved content, RAG models can still produce hallucinations by generating outputs th…

Cited by 8SourcePDFScholar
2025

RecFlow: An Industrial Full Flow Recommendation Dataset

ICLR 2025poster

Industrial recommendation systems (RS) rely on the multi-stage pipeline to balance effectiveness and efficiency when delivering items from a vast corpus to users. Existing RS benchmark datasets primarily focus on the exposure space, where novel RS algorithms are trained and evaluated. However, when…

2025

Trigger3:Refining Query Correction via Adaptive Model Selector

AAAI 2025technical

In search scenarios, user experience can be hindered by erroneous queries due to typos, voice errors, or knowledge gaps. Therefore, query correction is crucial for search engines. Current correction models, usually small models trained on specific data, often struggle with queries beyond their train…

2024

Talk With Human-like Agents: Empathetic Dialogue Through Perceptible Acoustic Reception and Reaction

ACL 2024long

Large Language Model (LLM)-enhanced agents become increasingly prevalent in Human-AI communication, offering vast potential from entertainment to professional domains. However, current multi-modal dialogue systems overlook the acoustic information present in speech, which is crucial for understandin…

2024

WizardLM: Empowering Large Pre-Trained Language Models to Follow Complex Instructions

ICLR 2024poster

Training large language models (LLMs) with open-domain instruction following data brings colossal success. However, manually creating such instruction data is very time-consuming and labor-intensive. Moreover, humans may struggle to produce high-complexity instructions. In this paper, we show an ave…

Cited by 175SourcePDFScholar
2023

Adversarial Knowledge Stimulated Contrastive Prompting for Few-shot Language Learners

ACL 2023findings

Prompt-based fine-tuning has boosted the performance of Pre-trained Language Models(PLMs) on few-shot Natural Language Understanding (NLU) tasks by employing task-specific prompts. Yet, PLMsare unfamiliar with prompt-style expressionsduring pre-training, which limits the few-shotlearning performance…

Cited by 0SourcePDFScholar
2022

Cache-Augmented Inbatch Importance Resampling for Training Recommender Retriever

NeurIPS 2022accept

Recommender retrievers aim to rapidly retrieve a fraction of items from the entire item corpus when a user query requests, with the representative two-tower model trained with the log softmax loss. For efficiently training recommender retrievers on modern hardwares, inbatch sampling, where the items…

Cited by 12SourcePDFScholar
2022

Knowledge Stimulated Contrastive Prompting for Low-Resource Stance Detection

EMNLP 2022finding

Stance Detection Task (SDT) aims at identifying the stance of the sentence towards a specific target and is usually modeled as a classification problem. Backgound knowledge is often necessary for stance detection with respect to a specific target, especially when there is no target explicitly mentio…

Cited by 14SourcePDFScholar
2022

MetaER-TTE: An Adaptive Meta-learning Model for En Route Travel Time Estimation

IJCAI 2022poster

En route travel time estimation (ER-TTE) aims to predict the travel time on the remaining route. Since the traveled and remaining parts of a trip usually have some common characteristics like driving speed, it is desirable to explore these characteristics for improved performance via effective adapt…

Cited by 15SourcePDFScholar
2021

Efficient Optimal Selection for Composited Advertising Creatives with Tree Structure

AAAI 2021technical

Ad creatives are one of the prominent mediums for online e-commerce advertisements. Ad creatives with enjoyable visual appearance may increase the click-through rate (CTR) of products. Ad creatives are typically handcrafted by advertisers and then delivered to the advertising platforms for advertise…

2021

MFNP: A Meta-optimized Model for Few-shot Next POI Recommendation

IJCAI 2021poster

Next Point-of-Interest (POI) recommendation is of great value for location-based services. Existing solutions mainly rely on extensive observed data and are brittle to users with few interactions. Unfortunately, the problem of few-shot next POI recommendation has not been well studied yet. In this p…

Cited by 49SourcePDFScholar
2020

(Locally) Differentially Private Combinatorial Semi-Bandits

ICML 2020poster

In this paper, we study Combinatorial Semi-Bandits (CSB) that is an extension of classic Multi-Armed Bandits (MAB) under Differential Privacy (DP) and stronger Local Differential Privacy (LDP) setting. Since the server receives more information from users in CSB, it usually causes additional depende…

Cited by 6SourcePDFScholar
2020

Bilinear Graph Neural Network with Neighbor Interactions

IJCAI 2020poster

Graph Neural Network (GNN) is a powerful model to learn representations and make predictions on graph data. Existing efforts on GNN have largely defined the graph convolution as a weighted sum of the features of the connected nodes to form the representation of the target node. Nevertheless, the ope…

2020

Discovering Subsequence Patterns for Next POI Recommendation

IJCAI 2020poster

Next Point-of-Interest (POI) recommendation plays an important role in location-based services. State-of-the-art methods learn the POI-level sequential patterns in the user's check-in sequence but ignore the subsequence patterns that often represent the socio-economic activities or coherence of pref…

Cited by 0SourcePDFScholar
2020

Locally Differentially Private (Contextual) Bandits Learning

NeurIPS 2020poster

We study locally differentially private (LDP) bandits learning in this paper. First, we propose simple black-box reduction frameworks that can solve a large family of context-free bandits learning problems with LDP guarantee. Based on our frameworks, we can improve previous best results for private…

2020

On Layer Normalization in the Transformer Architecture

ICML 2020poster

The Transformer is widely used in natural language processing tasks. To train a Transformer however, one usually needs a carefully designed learning rate warm-up stage, which is shown to be crucial to the final performance but will slow down the optimization and bring more hyper-parameter tunings. I…

Cited by 1272SourcePDFScholar
2017

Collect at Once, Use Effectively: Making Non-interactive Locally Private Learning Possible

ICML 2017poster

Non-interactive Local Differential Privacy (LDP) requires data analysts to collect data from users through noisy channel at once. In this paper, we extend the frontiers of Non-interactive LDP learning and estimation from several aspects. For learning with smooth generalized linear losses, we propose…

Cited by 56SourcePDFScholar