Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

Adaptive Activation Steering: A Tuning-Free LLM Truthfulness Improvement Method for Diverse Hallucinations Categories

About

Recent studies have indicated that Large Language Models (LLMs) harbor an inherent understanding of truthfulness, yet often fail to consistently express it and generate false statements. This gap between "knowing" and "telling" poses a challenge for ensuring the truthfulness of generated content. Inspired by recent work on the practice of encoding human-interpretable concepts linearly within large language models, we treat truthfulness as a specially linearly encoded concept within LLMs, and introduce Adaptive Activation Steering (ACT), a tuning-free method that adaptively shifts LLM's activations in the "truthful" direction during inference. ACT addresses diverse categories of hallucinations by utilizing diverse truthfulness-related steering vectors and adjusting the steering intensity adaptively. Applied as an add-on across various models, ACT significantly improves truthfulness in LLaMA ($\uparrow$ 142%), LLaMA2 ($\uparrow$ 24%), Alpaca ($\uparrow$ 36%), Vicuna ($\uparrow$ 28%), LLaMA2-Chat ($\uparrow$ 19%), and LLaMA3($\uparrow$ 34%). Furthermore, we verify ACT's scalability across larger models (13B, 33B, 65B), underscoring the adaptability of ACT to large-scale language models. Our code is available at https://github.com/tianlwang/ACT.

Tianlong Wang, Xianfeng Jiao, Yinghao Zhu, Zhongzhi Chen, Yifan He, Xu Chu, Junyi Gao, Yasha Wang, Liantao Ma• 2024

Related benchmarks

TaskDatasetResultRank
Commonsense ReasoningHellaSwag
Accuracy70.1
1891
Mathematical ReasoningGSM8K
Accuracy87.9
1362
Massive Multitask Language UnderstandingMMLU
Accuracy59.4
117
Factuality EvaluationTruthfulQA
MC293.7
73
Safety RefusalAdvBench
Refusal Rate83.9
46
AI Text DetectionHC3 AI-Prob
Finance Accuracy56.3
24
Sentiment AnalysisSST-2
Positive Rate48.6
24
Safety RefusalJailbreak Prompts
Refusal Rate78.5
15
Safety RefusalToxicChat
Refusal Rate59.8
15
Showing 9 of 9 rows

Other info

Follow for update