Saved in:
Bibliographic Details
Main Authors: Mohammadzadeh, Shahrad, Guerra, Juan David, Bonizzato, Marco, Rabbany, Reihaneh, Farnadi, Golnoosh
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2410.15460
Tags: Add Tag
No Tags, Be the first to tag this record!
Table of Contents:
  • As large language models (LLMs) become increasingly prevalent, concerns about their reliability, particularly due to hallucinations - factually inaccurate or irrelevant outputs - have grown. Our research investigates the relationship between the uncertainty in training dynamics and the emergence of hallucinations. Using models from the Pythia suite and several hallucination detection metrics, we analyze hallucination trends and identify significant variance during training. To address this, we propose Sensitivity Dropout (SenD), a novel training protocol designed to reduce hallucination variance during training by deterministically dropping embedding indices with significant variability. In addition, we develop an unsupervised hallucination detection metric, Efficient EigenScore (EES), which approximates the traditional EigenScore in 2x speed. This metric is integrated into our training protocol, allowing SenD to be both computationally scalable and effective at reducing hallucination variance. SenD improves test-time reliability of Pythia and Meta's Llama models by up to 17% and enhances factual accuracy in Wikipedia, Medical, Legal, and Coding domains without affecting downstream task performance.