Neighborhood Elites: Quality-Diversity Search for Uncertainty Quantification in Question Answering
Organizations: ETH Zurich, Zurich, Switzerland · University of St.Gallen, St.Gallen, Switzerland
Abstract
Large language models (LLMs) are increasingly capable, yet they produce wrong answers indistinguishable from correct ones. In question answering (QA), such an error propagates unnoticed into downstream tasks. Perturbation-based uncertainty quantification (UQ) addresses this by probing a model over a set of input rewrites and reading uncertainty from how much its predictions vary. Yet this set is usually built generically, by free paraphrasing, token-level perturbation, or adversarial search. Such rewrites leave two things uncontrolled that matter in QA: what kind of change is made to the context, and how explicitly the context supports the answer. When neither is controlled, the composition of the set is determined by the generator rather than by design, and the estimate inherits whatever that generator happens to produce. Quality-diversity (QD) search is built to cover a predefined exploration space, so we recast input perturbation as a QD problem and propose neighborhood elites (NE), a neighborhood constructor that spreads answer-preserving rewrites across two axes, the kind of edit and the explicitness of support. We compare NE against three other neighborhood constructors, holding the editor, QA model, uncertainty measure and search objective fixed, and against three scores read from the original context. Across five extractive-QA datasets and five instruction-tuned LLMs, NE improves failure detection against every other constructor in at least 24 of the 25 dataset-model pairs, reaching 0.726 AUROC and 0.211 AUPRC lift over the base rate. Additionally, NE detects failure better than semantic entropy computed over decoder samples, by 0.074 AUROC and 0.062 AUPRC lift, and this advantage widens as the neighborhood grows.
Figures & tables
| AUROC | AUPRC lift | ||||
| Method | Value | Value | |||
| Neighborhood constructed | |||||
| NE | — | — | |||
| U-ME | |||||
| SBS | |||||
| FP | |||||
| FP | SBS | U-ME | NE | |
|---|---|---|---|---|
| Cells occupied, of twelve | ||||
| Rewrites kept | ||||
| Rewrites per instance | ||||
| Rewrites per occupied cell | ||||
| Instances reaching cells |
| Method | AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC | AUROC | AUPRC |
|---|---|---|---|---|---|---|---|---|---|---|
| NE | 0.705 | 0.215 | 0.746 | 0.262 | 0.765 | 0.282 | 0.780 | 0.301 | 0.805 | 0.329 |
| U-ME | 0.637 | 0.150 | 0.667 | 0.184 | 0.683 | 0.206 | 0.685 | 0.212 | 0.713 | 0.252 |
| FP | 0.607 | 0.130 | 0.622 | 0.150 | 0.626 | 0.151 | 0.633 | 0.155 | 0.654 | 0.202 |
| PE | 0.677 | 0.133 | 0.677 | 0.139 | 0.674 | 0.137 | 0.684 | 0.156 | 0.686 | 0.166 |
| NLL | 0.672 | 0.127 | 0.670 | 0.129 | 0.666 | 0.124 | 0.678 | 0.144 | 0.684 | 0.170 |
Appendix figures & tables40 assets
Supplementary material from the paper’s appendix.
Appendix
| SQuAD v1.1 | Quoref | AdversarialQA | Nat. Questions | NewsQA | Total | Per pair | Base rate | |
|---|---|---|---|---|---|---|---|---|
| 3 | 3980 | 3587 | 3673 | 2566 | 1924 | 15730 | 629 | 0.300 |
| 4 | 2866 | 2637 | 2751 | 1822 | 1281 | 11357 | 454 | 0.284 |
| 5 | 2476 | 1924 | 2475 | 1257 | 830 | 8962 | 358 | 0.276 |
| 6 | 1786 | 1209 | 1819 | 675 | 441 | 5930 | 237 | 0.276 |
| 7 | 705 | 435 | 747 | 210 | 176 | 2273 | 91 | 0.300 |
| NE FP | NE U-ME | U-ME FP | ||||
| QA model | ||||||
| SQuAD | ||||||
| Falcon3-10B | ||||||
| Mistral-Nemo-12B | ||||||
| Gemma3-12B | ||||||
| OLMo2-13B | ||||||
| NE FP | NE U-ME | U-ME FP | ||||
| QA model | ||||||
| SQuAD | ||||||
| Falcon3-10B | ||||||
| Mistral-Nemo-12B | ||||||
| Gemma3-12B | ||||||
| OLMo2-13B | ||||||
| Constructor | |||||
|---|---|---|---|---|---|
| Semantic entropy | |||||
| FP | |||||
| U-ME | |||||
| NE | |||||
| Constructor | |||||
|---|---|---|---|---|---|
| Semantic entropy | |||||
| FP | |||||
| U-ME | |||||
| NE | |||||
| Neighborhood constructed | Original context | |||||||
| Dataset | Model | NE | U-ME | SBS | FP | PE | NLL | DSE |
| SQuAD v1.1 | Falcon3-10B | |||||||
| Gemma3-12B | ||||||||
| Mistral-NeMo-12B | ||||||||
| OLMo2-13B | ||||||||
| Phi-4-14B | ||||||||
| Neighborhood constructed | Original context | |||||||
| Dataset | Model | NE | U-ME | SBS | FP | PE | NLL | DSE |
| SQuAD v1.1 | Falcon3-10B | |||||||
| Gemma3-12B | ||||||||
| Mistral-NeMo-12B | ||||||||
| OLMo2-13B | ||||||||
| Phi-4-14B | ||||||||
| FP | SBS | U-ME | NE | |
| Per instance, mean over 5000 instances | ||||
| Editor calls | 15.43 | 25.00 | 24.26 | 24.21 |
| Proposals passing the gates | 3.69 | 10.35 | 21.67 | 12.12 |
| Retained in the archive | 3.69 | 2.01 | 5.90 | 4.43 |
| Distinct | 3.69 | 2.01 | 4.94 | 4.43 |
| Duplicate rate | 0.00 | 0.00 | 0.17 | 0.00 |
| Dataset | FP | SBS | U-ME | NE |
|---|---|---|---|---|
| SQuAD v1.1 | ||||
| Quoref | ||||
| AdversarialQA | ||||
| Natural Questions | ||||
| NewsQA | ||||
| Macro |
| Method | AUROC | AUPRC | Lift | Base | Cov. | sd | min – max | ||
|---|---|---|---|---|---|---|---|---|---|
| 1 | NE | 0.702 | 0.546 | 0.189 | 0.357 | 23739 | 0.950 | 0.053 | 0.602–0.776 |
| 2 | PE | 0.676 | 0.488 | 0.131 | 0.357 | 23739 | 0.950 | 0.047 | 0.582–0.772 |
| 3 | NLL | 0.674 | 0.492 | 0.131 | 0.361 | 24984 | 0.999 | 0.050 | 0.563–0.761 |
| 4 | U-ME | 0.658 | 0.496 | 0.146 | 0.351 | 24060 | 0.962 | 0.020 | 0.624–0.686 |
| 5 | DSE | 0.652 | 0.498 | 0.149 | 0.349 | 21920 | 0.877 | 0.066 | 0.518–0.734 |
| 6 | FP | 0.605 | 0.421 | 0.097 | 0.323 | 19569 | 0.783 | 0.031 | 0.546–0.651 |
| Method | AUROC | AUPRC | Lift | Base | Cov. | sd | min – max | ||
|---|---|---|---|---|---|---|---|---|---|
| 1 | NE | 0.726 | 0.561 | 0.211 | 0.349 | 21920 | 0.877 | 0.038 | 0.647–0.786 |
| 2 | PE | 0.676 | 0.480 | 0.131 | 0.349 | 21920 | 0.877 | 0.046 | 0.580–0.765 |
| 3 | NLL | 0.674 | 0.492 | 0.131 | 0.361 | 24984 | 0.999 | 0.050 | 0.563–0.761 |
| 4 | U-ME | 0.661 | 0.497 | 0.148 | 0.349 | 23920 | 0.957 | 0.019 | 0.627–0.689 |
| 5 | DSE | 0.652 | 0.498 | 0.149 | 0.349 | 21920 | 0.877 | 0.066 | 0.518–0.734 |
| 6 | FP | 0.621 | 0.428 | 0.113 | 0.315 | 17864 | 0.715 | 0.032 | 0.563–0.679 |
| Dataset | NE | U-ME | FP | DSE | PE | NLL |
|---|---|---|---|---|---|---|
| AUROC | ||||||
| SQuAD v1.1 | 0.742 | 0.671 | 0.616 | 0.676 | 0.673 | 0.666 |
| Quoref | 0.740 | 0.674 | 0.619 | 0.614 | 0.663 | 0.662 |
| NewsQA | 0.647 | 0.636 | 0.581 | 0.668 | 0.686 | 0.691 |
| Natural Questions | 0.638 | 0.652 | 0.594 | 0.643 | 0.728 | 0.731 |
| AdversarialQA | 0.741 | 0.659 | 0.613 | 0.658 | 0.628 | 0.620 |
| Dataset | NE | U-ME | FP | DSE | PE | NLL |
|---|---|---|---|---|---|---|
| AUROC | ||||||
| SQuAD v1.1 | 0.752 | 0.672 | 0.628 | 0.676 | 0.678 | 0.666 |
| Quoref | 0.756 | 0.674 | 0.626 | 0.614 | 0.665 | 0.662 |
| NewsQA | 0.687 | 0.641 | 0.611 | 0.668 | 0.687 | 0.691 |
| Natural Questions | 0.687 | 0.654 | 0.617 | 0.643 | 0.722 | 0.731 |
| AdversarialQA | 0.746 | 0.662 | 0.623 | 0.658 | 0.628 | 0.620 |
| QA model | NE | U-ME | FP | DSE | PE | NLL |
|---|---|---|---|---|---|---|
| AUROC | ||||||
| Gemma3-12B | 0.698 | 0.650 | 0.588 | 0.538 | 0.648 | 0.646 |
| OLMo2-13B-Instruct | 0.708 | 0.661 | 0.594 | 0.685 | 0.693 | 0.693 |
| Phi4-14B | 0.713 | 0.668 | 0.641 | 0.707 | 0.656 | 0.654 |
| Mistral-Nemo-12B | 0.692 | 0.648 | 0.578 | 0.670 | 0.706 | 0.705 |
| Falcon3-10B-Instruct | 0.698 | 0.665 | 0.621 | 0.660 | 0.676 | 0.672 |
| QA model | NE | U-ME | FP | DSE | PE | NLL |
|---|---|---|---|---|---|---|
| AUROC | ||||||
| Gemma3-12B | 0.721 | 0.652 | 0.601 | 0.538 | 0.647 | 0.646 |
| OLMo2-13B-Instruct | 0.730 | 0.664 | 0.609 | 0.685 | 0.696 | 0.693 |
| Phi4-14B | 0.738 | 0.670 | 0.662 | 0.707 | 0.659 | 0.654 |
| Mistral-Nemo-12B | 0.712 | 0.650 | 0.589 | 0.670 | 0.704 | 0.705 |
| Falcon3-10B-Instruct | 0.726 | 0.668 | 0.644 | 0.660 | 0.674 | 0.672 |
| AUROC | AUPRC | |||
|---|---|---|---|---|
| NE versus | Pairs won | Mean | Pairs won | Mean |
| U-ME | 25/25 | 24/25 | ||
| FP | 25/25 | 25/25 | ||
| DSE | 21/25 | 19/25 | ||
| PE | 18/25 | 21/25 | ||
| NLL | 18/25 | 19/25 | ||
| Instances | Share | Failure rate | |||||||
|---|---|---|---|---|---|---|---|---|---|
| Method | Scored | Unsc. | |||||||
| NE | 23739 | 1819 | 2582 | 19338 | 18.5% | 1261 | 0.462 | 0.460 | 0.330 |
| U-ME | 24060 | 140 | 505 | 23415 | 2.7% | 940 | 0.679 | 0.677 | 0.341 |
| FP | 19569 | 1705 | 1915 | 15949 | 18.5% | 5431 | 0.407 | 0.405 | 0.293 |
| DSE | 21920 | 0 | 2582 | 19338 | 11.8% | 3080 | – | 0.460 | 0.330 |
| PE | 23739 | 1819 | 2582 | 19338 | 18.5% | 1261 | 0.462 | 0.460 | 0.330 |
| AUROC | AUPRC | |||||||
|---|---|---|---|---|---|---|---|---|
| Dataset | Scored | Full | Full | |||||
| SQuAD v1.1 | 4980 | 8.2% | 0.742 | 0.759 | 0.480 | 0.485 | ||
| Quoref | 4943 | 14.1% | 0.740 | 0.767 | 0.510 | 0.527 | ||
| NewsQA | 4438 | 37.7% | 0.647 | 0.712 | 0.656 | 0.682 | ||
| Natural Questions | 4385 | 30.7% | 0.638 | 0.709 | 0.432 | 0.468 | ||
| AdversarialQA | 4993 | 5.6% | 0.741 | 0.748 | 0.654 | 0.656 | ||
| Method | AUROC | AUPRC | Lift | Base | Cov. | sd | min – max | ||
|---|---|---|---|---|---|---|---|---|---|
| 1 | NE | 0.731 | 0.579 | 0.222 | 0.357 | 24133 | 0.965 | 0.038 | 0.653–0.784 |
| 2 | NLL | 0.674 | 0.491 | 0.130 | 0.361 | 24993 | 1.000 | 0.050 | 0.563–0.760 |
| 3 | PE | 0.673 | 0.486 | 0.129 | 0.357 | 24133 | 0.965 | 0.047 | 0.584–0.770 |
| 4 | DSE | 0.658 | 0.511 | 0.158 | 0.353 | 23224 | 0.929 | 0.066 | 0.527–0.753 |
| 5 | FP | 0.638 | 0.455 | 0.127 | 0.328 | 21584 | 0.863 | 0.031 | 0.564–0.685 |
| Method | AUROC | AUPRC | Lift | Base | Cov. | sd | min – max | ||
|---|---|---|---|---|---|---|---|---|---|
| 1 | NE | 0.744 | 0.588 | 0.235 | 0.353 | 23224 | 0.929 | 0.030 | 0.676–0.786 |
| 2 | NLL | 0.674 | 0.491 | 0.130 | 0.361 | 24993 | 1.000 | 0.050 | 0.563–0.760 |
| 3 | PE | 0.673 | 0.483 | 0.129 | 0.353 | 23224 | 0.929 | 0.047 | 0.584–0.768 |
| 4 | DSE | 0.658 | 0.511 | 0.158 | 0.353 | 23224 | 0.929 | 0.066 | 0.527–0.753 |
| 5 | FP | 0.649 | 0.460 | 0.137 | 0.323 | 20600 | 0.824 | 0.031 | 0.568–0.693 |
| Dataset | NE | FP | DSE | PE | NLL |
|---|---|---|---|---|---|
| AUROC | |||||
| SQuAD v1.1 | 0.750 | 0.649 | 0.673 | 0.670 | 0.666 |
| Quoref | 0.749 | 0.651 | 0.616 | 0.660 | 0.662 |
| NewsQA | 0.677 | 0.619 | 0.673 | 0.683 | 0.691 |
| Natural Questions | 0.713 | 0.623 | 0.666 | 0.728 | 0.730 |
| AdversarialQA | 0.766 | 0.647 | 0.660 | 0.625 | 0.620 |
| Dataset | NE | FP | DSE | PE | NLL |
|---|---|---|---|---|---|
| AUROC | |||||
| SQuAD v1.1 | 0.758 | 0.659 | 0.673 | 0.672 | 0.666 |
| Quoref | 0.760 | 0.658 | 0.616 | 0.659 | 0.662 |
| NewsQA | 0.709 | 0.643 | 0.673 | 0.684 | 0.691 |
| Natural Questions | 0.729 | 0.631 | 0.666 | 0.724 | 0.730 |
| AdversarialQA | 0.766 | 0.657 | 0.660 | 0.626 | 0.620 |
| QA model | NE | FP | DSE | PE | NLL |
|---|---|---|---|---|---|
| AUROC | |||||
| Gemma3-12B | 0.745 | 0.634 | 0.543 | 0.648 | 0.646 |
| OLMo2-13B-Instruct | 0.738 | 0.639 | 0.691 | 0.689 | 0.693 |
| Phi4-14B | 0.736 | 0.663 | 0.711 | 0.658 | 0.653 |
| Mistral-Nemo-12B | 0.708 | 0.607 | 0.678 | 0.704 | 0.705 |
| Falcon3-10B-Instruct | 0.727 | 0.645 | 0.664 | 0.668 | 0.671 |
| QA model | NE | FP | DSE | PE | NLL |
|---|---|---|---|---|---|
| AUROC | |||||
| Gemma3-12B | 0.757 | 0.643 | 0.543 | 0.647 | 0.646 |
| OLMo2-13B-Instruct | 0.753 | 0.653 | 0.691 | 0.690 | 0.693 |
| Phi4-14B | 0.750 | 0.674 | 0.711 | 0.657 | 0.653 |
| Mistral-Nemo-12B | 0.719 | 0.617 | 0.678 | 0.705 | 0.705 |
| Falcon3-10B-Instruct | 0.743 | 0.660 | 0.664 | 0.666 | 0.671 |
| AUROC | AUPRC | |||
|---|---|---|---|---|
| NE versus | Pairs won | Mean | Pairs won | Mean |
| FP | 25/25 | 25/25 | ||
| DSE | 19/25 | 22/25 | ||
| PE | 18/25 | 21/25 | ||
| NLL | 17/25 | 20/25 | ||
| AUROC | AUPRC | |||
|---|---|---|---|---|
| NE versus | Pairs won | Mean | Pairs won | Mean |
| FP | 25/25 | 25/25 | ||
| DSE | 23/25 | 22/25 | ||
| PE | 21/25 | 23/25 | ||
| NLL | 21/25 | 22/25 | ||
| Method | AUROC | AUPRC | Lift | Base | Cov. | sd | min – max | ||
|---|---|---|---|---|---|---|---|---|---|
| 1 | NE | 0.724 | 0.568 | 0.211 | 0.356 | 24475 | 0.979 | 0.045 | 0.634–0.790 |
| 2 | PE | 0.674 | 0.486 | 0.129 | 0.356 | 24475 | 0.979 | 0.046 | 0.582–0.762 |
| 3 | NLL | 0.674 | 0.491 | 0.130 | 0.361 | 25000 | 1.000 | 0.049 | 0.563–0.762 |
| 4 | DSE | 0.662 | 0.512 | 0.162 | 0.350 | 23865 | 0.955 | 0.068 | 0.523–0.743 |
| Dataset | NE | DSE | PE | NLL |
|---|---|---|---|---|
| AUROC | ||||
| SQuAD v1.1 | 0.743 | 0.681 | 0.669 | 0.667 |
| Quoref | 0.754 | 0.616 | 0.660 | 0.661 |
| NewsQA | 0.660 | 0.677 | 0.683 | 0.691 |
| Natural Questions | 0.701 | 0.667 | 0.727 | 0.730 |
| AdversarialQA | 0.765 | 0.668 | 0.630 | 0.620 |
| QA model | NE | DSE | PE | NLL |
|---|---|---|---|---|
| AUROC | ||||
| Gemma3-12B | 0.727 | 0.544 | 0.644 | 0.646 |
| OLMo2-13B-Instruct | 0.723 | 0.698 | 0.689 | 0.693 |
| Phi4-14B | 0.733 | 0.718 | 0.654 | 0.654 |
| Mistral-Nemo-12B | 0.716 | 0.682 | 0.710 | 0.705 |
| Falcon3-10B-Instruct | 0.723 | 0.666 | 0.673 | 0.670 |
| AUROC | AUPRC | |||
|---|---|---|---|---|
| NE versus | Pairs won | Mean | Pairs won | Mean |
| DSE | 19/25 | 20/25 | ||
| PE | 16/25 | 21/25 | ||
| NLL | 15/25 | 18/25 | ||
| AUROC | Coverage | |||||
| Method | 8B | 14B | 32B | 8B | 14B | 32B |
| NE | 0.702 | 0.731 | 0.724 | 0.950 | 0.965 | 0.979 |
| PE | 0.676 | 0.673 | 0.674 | 0.950 | 0.965 | 0.979 |
| NLL | 0.674 | 0.674 | 0.674 | 0.999 | 1.000 | 1.000 |
| DSE | 0.652 | 0.658 | 0.662 | 0.877 | 0.929 | 0.955 |
| No 32B run | ||||||
| Model | SQuAD v1.1 | Quoref | NewsQA | Natural Questions | AdversarialQA | Mean |
|---|---|---|---|---|---|---|
| Accuracy | ||||||
| Gemma3-12B | ||||||
| OLMo2-13B-Instruct | ||||||
| Phi4-14B | ||||||
| Mistral-Nemo-12B | ||||||
| Falcon3-10B-Instruct | ||||||
| FP | SBS | U-ME | NE | |
| How candidates are generated | ||||
| Archive over | — | yes | yes | yes |
| Mutates an archived parent | — | — | yes | yes |
| Parent binds the offspring | — | — | no | yes |
| Targets empty cells | — | no | no | yes |
| Edit types requested | style only | all three | all three | all three |
| Rewrite type | Prompt strategies |
|---|---|
| distractor | distractor_support_proximal |
| distractor_question_overlap | |
| reorder | reorder_adjacent_swap |
| reorder_support_back | |
| reorder_global | |
| style_rewrite | support_direct_consolidation |
| Rewrite type | Prompt intention | Deterministic realization condition |
|---|---|---|
| distractor | Add plausible but answer-irrelevant material while preserving the evidence. | The child differs from the reference, adds at least one sentence, and retains every reference sentence as an exact ordered subsequence . |
| reorder | Change sentence order without adding, deleting, or paraphrasing sentences. | The reference has at least two sentences; the child differs from it and has the same normalized sentence multiset in a different order. |
| style_rewrite | Rewrite substantive wording or syntax. | The child’s token sequence changes, and the child is not merely a sentence permutation and not merely a sentence insertion. |
| Setting | Value |
|---|---|
| Random seed | |
| Editor (Appendix C.3 ) | |
| Model | Qwen3-8B (local MLX, 8-bit) |
| Decoding parameters | temperature ; sampling on; max new tokens |
| Support-explicitness rule (Appendix C.2.2 ) | |
| Sentence-split / token patterns | [.!?]+ / [A-Za-z0-9]+(’[A-Za-z0-9]+)? |
| Dataset | Context (mean/max) | Question (mean) | Answer (mean) | Multi-gold |
|---|---|---|---|---|
| SQuAD v1.1 | ||||
| Quoref | ||||
| NewsQA | ||||
| NaturalQuestions (MRQA) | ||||
| AdversarialQA |
Uncertainty Quantification
Uncertainty quantification estimates how reliable predictions and scientific inferences are, and how uncertainty propagates into decisions.
See all 663 papers