Zhefan Wang 0001

dblp:331/3618-1 · DBLP profile ↗
← Back
5ranked-venue papers
2as first author
5since 2021 · last 2025
0000-0001-7176-8234ORCID · verified

Domains — the database's venue-derived domains; a paper can count in several

Databases, data management, data science and information retrieval · 5 · 2 first-author · 5 since 2021Artificial intelligence and machine learning · 2 · 2 since 2021
YearPublicationVenuePosition
2025 StepTool: Enhancing Multi-Step Tool Usage in LLMs via Step-Grained Reinforcement Learning
abstract
Despite their powerful text generation capabilities, large language models (LLMs) still struggle to effectively utilize external tools to solve complex tasks, a challenge known as tool learning. Existing methods primarily rely on supervised fine-tuning, treating tool learning as a text generation problem while overlooking the decision-making complexities inherent in multi-step contexts. In this work, we propose modeling tool learning as a dynamic decision-making process and introduce StepTool, a novel step-grained reinforcement learning framework that enhances LLMs' capabilities in multi-step tool use. StepTool comprises two key components: Step-grained Reward Shaping, which assigns rewards to each tool interaction based on its invocation success and contribution to task completion; and Step-grained Optimization, which applies policy gradient methods to optimize the model across multiple decision steps. Extensive experiments across diverse benchmarks show that StepTool consistently outperforms both SFT-based and RL-based baselines in terms of task Pass Rate and Recall of relevant tools. Furthermore, our analysis suggests that StepTool helps models discover new tool-use strategies rather than merely re-weighting prior knowledge. These results highlight the importance of fine-grained decision modeling in tool learning and establish StepTool as a general and robust solution for enhancing multi-step tool use in LLMs. Code and data are available at https://github.com/yuyq18/StepTool.
Yuanqing Yu, Zhefan Wang 0001, Weizhi Ma, Shuai Wang 0020, Chuhan Wu, Zhiqiang Guo, Min Zhang 0006
CIKM2
2024 To Recommend or Not: Recommendability Identification in Conversations with Pre-trained Language Models
Zhefan Wang 0001, Weizhi Ma, Min Zhang 0006
DASFAA (5)1
2024 MACRec: A Multi-Agent Collaboration Framework for Recommendation
abstract
LLM-based agents have gained considerable attention for their decision-making skills and ability to handle complex tasks.Recognizing the current gap in leveraging agent capabilities for multiagent collaboration in recommendation systems, we introduce MACRec, a novel framework designed to enhance recommendation systems through multi-agent collaboration.Unlike existing work on using agents for user/item simulation, we aim to deploy multiagents to tackle recommendation tasks directly.In our framework, recommendation tasks are addressed through the collaborative efforts of various specialized agents, including Manager, User/Item Analyst, Reflector, Searcher, and Task Interpreter, with different working flows.Furthermore, we provide application examples of how developers can easily use MACRec on various recommendation tasks, including rating prediction, sequential recommendation, conversational recommendation, and explanation generation of recommendation results.The framework and demonstration video are publicly available at https://github.com/wzf2000/MACRec.
Zhefan Wang 0001, Yuanqing Yu, Wendi Zheng, Weizhi Ma, Min Zhang 0006
SIGIR1
2023 Intent-aware Ranking Ensemble for Personalized Recommendation
abstract
Ranking ensemble is a critical component in real recommender systems. When a user visits a platform, the system will prepare several item lists, each of which is generally from a single behavior objective recommendation model. As multiple behavior intents, e.g., both clicking and buying some specific item category, are commonly concurrent in a user visit, it is necessary to integrate multiple single-objective ranking lists into one. However, previous work on rank aggregation mainly focused on fusing homogeneous item lists with the same objective while ignoring ensemble of heterogeneous lists ranked with different objectives with various user intents.
Jiayu Li 0001, Peijie Sun, Zhefan Wang 0001, Weizhi Ma, Yangkun Li, Min Zhang 0006, Zhoutian Feng, Daiyue Xue
SIGIR3
2022 Target Interest Distillation for Multi-Interest Recommendation
abstract
Sequential recommendation aims at predicting the next item that the user may be interested in given the historical interaction sequence. Typical neural models derive a single history embedding to represent the user's interests. Moving one step forward, recent studies point out that multiple sequence embeddings can help to better capture multi-faceted user interests. However, when ranking candidate items, these methods usually adopt the greedy inference strategy. This approach uses the best matching interest for each candidate item to calculate the ranking score, neglecting the target interest distribution in different contexts, which might lead to incompatibility with the current user intent. In this paper, we propose to enhance multi-interest recommendation by predicting the target user interest with a separate interest predictor and a specifically designed distillation loss. The proposed framework consists of two modules: the 1) multi-interest extractor to generate multiple embeddings regarding different user interests; and the 2) target-interest predictor to predict the interest distribution in the current context, which will be further utilized to dynamically aggregate multi-interest embeddings. To provide explicit supervision signals to the target-interest predictor, we devise a target-interest distillation loss that uses the similarity between the target item and multi-interest embeddings as the soft label of the target interest. This helps the target-interest predictor to accurately predict the user interest at the inference stage and enhances its generalization ability. Extensive experiments on three real-world datasets show the effectiveness and flexibility of the proposed framework.
Chenyang Wang 0003, Zhefan Wang 0001, Yankai Liu, Weizhi Ma, Min Zhang 0006, Yiqun Liu 0001, Junlan Feng, Chao Deng 0002, Shaoping Ma
CIKM2