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

Kelong Mao

dblp:270/6458 · DBLP profile ↗
← Back
25ranked-venue papers
10as first author
24since 2021 · last 2025
0000-0002-5648-568XORCID · corroborated

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

Artificial intelligence and machine learning · 18 · 6 first-author · 18 since 2021Databases, data management, data science and information retrieval · 10 · 5 first-author · 9 since 2021Applied, interdisciplinary, general and emerging computing · 2 · 1 first-author · 2 since 2021Computer networks · 1 · 1 first-author · 1 since 2021Security and privacy · 1 · 1 since 2021Graphics, computer vision, multimedia, augmented reality and games · 1 · 1 first-author · 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
16 papers
Information retrieval · 88% Query processing and optimization · 6% Recommender systems · 4%
Artificial intelligence
12 papers
Language models and text generation · 60% Deep learning architectures and training · 16% Question answering and dialogue systems · 15%
Computer networks
1 paper
Network measurement and analytics · 100%

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

TopicWeightPapersLastEvidence papers
Information retrieval › interactive information retrieval › conversational information seeking
conversational search
6.692025
A Survey of Conversational Search · ACM Trans. Inf. Syst. 2025
UniConv: Unifying Retrieval and Response Generation for Large Language Models in Conversations · ACL (1) 2025
CHIQ: Contextual History Enhancement for Improving Query Rewriting in Conversational Search · EMNLP 2024
Information retrieval › interactive information retrieval › conversational information seeking › conversational search
conversational dense retrieval
3.552024
ChatRetriever: Adapting Large Language Models for Generalized and Robust Conversational Dense Retrieval · EMNLP 2024
Interpreting Conversational Dense Retrieval by Rewriting-Enhanced Inversion of Session Embedding · ACL (1) 2024
Generalizing Conversational Dense Retrieval via LLM-Cognition Data Augmentation · ACL (1) 2024
Information retrieval › retrieval models › neural retrieval
dense retrieval
2.952025
ChatRetriever: Adapting Large Language Models for Generalized and Robust Conversational Dense Retrieval · EMNLP 2024
Generalizing Conversational Dense Retrieval via LLM-Cognition Data Augmentation · ACL (1) 2024
Curriculum Contrastive Context Denoising for Few-shot Conversational Dense Retrieval · SIGIR 2022
Natural language and speech › Language models and text generation › language modeling › long-context language modeling › context utilization › long-context modeling
long-context language model
1.722025
MemoRAG: Boosting Long Context Processing with Global Memory-Enhanced Retrieval Augmentation · WWW 2025
A Silver Bullet or a Compromise for Full Attention? A Comprehensive Study of Gist Token-based Context Compression · ACL (1) 2025
Query processing and optimization
query rewriting
1.632024
CHIQ: Contextual History Enhancement for Improving Query Rewriting in Conversational Search · EMNLP 2024
Learning to Relate to Previous Turns in Conversational Search · KDD 2023
Interpreting Conversational Dense Retrieval by Rewriting-Enhanced Inversion of Session Embedding · ACL (1) 2024
Information retrieval
retrieval-augmented generation
1.622025
MemoRAG: Boosting Long Context Processing with Global Memory-Enhanced Retrieval Augmentation · WWW 2025
Grounding Language Model with Chunking-Free In-Context Retrieval · ACL (1) 2024
Information retrieval
query reformulation
1.522025
A Survey of Conversational Search · ACM Trans. Inf. Syst. 2025
ConvGQR: Generative Query Reformulation for Conversational Search · ACL (1) 2023
Information retrieval
retrieval models
0.922025
Learning to Relate to Previous Turns in Conversational Search · KDD 2023
MemoRAG: Boosting Long Context Processing with Global Memory-Enhanced Retrieval Augmentation · WWW 2025
Machine learning › Deep learning architectures and training
attention mechanism
0.912025
A Silver Bullet or a Compromise for Full Attention? A Comprehensive Study of Gist Token-based Context Compression · ACL (1) 2025
Machine learning › Efficient and distributed learning › inference efficiency
context compression
0.912025
A Silver Bullet or a Compromise for Full Attention? A Comprehensive Study of Gist Token-based Context Compression · ACL (1) 2025
Natural language and speech › Question answering and dialogue systems
knowledge-intensive tasks
0.912025
Tackling the Length Barrier: Dynamic Context Browsing for Knowledge-Intensive Task · KDD (1) 2025
Natural language and speech › Language models and text generation › language modeling
long-context language modeling
0.912025
Tackling the Length Barrier: Dynamic Context Browsing for Knowledge-Intensive Task · KDD (1) 2025
Natural language and speech › Language models and text generation › large language model reasoning
long-context reasoning
0.912025
Tackling the Length Barrier: Dynamic Context Browsing for Knowledge-Intensive Task · KDD (1) 2025
Natural language and speech › Language models and text generation
retrieval augmentation
0.912025
MemoRAG: Boosting Long Context Processing with Global Memory-Enhanced Retrieval Augmentation · WWW 2025
Information retrieval › query understanding
query clarification
0.912025
A Survey of Conversational Search · ACM Trans. Inf. Syst. 2025
Information retrieval › interactive information retrieval › conversational information seeking
response generation
0.912025
A Survey of Conversational Search · ACM Trans. Inf. Syst. 2025
Network measurement and analytics › traffic classification
encrypted traffic classification
0.912025
RBLJAN: Robust Byte-Label Joint Attention Network for Network Traffic Classification · IEEE Trans. Dependable Secur. Comput. 2025
Network measurement and analytics
traffic classification
0.912025
RBLJAN: Robust Byte-Label Joint Attention Network for Network Traffic Classification · IEEE Trans. Dependable Secur. Comput. 2025
Natural language and speech › Language models and text generation › large language model
large language model adaptation
0.812024
ChatRetriever: Adapting Large Language Models for Generalized and Robust Conversational Dense Retrieval · EMNLP 2024
Natural language and speech › Question answering and dialogue systems
open-domain question answering
0.812024
Grounding Language Model with Chunking-Free In-Context Retrieval · ACL (1) 2024
Information retrieval
evidence retrieval
0.812024
Grounding Language Model with Chunking-Free In-Context Retrieval · ACL (1) 2024
Information retrieval › document retrieval › domain-specific retrieval › legal information retrieval
legal case retrieval
0.812024
Learning Interpretable Legal Case Retrieval via Knowledge-Guided Case Reformulation · EMNLP 2024
Machine learning › Deep learning architectures and training
feature interaction
0.712023
FinalMLP: An Enhanced Two-Stream MLP Model for CTR Prediction · AAAI 2023
Recommender systems
click-through rate prediction
0.712023
FinalMLP: An Enhanced Two-Stream MLP Model for CTR Prediction · AAAI 2023
Information retrieval › query reformulation
query expansion
0.712023
Learning to Relate to Previous Turns in Conversational Search · KDD 2023
Natural language and speech › Language models and text generation
large language model
0.522025
UniConv: Unifying Retrieval and Response Generation for Large Language Models in Conversations · ACL (1) 2025
CHIQ: Contextual History Enhancement for Improving Query Rewriting in Conversational Search · EMNLP 2024
Recommender systems
item tagging
0.412020
Item Tagging for Information Retrieval: A Tripartite Graph Neural Network based Approach · SIGIR 2020
Information retrieval › retrieval models › query-document matching
relevance matching
0.412020
Item Tagging for Information Retrieval: A Tripartite Graph Neural Network based Approach · SIGIR 2020
Data mining › text mining › text classification
tag prediction
0.412020
Item Tagging for Information Retrieval: A Tripartite Graph Neural Network based Approach · SIGIR 2020
Network security › intrusion detection and prevention
intrusion detection
0.312025
RBLJAN: Robust Byte-Label Joint Attention Network for Network Traffic Classification · IEEE Trans. Dependable Secur. Comput. 2025

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

