Trigger3: Refining Query Correction via Adaptive Model Selector

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.

Cite

Text

Zhang et al. "Trigger3: Refining Query Correction via Adaptive Model Selector." AAAI Conference on Artificial Intelligence, 2025. doi:10.1609/AAAI.V39I12.33447

Markdown

[Zhang et al. "Trigger3: Refining Query Correction via Adaptive Model Selector." AAAI Conference on Artificial Intelligence, 2025.](https://mlanthology.org/aaai/2025/zhang2025aaai-trigger/) doi:10.1609/AAAI.V39I12.33447

BibTeX

@inproceedings{zhang2025aaai-trigger,
  title     = {{Trigger3: Refining Query Correction via Adaptive Model Selector}},
  author    = {Zhang, Kepu and Sun, Zhongxiang and Zhang, Xiao and Zang, Xiaoxue and Zheng, Kai and Song, Yang and Xu, Jun},
  booktitle = {AAAI Conference on Artificial Intelligence},
  year      = {2025},
  pages     = {13260-13268},
  doi       = {10.1609/AAAI.V39I12.33447},
  url       = {https://mlanthology.org/aaai/2025/zhang2025aaai-trigger/}
}