Model Editing for New Document Integration in Generative Information Retrieval

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Zhang, Zhen, Wang, Zihan, Ma, Xinyu, Wang, Shuaiqiang, Yin, Dawei, Xin, Xin, Ren, Pengjie, de Rijke, Maarten, Ren, Zhaochun
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866918493339779072
author Zhang, Zhen
Wang, Zihan
Ma, Xinyu
Wang, Shuaiqiang
Yin, Dawei
Xin, Xin
Ren, Pengjie
de Rijke, Maarten
Ren, Zhaochun
author_facet Zhang, Zhen
Wang, Zihan
Ma, Xinyu
Wang, Shuaiqiang
Yin, Dawei
Xin, Xin
Ren, Pengjie
de Rijke, Maarten
Ren, Zhaochun
contents Generative retrieval (GR) reformulates the Information Retrieval (IR) task as the generation of document identifiers (docIDs). Despite its promise, existing GR models exhibit poor generalization to newly added documents, often failing to generate the correct docIDs. While incremental training offers a straightforward remedy, it is computationally expensive, resource-intensive, and prone to catastrophic forgetting, thereby limiting the scalability and practicality of GR. In this paper, we identify the core bottleneck as the decoder's ability to map hidden states to the correct docIDs of newly added documents. Model editing, which enables targeted parameter modifications for docID mapping, represents a promising solution. However, applying model editing to current GR models is not trivial, which is severely hindered by indistinguishable edit vectors across queries, due to the high overlap of shared docIDs in retrieval results. To address this, we propose DOME (docID-oriented model editing), a novel method that effectively and efficiently adapts GR models to unseen documents. DOME comprises three stages: (1) identification of critical layers, (2) optimization of edit vectors, and (3) construction and application of updates. At its core, DOME employs a hybrid-label adaptive training strategy that learns discriminative edit vectors by combining soft labels, which preserve query-specific semantics for distinguishable updates, with hard labels that enforce precise mapping modifications. Experiments on widely used benchmarks, including NQ and MS MARCO, show that our method significantly improves retrieval performance on new documents while maintaining effectiveness on the original collection. Moreover, DOME achieves this with only about 60% of the training time required by incremental training, considerably reducing computational cost and enabling efficient, frequent model updates.
format Preprint
id arxiv_https___arxiv_org_abs_2603_02773
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Model Editing for New Document Integration in Generative Information Retrieval
Zhang, Zhen
Wang, Zihan
Ma, Xinyu
Wang, Shuaiqiang
Yin, Dawei
Xin, Xin
Ren, Pengjie
de Rijke, Maarten
Ren, Zhaochun
Information Retrieval
Generative retrieval (GR) reformulates the Information Retrieval (IR) task as the generation of document identifiers (docIDs). Despite its promise, existing GR models exhibit poor generalization to newly added documents, often failing to generate the correct docIDs. While incremental training offers a straightforward remedy, it is computationally expensive, resource-intensive, and prone to catastrophic forgetting, thereby limiting the scalability and practicality of GR. In this paper, we identify the core bottleneck as the decoder's ability to map hidden states to the correct docIDs of newly added documents. Model editing, which enables targeted parameter modifications for docID mapping, represents a promising solution. However, applying model editing to current GR models is not trivial, which is severely hindered by indistinguishable edit vectors across queries, due to the high overlap of shared docIDs in retrieval results. To address this, we propose DOME (docID-oriented model editing), a novel method that effectively and efficiently adapts GR models to unseen documents. DOME comprises three stages: (1) identification of critical layers, (2) optimization of edit vectors, and (3) construction and application of updates. At its core, DOME employs a hybrid-label adaptive training strategy that learns discriminative edit vectors by combining soft labels, which preserve query-specific semantics for distinguishable updates, with hard labels that enforce precise mapping modifications. Experiments on widely used benchmarks, including NQ and MS MARCO, show that our method significantly improves retrieval performance on new documents while maintaining effectiveness on the original collection. Moreover, DOME achieves this with only about 60% of the training time required by incremental training, considerably reducing computational cost and enabling efficient, frequent model updates.
title Model Editing for New Document Integration in Generative Information Retrieval
topic Information Retrieval
url https://arxiv.org/abs/2603.02773