large language model · 5.6contrastive learning · 3.6retrieval-augmented generation · 1.7retrieval augmentation · 1.7global memory · 1.7deep learning · 1.7byte-label joint attention · 1.7adversarial traffic generation · 1.7difficulty-adaptive sample filtering · 1.5cognition-aware prompting · 1.5prompting · 0.9oracle short-context extraction · 0.9gist token compression · 0.9vec2text · 0.8embedding inversion · 0.8autoregressive decoding · 0.8
YearPublicationVenuePosition
2025 A Silver Bullet or a Compromise for Full Attention? A Comprehensive Study of Gist Token-based Context Compression
abstract
Chenlong Deng, Zhisong Zhang, Kelong Mao, Shuaiyi Li, Xinting Huang, Dong Yu, Zhicheng Dou. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025.
Chenlong Deng, Zhisong Zhang, Kelong Mao, Shuaiyi Li, Xinting Huang, Dong Yu 0001, Zhicheng Dou
ACL (1)3
2025 UniConv: Unifying Retrieval and Response Generation for Large Language Models in Conversations
abstract
Fengran Mo, Yifan Gao, Chuan Meng, Xin Liu, Zhuofeng Wu, Kelong Mao, Zhengyang Wang, Pei Chen, Zheng Li, Xian Li, Bing Yin, Meng Jiang. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025.
Fengran Mo, Yifan Gao 0001, Chuan Meng, Xin Liu 0039, Zhuofeng Wu 0005, Kelong Mao, Zheng Li 0018, Meng Jiang 0001
ACL (1)6
2025 Tackling the Length Barrier: Dynamic Context Browsing for Knowledge-Intensive Task
abstract
Knowledge-intensive tasks often require complex reasoning and contextual understanding over long contexts. However, the learning and deployment of long-LLMs remains a challenging problem despite recent progresses. In this work, we propose that the short LLMs have great potentiality for solving knowledge-intensive tasks that have long context, i.e. they can be solved by purely working with oracle short-contexts within the input long-context. On top of this argument, we propose a framework called DCISO DynamiC knowledge-Intensive task S>Olver), which enables a short-LLM to address the knowledge-intensive tasks with long context via dynamic context browsing. In our framework, the short-LLM prompts itself to reason for two critical decisions: 1) how to access to the appropriate part of context within the input, 2) how to make effective use of the accessed context. By adaptively accessing and utilizing the context based on the presented tasks, DCISO can serve as a general framework to handle diversified knowledge-intensive long-context problems. We comprehensively evaluate different types of tasks from popular long-context benchmarks, where DCISO is able to achieve a substantially improved performance. Our codes will be released at this repository.
Hongjin Qian, Zheng Liu 0011, Peitian Zhang, Kelong Mao, Yujia Zhou 0002, Xu Chen 0017, Zhicheng Dou
KDD (1)4
2025 UniGist: Towards General and Hardware-aligned Sequence-level Long Context Compression
abstract
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 tendency to lose important details. We present UniGist, a gist token-based long context compression framework that removes the need for chunk-wise training, enabling the model to learn how to compress and utilize long-range context during training. To fully exploit the sparsity, we introduce a gist shift trick that transforms the attention layout into a right-aligned block structure and develop a block-table-free sparse attention kernel based on it. UniGist further supports one-pass training and flexible chunk sizes during inference, allowing efficient and adaptive context processing. Experiments across multiple long-context tasks show that UniGist significantly improves compression quality, with especially strong performance in recalling details and long-range dependency modeling.
Chenlong Deng, Zhisong Zhang, Kelong Mao, Shuaiyi Li, Tianqing Fang, Hongming Zhang 0009, Haitao Mi, Dong Yu 0001, Zhicheng Dou
NeurIPS3
2025 MemoRAG: Boosting Long Context Processing with Global Memory-Enhanced Retrieval Augmentation
abstract
Processing long contexts presents a significant challenge for large language models (LLMs). While recent advancements allow LLMs to handle much longer contexts than before (e.g., 32K or 128K tokens), it is computationally expensive and can still be insufficient for many applications. Retrieval-Augmented Generation (RAG) is considered a promising strategy to address this problem. However, conventional RAG methods face inherent limitations because of two underlying requirements: 1) explicitly stated queries, and 2) well-structured knowledge. These conditions, however, do not hold in general long-context processing tasks.
Hongjin Qian, Zheng Liu 0011, Peitian Zhang, Kelong Mao, Defu Lian, Zhicheng Dou, Tiejun Huang 0001
WWW4
2025 RBLJAN: Robust Byte-Label Joint Attention Network for Network Traffic Classification
abstract
Network traffic classification plays a crucial role in network management and cyberspace security. As the Internet evolves with new applications and protocols, traditional machine learning-based methods relying on feature mining have become obsolete. Instead, deep learning-based methods are becoming more popular in the field of traffic classification due to their end-to-end processing approach. However, the vulnerability of neural networks to adversarial examples significantly compromises their performance. In this paper, we propose Robust Byte-Label Joint Attention Network (RBLJAN), an efficient and robust deep learning-based framework for encrypted network traffic classification at both the packet-level and the flow-level. RBLJAN comprises a classifier and an adversarial traffic generator. The classifier utilizes mechanisms such as header-payload parallel processing and byte-label joint attention learning to capture implicit correlations between bytes and labels, enabling the construction of powerful packet representations. The generator produces adversarial examples that are fed to the classifier to enhance its robustness. Experimental results demonstrate that RBLJAN achieves over 99% average F1-score on real-world legitimate traffic datasets and achieves 97.86% average F1-score on malware identification. Moreover, RBLJAN exhibits superior performance in terms of detection speed and robustness compared to state-of-the-art methods in real-world scenarios.
Xi Xiao 0001, Shuo Wang 0012, Guangwu Hu, Qing Li 0006, Kelong Mao, Xiapu Luo, Bin Zhang 0048, Shutao Xia
IEEE Trans. Dependable Secur. Comput.5
2025 A Survey of Conversational Search
abstract
As a cornerstone of modern information access, search engines have become indispensable in everyday life. With the rapid advancements in AI and natural language processing (NLP) technologies, particularly large language models (LLMs), search engines have evolved to support more intuitive and intelligent interactions between users and systems. Conversational search, an emerging paradigm for next-generation search engines, leverages natural language dialogue to facilitate complex and precise information retrieval, thus attracting significant attention. Unlike traditional keyword-based search engines, conversational search systems enhance user experience by supporting intricate queries, maintaining context over multi-turn interactions, and providing robust information integration and processing capabilities. Key components such as query reformulation, search clarification, conversational retrieval, and response generation work in unison to enable these sophisticated interactions. In this survey, we explore the recent advancements and potential future directions in conversational search, examining the critical modules that constitute a conversational search system. We highlight the integration of LLMs in enhancing these systems and discuss the challenges and opportunities that lie ahead in this dynamic field. Additionally, we provide insights into real-world applications and robust evaluations of current conversational search systems, aiming to guide future research and development in conversational search.
Fengran Mo, Kelong Mao, Ziliang Zhao 0001, Hongjin Qian, Haonan Chen 0005, Yiruo Cheng, Xiaoxi Li 0005, Yutao Zhu 0001, Zhicheng Dou, Jian-Yun Nie
ACM Trans. Inf. Syst.2
2024 Generalizing Conversational Dense Retrieval via LLM-Cognition Data Augmentation
abstract
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 conversation in various ways.Consequently, they often struggle to generalize to diverse conversations in real-world scenarios.In this work, we propose a framework for generalizing Conversational dense retrieval via LLMcognition data Augmentation (CONVAUG).We first generate multi-level augmented conversations to capture the diverse nature of conversational contexts.Inspired by human cognition, we devise a cognition-aware prompting process to mitigate the generation of false positives, false negatives, and hallucinations.Moreover, we develop a difficulty-adaptive sample filter that selects challenging samples for complex conversations, thereby giving the model a larger learning space.A contrastive learning objective is then employed to train a better conversational context encoder.Extensive experiments conducted on four public datasets, under both normal and zero-shot settings, demonstrate the effectiveness, generalizability, and applicability of CONVAUG.The code is released at https://github.com/haon-chen/ConvAug.
Haonan Chen 0005, Zhicheng Dou, Kelong Mao, Jiongnan Liu 0001, Ziliang Zhao 0001
ACL (1)3
2024 Interpreting Conversational Dense Retrieval by Rewriting-Enhanced Inversion of Session Embedding
abstract
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 yet effective approach to shed light on interpretable conversational dense retrieval models.CONVINV transforms opaque conversational session embeddings into explicitly interpretable text while faithfully maintaining their original retrieval performance as much as possible.Such transformation is achieved by training a recently proposed Vec2Text model (Morris et al., 2023) based on the ad-hoc query encoder, leveraging the fact that the session and query embeddings share the same space in existing conversational dense retrieval.To further enhance interpretability, we propose to incorporate external interpretable query rewrites into the transformation process.Extensive evaluations on three conversational search benchmarks demonstrate that CONVINV can yield more interpretable text and faithfully preserve original retrieval performance than baselines.Our work connects opaque session embeddings with transparent query rewriting, paving the way toward trustworthy conversational search.Our code is available at this repository.
Yiruo Cheng, Kelong Mao, Zhicheng Dou
ACL (1)2
2024 Grounding Language Model with Chunking-Free In-Context Retrieval
abstract
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 and filtering out irrelevant content.Commonly employed solutions, such as document chunking and adapting language models to handle longer contexts, have their limitations.These methods either disrupt the semantic coherence of the text or fail to effectively address the issues of noise and inaccuracy in evidence retrieval.CFIC addresses these challenges by circumventing the conventional chunking process.It utilizes the encoded hidden states of documents for in-context retrieval, employing autoaggressive decoding to accurately identify the specific evidence text required for user queries, eliminating the need for chunking.CFIC is further enhanced by incorporating two decoding strategies, namely Constrained Sentence Prefix Decoding and Skip Decoding.These strategies not only improve the efficiency of the retrieval process but also ensure that the fidelity of the generated grounding text evidence is maintained.Our evaluations of CFIC on a range of open QA datasets demonstrate its superiority in retrieving relevant and accurate evidence, offering a significant improvement over traditional methods.By doing away with the need for document chunking, CFIC presents a more streamlined, effective, and efficient retrieval solution, making it a valuable advancement in the field of RAG systems.The codes will be released in this repository.
Hongjin Qian, Zheng Liu 0011, Kelong Mao, Yujia Zhou 0002, Zhicheng Dou
ACL (1)3
2024 Aligning Query Representation with Rewritten Query and Relevance Judgments in Conversational Search
abstract
Conversational search supports multi-turn user-system interactions to solve complex information needs. Different from the traditional single-turn ad-hoc search, conversational search encounters a more challenging problem of context-dependent query understanding with the lengthy and long-tail conversational history context. While conversational query rewriting (CQR) methods leverage explicit rewritten queries to train a rewriting model to transform the context-dependent query into a stand-stone search query, this is usually done without considering the quality of search results. Conversational dense retrieval (CDR) methods use fine-tuning to improve a pre-trained ad-hoc query encoder, but they are limited by the conversational search data available for training. In this paper, we leverage both rewritten queries and relevance judgments in the conversational search data to train a better query representation model. The key idea is to align the query representation with those of rewritten queries and relevant documents. The proposed model -- Query Representation Alignment Conversational Dense Retriever, QRACDR, is tested on eight datasets, including various settings in conversational search and ad-hoc search. The results demonstrate the strong performance of QRACDR compared with other state-of-the-art methods, and confirm the effectiveness of representation alignment.
Fengran Mo, Chen Qu 0001, Kelong Mao, Yihong Wu 0006, Zhan Su 0002, Jian-Yun Nie
CIKM3
2024 Learning Interpretable Legal Case Retrieval via Knowledge-Guided Case Reformulation
abstract
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 expert knowledge, which is crucial for accurately understanding and modeling legal cases, leading to unsatisfactory retrieval performance.This paper introduces KELLER, a legal knowledge-guided case reformulation approach based on large language models (LLMs) for effective and interpretable legal case retrieval.By incorporating professional legal knowledge about crimes and law articles, we enable large language models to accurately reformulate the original legal case into concise sub-facts of crimes, which contain the essential information of the case.Extensive experiments on two legal case retrieval benchmarks demonstrate superior retrieval performance and robustness on complex legal case queries of KELLER over existing methods.
Chenlong Deng, Kelong Mao, Zhicheng Dou
EMNLP2
2024 ChatRetriever: Adapting Large Language Models for Generalized and Robust Conversational Dense Retrieval
abstract
Conversational search requires accurate interpretation of user intent from complex multiturn 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 achieve this, we propose a simple and effective dual-learning approach that adapts LLM for retrieval via contrastive learning while enhancing the complex session understanding through masked instruction tuning on high-quality conversational instruction tuning data.Extensive experiments on five conversational search benchmarks demonstrate that ChatRetriever substantially outperforms existing conversational dense retrievers, achieving state-of-the-art performance on par with LLM-based rewriting approaches.Furthermore, ChatRetriever exhibits superior robustness in handling diverse conversational contexts.Our work highlights the potential of adapting LLMs for retrieval with complex inputs like conversational search sessions and proposes an effective approach to advance this research direction.
Kelong Mao, Chenlong Deng, Haonan Chen 0005, Fengran Mo, Zheng Liu 0011, Tetsuya Sakai, Zhicheng Dou
EMNLP1
2024 CHIQ: Contextual History Enhancement for Improving Query Rewriting in Conversational Search
abstract
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 conversation history before query rewriting.This approach contrasts with prior studies that predominantly use closed-source LLMs to directly generate search queries from conversation history.We demonstrate on five well-established benchmarks that CHIQ leads to state-of-the-art results across most settings, showing highly competitive performances with systems leveraging closed-source LLMs.Our study provides a first step towards leveraging open-source LLMs in conversational search, as a competitive alternative to the prevailing reliance on commercial LLMs for query rewriting.
Fengran Mo, Abbas Ghaddar, Kelong Mao, Mehdi Rezagholizadeh, Boxing Chen, Qun Liu 0001, Jian-Yun Nie
EMNLP3
2023 FinalMLP: An Enhanced Two-Stream MLP Model for CTR Prediction
abstract
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 complex feature interactions. As such, many two-stream models (e.g., Wide&Deep, DeepFM, and DCN) have recently been proposed, aiming to integrate two parallel sub-networks to learn feature interactions from two different views for enhanced CTR prediction. In addition to one MLP stream that learns feature interactions implicitly, most of the existing research focuses on designing another stream to complement the MLP stream with explicitly enhanced feature interactions. Instead, this paper presents a simple two-stream feature interaction model, namely FinalMLP, which employs only MLPs in both streams yet achieves surprisingly strong performance. In contrast to sophisticated network design in each stream, our work enhances CTR modeling through a feature selection module, which produces differentiated feature inputs to two streams, and a group-wise bilinear fusion module, which effectively captures stream-level interactions across two streams. We show that FinalMLP achieves competitive or even better performance against many existing two-stream CTR models on four open benchmark datasets and also brings significant CTR improvements during an online A/B test in our industrial news recommender system. We envision that the simple yet effective FinalMLP model could serve as a new strong baseline for future development of two-stream CTR models. Our source code will be available at MindSpore/models and FuxiCTR/model_zoo.
Kelong Mao, Jieming Zhu, Liangcai Su, Guohao Cai, Yuru Li, Zhenhua Dong
AAAI1
2023 ConvGQR: Generative Query Reformulation for Conversational Search
abstract
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 methods try to learn a rewriting model to de-contextualize the current query by mimicking the manual query rewriting.However, manually rewritten queries are not always the best search queries.Thus, training a rewriting model on them would lead to sub-optimal queries.Another useful information to enhance the search query is the potential answer to the question.In this paper, we propose ConvGQR, a new framework to reformulate conversational queries based on generative pre-trained language models (PLMs), one for query rewriting and another for generating potential answers.By combining both, ConvGQR can produce better search queries.In addition, to relate query reformulation to the retrieval task, we propose a knowledge infusion mechanism to optimize both query reformulation and retrieval.Extensive experiments on four conversational search datasets demonstrate the effectiveness of ConvGQR.
Fengran Mo, Kelong Mao, Yutao Zhu 0001, Yihong Wu 0006, Jian-Yun Nie
ACL (1)2
2023 Learning to Relate to Previous Turns in Conversational Search
abstract
Conversational search allows a user to interact with a search system in multiple turns. A query is strongly dependent on the conversation context. An effective way to improve retrieval effectiveness is to expand the current query with historical queries. However, not all the previous queries are related to, and useful for expanding the current query. In this paper, we propose a new method to select relevant historical queries that are useful for the current query. To cope with the lack of labeled training data, we use a pseudo-labeling approach to annotate useful historical queries based on their impact on the retrieval results. The pseudo-labeled data are used to train a selection model. We further propose a multi-task learning framework to jointly train the selector and the retriever during fine-tuning, allowing us to mitigate the possible inconsistency between the pseudo labels and the changed retriever. Extensive experiments on four conversational search datasets demonstrate the effectiveness and broad applicability of our method compared with several strong baselines.
Fengran Mo, Jian-Yun Nie, Kelong Mao, Yutao Zhu 0001, Peng Li 0030, Yang Liu 0005
KDD4
2023 Learning Denoised and Interpretable Session Representation for Conversational Search
abstract
Conversational search supports multi-turn user-system interactions to solve complex information needs. Compared with the traditional single-turn ad-hoc search, conversational search faces a more complex search intent understanding problem because a conversational search session is much longer and contains many noisy tokens. However, existing conversational dense retrieval solutions simply fine-tune the pre-trained ad-hoc query encoder on limited conversational search data, which are hard to achieve satisfactory performance in such a complex conversational search scenario. Meanwhile, the learned latent representation also lacks interpretability that people cannot perceive how the model understands the session. To tackle the above drawbacks, we propose a sparse Lexical-based Conversational REtriever (LeCoRE), which extends the SPLADE model with two well-matched multi-level denoising methods uniformly based on knowledge distillation and external query rewrites to generate denoised and interpretable lexical session representation. Extensive experiments on four public conversational search datasets in both normal and zero-shot evaluation settings demonstrate the strong performance of LeCoRE towards more effective and interpretable conversational search.
Kelong Mao, Hongjin Qian, Fengran Mo, Zhicheng Dou, Bang Liu 0003, Xiaohua Cheng, Zhao Cao
WWW1
2022 ConvTrans: Transforming Web Search Sessions for Conversational Dense Retrieval
abstract
Conversational search provides users with natural and convenient new search experience.Recently, conversational dense retrieval has shown to be a promising technique to realize conversational search.However, as conversational search systems have not been widely deployed, it is hard to get large-scale real conversational search sessions and relevance labels to support the training of conversational dense retrieval.To tackle this data scarcity problem, previous methods focus on developing better few-shot learning approaches or generating pseudo relevance labels, but the data they use for training still heavily rely on manual generation.In this paper, we present ConvTrans, a data augmentation method that can automatically transform easily-accessible web search sessions into conversational search sessions to fundamentally alleviate the data scarcity problem for conversational dense retrieval.ConvTrans eliminates the gaps between these two types of sessions in terms of session quality and query form to achieve effective session transformation.Extensive evaluations on two widely used conversational search benchmarks, i.e., CAsT-19 and CAsT-20, demonstrate that the same model trained on the data generated by Con-vTrans can achieve comparable retrieval performance as it trained on high-quality but expensive artificial conversational search data.
Kelong Mao, Zhicheng Dou, Hongjin Qian, Fengran Mo, Xiaohua Cheng, Zhao Cao
EMNLP1
2022 Curriculum Contrastive Context Denoising for Few-shot Conversational Dense Retrieval
abstract
Conversational search is a crucial and promising branch in information retrieval. In this paper, we reveal that not all historical conversational turns are necessary for understanding the intent of the current query. The redundant noisy turns in the context largely hinder the improvement of search performance. However, enhancing the context denoising ability for conversational search is quite challenging due to data scarcity and the steep difficulty for simultaneously learning conversational query encoding and context denoising. To address these issues, in this paper, we present a novel Curriculum cOntrastive conTExt Denoising framework, COTED, towards few-shot conversational dense retrieval. Under a curriculum training order, we progressively endow the model with the capability of context denoising via contrastive learning between noised samples and denoised samples generated by a new conversation data augmentation strategy. Three curriculums tailored to conversational search are exploited in our framework. Extensive experiments on two few-shot conversational search datasets, i.e., CAsT-19 and CAsT-20, validate the effectiveness and superiority of our method compared with the state-of-the-art baselines.
Kelong Mao, Zhicheng Dou, Hongjin Qian
SIGIR1
2021 SimpleX: A Simple and Strong Baseline for Collaborative Filtering
abstract
Collaborative filtering (CF) is a widely studied research topic in recommender systems. The learning of a CF model generally depends on three major components, namely interaction encoder, loss function, and negative sampling. While many existing studies focus on the design of more powerful interaction encoders, the impacts of loss functions and negative sampling ratios have not yet been well explored. In this work, we show that the choice of loss function as well as negative sampling ratio is equivalently important. More specifically, we propose the cosine contrastive loss (CCL) and further incorporate it to a simple unified CF model, dubbed SimpleX. Extensive experiments have been conducted on 10 benchmark datasets and compared with 28 existing CF models in total. Surprisingly, the results show that, under our CCL loss and a large negative sampling ratio, SimpleX can surpass most sophisticated state-of-the-art models by a large margin (e.g., max 48.5% improvement in [email protected] over LightGCN). We believe that SimpleX could not only serve as a simple strong baseline to foster future research on CF, but also shed light on the potential research direction towards improving loss function and negative sampling.
Kelong Mao, Jieming Zhu, Jinpeng Wang 0002, Quanyu Dai, Zhenhua Dong, Xi Xiao 0001, Xiuqiang He 0001
CIKM1
2021 UltraGCN: Ultra Simplification of Graph Convolutional Networks for Recommendation
abstract
With the recent success of graph convolutional networks (GCNs), they have been widely applied for recommendation, and achieved impressive performance gains. The core of GCNs lies in its message passing mechanism to aggregate neighborhood information. However, we observed that message passing largely slows down the convergence of GCNs during training, especially for large-scale recommender systems, which hinders their wide adoption. LightGCN makes an early attempt to simplify GCNs for collaborative filtering by omitting feature transformations and nonlinear activations. In this paper, we take one step further to propose an ultra-simplified formulation of GCNs (dubbed UltraGCN), which skips infinite layers of message passing for efficient recommendation. Instead of explicit message passing, UltraGCN resorts to directly approximate the limit of infinite-layer graph convolutions via a constraint loss. Meanwhile, UltraGCN allows for more appropriate edge weight assignments and flexible adjustment of the relative importances among different types of relationships. This finally yields a simple yet effective UltraGCN model, which is easy to implement and efficient to train. Experimental results on four benchmark datasets show that UltraGCN not only outperforms the state-of-the-art GCN models but also achieves more than 10x speedup over LightGCN.
Kelong Mao, Jieming Zhu, Xi Xiao 0001, Biao Lu 0005, Zhaowei Wang 0002, Xiuqiang He 0001
CIKM1
2021 Byte-Label Joint Attention Learning for Packet-grained Network Traffic Classification
abstract
Network traffic classification (TC) is to classify network traffic into a specific class which plays a fundamental role in terms of network measurement, network management, and so on. In this work, we focus on packet-grained traffic classification. We find that previous packet-grained methods based on the analogy between traffic packet and image or text are not sufficiently reasonable, leading to a sub-optimal performance on both accuracy and efficiency that still can be largely improved. In this paper, we devise a new method, called BLJAN, to jointly learn from byte sequence and labels for packet-grained traffic classification. BLJAN embeds the packet’s bytes and all labels into a joint embedding space to capture their implicit correlations with a dual attention mechanism. It finally builds a more powerful packet representation with an enhancement from label embeddings to achieve high classification accuracy and interpretability. Extensive experiments on two benchmark traffic classification tasks, including application identification and traffic characterization, with three real-world datasets, demonstrate that BLJAN can achieve high performance (96.2%, 96.7%, and 99.7% Macro F1-scores on three datasets) for packet-grained traffic classification, outperforming six representative state-of-the-art baselines in terms of both accuracy and detection speed.
Kelong Mao, Xi Xiao 0001, Guangwu Hu, Xiapu Luo, Bin Zhang 0048, Shutao Xia
IWQoS1
2021 Molecular graph enhanced transformer for retrosynthesis prediction
Kelong Mao, Xi Xiao 0001, Tingyang Xu, Yu Rong 0001, Junzhou Huang, Peilin Zhao
Neurocomputing1
2020 Item Tagging for Information Retrieval: A Tripartite Graph Neural Network based Approach
abstract
Tagging has been recognized as a successful practice to boost relevance matching for information retrieval (IR), especially when items lack rich textual descriptions. A lot of research has been done for either multi-label text categorization or image annotation. However, there is a lack of published work that targets at item tagging specifically for IR. Directly applying a traditional multi-label classification model for item tagging is sub-optimal, due to the ignorance of unique characteristics in IR. In this work, we propose to formulate item tagging as a link prediction problem between item nodes and tag nodes. To enrich the representation of items, we leverage the query logs available in IR tasks, and construct a query-item-tag tripartite graph. This formulation results in a TagGNN model that utilizes heterogeneous graph neural networks with multiple types of nodes and edges. Different from previous research, we also optimize both full tag prediction and partial tag completion cases in a unified framework via a primary-dual loss mechanism. Experimental results on both open and industrial datasets show that our TagGNN approach outperforms the state-of-the-art multi-label classification approaches.
Kelong Mao, Xi Xiao 0001, Jieming Zhu, Biao Lu 0005, Ruiming Tang, Xiuqiang He 0001
SIGIR1