IncDSI: Incrementally Updatable Document Retrieval

التفاصيل البيبلوغرافية
العنوان: IncDSI: Incrementally Updatable Document Retrieval
المؤلفون: Kishore, Varsha, Wan, Chao, Lovelace, Justin, Artzi, Yoav, Weinberger, Kilian Q.
سنة النشر: 2023
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Information Retrieval, Computer Science - Computation and Language, Computer Science - Machine Learning
الوصف: Differentiable Search Index is a recently proposed paradigm for document retrieval, that encodes information about a corpus of documents within the parameters of a neural network and directly maps queries to corresponding documents. These models have achieved state-of-the-art performances for document retrieval across many benchmarks. These kinds of models have a significant limitation: it is not easy to add new documents after a model is trained. We propose IncDSI, a method to add documents in real time (about 20-50ms per document), without retraining the model on the entire dataset (or even parts thereof). Instead we formulate the addition of documents as a constrained optimization problem that makes minimal changes to the network parameters. Although orders of magnitude faster, our approach is competitive with re-training the model on the whole dataset and enables the development of document retrieval systems that can be updated with new information in real-time. Our code for IncDSI is available at https://github.com/varshakishore/IncDSI.
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2307.10323
رقم الأكسشن: edsarx.2307.10323
قاعدة البيانات: arXiv