Optimization Methods for Personalizing Large Language Models through Retrieval Augmentation
On the map: Retrieval-augmented generation → RAG pipeline design and evaluation
Abstract (arXiv)
This paper studies retrieval-augmented approaches for personalizing large language models (LLMs), which potentially have a substantial impact on various applications and domains. We propose the first attempt to optimize the retrieval models that deliver a limited number of personal documents to large language models for the purpose of personalized generation. We develop two optimization algorithms that solicit feedback from the downstream personalized generation tasks for retrieval optimization -- one based on reinforcement learning whose reward function is defined using any arbitrary metric for personalized generation and another based on knowledge distillation from the downstream LLM to the retrieval model. This paper also introduces a pre- and post-generation retriever selection model that decides what retriever to choose for each LLM input. Extensive experiments on diverse tasks from the language model personalization (LaMP) benchmark reveal statistically significant improvements in six out of seven datasets.
Key facts. Cited by 98 papers in the map (data through 7 Oct 2026). Placed in RAG pipeline design and evaluation, one of 15 directions in Retrieval-augmented generation. Builds on 24 of its 60 references that are in the map. First posted on arXiv 9 Apr 2024.
Builds on
24 of 60 references.
References · 24
- 2014Adam: A Method for Stochastic Optimizationcited by 36,960
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,160
- 2013Distributed Representations of Words and Phrases and their Compositionalitycited by 3,462
- 2020Longformer: The Long-Document Transformercited by 3,362
- 2021Unsupervised Dense Information Retrieval with Contrastive Learningcited by 1,203
- 2023LaMP: When Large Language Models Meet Personalizationcited by 425
- 2019Deep Learning Recommendation Model for Personalization and Recommendation Systemscited by 327
- 2017Beam Search Strategies for Neural Machine Translationcited by 232
- 2020Distilling Knowledge from Reader to Retriever for Question Answeringcited by 226
- 2018Training Millions of Personalized Dialogue Agentscited by 126
- 2019Generating Personalized Recipes from Historical User Preferencescited by 52
- 2023Teach LLMs to Personalize -- An Approach inspired by Writing Educationcited by 48
12 more
- 2022Less is More: Learning to Refine Dialogue History for Personalized Dialogue Generationcited by 47
- 2022Retrieval-Enhanced Machine Learningcited by 46
- 2020Is Retriever Merely an Approximator of Reader?cited by 30
- 2021UserIdentifier: Implicit User Representations for Simple and Effective Personalized Sentiment Analysiscited by 28
- 2020Pchatbot: A Large-Scale Dataset for Personalized Chatbotcited by 26
- 2018Personalized Language Model for Query Auto-Completioncited by 25
- 2019Towards Controllable and Personalized Review Generationcited by 24
- 2018Compact Personalized Models for Neural Machine Translationcited by 19
- 2021Predicting Efficiency/Effectiveness Trade-offs for Dense vs. Sparse Retrieval Strategy Selectioncited by 15
- 2017Relevance-based Word Embeddingcited by 12
- 2023Selecting which Dense Retriever to use for Zero-Shot Searchcited by 7
- 2023A Personalized Dense Retrieval Framework for Unified Information Accesscited by 6
Built on by
Citing papers per year: 2024: 17; 2025: 47; 2026: 34.
98 papers.
Sort:
By area:
- 2024LongLaMP: A Benchmark for Personalized Long-form Text Generationcited by 91
- 2024Evaluating Retrieval Quality in Retrieval-Augmented Generationcited by 90
- 2024A Survey on Retrieval-Augmented Text Generation for Large Language Modelscited by 77
- 2024Personalized Pieces: Efficient Personalized Large Language Models through Collaborative Effortscited by 71
- 2025A Survey of Personalized Large Language Models: Progress and Future Directionscited by 65
- 2024HYDRA: Model Factorization Framework for Black-Box LLM Personalizationcited by 59
- 2025Measuring What Makes You Unique: Difference-Aware User Modeling for Enhancing LLM Personalizationcited by 44
- 2025PersonaLens: A Benchmark for Personalization Evaluation in Conversational AI Assistantscited by 44
- 2025Personalized Text Generation with Contrastive Activation Steeringcited by 30
- 2025LaMP-QA: A Benchmark for Personalized Long-form Question Answeringcited by 30
- 2025O-Mem: Omni Memory System for Personalized, Long Horizon, Self-Evolving Agentscited by 30
- 2024Stochastic RAG: End-to-End Retrieval-Augmented Generation through Expected Utility Maximizationcited by 29
38 more
- 2024LLMs + Persona-Plug = Personalized LLMscited by 29
- 2025PR-Attack: Coordinated Prompt-RAG Attacks on Retrieval-Augmented Generation in Large Language Models via Bilevel Optimizationcited by 29
- 2025Think-While-Generating: On-the-Fly Reasoning for Personalized Long-Form Generationcited by 29
- 2025From 1,000,000 Users to Every User: Scaling Up Personalized Preference for User-level Alignmentcited by 28
- 2025ExPerT: Effective and Explainable Evaluation of Personalized Long-Form Text Generationcited by 26
- 2025A Survey of Personalization: From RAG to Agentcited by 26
- 2024Towards a Search Engine for Machines: Unified Ranking for Multiple Retrieval-Augmented Large Language Modelscited by 23
- 2024Data Extraction Attacks in Retrieval-Augmented Generation via Backdoorscited by 23
- 2025Retrieval Augmented Generation with Collaborative Filtering for Personalized Text Generationcited by 23
- 2025Teaching Language Models to Evolve with Users: Dynamic Profile Modeling for Personalized Alignmentcited by 22
- 2024On the Way to LLM Personalization: Learning to Remember User Conversationscited by 21
- 2024Comparing Retrieval-Augmentation and Parameter-Efficient Fine-Tuning for Privacy-Preserving Personalization of Large Language Modelscited by 20
- 2025CUPID: Evaluating Personalized and Contextualized Alignment of LLMs from Interactionscited by 20
- 2024Planning Ahead in Generative Retrieval: Guiding Autoregressive Generation through Simultaneous Decodingcited by 19
- 2025Exploring the Potential of LLMs as Personalized Assistants: Dataset, Evaluation, and Analysiscited by 16
- 2026Learning Personalized Agents from Human Feedbackcited by 16
- 2024Towards Fair RAG: On the Impact of Fair Ranking in Retrieval-Augmented Generationcited by 15
- 2025Instant Personalized Large Language Model Adaptation via Hypernetworkcited by 15
- 2024Learning to Rank for Multiple Retrieval-Augmented Models through Iterative Utility Maximizationcited by 14
- 2024Accelerating Retrieval-Augmented Generationcited by 14
- 2025MMPB: It's Time for Multi-Modal Personalizationcited by 14
- 2025PersonaFeedback: A Large-scale Human-annotated Benchmark For Personalizationcited by 13
- 2025The Era of Real-World Human Interaction: RL from User Conversationscited by 13
- 2024Retrieval-Enhanced Machine Learning: Synthesis and Opportunitiescited by 11
- 2025PersonaAgent: Bridging Memory and Action for Personalized LLM Agentscited by 11
- 2026VitaBench 2.0: Evaluating Personalized and Proactive Agents in Long-Term User Interactionscited by 11
- 2024Understanding the Role of User Profile in the Personalization of Large Language Modelscited by 10
- 2025Personalized Graph-Based Retrieval for Large Language Modelscited by 10
- 2025MemWeaver: A Hierarchical Memory from Textual Interactive Behaviors for Personalized Generationcited by 8
- 2025Memory Assisted LLM for Personalized Recommendation Systemcited by 7
- 2025CAIM: Development and Evaluation of a Cognitive AI Memory Framework for Long-Term Interaction with Intelligent Agentscited by 7
- 2025SteerX: Disentangled Steering for LLM Personalizationcited by 7
- 2025T-POP: Test-Time Personalization with Online Preference Feedbackcited by 6
- 2026SpecSteer: Synergizing Local Context and Global Reasoning for Efficient Personalized Generationcited by 6
- 2025Retrieval Augmented Generation with Multi-Modal LLM Framework for Wireless Environmentscited by 5
- 2025Learning from Natural Language Feedback for Personalized Question Answeringcited by 5
- 2025GMTRouter: Personalized LLM Router over Multi-turn User Interactionscited by 5
- 2026Long Context, Less Focus: A Scaling Gap in LLMs Revealed through Privacy and Personalizationcited by 5
50 most cited of 98 All 98 (JSON)
- 2026ProTTT: Learning to Learn Semantic User Memory with Test-Time Training26 Sep 2026
- 2026Kernel-Managed Shared Memory for System-Wide Personalization9 Sep 2026
- 2026HyperTrace: Hypothesis-Based Preference Tracing for Online LLM Personalization9 Sep 2026
- 2026Privacy Personalization Trade offs in LLMs: The Impact of Stylometric Signal Reduction on User-Specific Text Generation18 Aug 2026
- 2026Learning Preference Adaptation for Large Language Model Personalization via Verbal Reinforcement Learning10 Aug 2026
- 2026TRACE-Memory: Public-Conditioned Retrieval and Utility-Aware Evidence Admission for Personalized Generation9 Aug 2026
- 2026Forgotten History or Test-of-Time? Retrospect and Prospect on RAG from an IR Perspective9 Aug 2026
- 2026Personalized Deep Research Query Refinement with Graph-Scaffolded Evidence Grounding6 Aug 2026
- 2026LUNAR: Benchmarking Personalized Large Language Models on UNiversal User BehAvioR Logs5 Aug 2026
- 2026PrefReward: Learning User Preference Matrix for Personalized Text Generation23 Jul 2026
- 2026Bounded Personas Match Retrieval on Classification but Not Regression for a Frozen Agent25 Jun 2026
- 2026Prompt-Space Meta-Learning Does Not Transfer Across Users: A Frozen-LLM Negative Result25 Jun 2026
38 more
- 2026Self-supervised User Profile Generation for Personalization3 Jun 2026
- 2026Beyond Retrieval: Learning Compact User Representations for Scalable LLM Personalization3 Jun 2026
- 2026Beyond Isolated Behaviors: Hierarchical User Modeling for LLM Personalization1 Jun 2026
- 2026VitaBench 2.0: Evaluating Personalized and Proactive Agents in Long-Term User Interactions26 May 2026
- 2026LATTE: Forecasting Peer Anchored Preference Trajectories for Personalized LLM Generation26 May 2026
- 2026Personal Visual Context Learning in Large Multimodal Models11 May 2026
- 2026Beyond Static Personas: Situational Personality Steering for Large Language Models15 Apr 2026
- 2026ClusterRAG: Cluster-Based Collaborative Filtering for Personalized Retrieval-Augmented Generation14 Apr 2026
- 2026PAC-BENCH: Evaluating Multi-Agent Collaboration under Privacy Constraints13 Apr 2026
- 2026PSPA-Bench: A Personalized Benchmark for Smartphone GUI Agent31 Mar 2026
- 2026Not All Subjectivity Is the Same! Defining Desiderata for the Evaluation of Subjectivity in NLP30 Mar 2026
- 2026SpecSteer: Synergizing Local Context and Global Reasoning for Efficient Personalized Generation17 Mar 2026
- 2026BenchBrowser: Retrieving Evidence for Evaluating Benchmark Validity25 Feb 2026
- 2026Learning Personalized Agents from Human Feedback18 Feb 2026
- 2026Long Context, Less Focus: A Scaling Gap in LLMs Revealed through Privacy and Personalization16 Feb 2026
- 2026Neuromem: A Granular Decomposition of the Streaming Lifecycle in External Memory for LLMs15 Feb 2026
- 2026Contextualized Visual Personalization in Vision-Language Models3 Feb 2026
- 2026Improving User Privacy in Personalized Generation: Client-Side Retrieval-Augmented Modification of Server-Side Generated Speculations24 Jan 2026
- 2026Do Implicit Personalization and Explicit Styles Conflict? PsPLUG: A Lightweight Plug-in for Balancing Personalization and Style in Customized LLMs10 Jan 2026
- 2026CARD: Cluster-level Adaptation with Reward-guided Decoding for Personalized Text Generation9 Jan 2026
- 2026Learning to Retrieve User History and Generate User Profiles for Personalized Persuasiveness Prediction9 Jan 2026
- 2026Text as a Universal Interface for Transferable Personalization8 Jan 2026
- 2025Think-While-Generating: On-the-Fly Reasoning for Personalized Long-Form Generation7 Dec 2025
- 2025WebRec: Enhancing LLM-based Recommendations with Attention-guided RAG from Web18 Nov 2025
- 2025O-Mem: Omni Memory System for Personalized, Long Horizon, Self-Evolving Agents17 Nov 2025
- 2025Reflective Personalization Optimization: A Post-hoc Rewriting Framework for Black-Box Large Language Models7 Nov 2025
- 2025Fints: Efficient Inference-Time Personalization for LLMs with Fine-Grained Instance-Tailored Steering31 Oct 2025
- 2025GMTRouter: Personalized LLM Router over Multi-turn User Interactions29 Oct 2025
- 2025SteerX: Disentangled Steering for LLM Personalization25 Oct 2025
- 2025Resolution-Aware Retrieval Augmented Zero-Shot Forecasting19 Oct 2025
- 2025Instant Personalized Large Language Model Adaptation via Hypernetwork18 Oct 2025
- 2025YNTP-100: A Benchmark for Your Next Token Prediction with 100 People16 Oct 2025
- 2025MemWeaver: A Hierarchical Memory from Textual Interactive Behaviors for Personalized Generation9 Oct 2025
- 2025The Era of Real-World Human Interaction: RL from User Conversations29 Sep 2025
- 2025T-POP: Test-Time Personalization with Online Preference Feedback29 Sep 2025
- 2025From Personal to Collective: On the Role of Local and Global Memory in LLM Personalization28 Sep 2025
- 2025Tagging the Thought: Unlocking Personalization Reasoning via Reinforcement Learning27 Sep 2025
- 2025MMPB: It's Time for Multi-Modal Personalization26 Sep 2025
50 newest of 98 All 98 (JSON)
- 2024Evaluating Retrieval Quality in Retrieval-Augmented Generationcited by 90
- 2024A Survey on Retrieval-Augmented Text Generation for Large Language Modelscited by 77
- 2024Stochastic RAG: End-to-End Retrieval-Augmented Generation through Expected Utility Maximizationcited by 29
- 2025PR-Attack: Coordinated Prompt-RAG Attacks on Retrieval-Augmented Generation in Large Language Models via Bilevel Optimizationcited by 29
- 2025A Survey of Personalization: From RAG to Agentcited by 26
- 2024Towards a Search Engine for Machines: Unified Ranking for Multiple Retrieval-Augmented Large Language Modelscited by 23
- 2024Data Extraction Attacks in Retrieval-Augmented Generation via Backdoorscited by 23
- 2025Retrieval Augmented Generation with Collaborative Filtering for Personalized Text Generationcited by 23
- 2024Towards Fair RAG: On the Impact of Fair Ranking in Retrieval-Augmented Generationcited by 15
- 2024Learning to Rank for Multiple Retrieval-Augmented Models through Iterative Utility Maximizationcited by 14
- 2024Accelerating Retrieval-Augmented Generationcited by 14
- 2025Retrieval Augmented Generation with Multi-Modal LLM Framework for Wireless Environmentscited by 5
12 most cited of 21 All 21 (JSON)
- 2026Forgotten History or Test-of-Time? Retrospect and Prospect on RAG from an IR Perspective9 Aug 2026
- 2026ClusterRAG: Cluster-Based Collaborative Filtering for Personalized Retrieval-Augmented Generation14 Apr 2026
- 2026Improving User Privacy in Personalized Generation: Client-Side Retrieval-Augmented Modification of Server-Side Generated Speculations24 Jan 2026
- 2025WebRec: Enhancing LLM-based Recommendations with Attention-guided RAG from Web18 Nov 2025
- 2025Resolution-Aware Retrieval Augmented Zero-Shot Forecasting19 Oct 2025
- 2025A Survey of Personalization: From RAG to Agent14 Apr 2025
- 2025Plan-and-Refine: Diverse and Comprehensive Retrieval-Augmented Generation10 Apr 2025
- 2025PR-Attack: Coordinated Prompt-RAG Attacks on Retrieval-Augmented Generation in Large Language Models via Bilevel Optimization10 Apr 2025
- 2025Retrieval Augmented Generation with Collaborative Filtering for Personalized Text Generation8 Apr 2025
- 2025Improving RAG for Personalization with Author Features and Contrastive Examples24 Mar 2025
- 2025Retrieval Augmented Generation with Multi-Modal LLM Framework for Wireless Environments9 Mar 2025
- 2025Towards Efficient Educational Chatbots: Benchmarking RAG Frameworks2 Mar 2025
12 newest of 21 All 21 (JSON)
- 2025O-Mem: Omni Memory System for Personalized, Long Horizon, Self-Evolving Agentscited by 30
- 2025PersonaAgent: Bridging Memory and Action for Personalized LLM Agentscited by 11
- 2026VitaBench 2.0: Evaluating Personalized and Proactive Agents in Long-Term User Interactionscited by 11
- 2025CAIM: Development and Evaluation of a Cognitive AI Memory Framework for Long-Term Interaction with Intelligent Agentscited by 7
- 2026PSPA-Bench: A Personalized Benchmark for Smartphone GUI Agentcited by 4
- 2026Personalized Deep Research Query Refinement with Graph-Scaffolded Evidence Groundingcited by 1
- 2025Dynamic Evaluation Framework for Personalized and Trustworthy Agents: A Multi-Session Approach to Preference Adaptabilitycited by 0
- 2026Personalized Deep Research Query Refinement with Graph-Scaffolded Evidence Grounding6 Aug 2026
- 2026VitaBench 2.0: Evaluating Personalized and Proactive Agents in Long-Term User Interactions26 May 2026
- 2026PSPA-Bench: A Personalized Benchmark for Smartphone GUI Agent31 Mar 2026
- 2025O-Mem: Omni Memory System for Personalized, Long Horizon, Self-Evolving Agents17 Nov 2025
- 2025PersonaAgent: Bridging Memory and Action for Personalized LLM Agents6 Jun 2025
- 2025CAIM: Development and Evaluation of a Cognitive AI Memory Framework for Long-Term Interaction with Intelligent Agents19 May 2025
- 2025Dynamic Evaluation Framework for Personalized and Trustworthy Agents: A Multi-Session Approach to Preference Adaptability8 Mar 2025
- 2025From 1,000,000 Users to Every User: Scaling Up Personalized Preference for User-level Alignmentcited by 28
- 2026Learning Personalized Agents from Human Feedbackcited by 16
- 2025T-POP: Test-Time Personalization with Online Preference Feedbackcited by 6
- 2025Group Preference Alignment: Customized LLM Response Generation from In-Situ Conversationscited by 2
- 2026CARD: Cluster-level Adaptation with Reward-guided Decoding for Personalized Text Generationcited by 2
- 2026PrefReward: Learning User Preference Matrix for Personalized Text Generationcited by 0
- 2026HyperTrace: Hypothesis-Based Preference Tracing for Online LLM Personalizationcited by 0
- 2026HyperTrace: Hypothesis-Based Preference Tracing for Online LLM Personalization9 Sep 2026
- 2026PrefReward: Learning User Preference Matrix for Personalized Text Generation23 Jul 2026
- 2026Learning Personalized Agents from Human Feedback18 Feb 2026
- 2026CARD: Cluster-level Adaptation with Reward-guided Decoding for Personalized Text Generation9 Jan 2026
- 2025T-POP: Test-Time Personalization with Online Preference Feedback29 Sep 2025
- 2025From 1,000,000 Users to Every User: Scaling Up Personalized Preference for User-level Alignment19 Mar 2025
- 2025Group Preference Alignment: Customized LLM Response Generation from In-Situ Conversations11 Mar 2025
- 2025Think-While-Generating: On-the-Fly Reasoning for Personalized Long-Form Generationcited by 29
- 2026SpecSteer: Synergizing Local Context and Global Reasoning for Efficient Personalized Generationcited by 6
- 2025Tagging the Thought: Unlocking Personalization Reasoning via Reinforcement Learningcited by 3
- 2025RPM: Reasoning-Level Personalization for Black-Box Large Language Modelscited by 2
- 2025Does Knowledge Distillation Matter for Large Language Model based Bundle Generation?cited by 1
- 2026SpecSteer: Synergizing Local Context and Global Reasoning for Efficient Personalized Generation17 Mar 2026
- 2025Think-While-Generating: On-the-Fly Reasoning for Personalized Long-Form Generation7 Dec 2025
- 2025Tagging the Thought: Unlocking Personalization Reasoning via Reinforcement Learning27 Sep 2025
- 2025RPM: Reasoning-Level Personalization for Black-Box Large Language Models27 May 2025
- 2025Does Knowledge Distillation Matter for Large Language Model based Bundle Generation?24 Apr 2025
- 2024LongLaMP: A Benchmark for Personalized Long-form Text Generationcited by 91
- 2024Personalized Pieces: Efficient Personalized Large Language Models through Collaborative Effortscited by 71
- 2025A Survey of Personalized Large Language Models: Progress and Future Directionscited by 65
- 2024HYDRA: Model Factorization Framework for Black-Box LLM Personalizationcited by 59
- 2025Measuring What Makes You Unique: Difference-Aware User Modeling for Enhancing LLM Personalizationcited by 44
- 2025PersonaLens: A Benchmark for Personalization Evaluation in Conversational AI Assistantscited by 44
- 2025Personalized Text Generation with Contrastive Activation Steeringcited by 30
- 2025LaMP-QA: A Benchmark for Personalized Long-form Question Answeringcited by 30
- 2024LLMs + Persona-Plug = Personalized LLMscited by 29
- 2025ExPerT: Effective and Explainable Evaluation of Personalized Long-Form Text Generationcited by 26
- 2025Teaching Language Models to Evolve with Users: Dynamic Profile Modeling for Personalized Alignmentcited by 22
- 2024On the Way to LLM Personalization: Learning to Remember User Conversationscited by 21
12 most cited of 56 All 56 (JSON)
- 2026ProTTT: Learning to Learn Semantic User Memory with Test-Time Training26 Sep 2026
- 2026Kernel-Managed Shared Memory for System-Wide Personalization9 Sep 2026
- 2026Privacy Personalization Trade offs in LLMs: The Impact of Stylometric Signal Reduction on User-Specific Text Generation18 Aug 2026
- 2026Learning Preference Adaptation for Large Language Model Personalization via Verbal Reinforcement Learning10 Aug 2026
- 2026TRACE-Memory: Public-Conditioned Retrieval and Utility-Aware Evidence Admission for Personalized Generation9 Aug 2026
- 2026LUNAR: Benchmarking Personalized Large Language Models on UNiversal User BehAvioR Logs5 Aug 2026
- 2026Bounded Personas Match Retrieval on Classification but Not Regression for a Frozen Agent25 Jun 2026
- 2026Prompt-Space Meta-Learning Does Not Transfer Across Users: A Frozen-LLM Negative Result25 Jun 2026
- 2026Self-supervised User Profile Generation for Personalization3 Jun 2026
- 2026Beyond Retrieval: Learning Compact User Representations for Scalable LLM Personalization3 Jun 2026
- 2026Beyond Isolated Behaviors: Hierarchical User Modeling for LLM Personalization1 Jun 2026
- 2026LATTE: Forecasting Peer Anchored Preference Trajectories for Personalized LLM Generation26 May 2026
12 newest of 56 All 56 (JSON)
Similar, not cited
Close in wording, no citation either way
- 2026Optimizing User Profiles via Contextual Bandits for Retrieval-Augmented LLM Personalization
- 2024Bridging the Preference Gap between Retrievers and LLMs
- 2026Less Is Personal: Learning Minimal Sufficient User Profiles for Personalized Language Models
- 2025PrLM: Learning Explicit Reasoning for Personalized RAG via Contrastive Reward Optimization
- 2025Direct Retrieval-augmented Optimization: Synergizing Knowledge Selection and Language Models
- 2023Integrating Summarization and Retrieval for Enhanced Personalization via Large Language Models
- 2025BESPOKE: Benchmark for Search-Augmented Large Language Model Personalization via Diagnostic Feedback
- 2025Personalizing Large Language Models using Retrieval Augmented Generation and Knowledge Graph