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

Weihang Su

dblp:301/7966 · DBLP profile ↗
shortlisted ← Back
22ranked-venue papers
8as first author
22since 2021 · last 2026
0000-0002-8718-9402ORCID · verified

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

Databases, data management, data science and information retrieval · 14 · 6 first-author · 14 since 2021Artificial intelligence and machine learning · 9 · 2 first-author · 9 since 2021Graphics, computer vision, multimedia, augmented reality and games · 3 · 1 first-author · 3 since 2021Applied, interdisciplinary, general and emerging computing · 2 · 2 since 2021Software engineering, systems software and programming languages · 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.

Databases, data mining, and information retrieval
15 papers
Information retrieval · 90% Recommender systems · 6% Query processing and optimization · 4%
Artificial intelligence
13 papers
Language models and text generation · 79% Reinforcement learning · 9% Trustworthy machine learning · 4%
Interdisciplinary, comprehensive, and emerging computing
2 papers
Computational social science and digital humanities · 100%

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

TopicWeightPapersLastEvidence papers
Information retrieval
retrieval-augmented generation
6.272026
Analytical Search · SIGIR 2026
SurGE: A Benchmark and Evaluation Framework for Scientific Survey Generation · SIGIR 2026
Robust Fine-tuning for Retrieval Augmented Generation against Retrieval Defects · SIGIR 2025
Natural language and speech › Language models and text generation
retrieval-augmented generation
3.542026
Enhancing Judgment Document Generation via Agentic Legal Information Collection and Rubric-Guided Optimization · SIGIR 2026
Decoupling Knowledge and Context: An Efficient and Effective Retrieval Augmented Generation Framework via Cross Attention · WWW 2025
Robust Fine-tuning for Retrieval Augmented Generation against Retrieval Defects · SIGIR 2025
Natural language and speech › Language models and text generation › large language model › knowledge in language models
knowledge injection
1.722025
Decoupling Knowledge and Context: An Efficient and Effective Retrieval Augmented Generation Framework via Cross Attention · WWW 2025
Parametric Retrieval Augmented Generation · SIGIR 2025
Information retrieval › retrieval-augmented generation
parametric retrieval-augmented generation
1.722025
Parametric Retrieval Augmented Generation · SIGIR 2025
Dynamic and Parametric Retrieval-Augmented Generation · SIGIR 2025
Information retrieval › retrieval models › neural retrieval
dense retrieval
1.622025
Pre-training for Legal Case Retrieval Based on Inter-Case Distinctions · ACM Trans. Inf. Syst. 2025
Scaling Laws For Dense Retrieval · SIGIR 2024
Information retrieval
retrieval models
1.622025
Pre-training for Legal Case Retrieval Based on Inter-Case Distinctions · ACM Trans. Inf. Syst. 2025
Wikiformer: Pre-training with Structured Information of Wikipedia for Ad-Hoc Retrieval · AAAI 2024
Information retrieval › document retrieval › domain-specific retrieval › legal information retrieval
legal case retrieval
1.522025
Pre-training for Legal Case Retrieval Based on Inter-Case Distinctions · ACM Trans. Inf. Syst. 2025
CaseEncoder: A Knowledge-enhanced Pre-trained Model for Legal Case Encoding · EMNLP 2023
Natural language and speech › Language models and text generation › retrieval-augmented generation
agentic retrieval-augmented generation
1.012026
Enhancing Judgment Document Generation via Agentic Legal Information Collection and Rubric-Guided Optimization · SIGIR 2026
Natural language and speech › Language models and text generation
hallucination detection
1.012026
Joint Evaluation of Answer and Reasoning Consistency for Hallucination Detection in Large Reasoning Models · AAAI 2026
Machine learning › Trustworthy machine learning
interpretability
1.012026
Joint Evaluation of Answer and Reasoning Consistency for Hallucination Detection in Large Reasoning Models · AAAI 2026
Natural language and speech › Language models and text generation
large language model evaluation
1.012026
Auto-PRE: An Automatic and Cost-Efficient Peer-Review Framework for Language Generation Evaluation · AAAI 2026
Natural language and speech › Language models and text generation › large language model
large reasoning model
1.012026
Joint Evaluation of Answer and Reasoning Consistency for Hallucination Detection in Large Reasoning Models · AAAI 2026
Natural language and speech › Language models and text generation
LLM agents
1.012026
SurGE: A Benchmark and Evaluation Framework for Scientific Survey Generation · SIGIR 2026
Natural language and speech › Language models and text generation › large language model evaluation
LLM-as-a-judge
1.012026
Auto-PRE: An Automatic and Cost-Efficient Peer-Review Framework for Language Generation Evaluation · AAAI 2026
Information retrieval › ranking › multi-objective ranking
fair ranking
1.012026
Equity vs. Equality: Optimizing Ranking Fairness for Tailored Provider Needs · SIGIR 2026
Recommender systems › fairness-aware recommendation
provider fairness
1.012026
Equity vs. Equality: Optimizing Ranking Fairness for Tailored Provider Needs · SIGIR 2026
Information retrieval › relevance feedback
pseudo-relevance feedback
1.012026
Generalized Pseudo-Relevance Feedback · WWW 2026
Query processing and optimization
query rewriting
1.012026
Generalized Pseudo-Relevance Feedback · WWW 2026
Natural language and speech › Language models and text generation
chain-of-thought reasoning
0.912025
Knowledge Editing through Chain-of-Thought · EMNLP 2025
Machine learning › Reinforcement learning › multi-agent reinforcement learning › multi-agent communication
communication protocols
0.912025
Augmenting Multi-Agent Communication with State Delta Trajectory · EMNLP 2025
Natural language and speech › Language models and text generation › knowledge editing
in-context knowledge editing
0.912025
Knowledge Editing through Chain-of-Thought · EMNLP 2025
Natural language and speech › Language models and text generation
knowledge editing
0.912025
Knowledge Editing through Chain-of-Thought · EMNLP 2025
Natural language and speech › Language models and text generation
large language model reasoning
0.912025
Augmenting Multi-Agent Communication with State Delta Trajectory · EMNLP 2025
Machine learning › Reinforcement learning › multi-agent reinforcement learning
multi-agent communication
0.912025
Augmenting Multi-Agent Communication with State Delta Trajectory · EMNLP 2025
Machine learning › Transfer learning and domain adaptation › fine-tuning
robust fine-tuning
0.912025
Robust Fine-tuning for Retrieval Augmented Generation against Retrieval Defects · SIGIR 2025
Information retrieval › reranking
neural re-ranking
0.912025
Pre-training for Legal Case Retrieval Based on Inter-Case Distinctions · ACM Trans. Inf. Syst. 2025
Information retrieval › retrieval models
neural retrieval
0.912025
Pre-training for Legal Case Retrieval Based on Inter-Case Distinctions · ACM Trans. Inf. Syst. 2025
Information retrieval › interactive information retrieval
adaptive retrieval
0.812024
DRAGIN: Dynamic Retrieval Augmented Generation based on the Real-time Information Needs of Large Language Models · ACL (1) 2024
Information retrieval › retrieval models
ad-hoc retrieval
0.812024
Wikiformer: Pre-training with Structured Information of Wikipedia for Ad-Hoc Retrieval · AAAI 2024
Information retrieval › reranking
document re-ranking
0.812024
Relevance Feedback with Brain Signals · ACM Trans. Inf. Syst. 2024

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

