mlsanityData to 14 Oct 2026

Energy Landscapes Enable Reliable Abstention in Retrieval-Augmented Large Language Models for Healthcare

arXiv 2509.04482 ↗First posted 31 Aug 2025cs.CL cs.AIcited by 3

On the map: Retrieval-augmented generation → Medical and biomedical RAG

Abstract (arXiv)

Reliable abstention is critical for retrieval-augmented generation (RAG) systems, particularly in safety-critical domains such as women's health, where incorrect answers can lead to harm. We present an energy-based model (EBM) that learns a smooth energy landscape over a dense semantic corpus of 2.6M guideline-derived questions, enabling the system to decide when to generate or abstain. We benchmark the EBM against a calibrated softmax baseline and a k-nearest neighbour (kNN) density heuristic across both easy and hard abstention splits, where hard cases are semantically challenging near-distribution queries. The EBM achieves superior abstention performance abstention on semantically hard cases, reaching AUROC 0.961 versus 0.950 for softmax, while also reducing FPR@95 (0.235 vs 0.331). On easy negatives, performance is comparable across methods, but the EBM's advantage becomes most pronounced in safety-critical hard distributions. A comprehensive ablation with controlled negative sampling and fair data exposure shows that robustness stems primarily from the energy scoring head, while the inclusion or exclusion of specific negative types (hard, easy, mixed) sharpens decision boundaries but is not essential for generalisation to hard cases. These results demonstrate that energy-based abstention scoring offers a more reliable confidence signal than probability-based softmax confidence, providing a scalable and interpretable foundation for safe RAG systems.

Key facts. Cited by 3 papers in the map (data through 14 Oct 2026). Placed in Medical and biomedical RAG, one of 15 directions in Retrieval-augmented generation. Builds on 26 of its 35 references that are in the map. First posted on arXiv 31 Aug 2025.

Builds on

26 of 35 references.

References · 26

  1. 2020Retrieval-Augmented Generation for Knowledge-Intensive NLP Taskscited by 9,297
  2. 2017On Calibration of Modern Neural Networkscited by 4,960
  3. 2016SQuAD: 100,000+ Questions for Machine Comprehension of Textcited by 4,148
  4. 2020Dense Passage Retrieval for Open-Domain Question Answeringcited by 3,824
  5. 2015FaceNet: A Unified Embedding for Face Recognition and Clusteringcited by 2,849
  6. 2016A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networkscited by 2,385
  7. 2022Language Models (Mostly) Know What They Knowcited by 1,856
  8. 2023Self-RAG: Learning to Retrieve, Generate, and Critique through Self-Reflectioncited by 1,839
  9. 2018Know What You Don't Know: Unanswerable Questions for SQuADcited by 1,554
  10. 2018A Simple Unified Framework for Detecting Out-of-Distribution Samples and Adversarial Attackscited by 1,485
  11. 2019MixMatch: A Holistic Approach to Semi-Supervised Learningcited by 1,391
  12. 2017Enhancing The Reliability of Out-of-distribution Image Detection in Neural Networkscited by 1,328
14 more
  1. 2020Energy-based Out-of-distribution Detectioncited by 1,299
  2. 2017Selective Classification for Deep Neural Networkscited by 852
  3. 2020Approximate Nearest Neighbor Negative Contrastive Learning for Dense Text Retrievalcited by 849
  4. 2021Generalized Out-of-Distribution Detection: A Surveycited by 802
  5. 2021A Gentle Introduction to Conformal Prediction and Distribution-Free Uncertainty Quantificationcited by 792
  6. 2017In Defense of the Triplet Loss for Person Re-Identificationcited by 786
  7. 2023Capabilities of GPT-4 on Medical Challenge Problemscited by 638
  8. 2022Out-of-Distribution Detection with Deep Nearest Neighborscited by 600
  9. 2022MedMCQA : A Large-scale Multi-Subject Multi-Choice Dataset for Medical domain Question Answeringcited by 543
  10. 2019Your Classifier is Secretly an Energy Based Model and You Should Treat it Like Onecited by 405
  11. 2019SelectiveNet: A Deep Neural Network with an Integrated Reject Optioncited by 350
  12. 2017Sampling Matters in Deep Embedding Learningcited by 285
  13. 2020Selective Question Answering under Domain Shiftcited by 238
  14. 2025Reducing Large Language Model Safety Risks in Women's Health using Semantic Entropycited by 8

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2026: 3.

3 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Healthier LLMs: Retrieval-Augmented Generation for Public Health Question Answering
  2. 2025Retrieval-Augmented Generation: A Comprehensive Survey of Architectures, Enhancements, and Robustness Frontiers
  3. 2026AB-RAG: Adaptive Budgeted Retrieval-Augmented Generation for Reliable Question Answering
  4. 2025On the Influence of Context Size and Model Choice in Retrieval-Augmented Generation Systems
  5. 2025Evaluating the Robustness of Retrieval-Augmented Generation to Adversarial Evidence in the Health Domain
  6. 2024W-RAG: Weakly Supervised Dense Retrieval in RAG for Open-domain Question Answering
  7. 2025Investigating the Robustness of Retrieval-Augmented Generation at the Query Level
  8. 2024Rationale-Guided Retrieval Augmented Generation for Medical Question Answering