Agent4Ranking: Semantic Robust Ranking via Personalized Query Rewriting Using Multi-agent LLM
On the map: LLM agents and tool use → Web, search, and research agents
Abstract (arXiv)
Search engines are crucial as they provide an efficient and easy way to access vast amounts of information on the internet for diverse information needs. User queries, even with a specific need, can differ significantly. Prior research has explored the resilience of ranking models against typical query variations like paraphrasing, misspellings, and order changes. Yet, these works overlook how diverse demographics uniquely formulate identical queries. For instance, older individuals tend to construct queries more naturally and in varied order compared to other groups. This demographic diversity necessitates enhancing the adaptability of ranking models to diverse query formulations. To this end, in this paper, we propose a framework that integrates a novel rewriting pipeline that rewrites queries from various demographic perspectives and a novel framework to enhance ranking robustness. To be specific, we use Chain of Thought (CoT) technology to utilize Large Language Models (LLMs) as agents to emulate various demographic profiles, then use them for efficient query rewriting, and we innovate a robust Multi-gate Mixture of Experts (MMoE) architecture coupled with a hybrid loss function, collectively strengthening the ranking models' robustness. Our extensive experimentation on both public and industrial datasets assesses the efficacy of our query rewriting approach and the enhanced accuracy and robustness of the ranking model. The findings highlight the sophistication and effectiveness of our proposed model.
Key facts. Cited by 18 papers in the map (data through 14 Oct 2026). Placed in Web, search, and research agents, one of 16 directions in LLM agents and tool use. Builds on 42 of its 85 references that are in the map. First posted on arXiv 24 Dec 2023.
Builds on
42 of 85 references.
References · 42
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,372
- 2019Sentence-BERT: Sentence Embeddings using Siamese BERT-Networkscited by 9,170
- 2023Judging LLM-as-a-Judge with MT-Bench and Chatbot Arenacited by 8,626
- 2019Parameter-Efficient Transfer Learning for NLPcited by 4,236
- 2021SimCSE: Simple Contrastive Learning of Sentence Embeddingscited by 2,173
- 2021BEIR: A Heterogenous Benchmark for Zero-shot Evaluation of Information Retrieval Modelscited by 1,252
- 2020Leveraging Passage Retrieval with Generative Models for Open Domain Question Answeringcited by 1,210
- 2020Approximate Nearest Neighbor Negative Contrastive Learning for Dense Text Retrievalcited by 849
- 2022Precise Zero-Shot Dense Retrieval without Relevance Labelscited by 517
- 2021Sentence-T5: Scalable Sentence Encoders from Pre-trained Text-to-Text Modelscited by 463
- 2023Character-LLM: A Trainable Agent for Role-Playingcited by 339
30 more
- 2020K-Adapter: Infusing Knowledge into Pre-Trained Models with Adapterscited by 319
- 2022Generate rather than Retrieve: Large Language Models are Strong Context Generatorscited by 298
- 2019ERNIE 2.0: A Continual Pre-training Framework for Language Understandingcited by 231
- 2023Query Rewriting for Retrieval-Augmented Large Language Modelscited by 213
- 2023Query2doc: Query Expansion with Large Language Modelscited by 149
- 2017A Deep Relevance Matching Model for Ad-hoc Retrievalcited by 126
- 2019Deeper Text Understanding for IR with Contextual Neural Language Modelingcited by 113
- 2023When MOE Meets LLMs: Parameter Efficient Fine-tuning for Multi-task Medical Applicationscited by 111
- 2018Deep Reinforcement Learning for Page-wise Recommendationscited by 100
- 2020Speaker-Aware BERT for Multi-Turn Response Selection in Retrieval-Based Chatbotscited by 73
- 2018Recommendations with Negative Feedback via Pairwise Deep Reinforcement Learningcited by 71
- 2023Enhancing Conversational Search: Large Language Model-Aided Informative Query Rewritingcited by 69
- 2023Query Expansion by Prompting Large Language Modelscited by 57
- 2021Evaluating the Robustness of Retrieval Pipelines with Query Variation Generatorscited by 34
- 2023KuaiSim: A Comprehensive Simulator for Recommender Systemscited by 30
- 2023Large Language Models are Strong Zero-Shot Retrievercited by 26
- 2021B-PROP: Bootstrapped Pre-training with Representative Words Prediction for Ad-hoc Retrievalcited by 25
- 2023AutoDenoise: Automatic Data Instance Denoising for Recommendationscited by 23
- 2023MILL: Mutual Verification with Large Language Models for Zero-Shot Query Expansioncited by 23
- 2022CharacterBERT and Self-Teaching for Improving the Robustness of Dense Retrievers on Queries with Typoscited by 22
- 2022QUILL: Query Intent with Large Language Models using Retrieval Augmentation and Multi-stage Distillationcited by 21
- 2023HAMUR: Hyper Adapter for Multi-Domain Recommendationcited by 21
- 2023Exploration and Regularization of the Latent Action Space in Recommendationcited by 17
- 2021Are Neural Ranking Models Robust?cited by 16
- 2023User Retention-oriented Recommendation with Decision Transformercited by 14
- 2023A Study on FGSM Adversarial Training for Neural Retrievalcited by 10
- 2023Generative and Pseudo-Relevant Feedback for Sparse, Dense and Learned Sparse Retrievalcited by 9
- 2023Query Understanding in the Age of Large Language Modelscited by 7
- 2023Typo-Robust Representation Learning for Dense Retrievalcited by 6
- 2023Noise-Robust Dense Retrieval via Contrastive Alignment Post Trainingcited by 5
Built on by
Citing papers per year: 2023: 1; 2024: 6; 2025: 8; 2026: 3.
18 papers.
Sort:
By area:
- 2023Large Language Models for Generative Information Extraction: A Surveycited by 185
- 2024When Search Engine Services meet Large Language Models: Visions and Challengescited by 59
- 2024G3: An Effective and Adaptive Framework for Worldwide Geolocalization Using Large Multi-Modality Modelscited by 34
- 2025A Survey of Large Language Model Empowered Agents for Recommendation and Search: Towards Next-Generation Information Retrievalcited by 33
- 2025Rec-R1: Bridging Generative Large Language Models and User-Centric Recommendation Systems via Reinforcement Learningcited by 33
- 2024BlendFilter: Advancing Retrieval-Augmented Large Language Models via Query Generation Blending and Knowledge Filteringcited by 29
- 2024Bridging Relevance and Reasoning: Rationale Distillation in Retrieval-Augmented Generationcited by 13
- 2026DeepResearch-9K: A Challenging Benchmark Dataset of Deep-Research Agentcited by 10
- 2024ERASE: Benchmarking Feature Selection Methods for Deep Recommender Systemscited by 9
- 2024Stepwise Reasoning Error Disruption Attack of LLMscited by 8
- 2025TeaRAG: A Token-Efficient Agentic Retrieval-Augmented Generation Frameworkcited by 7
- 2025Prompting Large Language Models with Partial Knowledge for Answering Questions with Unseen Entitiescited by 4
6 more
- 2025TAPO: Task-Referenced Adaptation for Prompt Optimizationcited by 3
- 2025MassTool: A Multi-Task Search-Based Tool Retrieval Framework for Large Language Modelscited by 3
- 2025Scaling Down, Serving Fast: Compressing and Deploying Efficient LLMs for Recommendation Systemscited by 2
- 2025Model Merging for Knowledge Editingcited by 2
- 2026LegalMALR:Multi-Agent Query Understanding and LLM-Based Reranking for Chinese Statute Retrievalcited by 0
- 2026Improving Neural Retrieval with Attribution-Guided Query Rewritingcited by 0
- 2026DeepResearch-9K: A Challenging Benchmark Dataset of Deep-Research Agent1 Mar 2026
- 2026Improving Neural Retrieval with Attribution-Guided Query Rewriting12 Feb 2026
- 2026LegalMALR:Multi-Agent Query Understanding and LLM-Based Reranking for Chinese Statute Retrieval25 Jan 2026
- 2025TeaRAG: A Token-Efficient Agentic Retrieval-Augmented Generation Framework7 Nov 2025
- 2025Prompting Large Language Models with Partial Knowledge for Answering Questions with Unseen Entities2 Aug 2025
- 2025MassTool: A Multi-Task Search-Based Tool Retrieval Framework for Large Language Models1 Jul 2025
- 2025Model Merging for Knowledge Editing14 Jun 2025
- 2025Rec-R1: Bridging Generative Large Language Models and User-Centric Recommendation Systems via Reinforcement Learning31 Mar 2025
- 2025A Survey of Large Language Model Empowered Agents for Recommendation and Search: Towards Next-Generation Information Retrieval7 Mar 2025
- 2025Scaling Down, Serving Fast: Compressing and Deploying Efficient LLMs for Recommendation Systems20 Feb 2025
- 2025TAPO: Task-Referenced Adaptation for Prompt Optimization12 Jan 2025
- 2024Stepwise Reasoning Error Disruption Attack of LLMs16 Dec 2024
6 more
- 2024Bridging Relevance and Reasoning: Rationale Distillation in Retrieval-Augmented Generation11 Dec 2024
- 2024When Search Engine Services meet Large Language Models: Visions and Challenges28 Jun 2024
- 2024G3: An Effective and Adaptive Framework for Worldwide Geolocalization Using Large Multi-Modality Models23 May 2024
- 2024ERASE: Benchmarking Feature Selection Methods for Deep Recommender Systems19 Mar 2024
- 2024BlendFilter: Advancing Retrieval-Augmented Large Language Models via Query Generation Blending and Knowledge Filtering16 Feb 2024
- 2023Large Language Models for Generative Information Extraction: A Survey29 Dec 2023
- 2024BlendFilter: Advancing Retrieval-Augmented Large Language Models via Query Generation Blending and Knowledge Filteringcited by 29
- 2025Prompting Large Language Models with Partial Knowledge for Answering Questions with Unseen Entitiescited by 4
- 2026LegalMALR:Multi-Agent Query Understanding and LLM-Based Reranking for Chinese Statute Retrievalcited by 0
- 2026LegalMALR:Multi-Agent Query Understanding and LLM-Based Reranking for Chinese Statute Retrieval25 Jan 2026
- 2025Prompting Large Language Models with Partial Knowledge for Answering Questions with Unseen Entities2 Aug 2025
- 2024BlendFilter: Advancing Retrieval-Augmented Large Language Models via Query Generation Blending and Knowledge Filtering16 Feb 2024
- 2023Large Language Models for Generative Information Extraction: A Surveycited by 185
- 2024When Search Engine Services meet Large Language Models: Visions and Challengescited by 59
- 2024G3: An Effective and Adaptive Framework for Worldwide Geolocalization Using Large Multi-Modality Modelscited by 34
- 2025Rec-R1: Bridging Generative Large Language Models and User-Centric Recommendation Systems via Reinforcement Learningcited by 33
- 2024ERASE: Benchmarking Feature Selection Methods for Deep Recommender Systemscited by 9
- 2025TAPO: Task-Referenced Adaptation for Prompt Optimizationcited by 3
- 2025MassTool: A Multi-Task Search-Based Tool Retrieval Framework for Large Language Modelscited by 3
- 2025Scaling Down, Serving Fast: Compressing and Deploying Efficient LLMs for Recommendation Systemscited by 2
- 2025Model Merging for Knowledge Editingcited by 2
- 2026Improving Neural Retrieval with Attribution-Guided Query Rewritingcited by 0
- 2026Improving Neural Retrieval with Attribution-Guided Query Rewriting12 Feb 2026
- 2025MassTool: A Multi-Task Search-Based Tool Retrieval Framework for Large Language Models1 Jul 2025
- 2025Model Merging for Knowledge Editing14 Jun 2025
- 2025Rec-R1: Bridging Generative Large Language Models and User-Centric Recommendation Systems via Reinforcement Learning31 Mar 2025
- 2025Scaling Down, Serving Fast: Compressing and Deploying Efficient LLMs for Recommendation Systems20 Feb 2025
- 2025TAPO: Task-Referenced Adaptation for Prompt Optimization12 Jan 2025
- 2024When Search Engine Services meet Large Language Models: Visions and Challenges28 Jun 2024
- 2024G3: An Effective and Adaptive Framework for Worldwide Geolocalization Using Large Multi-Modality Models23 May 2024
- 2024ERASE: Benchmarking Feature Selection Methods for Deep Recommender Systems19 Mar 2024
- 2023Large Language Models for Generative Information Extraction: A Survey29 Dec 2023
Similar, not cited
Close in wording, no citation either way
- 2025A Multi-Agent Perspective on Modern Information Retrieval
- 2026Prism-Reranker: Beyond Relevance Scoring -- Jointly Producing Contributions and Evidence for Agentic Retrieval
- 2026Agent-Orchestrated Adaptive RAG: A Comparative Study on Structured and Multi-Hop Retrieval
- 2024Learning to Rank for Multiple Retrieval-Augmented Models through Iterative Utility Maximization
- 2026Beyond Top-k Skill Retrieval: Diversity-Aware Skill Routing for LLM Agents
- 2025REARANK: Reasoning Re-ranking Agent via Reinforcement Learning
- 2025DynamicRAG: Leveraging Outputs of Large Language Model as Feedback for Dynamic Reranking in Retrieval-Augmented Generation
- 2026LLM-Confidence Reranker: A Training-Free Approach for Enhancing Retrieval-Augmented Generation Systems