Transformers
PyTorch
t5
generative-retrieval
information-retrieval
msmarco
robustness
reproducibility
text-generation-inference
Instructions to use kiyam/lost-in-decoding-pag with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use kiyam/lost-in-decoding-pag with Transformers:
# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, T5ForLexicalSemanticGeneration tokenizer = AutoTokenizer.from_pretrained("kiyam/lost-in-decoding-pag") model = T5ForLexicalSemanticGeneration.from_pretrained("kiyam/lost-in-decoding-pag", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Upload PAG checkpoint, document identifiers, and Lost-in-Decoding evaluation artifacts
Browse files- .gitattributes +2 -0
- README.md +116 -0
- args.json +69 -0
- config.json +57 -0
- identifiers/README.md +43 -0
- identifiers/rq_docids.json +3 -0
- identifiers/set_docids.json +3 -0
- lost_in_decoding_artifacts/README.md +53 -0
- lost_in_decoding_artifacts/metadata/perturbed_queries_metadata.json +19 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_1999_attack_method_mispelling.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_1999_attack_method_naturality.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_1999_attack_method_ordering.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_1999_attack_method_paraphrase.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_1999_attack_method_synonym.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2016_attack_method_mispelling.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2016_attack_method_naturality.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2016_attack_method_ordering.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2016_attack_method_paraphrase.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2016_attack_method_synonym.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2026_attack_method_mispelling.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2026_attack_method_naturality.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2026_attack_method_ordering.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2026_attack_method_paraphrase.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2026_attack_method_synonym.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_27_attack_method_mispelling.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_27_attack_method_naturality.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_27_attack_method_ordering.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_27_attack_method_paraphrase.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_27_attack_method_synonym.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_5_attack_method_mispelling.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_5_attack_method_naturality.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_5_attack_method_ordering.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_5_attack_method_paraphrase.json +0 -0
- lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_5_attack_method_synonym.json +0 -0
- model_args.bin +3 -0
- pytorch_model.bin +3 -0
- special_tokens_map.json +1 -0
- spiece.model +3 -0
- tokenizer.json +0 -0
- tokenizer_config.json +1 -0
.gitattributes
CHANGED
|
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
|
|
|
|
|
| 33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
| 34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
| 35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
| 36 |
+
identifiers/rq_docids.json filter=lfs diff=lfs merge=lfs -text
|
| 37 |
+
identifiers/set_docids.json filter=lfs diff=lfs merge=lfs -text
|
README.md
ADDED
|
@@ -0,0 +1,116 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
tags:
|
| 4 |
+
- generative-retrieval
|
| 5 |
+
- information-retrieval
|
| 6 |
+
- t5
|
| 7 |
+
- msmarco
|
| 8 |
+
- robustness
|
| 9 |
+
- reproducibility
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
# Lost in Decoding? — PAG Checkpoint & Evaluation Artifacts
|
| 13 |
+
|
| 14 |
+
This repository hosts the model checkpoint, document identifiers, and evaluation
|
| 15 |
+
artifacts used by:
|
| 16 |
+
|
| 17 |
+
> **Lost in Decoding? Reproducing and Stress-Testing the Look-Ahead Prior in
|
| 18 |
+
> Generative Retrieval** (SIGIR 2026 Reproducibility Track, submission #298)
|
| 19 |
+
> [arXiv:2604.23396](https://arxiv.org/abs/2604.23396) ·
|
| 20 |
+
> [Code](https://github.com/kidist-amde/lost-in-decoding)
|
| 21 |
+
|
| 22 |
+
It is a **reproducibility and robustness study** of the PAG (Planner-Assisted
|
| 23 |
+
Generative retrieval) system. It does not introduce a new trained model — the
|
| 24 |
+
checkpoint and document identifiers below are the **original artifacts released
|
| 25 |
+
by the PAG authors**, re-hosted here for convenience and long-term availability.
|
| 26 |
+
The perturbed-query evaluation sets are **new artifacts created for this paper**.
|
| 27 |
+
|
| 28 |
+
## What's in this repository, and who made it
|
| 29 |
+
|
| 30 |
+
| Artifact | Created by | Description |
|
| 31 |
+
|---|---|---|
|
| 32 |
+
| `pytorch_model.bin`, `config.json`, tokenizer files | **PAG authors** (Zeng & Zamani, 2024) | Original PAG checkpoint (`ripor_direct_lng_knp_seq2seq_1`), unmodified. |
|
| 33 |
+
| `identifiers/rq_docids.json` | **PAG authors** | Sequential ("RQ"/AQ) document identifiers (`aq_smtid/docid_to_tokenids.json` in the upstream release). |
|
| 34 |
+
| `identifiers/set_docids.json` | **PAG authors** | Set-based ("lexical"/bag-of-words) document identifiers (`top_bow/docid_to_tokenids.json` in the upstream release). |
|
| 35 |
+
| `lost_in_decoding_artifacts/perturbed_queries/` | **This paper (Lost in Decoding)** | Perturbed MS MARCO Dev queries (misspelling, reordering, synonym replacement, paraphrasing, naturality normalization) used for the RQ2 robustness evaluation. |
|
| 36 |
+
| `lost_in_decoding_artifacts/metadata/` | **This paper (Lost in Decoding)** | Metadata describing generation settings (seeds, attack methods) for the perturbed-query sets. |
|
| 37 |
+
|
| 38 |
+
Multilingual (mMARCO) queries used for RQ3 are **not** re-hosted here; they are
|
| 39 |
+
third-party data. See the *Source data* section below.
|
| 40 |
+
|
| 41 |
+
## Model
|
| 42 |
+
|
| 43 |
+
- **Architecture:** `T5ForLexicalSemanticGeneration`, 12-layer encoder/decoder, 768 hidden size (T5-base scale).
|
| 44 |
+
- **Format:** `pytorch_model.bin` (`transformers` 4.17 era), not safetensors — uploaded as originally released.
|
| 45 |
+
- **Task:** Generative document retrieval over MS MARCO, decoding a two-stage (set-based + sequential) document identifier per query.
|
| 46 |
+
|
| 47 |
+
```python
|
| 48 |
+
from transformers import AutoTokenizer, AutoConfig
|
| 49 |
+
# Model class (T5ForLexicalSemanticGeneration) is defined in the Lost-in-Decoding /
|
| 50 |
+
# upstream PAG codebase, not in transformers. Load config/tokenizer directly,
|
| 51 |
+
# and use the repository's modeling code to load the checkpoint:
|
| 52 |
+
# https://github.com/kidist-amde/lost-in-decoding/blob/main/t5_pretrainer/modeling/t5_generative_retriever.py
|
| 53 |
+
|
| 54 |
+
tokenizer = AutoTokenizer.from_pretrained("kiyam/lost-in-decoding-pag")
|
| 55 |
+
config = AutoConfig.from_pretrained("kiyam/lost-in-decoding-pag")
|
| 56 |
+
```
|
| 57 |
+
|
| 58 |
+
See the [Lost-in-Decoding repository](https://github.com/kidist-amde/lost-in-decoding)
|
| 59 |
+
for the full two-stage constrained-decoding pipeline (lexical planning →
|
| 60 |
+
sequential constrained beam search) required to actually run retrieval with
|
| 61 |
+
this checkpoint.
|
| 62 |
+
|
| 63 |
+
## Document identifiers (`identifiers/`)
|
| 64 |
+
|
| 65 |
+
Both files map `docid (str) -> list[int]` (token ID sequences forming each
|
| 66 |
+
document's identifier). See [`identifiers/README.md`](identifiers/README.md)
|
| 67 |
+
for details.
|
| 68 |
+
|
| 69 |
+
## Evaluation artifacts (`lost_in_decoding_artifacts/`)
|
| 70 |
+
|
| 71 |
+
Perturbed MS MARCO Dev queries used in the RQ2 robustness evaluation
|
| 72 |
+
(Table 6 in the paper, corrected — see the main repository README for the
|
| 73 |
+
correction notice). See
|
| 74 |
+
[`lost_in_decoding_artifacts/perturbed_queries/README.md`](lost_in_decoding_artifacts/README.md)
|
| 75 |
+
for format and generation details.
|
| 76 |
+
|
| 77 |
+
## Source data
|
| 78 |
+
|
| 79 |
+
- **MS MARCO** passage ranking dataset (queries, qrels, collection): [microsoft/MSMARCO](https://microsoft.github.io/msmarco/) — used under its original terms.
|
| 80 |
+
- **TREC Deep Learning 2019 / 2020** query sets and qrels: [NIST TREC-DL](https://microsoft.github.io/msmarco/TREC-Deep-Learning.html).
|
| 81 |
+
- **mMARCO** (multilingual queries, RQ3): [unicamp-dl/mmarco](https://huggingface.co/datasets/unicamp-dl/mmarco) — not redistributed here; download via `cross_lingual/scripts/download_mmarco.sh` in the code repository.
|
| 82 |
+
|
| 83 |
+
## Citation
|
| 84 |
+
|
| 85 |
+
If you use the perturbed-query artifacts or the RQ2/RQ3 evaluation results, please cite:
|
| 86 |
+
|
| 87 |
+
```bibtex
|
| 88 |
+
@article{lost-in-decoding-2026,
|
| 89 |
+
title = {Lost in Decoding? Reproducing and Stress-Testing the Look-Ahead Prior in Generative Retrieval},
|
| 90 |
+
journal = {arXiv preprint arXiv:2604.23396},
|
| 91 |
+
year = {2026},
|
| 92 |
+
note = {SIGIR 2026 Reproducibility Track, submission \#298}
|
| 93 |
+
}
|
| 94 |
+
```
|
| 95 |
+
|
| 96 |
+
If you use the PAG checkpoint or document identifiers, please also cite the
|
| 97 |
+
original PAG paper:
|
| 98 |
+
|
| 99 |
+
```bibtex
|
| 100 |
+
@article{zeng2024pag,
|
| 101 |
+
title = {Planning Ahead in Generative Retrieval: Guiding Autoregressive Generation through Simultaneous Decoding},
|
| 102 |
+
author = {Zeng, Hansi and Zamani, Hamed},
|
| 103 |
+
journal = {arXiv preprint arXiv:2404.14600},
|
| 104 |
+
year = {2024}
|
| 105 |
+
}
|
| 106 |
+
```
|
| 107 |
+
|
| 108 |
+
- PAG paper: [arXiv:2404.14600](https://arxiv.org/pdf/2404.14600)
|
| 109 |
+
- PAG upstream repository: [github.com/HansiZeng/PAG](https://github.com/HansiZeng/PAG/tree/main)
|
| 110 |
+
|
| 111 |
+
## License
|
| 112 |
+
|
| 113 |
+
Apache 2.0, matching the [Lost-in-Decoding repository license](https://github.com/kidist-amde/lost-in-decoding/blob/main/LICENSE).
|
| 114 |
+
The PAG checkpoint and document identifiers are redistributed under the same
|
| 115 |
+
terms as the upstream PAG repository; consult the upstream repository for its
|
| 116 |
+
original license terms.
|
args.json
ADDED
|
@@ -0,0 +1,69 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_name_or_path": "t5-base",
|
| 3 |
+
"num_decoder_layers": null,
|
| 4 |
+
"teacher_score_path": "\/home\/ec2-user\/quic-efs\/user\/hansizeng\/work\/term_generative_retriever\/experiments-full-lexical-ripor\/t5-full-dense-1-5e-4-12l\/out\/MSMARCO_TRAIN\/qrel_added_merged_teacher_scores.json",
|
| 5 |
+
"collection_path": "\/home\/ec2-user\/quic-efs\/user\/hansizeng\/work\/data\/msmarco-full\/full_collection\/",
|
| 6 |
+
"queries_path": "\/home\/ec2-user\/quic-efs\/user\/hansizeng\/work\/data\/msmarco-full\/all_train_queries\/train_queries",
|
| 7 |
+
"qrels_path": "\/home\/ec2-user\/quic-efs\/user\/hansizeng\/work\/data\/msmarco\/train_queries\/qrels.json",
|
| 8 |
+
"output_dir": "\/home\/ec2-user\/quic-efs\/user\/hansizeng\/work\/term_generative_retriever\/experiments-full-lexical-ripor\/lexical_ripor_direct_lng_knp_seq2seq_1\/checkpoint",
|
| 9 |
+
"example_path": null,
|
| 10 |
+
"pseudo_queries_to_docid_path": null,
|
| 11 |
+
"pseudo_queries_to_mul_docid_path": null,
|
| 12 |
+
"docid_to_smtid_path": "",
|
| 13 |
+
"docid_to_tokenids_path": null,
|
| 14 |
+
"qid_to_smtid_path": null,
|
| 15 |
+
"first_centroid_path": null,
|
| 16 |
+
"second_centroid_path": null,
|
| 17 |
+
"third_centroid_path": null,
|
| 18 |
+
"qid_to_rrpids_path": null,
|
| 19 |
+
"docid_decode_eval_path": null,
|
| 20 |
+
"centroid_path": null,
|
| 21 |
+
"centroid_idx": null,
|
| 22 |
+
"triple_margin_mse_path": null,
|
| 23 |
+
"query_to_docid_path": null,
|
| 24 |
+
"teacher_rerank_nway_path": null,
|
| 25 |
+
"bce_example_path": null,
|
| 26 |
+
"smt_docid_to_smtid_path": "\/home\/ec2-user\/quic-efs\/user\/hansizeng\/work\/term_generative_retriever\/experiments-full-lexical-ripor\/t5-full-dense-1-5e-4-12l\/aq_smtid\/docid_to_tokenids.json",
|
| 27 |
+
"lex_docid_to_smtid_path": "\/home\/ec2-user\/quic-efs\/user\/hansizeng\/work\/term_generative_retriever\/experiments-splade\/t5-splade-0-12l\/\/top_bow\/docid_to_tokenids.json",
|
| 28 |
+
"run_name": "lexical_ripor_direct_lng_knp_seq2seq_1",
|
| 29 |
+
"pretrained_path": "\/home\/ec2-user\/quic-efs\/user\/hansizeng\/work\/term_generative_retriever\/experiments-full-lexical-ripor\/ripor_direct_lng_knp_seq2seq_1\/checkpoint\/",
|
| 30 |
+
"loss_type": "direct_lng_knp_margin_mse",
|
| 31 |
+
"model_type": "lexical_ripor",
|
| 32 |
+
"do_eval": false,
|
| 33 |
+
"max_length": 64,
|
| 34 |
+
"learning_rate": 0.0005,
|
| 35 |
+
"warmup_ratio": 0.04,
|
| 36 |
+
"per_device_train_batch_size": 64,
|
| 37 |
+
"logging_steps": 50,
|
| 38 |
+
"max_steps": -1,
|
| 39 |
+
"epochs": 120,
|
| 40 |
+
"local_rank": 0,
|
| 41 |
+
"task_names": [
|
| 42 |
+
"rank_4",
|
| 43 |
+
"rank",
|
| 44 |
+
"lexical_rank"
|
| 45 |
+
],
|
| 46 |
+
"ln_to_weight": {
|
| 47 |
+
"rank_4": 1.0,
|
| 48 |
+
"rank": 1.0,
|
| 49 |
+
"lexical_rank": 1.0
|
| 50 |
+
},
|
| 51 |
+
"multi_weights": null,
|
| 52 |
+
"nway_label_type": null,
|
| 53 |
+
"nway_rrpids": 24,
|
| 54 |
+
"nway": 12,
|
| 55 |
+
"eval_steps": 50,
|
| 56 |
+
"use_fp16": true,
|
| 57 |
+
"multi_vocab_sizes": false,
|
| 58 |
+
"save_steps": 15000,
|
| 59 |
+
"wandb_project_name": "full_lexical_ripor",
|
| 60 |
+
"pad_token_id": null,
|
| 61 |
+
"smtid_as_docid": false,
|
| 62 |
+
"apply_lex_loss": false,
|
| 63 |
+
"eval_collection_path": null,
|
| 64 |
+
"eval_queries_path": null,
|
| 65 |
+
"full_rank_eval_qrel_path": null,
|
| 66 |
+
"full_rank_eval_topk": 200,
|
| 67 |
+
"full_rank_index_dir": null,
|
| 68 |
+
"full_rank_out_dir": null
|
| 69 |
+
}
|
config.json
ADDED
|
@@ -0,0 +1,57 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_name_or_path": "/home/ec2-user/quic-efs/user/hansizeng/work/term_generative_retriever/experiments-full-lexical-ripor/ripor_direct_lng_knp_seq2seq_1/checkpoint/",
|
| 3 |
+
"architectures": [
|
| 4 |
+
"T5ForLexicalSemanticGeneration"
|
| 5 |
+
],
|
| 6 |
+
"d_ff": 3072,
|
| 7 |
+
"d_kv": 64,
|
| 8 |
+
"d_model": 768,
|
| 9 |
+
"decoder_start_token_id": 0,
|
| 10 |
+
"dropout_rate": 0.1,
|
| 11 |
+
"eos_token_id": 1,
|
| 12 |
+
"feed_forward_proj": "relu",
|
| 13 |
+
"initializer_factor": 1.0,
|
| 14 |
+
"is_encoder_decoder": true,
|
| 15 |
+
"layer_norm_epsilon": 1e-06,
|
| 16 |
+
"model_type": "t5",
|
| 17 |
+
"n_positions": 512,
|
| 18 |
+
"num_decoder_layers": 12,
|
| 19 |
+
"num_heads": 12,
|
| 20 |
+
"num_layers": 12,
|
| 21 |
+
"output_past": true,
|
| 22 |
+
"pad_token_id": 0,
|
| 23 |
+
"relative_attention_num_buckets": 32,
|
| 24 |
+
"task_specific_params": {
|
| 25 |
+
"summarization": {
|
| 26 |
+
"early_stopping": true,
|
| 27 |
+
"length_penalty": 2.0,
|
| 28 |
+
"max_length": 200,
|
| 29 |
+
"min_length": 30,
|
| 30 |
+
"no_repeat_ngram_size": 3,
|
| 31 |
+
"num_beams": 4,
|
| 32 |
+
"prefix": "summarize: "
|
| 33 |
+
},
|
| 34 |
+
"translation_en_to_de": {
|
| 35 |
+
"early_stopping": true,
|
| 36 |
+
"max_length": 300,
|
| 37 |
+
"num_beams": 4,
|
| 38 |
+
"prefix": "translate English to German: "
|
| 39 |
+
},
|
| 40 |
+
"translation_en_to_fr": {
|
| 41 |
+
"early_stopping": true,
|
| 42 |
+
"max_length": 300,
|
| 43 |
+
"num_beams": 4,
|
| 44 |
+
"prefix": "translate English to French: "
|
| 45 |
+
},
|
| 46 |
+
"translation_en_to_ro": {
|
| 47 |
+
"early_stopping": true,
|
| 48 |
+
"max_length": 300,
|
| 49 |
+
"num_beams": 4,
|
| 50 |
+
"prefix": "translate English to Romanian: "
|
| 51 |
+
}
|
| 52 |
+
},
|
| 53 |
+
"torch_dtype": "float32",
|
| 54 |
+
"transformers_version": "4.17.0",
|
| 55 |
+
"use_cache": true,
|
| 56 |
+
"vocab_size": 48484
|
| 57 |
+
}
|
identifiers/README.md
ADDED
|
@@ -0,0 +1,43 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Document Identifiers
|
| 2 |
+
|
| 3 |
+
**Provenance:** created by the PAG authors (Zeng & Zamani, 2024) and released
|
| 4 |
+
in the [upstream PAG repository](https://github.com/HansiZeng/PAG). Re-hosted
|
| 5 |
+
here unmodified.
|
| 6 |
+
|
| 7 |
+
Both files are JSON objects mapping `docid (str) -> list[int]`, where the list
|
| 8 |
+
is the sequence of token IDs that make up that document's generative
|
| 9 |
+
identifier under the PAG tokenizer/vocabulary.
|
| 10 |
+
|
| 11 |
+
## `rq_docids.json`
|
| 12 |
+
|
| 13 |
+
Sequential document identifiers produced by PAG's residual-quantization (AQ)
|
| 14 |
+
stage. Corresponds to `aq_smtid/docid_to_tokenids.json` in the upstream
|
| 15 |
+
release. Used by the sequential constrained-decoding stage of the PAG
|
| 16 |
+
pipeline (Stage 2).
|
| 17 |
+
|
| 18 |
+
## `set_docids.json`
|
| 19 |
+
|
| 20 |
+
Set-based (bag-of-words) document identifiers produced by PAG's lexical/SPLADE
|
| 21 |
+
planning stage. Corresponds to `top_bow/docid_to_tokenids.json` in the
|
| 22 |
+
upstream release. Used by the lexical planning stage of the PAG pipeline
|
| 23 |
+
(Stage 1).
|
| 24 |
+
|
| 25 |
+
## Usage
|
| 26 |
+
|
| 27 |
+
See the two-stage constrained-decoding pipeline in the
|
| 28 |
+
[Lost-in-Decoding repository](https://github.com/kidist-amde/lost-in-decoding)
|
| 29 |
+
(`t5_pretrainer/evaluate.py`) for how these identifier files are consumed
|
| 30 |
+
during retrieval.
|
| 31 |
+
|
| 32 |
+
## Citation
|
| 33 |
+
|
| 34 |
+
If you use these identifiers, please cite the original PAG paper:
|
| 35 |
+
|
| 36 |
+
```bibtex
|
| 37 |
+
@article{zeng2024pag,
|
| 38 |
+
title = {Planning Ahead in Generative Retrieval: Guiding Autoregressive Generation through Simultaneous Decoding},
|
| 39 |
+
author = {Zeng, Hansi and Zamani, Hamed},
|
| 40 |
+
journal = {arXiv preprint arXiv:2404.14600},
|
| 41 |
+
year = {2024}
|
| 42 |
+
}
|
| 43 |
+
```
|
identifiers/rq_docids.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:fe2b082d95316560bc8637d36d96e32f5559712602dd863db1ad21e07c5b0f56
|
| 3 |
+
size 529398271
|
identifiers/set_docids.json
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:29e6b3b0cb74db6dcbc34f5f7e9cb6c928e60e0a381bd460da5de5a7d7a624b2
|
| 3 |
+
size 2970761161
|
lost_in_decoding_artifacts/README.md
ADDED
|
@@ -0,0 +1,53 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Lost in Decoding — Evaluation Artifacts
|
| 2 |
+
|
| 3 |
+
**Provenance:** created for this paper (*Lost in Decoding? Reproducing and
|
| 4 |
+
Stress-Testing the Look-Ahead Prior in Generative Retrieval*,
|
| 5 |
+
[arXiv:2604.23396](https://arxiv.org/abs/2604.23396)), not part of the
|
| 6 |
+
original PAG release.
|
| 7 |
+
|
| 8 |
+
## `perturbed_queries/`
|
| 9 |
+
|
| 10 |
+
Perturbed MS MARCO Dev query sets used for the RQ2 robustness evaluation
|
| 11 |
+
(query-perturbation sensitivity of PAG's look-ahead prior).
|
| 12 |
+
|
| 13 |
+
- **Split:** MS MARCO Dev (6980 queries).
|
| 14 |
+
- **Attack methods:** `mispelling`, `ordering` (word reordering), `synonym`
|
| 15 |
+
(synonym replacement), `paraphrase`, `naturality` (naturality
|
| 16 |
+
normalization). Implemented in
|
| 17 |
+
[`robustness/query_variations/penha/`](https://github.com/kidist-amde/lost-in-decoding/tree/main/robustness/query_variations/penha)
|
| 18 |
+
in the code repository.
|
| 19 |
+
- **Seeds:** `1999`, `5`, `27`, `2016`, `2026`.
|
| 20 |
+
- **File naming:** `msmarco_test_attacked_queries_seed_{seed}_attack_method_{attack_method}.json`
|
| 21 |
+
- **Format:** JSON object, `{qid (str): perturbed_query_text (str)}`.
|
| 22 |
+
- **Note (`naturality`):** this transformation is a deterministic
|
| 23 |
+
style-normalization rule rather than a randomized perturbation, so the five
|
| 24 |
+
seed files for `naturality` are identical to each other by construction.
|
| 25 |
+
|
| 26 |
+
**TREC-DL 2019 / TREC-DL 2020 perturbed queries are not included as static
|
| 27 |
+
files.** They are produced by the same code path
|
| 28 |
+
(`robustness/scripts/generate_query_variations_sub.sh`) using the same seeds
|
| 29 |
+
and attack methods, and were consumed directly during RQ2 evaluation without
|
| 30 |
+
being cached to disk. Regenerate them with:
|
| 31 |
+
|
| 32 |
+
```bash
|
| 33 |
+
bash robustness/scripts/generate_query_variations_sub.sh
|
| 34 |
+
```
|
| 35 |
+
|
| 36 |
+
## `metadata/`
|
| 37 |
+
|
| 38 |
+
`perturbed_queries_metadata.json` documents the generation settings (split,
|
| 39 |
+
seeds, attack methods, file naming convention) for the released perturbed
|
| 40 |
+
query sets.
|
| 41 |
+
|
| 42 |
+
## Citation
|
| 43 |
+
|
| 44 |
+
If you use these artifacts, please cite:
|
| 45 |
+
|
| 46 |
+
```bibtex
|
| 47 |
+
@article{lost-in-decoding-2026,
|
| 48 |
+
title = {Lost in Decoding? Reproducing and Stress-Testing the Look-Ahead Prior in Generative Retrieval},
|
| 49 |
+
journal = {arXiv preprint arXiv:2604.23396},
|
| 50 |
+
year = {2026},
|
| 51 |
+
note = {SIGIR 2026 Reproducibility Track, submission \#298}
|
| 52 |
+
}
|
| 53 |
+
```
|
lost_in_decoding_artifacts/metadata/perturbed_queries_metadata.json
ADDED
|
@@ -0,0 +1,19 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"description": "Metadata for the perturbed MS MARCO Dev query sets released alongside 'Lost in Decoding? Reproducing and Stress-Testing the Look-Ahead Prior in Generative Retrieval' (SIGIR 2026, submission #298).",
|
| 3 |
+
"split": "msmarco_dev",
|
| 4 |
+
"num_queries_per_file": 6980,
|
| 5 |
+
"attack_methods": [
|
| 6 |
+
"mispelling",
|
| 7 |
+
"ordering",
|
| 8 |
+
"synonym",
|
| 9 |
+
"paraphrase",
|
| 10 |
+
"naturality"
|
| 11 |
+
],
|
| 12 |
+
"seeds": [1999, 5, 27, 2016, 2026],
|
| 13 |
+
"file_naming": "msmarco_test_attacked_queries_seed_{seed}_attack_method_{attack_method}.json",
|
| 14 |
+
"format": "JSON object mapping qid (string) -> perturbed query text (string)",
|
| 15 |
+
"generation_code": "robustness/query_variations (penha transformations) in the Lost-in-Decoding repository",
|
| 16 |
+
"note": "TREC-DL 2019 and TREC-DL 2020 perturbed queries are not included as static files here. They are generated on demand by the same code path (robustness/scripts/generate_query_variations_sub.sh) using the same seeds and attack methods, and were consumed directly by the RQ2 evaluation pipeline without being cached to disk.",
|
| 17 |
+
"source_paper": "Lost in Decoding? Reproducing and Stress-Testing the Look-Ahead Prior in Generative Retrieval (arXiv:2604.23396)",
|
| 18 |
+
"source_repository": "https://github.com/kidist-amde/lost-in-decoding"
|
| 19 |
+
}
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_1999_attack_method_mispelling.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_1999_attack_method_naturality.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_1999_attack_method_ordering.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_1999_attack_method_paraphrase.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_1999_attack_method_synonym.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2016_attack_method_mispelling.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2016_attack_method_naturality.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2016_attack_method_ordering.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2016_attack_method_paraphrase.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2016_attack_method_synonym.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2026_attack_method_mispelling.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2026_attack_method_naturality.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2026_attack_method_ordering.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2026_attack_method_paraphrase.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2026_attack_method_synonym.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_27_attack_method_mispelling.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_27_attack_method_naturality.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_27_attack_method_ordering.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_27_attack_method_paraphrase.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_27_attack_method_synonym.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_5_attack_method_mispelling.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_5_attack_method_naturality.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_5_attack_method_ordering.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_5_attack_method_paraphrase.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_5_attack_method_synonym.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
model_args.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c71b3f8892a3ca56cac5615e641aa34a8f2500bbebbdbc7406d99166dad49555
|
| 3 |
+
size 431
|
pytorch_model.bin
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:9069588676ced21d94df2234690a93aaafa9bac4b2a4018b11c11874c480c4a1
|
| 3 |
+
size 941976511
|
special_tokens_map.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"eos_token": "</s>", "unk_token": "<unk>", "pad_token": "<pad>", "additional_special_tokens": ["<extra_id_0>", "<extra_id_1>", "<extra_id_2>", "<extra_id_3>", "<extra_id_4>", "<extra_id_5>", "<extra_id_6>", "<extra_id_7>", "<extra_id_8>", "<extra_id_9>", "<extra_id_10>", "<extra_id_11>", "<extra_id_12>", "<extra_id_13>", "<extra_id_14>", "<extra_id_15>", "<extra_id_16>", "<extra_id_17>", "<extra_id_18>", "<extra_id_19>", "<extra_id_20>", "<extra_id_21>", "<extra_id_22>", "<extra_id_23>", "<extra_id_24>", "<extra_id_25>", "<extra_id_26>", "<extra_id_27>", "<extra_id_28>", "<extra_id_29>", "<extra_id_30>", "<extra_id_31>", "<extra_id_32>", "<extra_id_33>", "<extra_id_34>", "<extra_id_35>", "<extra_id_36>", "<extra_id_37>", "<extra_id_38>", "<extra_id_39>", "<extra_id_40>", "<extra_id_41>", "<extra_id_42>", "<extra_id_43>", "<extra_id_44>", "<extra_id_45>", "<extra_id_46>", "<extra_id_47>", "<extra_id_48>", "<extra_id_49>", "<extra_id_50>", "<extra_id_51>", "<extra_id_52>", "<extra_id_53>", "<extra_id_54>", "<extra_id_55>", "<extra_id_56>", "<extra_id_57>", "<extra_id_58>", "<extra_id_59>", "<extra_id_60>", "<extra_id_61>", "<extra_id_62>", "<extra_id_63>", "<extra_id_64>", "<extra_id_65>", "<extra_id_66>", "<extra_id_67>", "<extra_id_68>", "<extra_id_69>", "<extra_id_70>", "<extra_id_71>", "<extra_id_72>", "<extra_id_73>", "<extra_id_74>", "<extra_id_75>", "<extra_id_76>", "<extra_id_77>", "<extra_id_78>", "<extra_id_79>", "<extra_id_80>", "<extra_id_81>", "<extra_id_82>", "<extra_id_83>", "<extra_id_84>", "<extra_id_85>", "<extra_id_86>", "<extra_id_87>", "<extra_id_88>", "<extra_id_89>", "<extra_id_90>", "<extra_id_91>", "<extra_id_92>", "<extra_id_93>", "<extra_id_94>", "<extra_id_95>", "<extra_id_96>", "<extra_id_97>", "<extra_id_98>", "<extra_id_99>"]}
|
spiece.model
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d60acb128cf7b7f2536e8f38a5b18a05535c9e14c7a355904270e15b0945ea86
|
| 3 |
+
size 791656
|
tokenizer.json
ADDED
|
The diff for this file is too large to render.
See raw diff
|
|
|
tokenizer_config.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"eos_token": "</s>", "unk_token": "<unk>", "pad_token": "<pad>", "extra_ids": 100, "additional_special_tokens": ["<extra_id_0>", "<extra_id_1>", "<extra_id_2>", "<extra_id_3>", "<extra_id_4>", "<extra_id_5>", "<extra_id_6>", "<extra_id_7>", "<extra_id_8>", "<extra_id_9>", "<extra_id_10>", "<extra_id_11>", "<extra_id_12>", "<extra_id_13>", "<extra_id_14>", "<extra_id_15>", "<extra_id_16>", "<extra_id_17>", "<extra_id_18>", "<extra_id_19>", "<extra_id_20>", "<extra_id_21>", "<extra_id_22>", "<extra_id_23>", "<extra_id_24>", "<extra_id_25>", "<extra_id_26>", "<extra_id_27>", "<extra_id_28>", "<extra_id_29>", "<extra_id_30>", "<extra_id_31>", "<extra_id_32>", "<extra_id_33>", "<extra_id_34>", "<extra_id_35>", "<extra_id_36>", "<extra_id_37>", "<extra_id_38>", "<extra_id_39>", "<extra_id_40>", "<extra_id_41>", "<extra_id_42>", "<extra_id_43>", "<extra_id_44>", "<extra_id_45>", "<extra_id_46>", "<extra_id_47>", "<extra_id_48>", "<extra_id_49>", "<extra_id_50>", "<extra_id_51>", "<extra_id_52>", "<extra_id_53>", "<extra_id_54>", "<extra_id_55>", "<extra_id_56>", "<extra_id_57>", "<extra_id_58>", "<extra_id_59>", "<extra_id_60>", "<extra_id_61>", "<extra_id_62>", "<extra_id_63>", "<extra_id_64>", "<extra_id_65>", "<extra_id_66>", "<extra_id_67>", "<extra_id_68>", "<extra_id_69>", "<extra_id_70>", "<extra_id_71>", "<extra_id_72>", "<extra_id_73>", "<extra_id_74>", "<extra_id_75>", "<extra_id_76>", "<extra_id_77>", "<extra_id_78>", "<extra_id_79>", "<extra_id_80>", "<extra_id_81>", "<extra_id_82>", "<extra_id_83>", "<extra_id_84>", "<extra_id_85>", "<extra_id_86>", "<extra_id_87>", "<extra_id_88>", "<extra_id_89>", "<extra_id_90>", "<extra_id_91>", "<extra_id_92>", "<extra_id_93>", "<extra_id_94>", "<extra_id_95>", "<extra_id_96>", "<extra_id_97>", "<extra_id_98>", "<extra_id_99>"], "model_max_length": 512, "special_tokens_map_file": null, "name_or_path": "t5-base", "tokenizer_class": "T5Tokenizer"}
|