Hallucination reduction with CASAL: Contrastive Activation Steering For Amortized Learning

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Wannan, Yang, Qiu, Xinchi, Yu, Lei, Zhang, Yuchen, Yang, Aobo, Kokhlikyan, Narine, Cancedda, Nicola, Garcia-Olano, Diego
Format: Preprint
Publié: 2025
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866912752911515648
author Wannan
Yang
Qiu, Xinchi
Yu, Lei
Zhang, Yuchen
Yang, Aobo
Kokhlikyan, Narine
Cancedda, Nicola
Garcia-Olano, Diego
author_facet Wannan
Yang
Qiu, Xinchi
Yu, Lei
Zhang, Yuchen
Yang, Aobo
Kokhlikyan, Narine
Cancedda, Nicola
Garcia-Olano, Diego
contents Large Language Models (LLMs) exhibit impressive capabilities but often hallucinate, confidently providing incorrect answers instead of admitting ignorance. Prior work has shown that models encode linear representations of their own knowledge and that activation steering can reduce hallucinations. These approaches, however, require real-time monitoring and intervention during inference. We introduce Contrastive Activation Steering for Amortized Learning (CASAL), an efficient algorithm that connects interpretability with amortized optimization. CASAL directly bakes the benefits of activation steering into model's weights. Once trained, LLMs answer questions they know while abstaining from answering those they do not. CASAL's light-weight design requires training only a submodule of a single transformer layer and yet reduces hallucination by 30%-40% across multiple short-form QA benchmarks. CASAL is 30x more compute-efficient and 20x more data-efficient than strong LoRA-based baselines such as SFT and DPO, boosting its practical applicability in data scarce domains. Importantly, CASAL also generalizes effectively to out-of-distribution (OOD) domains. We showcase CASAL's flexibility in mitigating hallucinations in both text-only and vision-language models. To our knowledge, CASAL is the first steering-based training method that has been shown to be effective for both dense and Mixture-of-Experts (MoE) models. CASAL represents a promising step forward for applying interpretability-inspired method for practical deployment in production systems.
format Preprint
id arxiv_https___arxiv_org_abs_2510_02324
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Hallucination reduction with CASAL: Contrastive Activation Steering For Amortized Learning
Wannan
Yang
Qiu, Xinchi
Yu, Lei
Zhang, Yuchen
Yang, Aobo
Kokhlikyan, Narine
Cancedda, Nicola
Garcia-Olano, Diego
Computation and Language
Artificial Intelligence
Large Language Models (LLMs) exhibit impressive capabilities but often hallucinate, confidently providing incorrect answers instead of admitting ignorance. Prior work has shown that models encode linear representations of their own knowledge and that activation steering can reduce hallucinations. These approaches, however, require real-time monitoring and intervention during inference. We introduce Contrastive Activation Steering for Amortized Learning (CASAL), an efficient algorithm that connects interpretability with amortized optimization. CASAL directly bakes the benefits of activation steering into model's weights. Once trained, LLMs answer questions they know while abstaining from answering those they do not. CASAL's light-weight design requires training only a submodule of a single transformer layer and yet reduces hallucination by 30%-40% across multiple short-form QA benchmarks. CASAL is 30x more compute-efficient and 20x more data-efficient than strong LoRA-based baselines such as SFT and DPO, boosting its practical applicability in data scarce domains. Importantly, CASAL also generalizes effectively to out-of-distribution (OOD) domains. We showcase CASAL's flexibility in mitigating hallucinations in both text-only and vision-language models. To our knowledge, CASAL is the first steering-based training method that has been shown to be effective for both dense and Mixture-of-Experts (MoE) models. CASAL represents a promising step forward for applying interpretability-inspired method for practical deployment in production systems.
title Hallucination reduction with CASAL: Contrastive Activation Steering For Amortized Learning
topic Computation and Language
Artificial Intelligence
url https://arxiv.org/abs/2510.02324