Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Yan Zhang 0117

dblp:04/3348-117 · DBLP profile ↗
← Back
26ranked-venue papers
0as first author
25since 2021 · last 2026
0000-0003-4003-0290ORCID · conflict

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

Artificial intelligence and machine learning · 20 · 19 since 2021Databases, data management, data science and information retrieval · 12 · 12 since 2021Graphics, computer vision, multimedia, augmented reality and games · 2 · 2 since 2021Computer networks · 1 · 1 since 2021Applied, interdisciplinary, general and emerging computing · 1 · 1 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
18 papers
Language models and text generation · 35% Graph learning · 21% Question answering and dialogue systems · 12%
Databases, data mining, and information retrieval
10 papers
Information retrieval · 70% Recommender systems · 23% Data models and query languages · 6%
Computer architecture, parallel and distributed computing, and storage systems
1 paper
Cloud and datacenter computing · 50% Performance modeling and evaluation · 50%

Topics — the 30 heaviest of 57, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Information retrieval
retrieval-augmented generation
2.132026
Graph Retrieval-Augmented Generation: A Survey · ACM Trans. Inf. Syst. 2026
Enhancing Retrieval-Augmented Generation via Evidence Tree Search · ACL (1) 2025
Towards Verifiable Text Generation with Evolving Memory and Self-Reflection · EMNLP 2024
Knowledge, reasoning and agents › Planning, search and constraint satisfaction › game tree search
monte carlo tree search
1.722025
DecoupleSearch: Decouple Planning and Search via Hierarchical Reward Modeling · EMNLP 2025
Enhancing Retrieval-Augmented Generation via Evidence Tree Search · ACL (1) 2025
Natural language and speech › Question answering and dialogue systems
knowledge base question answering
1.522025
M³GQA: A Multi-Entity Multi-Hop Multi-Setting Graph Question Answering Benchmark · ACL (1) 2025
History Semantic Graph Enhanced Conversational KBQA with Temporal Information Modeling · ACL (1) 2023
Knowledge, reasoning and agents › Knowledge representation and reasoning › knowledge graph
knowledge graph querying
1.012026
COSMOS: Connectivity-Oriented Submodular Maximization for Optimal Subgraph Retrieval · ACL (1) 2026
Natural language and speech › Language models and text generation › large language model
large language model augmentation
1.012026
Graph Retrieval-Augmented Generation: A Survey · ACM Trans. Inf. Syst. 2026
Information retrieval › retrieval models
graph-based retrieval
1.012026
Graph Retrieval-Augmented Generation: A Survey · ACM Trans. Inf. Syst. 2026
Information retrieval › retrieval-augmented generation
graph-based retrieval-augmented generation
1.012026
Graph Retrieval-Augmented Generation: A Survey · ACM Trans. Inf. Syst. 2026
Cloud and datacenter computing
inference serving
1.012026
ServeGen: Workload Characterization and Generation of Large Language Model Serving in Production · NSDI 2026
Performance modeling and evaluation
workload characterization
1.012026
ServeGen: Workload Characterization and Generation of Large Language Model Serving in Production · NSDI 2026
Mathematical optimization › submodular optimization
submodular maximization
1.012026
COSMOS: Connectivity-Oriented Submodular Maximization for Optimal Subgraph Retrieval · ACL (1) 2026
Natural language and speech › Language models and text generation › retrieval-augmented generation
agentic retrieval-augmented generation
0.912025
DecoupleSearch: Decouple Planning and Search via Hierarchical Reward Modeling · EMNLP 2025
Natural language and speech › Language models and text generation
retrieval-augmented generation
0.912025
DecoupleSearch: Decouple Planning and Search via Hierarchical Reward Modeling · EMNLP 2025
Information retrieval
document retrieval
0.912025
Unveil: Unified Visual-Textual Integration and Distillation for Multi-modal Document Retrieval · ACL (1) 2025
Information retrieval
evidence retrieval
0.912025
Enhancing Retrieval-Augmented Generation via Evidence Tree Search · ACL (1) 2025
Information retrieval › multimodal retrieval
multimodal document retrieval
0.912025
Unveil: Unified Visual-Textual Integration and Distillation for Multi-modal Document Retrieval · ACL (1) 2025
Machine learning › Graph learning
graph neural network
0.922023
On Manipulating Signals of User-Item Graph: A Jacobi Polynomial-based Graph Collaborative Filtering · KDD 2023
Efficiently Leveraging Multi-level User Intent for Session-based Recommendation via Atten-Mixer Network · WSDM 2023
Natural language and speech › Language models and text generation › text generation › scientific text generation
citation generation
0.812024
Towards Verifiable Text Generation with Evolving Memory and Self-Reflection · EMNLP 2024
Machine learning › Efficient and distributed learning › distributed inference › collaborative inference
cloud-edge collaborative inference
0.812024
AdaSwitch: Adaptive Switching between Small and Large Agents for Effective Cloud-Local Collaborative Learning · EMNLP 2024
Machine learning › Generative modeling › diffusion model
conditional diffusion model
0.812024
Generative Auto-bidding via Conditional Diffusion Modeling · KDD 2024
Machine learning › Generative modeling
diffusion model
0.812024
Generative Auto-bidding via Conditional Diffusion Modeling · KDD 2024
Natural language and speech › Language models and text generation
in-context learning
0.812024
Retrieved In-Context Principles from Previous Mistakes · EMNLP 2024
Natural language and speech › Language models and text generation › LLM agents
LLM collaboration
0.812024
AdaSwitch: Adaptive Switching between Small and Large Agents for Effective Cloud-Local Collaborative Learning · EMNLP 2024
Natural language and speech › Language models and text generation
prompting
0.812024
Retrieved In-Context Principles from Previous Mistakes · EMNLP 2024
Information retrieval › online advertising
auto-bidding
0.812024
Generative Auto-bidding via Conditional Diffusion Modeling · KDD 2024
Information retrieval › retrieval models › neural retrieval
dense retrieval
0.812024
LEAD: Liberal Feature-based Distillation for Dense Retrieval · WSDM 2024
Information retrieval › retrieval models
knowledge distillation for retrieval
0.812024
LEAD: Liberal Feature-based Distillation for Dense Retrieval · WSDM 2024
Information retrieval
online advertising
0.812024
Generative Auto-bidding via Conditional Diffusion Modeling · KDD 2024
Natural language and speech › Question answering and dialogue systems › interactive question answering
conversational question answering
0.712023
History Semantic Graph Enhanced Conversational KBQA with Temporal Information Modeling · ACL (1) 2023
Machine learning › Graph learning › graph neural network
graph rewiring
0.712023
Homophily-oriented Heterogeneous Graph Rewiring · WWW 2023
Machine learning › Graph learning › graph neural network
heterogeneous graph neural network
0.712023
Homophily-oriented Heterogeneous Graph Rewiring · WWW 2023