fine-tuning · 5.1large language model · 2.8rubric-guided optimization · 2.0group relative policy optimization · 2.0dynamic planning agent · 2.0LLM-based judgment · 2.0contrastive learning · 1.6semantic alignment · 1.0reinforcement learning · 1.0peer review process · 1.0multi-dimensional evaluation protocol · 1.0inter-sample consistency · 1.0gradient-based optimization · 1.0entropy-based uncertainty · 1.0automatic evaluator selection · 1.0parameter-level knowledge injection · 0.9in-context knowledge injection · 0.9few-shot in-context learning · 0.9
YearPublicationVenuePosition
2026 Auto-PRE: An Automatic and Cost-Efficient Peer-Review Framework for Language Generation Evaluation
abstract
The rapid development of large language models (LLMs) has highlighted the need for efficient and reliable methods to evaluate their performance. Traditional evaluation methods often face challenges like high costs, limited task formats, dependence on human references, and systematic biases. To address these limitations, we propose Auto-PRE, an automatic LLM evaluation framework inspired by the peer review process. Unlike previous approaches that rely on human annotations, Auto-PRE automatically selects evaluator LLMs based on three core traits: consistency, pertinence, and self-confidence, which correspond to the instruction, content, and response stages, respectively, and collectively cover the entire evaluation process. Experiments on three representative tasks, including summarization, non-factoid QA, and dialogue generation, demonstrate that Auto-PRE achieves state-of-the-art performance while significantly reducing evaluation costs. Furthermore, the structured and scalable design of our automatic qualification exam framework provides valuable insights into automating the evaluation of LLMs-as-judges, paving the way for more advanced LLM-based evaluation frameworks.
Weihang Su, Zhumin Chu, Haitao Li 0006, Yujia Zhou 0002, Dingbo Yuan, Yiqun Liu 0001, Min Zhang 0006, Shaoping Ma, Qingyao Ai
AAAI2
2026 Joint Evaluation of Answer and Reasoning Consistency for Hallucination Detection in Large Reasoning Models
abstract
Large Reasoning Models (LRMs) extend large language models with explicit, multi-step reasoning traces to enhance transparency and performance on complex tasks. However, these reasoning traces can be redundant or logically inconsistent, becoming a new and hard-to-detect source of hallucination. Existing hallucination detection methods focus primarily on answer-level uncertainty and often fail to detect hallucinations or logical inconsistencies arising from the model’s reasoning trace. This oversight is particularly problematic for LRMs, where the explicit thinking trace is not only an important support to the model's decision-making process but also a key source of potential hallucination. To this end, we propose RACE (Reasoning and Answer Consistency Evaluation), a novel framework specifically tailored for hallucination detection in LRMs. RACE operates by extracting essential reasoning steps and computing four diagnostic signals: inter-sample consistency of reasoning traces, entropy-based answer uncertainty, semantic alignment between reasoning and answers, and internal coherence of reasoning. This joint analysis enables fine-grained hallucination detection even when the final answer appears correct. Experiments across datasets and different LLMs demonstrate that RACE outperforms existing hallucination detection baselines, offering a robust and generalizable solution for evaluating LRMs.
Changyue Wang 0001, Weihang Su, Qingyao Ai, Yiqun Liu 0001
AAAI2
2026 Enhancing Judgment Document Generation via Agentic Legal Information Collection and Rubric-Guided Optimization
abstract
Automating the drafting of judgment documents is pivotal to judicial efficiency, yet it remains challenging due to the dual requirements of comprehensive retrieval of legal information and rigorous logical reasoning. Existing approaches, typically relying on standard Retrieval-Augmented Generation and Supervised Fine-Tuning, often suffer from insufficient evidence recall, hallucinated statutory references, and logically flawed legal reasoning. To bridge this gap, we propose Judge-R1, a unified framework designed to enhance LLM-based judgment document generation by jointly improving legal information collection and judgment document generation. First, we introduce Agentic Legal Information Collection, which employs a dynamic planning agent to retrieve precise statutes and precedents from multiple sources. Second, we implement Rubric-Guided Optimization, a reinforcement learning phase utilizing Group Relative Policy Optimization (GRPO) with a comprehensive legal reward function to enforce adherence to judicial standards and reasoning logic. Extensive experiments on the JuDGE benchmark demonstrate that Judge-R1 significantly outperforms state-of-the-art baselines in both legal accuracy and generation quality.
Weihang Su, Xuanyi Chen, Yueyue Wu, Qingyao Ai, Yiqun Liu 0001
SIGIR1
2026 SurGE: A Benchmark and Evaluation Framework for Scientific Survey Generation
abstract
The exponential growth of scientific literature has created a pressing need for automated survey generation. Although recent LLM-based agents have shown promise in automating this task, current progress is hindered by the lack of a standardized, scalable evaluation protocol. Existing evaluation methods typically rely on either human evaluation or custom metrics designed to validate specific pipelines, which restricts scalability and hinders fair comparison. To address this, we introduce SurGE, a benchmark and evaluation framework tailored for scientific survey generation. SurGE provides a large-scale retrieval corpus of over one million papers and expert-validated ground-truth surveys. Furthermore, we propose a robust multi-dimensional evaluation protocol that integrates both objective metrics and LLM-based judgments, and empirically verify its high alignment with human experts. Our experiments reveal that while agentic pipelines outperform RAG baselines in fluency and structural quality, they still struggle with citation accuracy, highlighting key directions for future research.
Weihang Su, Anzhe Xie, Qingyao Ai, Jianming Long, Xuanyi Chen, Jiaxin Mao, Ziyi Ye, Yiqun Liu 0001
SIGIR1
2026 Analytical Search
abstract
Analytical information needs, such as trend analysis and causal impact assessment, are prevalent across various domains including law, finance, science, and much more. However, existing information retrieval paradigms, whether based on relevance-oriented document ranking or retrieval-augmented generation (RAG) with large language models (LLMs), often struggle to meet the end-to-end requirements of such tasks at the corpus scale. They either emphasize information finding rather than end-to-end problem solving, or simply treat everything as question answering, offering limited control over reasoning, evidence usage, and verifiability. As a result, they struggle to support analytical queries that have diverse utility concepts and high accountability requirements.
Yiteng Tu, Shuo Miao, Weihang Su, Yiqun Liu 0001, Qingyao Ai
SIGIR3
2026 Equity vs. Equality: Optimizing Ranking Fairness for Tailored Provider Needs
abstract
Ranking plays a central role in connecting users and providers in Information Retrieval (IR) systems, making provider-side fairness an important challenge. While recent research has begun to address fairness in ranking, most existing approaches adopt an equality-based perspective, aiming to ensure that providers with similar content receive similar exposure. However, it overlooks the diverse needs of real-world providers, whose utility from ranking may depend not only on exposure but also on outcomes like sales or engagement. Consequently, exposure-based fairness may not accurately capture the true utility perceived by different providers with varying priorities. To this end, we introduce an equity-oriented fairness framework that explicitly models each provider's preferences over key outcomes such as exposure and sales, thus evaluating whether a ranking algorithm can fulfill these individualized goals while maintaining overall fairness across providers. Based on this framework, we develop EquityRank, a gradient-based algorithm that jointly optimizes user-side effectiveness and provider-side equity. Extensive offline and online simulations demonstrate that EquityRank offers improved trade-offs between effectiveness and fairness and adapts to heterogeneous provider needs.
Yiteng Tu, Weihang Su, Shuguang Han, Yiqun Liu 0001, Qingyao Ai
SIGIR2
2026 Generalized Pseudo-Relevance Feedback
abstract
Query rewriting is a fundamental technique in information retrieval (IR). It typically employs the retrieval result as relevance feedback to refine the query and thereby addresses the vocabulary mismatch between user queries and relevant documents. Traditional pseudo-relevance feedback (PRF) and its vector-based extension (VPRF) improve retrieval performance by leveraging top-retrieved documents as relevance feedback. However, they are constructed based on two major hypotheses: the relevance assumption (top documents are relevant) and the model assumption (rewriting methods need to be designed specifically for particular model architectures). While recent large language models (LLMs)-based generative relevance feedback (GRF) enables model-free query reformulation, it either suffers from severe LLM hallucination or, again, relies on the relevance assumption to guarantee the effectiveness of rewriting quality. To overcome these limitations, we introduce an assumption-relaxed framework: Generalized Pseudo Relevance Feedback (GPRF), which performs model-free, natural language rewriting based on retrieved documents, not only eliminating the model assumption but also reducing dependence on the relevance assumption. Specifically, we design a utility-oriented training pipeline with reinforcement learning to ensure robustness against noisy feedback. Extensive experiments across multiple benchmarks and retrievers demonstrate that GPRF consistently outperforms strong baselines, establishing it as an effective and generalizable framework for query rewriting.
Yiteng Tu, Weihang Su, Yujia Zhou 0002, Yiqun Liu 0001, Fen Lin 0002, Qin Liu 0022, Qingyao Ai
WWW2
2025 Augmenting Multi-Agent Communication with State Delta Trajectory
abstract
Multi-agent techniques such as role playing or multi-turn debates have been shown to be effective in improving the performance of large language models (LLMs) in downstream tasks.Despite their differences in workflows, existing multi-agent systems constructed from a single base LLM mostly use natural language for agent communication.While this is appealing for its simplicity and interpretability, it also introduces inevitable information loss as one model must down sample its continuous state vectors to discrete tokens before transferring them to the other model.Such losses are particularly significant when the information to transfer is not simple facts, but reasoning logics or abstractive thoughts.To tackle this problem, we propose a new communication protocol that transfers both natural language tokens and token-wise state transition trajectory from one agent to another.Particularly, compared to the actual state value, we find that the sequence of state changes in LLMs after generating each token can better reflect the information hidden behind the inference process.We propose a State Delta Encoding (SDE) method to represent state transition trajectories.The experimental results show that multi-agent systems with SDE achieve SOTA performance compared to other communication protocols, particularly in tasks that involve complex reasoning.
Yichen Tang 0001, Weihang Su, Yujia Zhou 0002, Yiqun Liu 0001, Min Zhang 0006, Shaoping Ma, Qingyao Ai
EMNLP2
2025 Knowledge Editing through Chain-of-Thought
abstract
Knowledge Editing is a technique that updates large language models (LLMs) with new information to maintain their world knowledge.This approach avoids the need to rebuild the model from scratch, thereby addressing the high costs associated with frequent retraining.Among these, the in-context editing paradigm stands out for its effectiveness in integrating new knowledge while preserving the model's original capabilities.Despite its potential, existing in-context knowledge editing methods are often task-specific, focusing primarily on multi-hop QA tasks using structured knowledge triples.Moreover, their reliance on fewshot prompting for task decomposition makes them unstable and less effective in generalizing across diverse tasks.In response to these limitations, we propose EditCoT, a novel knowledge editing framework that flexibly and efficiently updates LLMs across various tasks without retraining.EditCoT works by generating a chain-of-thought (CoT) for a given input and then iteratively refining this CoT process using a CoT editor based on updated knowledge.We evaluate EditCoT across a diverse range of benchmarks, covering multiple languages and tasks.The results demonstrate that our approach achieves state-of-the-art performance while offering superior generalization, effectiveness, and stability compared to existing methods, marking a significant advancement in the field of knowledge updating 1 .
Changyue Wang 0001, Weihang Su, Qingyao Ai, Yichen Tang 0001, Yiqun Liu 0001
EMNLP2
2025 Dynamic and Parametric Retrieval-Augmented Generation
abstract
Retrieval-Augmented Generation (RAG) has become a foundational paradigm for enhancing large language models (LLMs) with external knowledge, playing an important role in modern information retrieval and knowledge-intensive NLP applications. Standard RAG systems typically adopt a static retrieve-then-generate pipeline and rely on in-context knowledge injection, which can be suboptimal for complex tasks that require multihop reasoning, adaptive information access, and deeper integration of external knowledge. Motivated by these limitations, the research community has moved beyond static retrieval and in-context knowledge injection. Among the emerging directions, this tutorial delves into two rapidly growing and complementary research directions on RAG: Dynamic RAG and Parametric RAG. Dynamic RAG explores how LLMs can actively decide when and what to retrieve during generation, enabling real-time adaptation to evolving information needs. Parametric RAG rethinks how the retrieved knowledge should be incorporated, moving from input-level to parameter-level knowledge injection for improved efficiency and effectiveness. This tutorial offers a comprehensive overview of recent advances in both directions. It provides participants with the theoretical foundations and actionable insights needed to build flexible and scalable RAG systems.
Weihang Su, Qingyao Ai, Jingtao Zhan, Yiqun Liu 0001
SIGIR1
2025 Parametric Retrieval Augmented Generation
abstract
Retrieval-augmented generation (RAG) has emerged as a promising solution to enhance the reliability of large language models (LLMs) with external knowledge. Existing RAG methods share a common strategy for knowledge injection: they place the retrieved documents into the input context of the LLM, which we refer to as the in-context knowledge injection method. While this approach is simple and often effective, it has inherent limitations. Firstly, increasing the context length and number of relevant documents can lead to higher computational overhead and degraded performance, especially in complex reasoning tasks. More importantly, in-context knowledge injection operates primarily at the input level, but LLMs store their internal knowledge in their parameters. This gap fundamentally limits the capacity of in-context methods. To this end, we introduce Parametric RAG, a new RAG paradigm that integrates external knowledge directly into the feed-forward networks of an LLM through document parameterization. This approach not only reduces online computational costs by shortening the input context length, but also deepens the integration of external knowledge by enabling LLMs to utilize it in the same way as internal parametric knowledge. Experimental results demonstrate that Parametric RAG substantially enhances the effectiveness and efficiency of knowledge augmentation in LLMs. Also, it can be combined with in-context RAG methods to achieve even better performance. We have open-sourced all the code, data, and models in the following GitHub link: https://github.com/oneal2000/PRAG
Weihang Su, Yichen Tang 0001, Qingyao Ai, Junxi Yan, Changyue Wang 0001, Hongning Wang, Ziyi Ye, Yujia Zhou 0002, Yiqun Liu 0001
SIGIR1
2025 JuDGE: Benchmarking Judgment Document Generation for Chinese Legal System
abstract
This paper introduces JuDGE (Judgment Document Generation Evaluation), a novel benchmark for evaluating the performance of judgment document generation in the Chinese legal system. We define the task as generating a complete legal judgment document from the given factual description of the case. To facilitate this benchmark, we construct a comprehensive dataset consisting of factual descriptions from real legal cases, paired with their corresponding full judgment documents, which serve as the ground truth for evaluating the quality of generated documents. This dataset is further augmented by two external legal corpora that provide additional legal knowledge for the task: one comprising statutes and regulations, and the other consisting of a large collection of past judgment documents. In collaboration with legal professionals, we establish a comprehensive automated evaluation framework to assess the quality of generated judgment documents across various dimensions. We evaluate various baseline approaches, including few-shot in-context learning, fine-tuning, and a multi-source retrieval-augmented generation (RAG) approach, using both general and legal-domain LLMs. The experimental results demonstrate that, while RAG approaches can effectively improve performance in this task, there is still substantial room for further improvement. All the codes and datasets are available at: https://github.com/oneal2000/JuDGE
Weihang Su, Baoqing Yue, Qingyao Ai, Yiran Hu, Changyue Wang 0001, Yueyue Wu, Yiqun Liu 0001
SIGIR1
2025 Robust Fine-tuning for Retrieval Augmented Generation against Retrieval Defects
abstract
Retrieval-augmented generation (RAG) enhances large language models (LLMs) by integrating external knowledge retrieved from a knowledge base. However, its effectiveness is fundamentally constrained by the reliability of both the retriever and the knowledge base (i.e., the retrieval system). In real-world scenarios, imperfections in these components often lead to the retrieval of noisy, irrelevant, or misleading counterfactual information, ultimately undermining the trustworthiness of RAG systems. To address this challenge, we propose Robust Fine-Tuning (RbFT), a method designed to enhance the resilience of LLMs against retrieval defects through two targeted fine-tuning tasks. Experimental results demonstrate that RbFT significantly improves the robustness of RAG systems across diverse retrieval conditions, surpassing existing methods while maintaining high inference efficiency and compatibility with other robustness techniques.
Yiteng Tu, Weihang Su, Yujia Zhou 0002, Yiqun Liu 0001, Qingyao Ai
SIGIR2
2025 Decoupling Knowledge and Context: An Efficient and Effective Retrieval Augmented Generation Framework via Cross Attention
abstract
Retrieval-Augmented Generation (RAG) systems have become a crucial tool to augment large language models (LLMs) with external knowledge for better task performance.However, existing traditional RAG methods inject knowledge directly into the context, resulting in several limitations.First, these methods highly rely on the in-context learning capability of LLMs, which often leads to excessively long contexts.This is inefficient due to the quadratic complexity of self-attention, leading to significant increase in inference time.Second, the extended context and the nature of self-attention can cause the LLMs to lose important information in the context, thereby degrading the original capabilities of LLMs.Third, the effectiveness of knowledge injection is perturbed by the permutation of knowledge within the extended context, reducing the robustness of existing RAG methods.To tackle the above problems, we propose DecoupledRAG, a method that decouples external knowledge from the context within the RAG framework.Specifically, we introduce a cross-attention based method that injects retrieved knowledge directly into the inference process of LLM on the fly, without modifying its parameters or the input context, so that the external knowledge can be utilized robustly in a permutation-independent manner.To the best of our knowledge, this is the first work that explore how to utilize cross-attention to inject knowledge with low training cost in decoder-only LLM era.By leveraging cross-attention operation, DecoupledRAG enables seamless knowledge aggregation without creating extended context.Experimental results demonstrate that our method could achieve
Qingyao Ai, Hongning Wang, Haitao Li 0006, Weihang Su, Yiqun Liu 0001, Tat-Seng Chua, Shaoping Ma
WWW6
2025 Pre-training for Legal Case Retrieval Based on Inter-Case Distinctions
abstract
Legal case retrieval aims to help legal workers find relevant cases related to their cases at hand, which is important for the guarantee of fairness and justice in legal judgments. While recent advances in neural retrieval methods have significantly improved the performance of open-domain retrieval tasks (e.g., Web search), their advantages haven’t been observed in legal case retrieval due to their thirst for annotated data. As annotating large-scale training data in legal domains is prohibitive due to the need for domain expertise, traditional search techniques based on lexical matching such as TF-IDF, BM25, and Query Likelihood are still prevalent in legal case retrieval systems. While previous studies have designed several pre-training methods for IR models in open-domain tasks, these methods are usually suboptimal in legal case retrieval because they cannot understand and capture the key knowledge and data structures in the legal corpus. To this end, we propose a novel pre-training framework named Caseformer that enables the pre-trained models to learn legal knowledge and domain-specific relevance-matching patterns in legal case retrieval without any human-labeled data. This framework is designed to support both dense retrieval models and neural re-ranking models. Through three unsupervised learning tasks, Caseformer is able to capture the special language, document structure, and relevance-matching patterns of legal case documents, making it a strong backbone for downstream legal case retrieval tasks. Experimental results show that our model has achieved state-of-the-art performance in both zero-shot and fine-tuning settings. Also, experiments on both Chinese and English legal datasets demonstrate that the effectiveness of Caseformer is language-independent in legal case retrieval.
Weihang Su, Qingyao Ai, Yueyue Wu, Anzhe Xie, Changyue Wang 0001, Haitao Li 0006, Zhijing Wu 0001, Yiqun Liu 0001, Min Zhang 0006
ACM Trans. Inf. Syst.1
2024 Wikiformer: Pre-training with Structured Information of Wikipedia for Ad-Hoc Retrieval
abstract
With the development of deep learning and natural language processing techniques, pre-trained language models have been widely used to solve information retrieval (IR) problems. Benefiting from the pre-training and fine-tuning paradigm, these models achieve state-of-the-art performance. In previous works, plain texts in Wikipedia have been widely used in the pre-training stage. However, the rich structured information in Wikipedia, such as the titles, abstracts, hierarchical heading (multi-level title) structure, relationship between articles, references, hyperlink structures, and the writing organizations, has not been fully explored. In this paper, we devise four pre-training objectives tailored for IR tasks based on the structured knowledge of Wikipedia. Compared to existing pre-training methods, our approach can better capture the semantic knowledge in the training corpus by leveraging the human-edited structured data from Wikipedia. Experimental results on multiple IR benchmark datasets show the superior performance of our model in both zero-shot and fine-tuning settings compared to existing strong retrieval baselines. Besides, experimental results in biomedical and legal domains demonstrate that our approach achieves better performance in vertical domains compared to previous models, especially in scenarios where long text similarity matching is needed. The code is available at https://github.com/oneal2000/Wikiformer.
Weihang Su, Qingyao Ai, Xiangsheng Li, Jia Chen 0003, Yiqun Liu 0001, Shengluan Hou
AAAI1
2024 DRAGIN: Dynamic Retrieval Augmented Generation based on the Real-time Information Needs of Large Language Models
abstract
Dynamic retrieval augmented generation (RAG) paradigm actively decides when and what to retrieve during the text generation process of Large Language Models (LLMs).There are two key elements of this paradigm: identifying the optimal moment to activate the retrieval module (deciding when to retrieve) and crafting the appropriate query once retrieval is triggered (determining what to retrieve).However, current dynamic RAG methods fall short in both aspects.Firstly, the strategies for deciding when to retrieve often rely on static rules.Moreover, the strategies for deciding what to retrieve typically limit themselves to the LLM's most recent sentence or the last few tokens, while the LLM's information needs may span across the entire context.To overcome these limitations, we introduce a new framework, DRAGIN, i.e., Dynamic Retrieval Augmented Generation based on the Information Needs of LLMs.Our framework is specifically designed to make decisions on when and what to retrieve based on the LLM's information needs during the text generation process.We evaluate DRAGIN along with existing methods comprehensively over 4 knowledge-intensive generation datasets.Experimental results show that DRAGIN achieves superior performance on all tasks, demonstrating the effectiveness of our method 1 .
Weihang Su, Yichen Tang 0001, Qingyao Ai, Zhijing Wu 0001, Yiqun Liu 0001
ACL (1)1
2024 Scaling Laws For Dense Retrieval
abstract
Scaling laws have been observed in a wide range of tasks, particularly in language generation. Previous studies have found that the performance of large language models adheres to predictable patterns with respect to the size of models and datasets. This helps us design training strategies effectively and efficiently, especially as large-scale training becomes increasingly resource-intensive. Yet, in dense retrieval, such scaling law has not been fully explored. In this study, we investigate how scaling affects the performance of dense retrieval models. We implement dense retrieval models with different numbers of parameters, and train them with various amounts of annotated data. We propose to use the contrastive entropy as the evaluation metric, which is continuous compared with discrete ranking metrics and thus can accurately reflect model performance. Results indicate that the performance of dense retrieval models follows a precise power-law scaling related to the model size and the number of annotations across different datasets and annotation methods. Additionally, we show that the scaling laws help optimize the training process, such as resolving the resource allocation problem under a budget constraint. We believe that these findings significantly contribute to understanding the scaling effect of dense retrieval models and offer meaningful guidance for future research.
Jingtao Zhan, Qingyao Ai, Jiaxin Mao, Weihang Su, Jia Chen 0003, Yiqun Liu 0001
SIGIR5
2024 Relevance Feedback with Brain Signals
abstract
The Relevance Feedback (RF) process relies on accurate and real-time relevance estimation of feedback documents to improve retrieval performance. Since collecting explicit relevance annotations imposes an extra burden on the user, extensive studies have explored using pseudo-relevance signals and implicit feedback signals as substitutes. However, such signals are indirect indicators of relevance and suffer from complex search scenarios where user interactions are absent or biased. Recently, the advances in portable and high-precision brain-computer interface (BCI) devices have shown the possibility to monitor user’s brain activities during search process. Brain signals can directly reflect user’s psychological responses to search results and thus it can act as additional and unbiased RF signals. To explore the effectiveness of brain signals in the context of RF, we propose a novel RF framework that combines BCI-based RF with pseudo-relevance signals and implicit signals to improve the performance of document re-ranking. The experimental results on the user study dataset show that incorporating brain signals leads to significant performance improvement in our RF framework. Besides, we observe that brain signals perform particularly well in several hard search scenarios, especially when implicit signals as feedback are missing or noisy. This reveals when and how to exploit brain signals in the context of RF.
Ziyi Ye, Xiaohui Xie, Qingyao Ai, Yiqun Liu 0001, Weihang Su, Min Zhang 0006
ACM Trans. Inf. Syst.6
2023 CaseEncoder: A Knowledge-enhanced Pre-trained Model for Legal Case Encoding
abstract
Legal case retrieval is a critical process for modern legal information systems.While recent studies have utilized pre-trained language models (PLMs) based on the general domain selfsupervised pre-training paradigm to build models for legal case retrieval, there are limitations in using general domain PLMs as backbones.Specifically, these models may not fully capture the underlying legal features in legal case documents.To address this issue, we propose CaseEncoder, a legal document encoder that leverages fine-grained legal knowledge in both the data sampling and pre-training phases.In the data sampling phase, we enhance the quality of the training data by utilizing fine-grained law article information to guide the selection of positive and negative examples.In the pretraining phase, we design legal-specific pretraining tasks that align with the judging criteria of relevant legal cases.Based on these tasks, we introduce an innovative loss function called Biased Circle Loss to enhance the model's ability to recognize case relevance in fine grains.Experimental results on multiple benchmarks
Yueyue Wu, Weihang Su, Qingyao Ai, Yiqun Liu 0001
EMNLP3
2022 Web Search via an Efficient and Effective Brain-Machine Interface
abstract
While search technologies have evolved to be robust and ubiquitous, the fundamental interaction paradigm has remained relatively stable for decades. With the maturity of the Brain-Machine Interface(BMI), we build an efficient and effective communication system between human beings and search engines based on electroencephalogram (EEG) signals, called Brain Machine Search Interface (BMSI)system. The BMSI system provides functions including query reformulation and search result interaction. In our system, users can perform search tasks without having to use the mouse and keyboard. Therefore, it is useful for application scenarios in which hand-based interactions are infeasible, e.g, for users with severe neuromuscular disorders. Besides, based on brain signals decoding, our system can provide abundant and valuable user-side context information (e.g., real-time satisfaction feedback, extensive context information, and a clearer description of information needs) to the search engine, which is hard to capture in the previous paradigm. In our implementation, the system can decode user satisfaction from brain signals in real-time during the interaction process and re-rank the search results list based on user satisfaction feedback.The demo video is available at http://www.thuir.cn/group/YQLiu/videos/BMSISystem.html
Xuesong Chen 0005, Ziyi Ye, Xiaohui Xie, Yiqun Liu 0001, Xiaorong Gao, Weihang Su, Shuqi Zhu, Yike Sun, Min Zhang 0006, Shaoping Ma
WSDM6
2021 Understanding the Impact of COVID-19 on Github Developers: A Preliminary Study
abstract
The ongoing COVID-19 pandemic has impact almost every aspect of human lives profoundly.This paper investigates the impact of COVID-19 on the activity and contribution of open source software (OSS) developers.Specifically, we make great efforts to harvest the information of all the developers (over 25 million) on GitHub and their contribution activities.With such a large-scale dataset, we perform analysis from four perspectives, including the overall ecosystem level, country level, organization level and developer level, to characterize the impact of COVID-19 on the OSS community.We have revealed a number of interesting observations and trends, which are crucial to understanding the OSS contributors and supporting the collaboration to combat global crisis like COVID-19.
Liu Wang 0002, Guangdong Bai, Weihang Su, Haoyu Wang 0001
SEKE5