Trigger3:Refining Query Correction via Adaptive Model Selector

Authors

  • Kepu Zhang Renmin University of China
  • Zhongxiang Sun Renmin University of China
  • Xiao Zhang Renmin University of China
  • Xiaoxue Zang Kuaishou Technology Co., Ltd.
  • Kai Zheng Kuaishou Technology Co., Ltd.
  • Yang Song Kuaishou Technology Co., Ltd.
  • Jun Xu Renmin University of China

DOI:

https://doi.org/10.1609/aaai.v39i12.33447

Abstract

In search scenarios, user experience can be hindered by erroneous queries due to typos, voice errors, or knowledge gaps. Therefore, query correction is crucial for search engines. Current correction models, usually small models trained on specific data, often struggle with queries beyond their training scope or those requiring contextual understanding. While the advent of Large Language Models (LLMs) offers a potential solution, they are still limited by their pre-training data and inference cost, particularly for complex queries, making them not always effective for query correction. To tackle these, we propose Trigger3, a large-small model collaboration framework that integrates the traditional correction model and LLM for query correction, capable of adaptively choosing the appropriate correction method based on the query and the correction results from the traditional correction model and LLM. Trigger3 first employs a correction trigger to filter out correct queries. Incorrect queries are then corrected by the traditional correction model. If this fails, an LLM trigger is activated to call the LLM for correction. Finally, for queries that no model can correct, a fallback trigger decides to return the original query. Extensive experiments demonstrate Trigger3 outperforms correction baselines while maintaining efficiency.

Published

2025-04-11

How to Cite

Zhang, K., Sun, Z., Zhang, X., Zang, X., Zheng, K., Song, Y., & Xu, J. (2025). Trigger3:Refining Query Correction via Adaptive Model Selector. Proceedings of the AAAI Conference on Artificial Intelligence, 39(12), 13260–13268. https://doi.org/10.1609/aaai.v39i12.33447

Issue

Section

AAAI Technical Track on Data Mining & Knowledge Management II