Methods — techniques the papers use, named apart from their topics

monte carlo tree search · 2.6submodular optimization · 2.0retrieval-augmented generation · 2.0graph-based indexing · 2.0connectivity-oriented retrieval · 2.0visual-textual embedding · 1.7knowledge distillation · 1.7graph neural network · 1.3workload generation · 1.0multi-hop reasoning · 0.9large language model · 0.9benchmark construction · 0.9beam search · 0.9self-reflection · 0.8evolving memory · 0.8active retrieval · 0.8multi-choice question interaction · 0.4
YearPublicationVenuePosition
2026 COSMOS: Connectivity-Oriented Submodular Maximization for Optimal Subgraph Retrieval
abstract
Boci Peng, Xiao Liu, Boren Hu, Yun Zhu, Xuanbo Fan, Yanwei Yue, Chunyu Yang, Yan Zhang. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026.
Boci Peng, Xiao Liu 0029, Boren Hu, Yun Zhu 0007, Xuanbo Fan, Yanwei Yue, Chunyu Yang 0005, Yan Zhang 0117
ACL (1)8
2026 ServeGen: Workload Characterization and Generation of Large Language Model Serving in Production
Yuxing Xiang, Xue Li 0024, Kun Qian 0004, Yan Zhang 0117, Wenyuan Yu, Ennan Zhai, Xin Jin 0008, Jingren Zhou 0001
NSDI4
2026 Graph Retrieval-Augmented Generation: A Survey
abstract
Recently, Retrieval-Augmented Generation (RAG) has achieved remarkable success in addressing the challenges of Large Language Models (LLMs) without necessitating retraining. By referencing an external knowledge base, RAG refines LLM outputs, effectively mitigating issues such as “hallucination,” lack of domain-specific knowledge, and outdated information. However, the complex structure of relationships among different entities in databases presents challenges for RAG systems. In response, GraphRAG leverages structural information across entities to enable more precise and comprehensive retrieval, capturing relational knowledge and facilitating more accurate, context-aware responses. Given the novelty and potential of GraphRAG, a systematic review of current technologies is imperative. This article provides the first comprehensive overview of GraphRAG methodologies. We formalize the GraphRAG workflow, encompassing Graph-Based Indexing, Graph-Guided Retrieval, and Graph-Enhanced Generation. We then outline the core technologies and training methods at each stage. Additionally, we examine downstream tasks, application domains, evaluation methodologies, and industrial use cases of GraphRAG. Finally, we explore future research directions to inspire further inquiries and advance progress in the field. In order to track recent progress, we set up a repository at https://github.com/pengboci/GraphRAG-Survey .
Boci Peng, Yun Zhu 0007, Yongchao Liu 0004, Xiaohe Bo, Haizhou Shi, Chuntao Hong, Yan Zhang 0117, Siliang Tang
ACM Trans. Inf. Syst.7
2025 Unveil: Unified Visual-Textual Integration and Distillation for Multi-modal Document Retrieval
abstract
Document retrieval in real-world scenarios faces significant challenges due to diverse document formats and modalities. Traditional text-based approaches rely on tailored parsing techniques that disregard layout information and are prone to errors, while recent parsing-free visual methods often struggle to capture fine-grained textual semantics in text-rich scenarios. To address these limitations, we propose \textbf{Unveil}, a novel visual-textual embedding framework that effectively integrates textual and visual features for robust document representation. Through knowledge distillation, we transfer the semantic understanding capabilities from the visual-textual embedding model to a purely visual model, enabling efficient parsing-free retrieval while preserving semantic fidelity. Experimental results demonstrate that our visual-textual embedding method surpasses existing approaches, while knowledge distillation successfully bridges the performance gap between visual-textual and visual-only methods, improving both retrieval accuracy and efficiency.
Hao Sun 0015, Yingyan Hou, Jiayan Guo, Bo Wang 0134, Chunyu Yang 0005, Jinsong Ni, Yan Zhang 0117
ACL (1)7
2025 M³GQA: A Multi-Entity Multi-Hop Multi-Setting Graph Question Answering Benchmark
abstract
Boci Peng, Yongchao Liu, Xiaohe Bo, Jiaxin Guo, Yun Zhu, Xuanbo Fan, Chuntao Hong, Yan Zhang. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025.
Boci Peng, Yongchao Liu 0004, Xiaohe Bo, Yun Zhu 0007, Xuanbo Fan, Chuntao Hong, Yan Zhang 0117
ACL (1)8
2025 Enhancing Retrieval-Augmented Generation via Evidence Tree Search
abstract
Retrieval-Augmented Generation (RAG) is widely used to enhance Large Language Models (LLMs) by grounding responses in external knowledge. However, in real-world applications, retrievers often return lengthy documents with redundant or irrelevant content, confusing downstream readers. While evidence retrieval aims to address this by extracting key information, it faces critical challenges: (1) inability to model synergistic inter-dependencies among evidence sentences, (2) lack of supervision for evaluating multi-sentence evidence quality, and (3) computational inefficiency in navigating exponentially growing search spaces of candidate evidence sets. To tackle these challenges, we propose ETS (Evidence Tree Search), a novel framework that reformulates evidence retrieval as a dynamic tree expansion process. Our approach first constructs an evidence tree where each path represents a candidate evidence set, explicitly modeling inter-sentence dependencies through context-aware node selection. We then leverage Monte Carlo Tree Search (MCTS) to efficiently assess evidence quality and introduce an Early-Terminating Beam Search strategy to efficiently accelerate the model inference. Extensive experiments on five datasets demonstrate that ETS significantly outperforms existing methods across different readers. Our code and datasets will be released to facilitate future research.
Hao Sun 0015, Hengyi Cai, Yuchen Li 0006, Xuanbo Fan, Xiaochi Wei, Shuaiqiang Wang, Yan Zhang 0117, Dawei Yin 0001
ACL (1)7
2025 DecoupleSearch: Decouple Planning and Search via Hierarchical Reward Modeling
abstract
Retrieval-Augmented Generation (RAG) systems have emerged as a pivotal methodology for enhancing Large Language Models (LLMs) through the dynamic integration of external knowledge.To further improve RAG's flexibility, Agentic RAG introduces autonomous agents into the workflow.However, Agentic RAG faces several challenges: (1) the success of each step depends on both high-quality planning and accurate search, (2) the lack of supervision for intermediate reasoning steps, and (3) the exponentially large candidate space for planning and searching.To address these challenges, we propose DecoupleSearch, a novel framework that decouples planning and search processes using dual value models, enabling independent optimization of plan reasoning and search grounding.Our approach constructs a reasoning tree, where each node represents planning and search steps.We leverage Monte Carlo Tree Search to assess the quality of each step.During inference, Hierarchical Beam Search iteratively refines planning and search candidates with dual value models.Extensive experiments across policy models of varying parameter sizes, demonstrate the effectiveness of our method.
Hao Sun 0015, Zile Qiao, Bo Wang 0134, Guoxin Chen, Yingyan Hou, Yong Jiang 0005, Pengjun Xie, Fei Huang 0002, Yan Zhang 0117
EMNLP9
2025 Advancing Session-Based Recommendations with Atten-Mixer+: Dynamic and Adaptive Multi-Level Intent Mining
abstract
Session-Based Recommendation (SBR) systems, traditionally reliant on complex Graph Neural Networks (GNNs), often face challenges with marginal performance improvements despite increased model complexity. In this article, we dissect the classical GNN-based SBR models and empirically find that the sophisticated GNN propagations might be redundant, given the readout module plays a significant role in GNN-based models. Based on this observation, we introduce Atten-Mixer+, an advanced iteration of our previously developed Multi-Level Attention Mixture Network (Atten-Mixer). Atten-Mixer+ forgoes GNN propagation in favor of a dynamic and adaptive readout process, tailored to the unique characteristics of each session. Different from the vanilla version, Atten-Mixer+ features the Adaptive Intent Scaler (AIS) layer, which dynamically determines the depth of multi-level user intent analysis and a soft allocation approach for generating user intent queries across entire user interaction sequences. This innovative design allows Atten-Mixer+ to capture a nuanced and comprehensive understanding of user behaviors, overcoming the limitations of fixed-length analysis. Empirical evaluations on benchmark datasets highlight Atten-Mixer+’s superior efficiency and effectiveness, marking a significant step forward in the predictive accuracy of SBR systems.
Peiyan Zhang, Jiayan Guo, Chaozhuo Li, Liying Kang, Jae Boum Kim, Jie Xu 0015, Xi Zhang 0008, Yan Zhang 0117, Haohan Wang, Sung Hun Kim 0003
ACM Trans. Intell. Syst. Technol.8
2024 AdaSwitch: Adaptive Switching between Small and Large Agents for Effective Cloud-Local Collaborative Learning
abstract
Recent advancements in large language models (LLMs) have been remarkable.Users face a choice between using cloud-based LLMs for generation quality and deploying local-based LLMs for lower computational cost.The former option is typically costly and inefficient, while the latter usually fails to deliver satisfactory performance for reasoning steps requiring deliberate thought processes.In this work, we propose a novel LLM utilization paradigm that facilitates the collaborative operation of large cloud-based LLMs and smaller local-deployed LLMs.Our framework comprises two primary modules: the local agent instantiated with a relatively smaller LLM, handling less complex reasoning steps, and the cloud agent equipped with a larger LLM, managing more intricate reasoning steps.This collaborative processing is enabled through an adaptive mechanism where the local agent introspectively identifies errors and proactively seeks assistance from the cloud agent, thereby effectively integrating the strengths of both locally-deployed and cloudbased LLMs, resulting in significant enhancements in task completion performance and efficiency.We evaluate ADASWITCH across 7 benchmarks, ranging from mathematical reasoning and complex question answering, using various types of LLMs to instantiate the local and cloud agents.The empirical results show that ADASWITCH effectively improves the performance of the local agent, and sometimes achieves competitive results compared to the cloud agent while utilizing much less computational overhead.Question: Joy is 2 years older than twice the age of Tom.If Tom is 10 years old, how old is Joy?Thought: Calculate twice the age of Tom.Action: R1 = Calculator(2 * 2) Observation: 4 Reflection: Is previous step wrong: Yes Thought: Calculate twice the age of Tom who is 10.Action: R1 = Calculator(10 * 2) Observation: 20 Reflection: Is previous step wrong: No Thought: Calculate the age of Joy.
Hao Sun 0015, Jiayi Wu 0001, Hengyi Cai, Xiaochi Wei, Yue Feng 0002, Bo Wang 0134, Shuaiqiang Wang, Yan Zhang 0117, Dawei Yin 0001
EMNLP8
2024 Towards Verifiable Text Generation with Evolving Memory and Self-Reflection
abstract
Despite the remarkable ability of large language models (LLMs) in language comprehension and generation, they often suffer from producing factually incorrect information, also known as hallucination.A promising solution to this issue is verifiable text generation, which prompts LLMs to generate content with citations for accuracy verification.However, verifiable text generation is non-trivial due to the focus-shifting phenomenon, the intricate reasoning needed to align the claim with correct citations, and the dilemma between the precision and breadth of retrieved documents.In this paper, we present VTG, an innovative framework for Verifiable Text Generation with evolving memory and self-reflection.VTG introduces evolving long short-term memory to retain both valuable documents and recent documents.A two-tier verifier equipped with an evidence finder is proposed to rethink and reflect on the relationship between the claim and citations.Furthermore, active retrieval and diverse query generation are utilized to enhance both the precision and breadth of the retrieved documents.We conduct extensive experiments on five datasets across three knowledge-intensive tasks and the results reveal that VTG significantly outperforms baselines.
Hao Sun 0015, Hengyi Cai, Bo Wang 0134, Yingyan Hou, Xiaochi Wei, Shuaiqiang Wang, Yan Zhang 0117, Dawei Yin 0001
EMNLP7
2024 Retrieved In-Context Principles from Previous Mistakes
abstract
In-context learning (ICL) has been instrumental in adapting large language models (LLMs) to downstream tasks using correct input-output examples.Recent advances have attempted to improve model performance through principles derived from mistakes, yet these approaches suffer from lack of customization and inadequate error coverage.To address these limitations, we propose Retrieved In-Context Principles (RICP), a novel teacherstudent framework.In RICP, the teacher model analyzes mistakes from the student model to generate reasons and insights for preventing similar mistakes.These mistakes are clustered based on their underlying reasons for developing task-level principles, enhancing the error coverage of principles.During inference, the most relevant mistakes for each question are retrieved to create question-level principles, improving the customization of the provided guidance.RICP is orthogonal to existing prompting methods and does not require intervention from the teacher model during inference.Experimental results across seven reasoning benchmarks reveal that RICP effectively enhances performance when applied to various prompting strategies.
Hao Sun 0015, Yong Jiang 0005, Bo Wang 0134, Yingyan Hou, Yan Zhang 0117, Pengjun Xie, Fei Huang 0002
EMNLP5
2024 Generative Auto-bidding via Conditional Diffusion Modeling
abstract
Auto-bidding plays a crucial role in facilitating online advertising by automatically providing bids for advertisers. Reinforcement learning (RL) has gained popularity for auto-bidding. However, most current RL auto-bidding methods are modeled through the Markovian Decision Process (MDP), which assumes the Markovian state transition. This assumption restricts the ability to perform in long horizon scenarios and makes the model unstable when dealing with highly random online advertising environments. To tackle this issue, this paper introduces AI-Generated Bidding (AIGB), a novel paradigm for auto-bidding through generative modeling. In this paradigm, we propose DiffBid, a conditional diffusion modeling approach for bid generation. DiffBid directly models the correlation between the return and the entire trajectory, effectively avoiding error propagation across time steps in long horizons. Additionally, DiffBid offers a versatile approach for generating trajectories that maximize given targets while adhering to specific constraints. Extensive experiments conducted on the real-world dataset and online A/B test on Alibaba advertising platform demonstrate the effectiveness of DiffBid, achieving 2.81% increase in GMV and 3.36% increase in ROI.
Jiayan Guo, Yusen Huo, Zhilin Zhang 0003, Tianyu Wang 0028, Chuan Yu 0002, Jian Xu 0015, Bo Zheng 0007, Yan Zhang 0117
KDD8
2024 Subgraph Retrieval Enhanced by Graph-Text Alignment for Commonsense Question Answering
Boci Peng, Yongchao Liu 0004, Xiaohe Bo, Baokun Wang, Chuntao Hong, Yan Zhang 0117
ECML/PKDD (6)7
2024 LEAD: Liberal Feature-based Distillation for Dense Retrieval
abstract
Knowledge distillation is often used to transfer knowledge from a strong teacher model to a relatively weak student model. Traditional methods include response-based methods and feature-based methods. Response-based methods are widely used but suffer from lower upper limits of performance due to their ignorance of intermediate signals, while feature-based methods have constraints on vocabularies, tokenizers and model architectures. In this paper, we propose a liberal feature-based distillation method (LEAD). LEAD aligns the distribution between the intermediate layers of teacher model and student model, which is effective, extendable, portable and has no requirements on vocabularies, tokenizers, or model architectures. Extensive experiments show the effectiveness of LEAD on widely-used benchmarks, including MS MARCO Passage Ranking, TREC 2019 DL Track, MS MARCO Document Ranking and TREC 2020 DL Track. Our code is available in https://github.com/microsoft/SimXNS/tree/main/LEAD.
Hao Sun 0015, Xiao Liu 0029, Yeyun Gong, Anlei Dong, Jingwen Lu, Yan Zhang 0117, Linjun Yang, Rangan Majumder, Nan Duan 0001
WSDM6
2023 History Semantic Graph Enhanced Conversational KBQA with Temporal Information Modeling
abstract
Hao Sun, Yang Li, Liwei Deng, Bowen Li, Binyuan Hui, Binhua Li, Yunshi Lan, Yan Zhang, Yongbin Li. Proceedings of the 61st Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2023.
Hao Sun 0015, Liwei Deng 0001, Binyuan Hui, Binhua Li, Yunshi Lan, Yan Zhang 0117
ACL (1)8
2023 On Manipulating Signals of User-Item Graph: A Jacobi Polynomial-based Graph Collaborative Filtering
abstract
Collaborative filtering (CF) is an important research direction in recommender systems that aims to make recommendations given the information on user-item interactions. Graph CF has attracted more and more attention in recent years due to its effectiveness in leveraging high-order information in the user-item bipartite graph for better recommendations. Specifically, recent studies show the success of graph neural networks (GNN) for CF is attributed to its low-pass filtering effects. However, current researches lack a study of how different signal components contributes to recommendations, and how to design strategies to properly use them well. To this end, from the view of spectral transformation, we analyze the important factors that a graph filter should consider to achieve better performance. Based on the discoveries, we design JGCF, an efficient and effective method for CF based on Jacobi polynomial bases and frequency decomposition strategies. Extensive experiments on four widely used public datasets show the effectiveness and efficiency of the proposed methods, which brings at most 27.06% performance gain on Alibaba-iFashion. Besides, the experimental results also show that JGCF is better at handling sparse datasets, which shows potential in making recommendations for cold-start users.
Jiayan Guo, Lun Du, Xu Chen 0022, Xiaojun Ma 0001, Qiang Fu 0015, Shi Han, Dongmei Zhang 0001, Yan Zhang 0117
KDD8
2023 Efficiently Leveraging Multi-level User Intent for Session-based Recommendation via Atten-Mixer Network
abstract
Session-based recommendation (SBR) aims to predict the user's next action based on short and dynamic sessions. Recently, there has been an increasing interest in utilizing various elaborately designed graph neural networks (GNNs) to capture the pair-wise relationships among items, seemingly suggesting the design of more complicated models is the panacea for improving the empirical performance. However, these models achieve relatively marginal improvements with exponential growth in model complexity. In this paper, we dissect the classical GNN-based SBR models and empirically find that some sophisticated GNN propagations are redundant, given the readout module plays a significant role in GNN-based models. Based on this observation, we intuitively propose to remove the GNN propagation part, while the readout module will take on more responsibility in the model reasoning process. To this end, we propose the Multi-Level Attention Mixture Network (Atten-Mixer), which leverages both concept-view and instance-view readouts to achieve multi-level reasoning over item transitions. As simply enumerating all possible high-level concepts is infeasible for large real-world recommender systems, we further incorporate SBR-related inductive biases, i.e., local invariance and inherent priority to prune the search space. Experiments on three benchmarks demonstrate the effectiveness and efficiency of our proposal. We also have already launched the proposed techniques to a large-scale e-commercial online service since April 2021, with significant improvements of top-tier business metrics demonstrated in the online experiments on live traffic.
Peiyan Zhang, Jiayan Guo, Chaozhuo Li, Yueqi Xie, Jae Boum Kim, Yan Zhang 0117, Xing Xie 0001, Haohan Wang, Sunghun Kim 0001
WSDM6
2023 Homophily-oriented Heterogeneous Graph Rewiring
abstract
With the rapid development of the World Wide Web (WWW), heterogeneous graphs (HG) have explosive growth. Recently, heterogeneous graph neural network (HGNN) has shown great potential in learning on HG. Current studies of HGNN mainly focus on some HGs with strong homophily properties (nodes connected by meta-path tend to have the same labels), while few discussions are made in those that are less homophilous. Recently, there have been many works on homogeneous graphs with heterophily. However, due to heterogeneity, it is non-trivial to extend their approach to deal with HGs with heterophily. In this work, based on empirical observations, we propose a meta-path-induced metric to measure the homophily degree of a HG. We also find that current HGNNs may have degenerated performance when handling HGs with less homophilous properties. Thus it is essential to increase the generalization ability of HGNNs on non-homophilous HGs. To this end, we propose HDHGR, a homophily-oriented deep heterogeneous graph rewiring approach that modifies the HG structure to increase the performance of HGNN. We theoretically verify HDHGR. In addition, experiments on real-world HGs demonstrate the effectiveness of HDHGR, which brings at most more than 10% relative gain.
Jiayan Guo, Lun Du, Wendong Bi, Qiang Fu 0015, Xiaojun Ma 0001, Xu Chen 0022, Shi Han, Dongmei Zhang 0001, Yan Zhang 0117
WWW9
2022 Neural Enhanced Dynamic Message Passing
abstract
Predicting stochastic spreading processes on complex networks is critical in epidemic control, opinion propagation, and viral marketing. We focus on the problem of inferring the time-dependent marginal probabilities of states for each node which collectively quantifies the spreading results. Dynamic Message Passing (DMP) has been developed as an efficient inference algorithm for several spreading models, and it is asymptotically exact on locally tree-like networks. However, DMP can struggle in diffusion networks with lots of local loops. We address this limitation by using Graph Neural Networks (GNN) to learn the dependency amongst messages implicitly. Specifically, we propose a hybrid model in which the GNN module runs jointly with DMP equations. The GNN module refines the aggregated messages in DMP iterations by learning from simulation data. We demonstrate numerically that after training, our model’s inference accuracy substantially outperforms DMP in conditions of various network structure and dynamics parameters. Moreover, compared to pure data-driven models, the proposed hybrid model has a better generalization ability for out-of-training cases, profiting from the explicitly utilized dynamics priors in the hybrid model. A PyTorch implementation of our model is at https://github.com/FeiGSSS/NEDMP.
Jiang Zhang 0006, Yan Zhang 0117
AISTATS3
2022 Evolutionary Preference Learning via Graph Nested GRU ODE for Session-based Recommendation
abstract
Session-based recommendation (SBR) aims to predict the user's next action based on the ongoing sessions. Recently, there has been an increasing interest in modeling the user preference evolution to capture the fine-grained user interests. While latent user preferences behind the sessions drift continuously over time, most existing approaches still model the temporal session data in discrete state spaces, which are incapable of capturing the fine-grained preference evolution and result in sub-optimal solutions. To this end, we propose Graph Nested GRU ordinary differential equation (ODE), namely GNG-ODE, a novel continuum model that extends the idea of neural ODEs to continuous-time temporal session graphs. The proposed model preserves the continuous nature of dynamic user preferences, encoding both temporal and structural patterns of item transitions into continuous-time dynamic embeddings. As the existing ODE solvers do not consider graph structure change and thus cannot be directly applied to the dynamic graph, we propose a time alignment technique, called t-Alignment, to align the updating time steps of the temporal session graphs within a batch. Empirical results on three benchmark datasets show that GNG-ODE significantly outperforms other baselines.
Jiayan Guo, Peiyan Zhang, Chaozhuo Li, Xing Xie 0001, Yan Zhang 0117, Sunghun Kim 0001
CIKM5
2022 AMCAD: Adaptive Mixed-Curvature Representation based Advertisement Retrieval System
abstract
Graph embedding based retrieval has become one of the most popular techniques in the information retrieval community and search engine industry. The classical paradigm mainly relies on the flat Euclidean geometry. In recent years, hyperbolic (negative curvature) and spherical (positive curvature) representation methods have shown their superiority to capture hierarchical and cyclic data structures respectively. However, in industrial scenarios such as e-commerce sponsored search platforms, the large-scale heterogeneous query-item-advertisement interaction graphs often have multiple structures coexisting. Existing methods either only consider a single geometry space, or combine several spaces manually, which are incapable and inflexible to model the complexity and heterogeneity in the real scenario. To tackle this challenge, we present a web-scale Adaptive Mixed-Curvature ADvertisement retrieval system (AM-CAD) to automatically capture the complex and heterogeneous graph structures in non-Euclidean spaces. Specifically, entities are represented in adaptive mixed-curvature spaces, where the types and curvatures of the subspaces are trained to be optimal combinations. Besides, an attentive edge-wise space projector is designed to model the similarities between heterogeneous nodes according to local graph structures and the relation types. Moreover, to deploy AMCAD in Taobao, one of the largest e-commerce platforms with hundreds of million users, we design an efficient two-layer online retrieval framework for the task of graph based advertisement retrieval. Extensive evaluations on real-world datasets and A/B tests on online traffic are conducted to illustrate the effectiveness of the proposed system.
Zhirong Xu, Shiyang Wen, Junshan Wang, Liang Wang 0001, Zhi Yang 0001, Yan Zhang 0117, Di Zhang 0026, Jian Xu 0015, Bo Zheng 0007
ICDE8
2022 Small Changes Make Big Differences: Improving Multi-turn Response Selection in Dialogue Systems via Fine-Grained Contrastive Learning
abstract
Retrieve-based dialogue response selection aims to find a proper response from a candidate set given a multi-turn context.Pre-trained language models (PLMs) based methods have yielded significant improvements on this task.The sequence representation plays a key role in the learning of matching degree between the dialogue context and the response.However, we observe that different context-response pairs sharing the same context always have a greater similarity in the sequence representations calculated by PLMs, which makes it hard to distinguish positive responses from negative ones.Motivated by this, we propose a novel Fine-Grained Contrastive (FGC) learning method for the response selection task based on PLMs.This FGC learning strategy helps PLMs to generate more distinguishable matching representations of each dialogue at fine grains, and further make better predictions on choosing positive responses.Empirical studies on two benchmark datasets demonstrate that the proposed FGC learning method can generally and significantly improve the model performance of existing PLMbased matching models. 1
Can Xu 0002, Huang Hu, Lei Sha, Yan Zhang 0117, Daxin Jiang
INTERSPEECH5
2022 ConLearn: Contextual-knowledge-aware Concept Prerequisite Relation Learning with Graph Neural Network
abstract
Prerequisite relations among concepts are important for a wide range of educational applications, such as intelligent tutoring and curriculum planning. However, concept prerequisite relation learning is not trivial due to the sparsity of prerequisite relations. In this paper, we propose a contextual-knowledge-aware concept prerequisite relation learning approach called ConLearn. Four unique properties of the proposed approach are: (1) It transfers knowledge from large language model BERT to improve contextual representations of concepts; (2) It captures concept prerequisite transition patterns by applying graph neural network on concept prerequisite graph; (3) It is equipped with self-attention mechanism to fuse information from related concepts for target concept prerequisite relation classification; (4) No handcrafted features are used in our model, which makes our model easy to implement in downstream applications. Extensive experiments on three representative datasets demonstrate that our approach significantly outperforms the state-of-the-art methods.
Hao Sun 0015, Yan Zhang 0117
SDM3
2021 SMAD: Scalable Multi-view Ad Retrieval System for E-Commerce Sponsored Search
abstract
Ad retrieval in sponsored search aims to understand user search intentions (user queries) and retrieves a set of ads inferred as being relevant to the queries. Due to the huge amount of search traffic and multiple views of relevance (such as co-clicking, co-bidding or textual similar), it is highly desirable but remain challenging to achieve a large-scale, multi-view matching between queries and ads, particularly in industrial settings. In this paper, we propose a scalable multi-view ad retrieval engine SMAD that we developed and deployed at Taobao, the largest e-commerce platform in China. We construct a multi-relation query-item-ad graph capturing different views of query-ad relevance, which is of large scale and with complex structure. Since in e-commerce platform, the queries and products are organized into a category tree, to deal with the large scale of the graph, we propose a category constrained graph sampling and partition method to enable distributed parallel offline training. To tackle the complex multi-view structure, we propose a multi-view parallel deep neural network (DNN) model to combine the information from different views in a principled way. According to offline experiments and online A/B tests, our framework significantly outperforms baselines in terms of relevance, coverage, and revenue.
Shiyang Wen, Yiran Chen 0012, Zhi Yang 0001, Yan Zhang 0117, Di Zhang 0026, Liang Wang 0001, Bo Zheng 0007
CIKM4
2021 Keep the Structure: A Latent Shift-Reduce Parser for Semantic Parsing
abstract
Traditional end-to-end semantic parsing models treat a natural language utterance as a holonomic structure. However, hierarchical structures exist in natural languages, which also align with the hierarchical structures of logical forms. In this paper, we propose a latent shift-reduce parser, called LASP, which decomposes both natural language queries and logical form expressions according to their hierarchical structures and finds local alignment between them to enhance semantic parsing. LASP consists of a base parser and a shift-reduce splitter. The splitter dynamically separates an NL query into several spans. The base parser converts the relevant simple spans into logical forms, which are further combined to obtain the final logical form. We conducted empirical studies on two datasets across different domains and different types of logical forms. The results demonstrate that the proposed method significantly improves the performance of semantic parsing, especially on unseen scenarios.
Bei Chen 0008, Qian Liu 0033, Yan Gao 0002, Jian-Guang Lou, Yan Zhang 0117, Dongmei Zhang 0001
IJCAI6
2020 "What Do You Mean by That?" A Parser-Independent Interactive Approach for Enhancing Text-to-SQL
abstract
In Natural Language Interfaces to Databases systems, the text-to-SQL technique allows users to query databases by using natural language questions. Though significant progress in this area has been made recently, most parsers may fall short when they are deployed in real systems. One main reason stems from the difficulty of fully understanding the users' natural language questions. In this paper, we include human in the loop and present a novel parser-independent interactive approach (PIIA) that interacts with users using multi-choice questions and can easily work with arbitrary parsers. Experiments were conducted on two cross-domain datasets, the WikiSQL and the more complex Spider, with five state-of-the-art parsers. These demonstrated that PIIA is capable of enhancing the text-to-SQL performance with limited interaction turns by using both simulation and human evaluation.
Bei Chen 0008, Qian Liu 0033, Yan Gao 0002, Jian-Guang Lou, Yan Zhang 0117, Dongmei Zhang 0001
EMNLP (1)6