AmazonQAC: A Large-Scale, Naturalistic Query Autocomplete Dataset

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Everaert, Dante, Patki, Rohit, Zheng, Tianqi, Potts, Christopher
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866929580277760000
author Everaert, Dante
Patki, Rohit
Zheng, Tianqi
Potts, Christopher
author_facet Everaert, Dante
Patki, Rohit
Zheng, Tianqi
Potts, Christopher
contents Query Autocomplete (QAC) is a critical feature in modern search engines, facilitating user interaction by predicting search queries based on input prefixes. Despite its widespread adoption, the absence of large-scale, realistic datasets has hindered advancements in QAC system development. This paper addresses this gap by introducing AmazonQAC, a new QAC dataset sourced from Amazon Search logs, comprising 395M samples. The dataset includes actual sequences of user-typed prefixes leading to final search terms, as well as session IDs and timestamps that support modeling the context-dependent aspects of QAC. We assess Prefix Trees, semantic retrieval, and Large Language Models (LLMs) with and without finetuning. We find that finetuned LLMs perform best, particularly when incorporating contextual information. However, even our best system achieves only half of what we calculate is theoretically possible on our test data, which implies QAC is a challenging problem that is far from solved with existing systems. This contribution aims to stimulate further research on QAC systems to better serve user needs in diverse environments. We open-source this data on Hugging Face at https://huggingface.co/datasets/amazon/AmazonQAC.
format Preprint
id arxiv_https___arxiv_org_abs_2411_04129
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle AmazonQAC: A Large-Scale, Naturalistic Query Autocomplete Dataset
Everaert, Dante
Patki, Rohit
Zheng, Tianqi
Potts, Christopher
Information Retrieval
Artificial Intelligence
Machine Learning
Query Autocomplete (QAC) is a critical feature in modern search engines, facilitating user interaction by predicting search queries based on input prefixes. Despite its widespread adoption, the absence of large-scale, realistic datasets has hindered advancements in QAC system development. This paper addresses this gap by introducing AmazonQAC, a new QAC dataset sourced from Amazon Search logs, comprising 395M samples. The dataset includes actual sequences of user-typed prefixes leading to final search terms, as well as session IDs and timestamps that support modeling the context-dependent aspects of QAC. We assess Prefix Trees, semantic retrieval, and Large Language Models (LLMs) with and without finetuning. We find that finetuned LLMs perform best, particularly when incorporating contextual information. However, even our best system achieves only half of what we calculate is theoretically possible on our test data, which implies QAC is a challenging problem that is far from solved with existing systems. This contribution aims to stimulate further research on QAC systems to better serve user needs in diverse environments. We open-source this data on Hugging Face at https://huggingface.co/datasets/amazon/AmazonQAC.
title AmazonQAC: A Large-Scale, Naturalistic Query Autocomplete Dataset
topic Information Retrieval
Artificial Intelligence
Machine Learning
url https://arxiv.org/abs/2411.04129