← Search

Kelong Mao

19 accepted papers

2025

A Silver Bullet or a Compromise for Full Attention? A Comprehensive Study of Gist Token-based Context Compression

ACL 2025long

In this work, we provide an empirical investigation of gist-based context compression methods to improve context processing in large language models. We focus on two key questions: (1) How well can these methods replace full attention models? and (2) What potential failure patterns arise due to comp…

2025

CORAL: Benchmarking Multi-turn Conversational Retrieval-Augmented Generation

NAACL 2025findings

Retrieval-Augmented Generation (RAG) has become a powerful paradigm for enhancing large language models (LLMs) through external knowledge retrieval. Despite its widespread attention, existing academic research predominantly focuses on single-turn RAG, leaving a significant gap in addressing the comp…

2025

UniConv: Unifying Retrieval and Response Generation for Large Language Models in Conversations

ACL 2025long

The rapid advancement of conversational search systems revolutionizes how information is accessed by enabling the multi-turn interaction between the user and the system. Existing conversational search systems are usually built with two different models. This separation restricts the system from leve…

Cited by 0SourcePDFScholar
2025

UniGist: Towards General and Hardware-aligned Sequence-level Long Context Compression

NeurIPS 2025poster

Large language models are increasingly capable of handling long-context inputs, but the memory overhead of KV cache remains a major bottleneck for general-purpose deployment. While many compression strategies have been explored, sequence-level compression is particularly challenging due to its tende…

Cited by 0SourceScholar
2024

An Element is Worth a Thousand Words: Enhancing Legal Case Retrieval by Incorporating Legal Elements

ACL 2024findings

Legal case retrieval plays an important role in promoting judicial justice and fairness. One of its greatest challenges is that the definition of relevance goes far beyond the common semantic relevance as in ad-hoc retrieval. In this paper, we reveal that the legal elements, which typically comprise…

2024

CHIQ: Contextual History Enhancement for Improving Query Rewriting in Conversational Search

EMNLP 2024main

In this paper, we study how open-source large language models (LLMs) can be effectively deployed for improving query rewriting in conversational search, especially for ambiguous queries. We introduce CHIQ, a two-step method that leverages the capabilities of LLMs to resolve ambiguities in the conver…

2024

ChatRetriever: Adapting Large Language Models for Generalized and Robust Conversational Dense Retrieval

EMNLP 2024main

Conversational search requires accurate interpretation of user intent from complex multi-turn contexts. This paper presents ChatRetriever, which inherits the strong generalization capability of large language models to robustly represent complex conversational sessions for dense retrieval. To achiev…

2024

Enabling Discriminative Reasoning in LLMs for Legal Judgment Prediction

EMNLP 2024finding

Legal judgment prediction is essential for enhancing judicial efficiency. In this work, we identify that existing large language models (LLMs) underperform in this domain due to challenges in understanding case complexities and distinguishing between similar charges. To adapt LLMs for effective lega…

2024

Generalizing Conversational Dense Retrieval via LLM-Cognition Data Augmentation

ACL 2024long

Conversational search utilizes muli-turn natural language contexts to retrieve relevant passages. Existing conversational dense retrieval models mostly view a conversation as a fixed sequence of questions and responses, overlooking the severe data sparsity problem – that is, users can perform a conv…

2024

Grounding Language Model with Chunking-Free In-Context Retrieval

ACL 2024long

This paper presents a novel Chunking-Free In-Context (CFIC) retrieval approach, specifically tailored for Retrieval-Augmented Generation (RAG) systems. Traditional RAG systems often struggle with grounding responses using precise evidence text due to the challenges of processing lengthy documents an…

Cited by 16SourcePDFScholar
2024

History-Aware Conversational Dense Retrieval

ACL 2024findings

Conversational search facilitates complex information retrieval by enabling multi-turn interactions between users and the system. Supporting such interactions requires a comprehensive understanding of the conversational inputs to formulate a good search query based on historical information. In part…

2024

Interpreting Conversational Dense Retrieval by Rewriting-Enhanced Inversion of Session Embedding

ACL 2024long

Conversational dense retrieval has shown to be effective in conversational search. However, a major limitation of conversational dense retrieval is their lack of interpretability, hindering intuitive understanding of model behaviors for targeted improvements. This paper presents CONVINV, a simple ye…

2024

Learning Interpretable Legal Case Retrieval via Knowledge-Guided Case Reformulation

EMNLP 2024main

Legal case retrieval for sourcing similar cases is critical in upholding judicial fairness. Different from general web search, legal case retrieval involves processing lengthy, complex, and highly specialized legal documents. Existing methods in this domain often overlook the incorporation of legal…

2024

RAG-Studio: Towards In-Domain Adaptation of Retrieval Augmented Generation Through Self-Alignment

EMNLP 2024finding

Retrieval-Augmented Generation (RAG) has proven to be an effective paradigm for enhancing the quality of text generation by integrating large language models (LLMs) with external knowledge. However, an off-the-shelf RAG system, which relies on generally pre-trained LLMs and retrievers, often falls s…

2023

ConvGQR: Generative Query Reformulation for Conversational Search

ACL 2023long

In conversational search, the user’s real search intent for the current conversation turn is dependent on the previous conversation history. It is challenging to determine a good search query from the whole conversation context. To avoid the expensive re-training of the query encoder, most existing…

2023

FinalMLP: An Enhanced Two-Stream MLP Model for CTR Prediction

AAAI 2023technical

Click-through rate (CTR) prediction is one of the fundamental tasks in online advertising and recommendation. Multi-layer perceptron (MLP) serves as a core component in many deep CTR prediction models, but it has been widely shown that applying a vanilla MLP network alone is ineffective in learning…

2023

Large Language Models Know Your Contextual Search Intent: A Prompting Framework for Conversational Search

EMNLP 2023long findings

Precisely understanding users' contextual search intent has been an important challenge for conversational search. As conversational search sessions are much more diverse and long-tailed, existing methods trained on limited data still show unsatisfactory effectiveness and robustness to handle real c…

Cited by 0SourceScholar
2023

Search-Oriented Conversational Query Editing

ACL 2023findings

Conversational query rewriting (CQR) realizes conversational search by reformulating the search dialogue into a standalone rewrite. However, existing CQR models either are not learned toward improving the downstream search performance or inefficiently generate the rewrite token-by-token from scratch…

2022

ConvTrans: Transforming Web Search Sessions for Conversational Dense Retrieval

EMNLP 2022main

Conversational search provides users with a natural and convenient new search experience. Recently, conversational dense retrieval has shown to be a promising technique for realizing conversational search. However, as conversational search systems have not been widely deployed, it is hard to get lar…