Noise augmented fine tuning for mitigating hallucinations in large language models

Read full text
Authors: Afshin Khadangi, Amir Sartipi, Igor Tchappi, Ramin Bahmani
Cited by: 0

Abstract

Large language models (LLMs) often produce inaccurate or misleading content— hallucinations . To address this challenge, we introduce Noise-Augmented Fine-Tuning (NoiseFiT) , a novel framework that leverages adaptive noise injection based on the signal-to-noise ratio (SNR) to enhance model robustness. Our contribution is threefold. First, NoiseFiT selectively perturbs layers identified as either high-SNR (more robust) or low-SNR (potentially under-regularized) using a dynamically scaled Gaussian noise. Second, we further propose a hybrid loss that combines standard cross-entropy, soft cross-entropy, and consistency regularization to ensure stable and accurate outputs under noisy training conditions. Third, a theoretical analysis proposed shows that adaptive noise injection is both unbiased and variance-preserving , providing strong guarantees for convergence in expectation. Moreover, empirical results on multiple test and benchmark datasets, demonstrate that NoiseFiT significantly reduces hallucination rates, often improving or matching baseline performance in key tasks. These findings highlight the promise of noise-driven strategies for achieving robust, trustworthy language modeling without incurring prohibitive computational overhead. We have publicly released the fine-tuning logs, benchmark evaluation artifacts, and source code online at W&B , Hugging Face , and GitHub , respectively, to foster further research, accessibility and reproducibility.

Fields of Study & Subjects

Machine Learning in Healthcare Anomaly Detection Techniques and Applications

Publication details

Journal / source
Neurocomputing
Volume
703
Pages
134636–134636
Publication date
30 July 2026
Language
EN