Trigger$^3$: Refining Query Correction via Adaptive Model Selector

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Zhang, Kepu, Sun, Zhongxiang, Zhang, Xiao, Zang, Xiaoxue, Zheng, Kai, Song, Yang, Xu, Jun
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929635098361856
author Zhang, Kepu
Sun, Zhongxiang
Zhang, Xiao
Zang, Xiaoxue
Zheng, Kai
Song, Yang
Xu, Jun
author_facet Zhang, Kepu
Sun, Zhongxiang
Zhang, Xiao
Zang, Xiaoxue
Zheng, Kai
Song, Yang
Xu, Jun
contents 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 Trigger$^3$, 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. Trigger$^3$ 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 Trigger$^3$ outperforms correction baselines while maintaining efficiency.
format Preprint
id arxiv_https___arxiv_org_abs_2412_12701
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Trigger$^3$: Refining Query Correction via Adaptive Model Selector
Zhang, Kepu
Sun, Zhongxiang
Zhang, Xiao
Zang, Xiaoxue
Zheng, Kai
Song, Yang
Xu, Jun
Computation and Language
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 Trigger$^3$, 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. Trigger$^3$ 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 Trigger$^3$ outperforms correction baselines while maintaining efficiency.
title Trigger$^3$: Refining Query Correction via Adaptive Model Selector
topic Computation and Language
url https://arxiv.org/abs/2412.12701