Skip to main navigation Skip to search Skip to main content

Personalize Before Retrieve: LLM-based Personalized Query Expansion for User-Centric Retrieval

  • Yingyi Zhang
  • , Pengyue Jia
  • , Derong Xu
  • , Yi Wen
  • , Xianneng Li*
  • , Yichao Wang*
  • , Wenlin Zhang
  • , Xiaopeng Li
  • , Weinan Gan
  • , Huifeng Guo
  • , Yong Liu
  • , Xiangyu Zhao*
  • *Corresponding author for this work
  • Dalian University of Technology
  • City University of Hong Kong
  • University of Science and Technology of China
  • Huawei Technologies Co., Ltd.

Research output: Contribution to journalConference articlepeer-review

Abstract

Retrieval-Augmented Generation (RAG) critically depends on effective query expansion to retrieve relevant information. However, existing expansion methods adopt uniform strategies that overlook user-specific semantics, ignoring individual expression styles, preferences, and historical context. In practice, identical queries in text can express vastly different intentions across users. This representational rigidity limits the ability of current RAG systems to generalize effectively in personalized settings. Specifically, we identify two core challenges for personalization: 1) user expression styles are inherently diverse, making it difficult for standard expansions to preserve personalized intent. 2) user corpora induce heterogeneous semantic structures—varying in topical focus and lexical organization—which hinders the effective anchoring of expanded queries within the user’s corpora space. To address these challenges, we propose Personalize Before Retrieve (PBR), a framework that incorporates user-specific signals into query expansion prior to retrieval. PBR consists of two components: P-PRF, which generates stylistically aligned pseudo feedback using user history for simulating user expression style, and P-Anchor, which performs graph-based structure alignment over user corpora to capture its structure. Together, they produce personalized query representations tailored for retrieval. Experiments on two personalized benchmarks show that PBR consistently outperforms strong baselines, with up to 10% gains on PersonaBench across retrievers. Our findings demonstrate the value of modeling personalization before retrieval to close the semantic gap in user-adaptive RAG systems.

Original languageEnglish
Pages (from-to)16406-16414
Number of pages9
JournalProceedings of the AAAI Conference on Artificial Intelligence
Volume40
Issue number19
DOIs
StatePublished - 2026
Externally publishedYes
Event40th AAAI Conference on Artificial Intelligence, AAAI 2026 - Singapore, Singapore
Duration: 20 Jan 202627 Jan 2026

Fingerprint

Dive into the research topics of 'Personalize Before Retrieve: LLM-based Personalized Query Expansion for User-Centric Retrieval'. Together they form a unique fingerprint.

Cite this