kiyam commited on
Commit
247baec
·
verified ·
1 Parent(s): 308d775

Upload PAG checkpoint, document identifiers, and Lost-in-Decoding evaluation artifacts

Browse files
Files changed (40) hide show
  1. .gitattributes +2 -0
  2. README.md +116 -0
  3. args.json +69 -0
  4. config.json +57 -0
  5. identifiers/README.md +43 -0
  6. identifiers/rq_docids.json +3 -0
  7. identifiers/set_docids.json +3 -0
  8. lost_in_decoding_artifacts/README.md +53 -0
  9. lost_in_decoding_artifacts/metadata/perturbed_queries_metadata.json +19 -0
  10. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_1999_attack_method_mispelling.json +0 -0
  11. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_1999_attack_method_naturality.json +0 -0
  12. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_1999_attack_method_ordering.json +0 -0
  13. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_1999_attack_method_paraphrase.json +0 -0
  14. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_1999_attack_method_synonym.json +0 -0
  15. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2016_attack_method_mispelling.json +0 -0
  16. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2016_attack_method_naturality.json +0 -0
  17. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2016_attack_method_ordering.json +0 -0
  18. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2016_attack_method_paraphrase.json +0 -0
  19. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2016_attack_method_synonym.json +0 -0
  20. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2026_attack_method_mispelling.json +0 -0
  21. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2026_attack_method_naturality.json +0 -0
  22. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2026_attack_method_ordering.json +0 -0
  23. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2026_attack_method_paraphrase.json +0 -0
  24. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_2026_attack_method_synonym.json +0 -0
  25. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_27_attack_method_mispelling.json +0 -0
  26. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_27_attack_method_naturality.json +0 -0
  27. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_27_attack_method_ordering.json +0 -0
  28. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_27_attack_method_paraphrase.json +0 -0
  29. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_27_attack_method_synonym.json +0 -0
  30. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_5_attack_method_mispelling.json +0 -0
  31. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_5_attack_method_naturality.json +0 -0
  32. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_5_attack_method_ordering.json +0 -0
  33. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_5_attack_method_paraphrase.json +0 -0
  34. lost_in_decoding_artifacts/perturbed_queries/msmarco_test_attacked_queries_seed_5_attack_method_synonym.json +0 -0
  35. model_args.bin +3 -0
  36. pytorch_model.bin +3 -0
  37. special_tokens_map.json +1 -0
  38. spiece.model +3 -0
  39. tokenizer.json +0 -0
  40. 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"}