AAAI 2026technical0 citations

Think Then Rewrite: Reasoning Enhanced Query Rewriting for Domain Specific Retrieval

Ang Li, Yufei Shi, Yuxuan Si, Yiquan Wu, Ming Cai, Xu Tan, Yi Wang, Changlong Sun

Abstract

Query rewriting is a crucial task for improving retrieval, especially in professional domains such as law and medicine, where user queries are often underspecified and ambiguous. While large language models (LLMs) offer strong understanding and generation capabilities, existing LLM-based approaches reduce the task to text transformation or expansion, neglecting reasoning to disambiguate queries, which fails to bridge the cognitive gap between user queries and specialized documents. In this paper, we propose Think-Then-Rewrite (TTR), a reinforcement learning based framework that unleashes LLMs

BibTeX
@inproceedings{aaai2026_thinkthenrewrite,
  title = {Think Then Rewrite: Reasoning Enhanced Query Rewriting for Domain Specific Retrieval},
  author = {Ang Li and Yufei Shi and Yuxuan Si and Yiquan Wu and Ming Cai and Xu Tan and Yi Wang and Changlong Sun and Xiaozhong Liu and Kun Kuang},
  booktitle = {AAAI 2026},
  year = {2026}
}
Think Then Rewrite: Reasoning Enhanced Query Rewriting for Domain Specific Retrieval · AAAI 2026