arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2411.02461cs.CLcs.AI
收藏

增强大语言模型可信度的多维度:基于稀疏激活控制

Enhancing Multiple Dimensions of Trustworthiness in LLMs via Sparse Activation Control

State Key Lab of CAD&CG, Zhejiang University(浙江大学CAD&CG国家重点实验室);Alibaba Cloud(阿里云);Hong Kong Baptist University(香港浸会大学);School of Software Technology, Zhejiang University(浙江大学软件学院);Zhiyuan Research Institute(智源研究院);Fabu Inc.(fabu公司)

Yuxin Xiao, Chaoqun Wan, Yonggang Zhang, Wenxiao Wang, Binbin Lin, Xiaofei He, Xu Shen, Jieping Ye

AI总结:

本文提出稀疏激活控制方法,通过识别注意力头实现无需训练的多维度可信度控制,在Llama系列模型上同时提升安全性、事实性和减少偏见。

摘要:

随着大语言模型(LLMs)的开发与应用持续快速发展,增强其可信度并使其与人类偏好对齐已成为一个关键的研究领域。传统方法严重依赖大量数据来进行基于人类反馈的强化学习(RLHF),而表示工程提供了一种新的、无需训练的方法。该技术利用语义特征来控制LLM中间隐藏状态的表示,使模型能够满足特定要求,如提高诚实度或增强安全意识。然而,在尝试同时满足多个要求时出现了一个重大挑战:将诚实和安全等不同语义内容编码到单一语义特征中较为困难,这限制了其实际应用性。在本工作中,我们通过“稀疏激活控制”来解决这一问题。通过深入探究LLM的内在机制,我们成功识别并定位了模型内与特定任务密切相关的组件,即注意力头。这些注意力头展现出稀疏特性,使得对不同任务的近独立控制成为可能。我们在开源Llama系列模型上进行的实验取得了令人鼓舞的结果。这些模型能够在安全性、事实性和偏见问题上同时与人类偏好对齐。

英文摘要:

As the development and application of Large Language Models (LLMs) continue to advance rapidly, enhancing their trustworthiness and aligning them with human preferences has become a critical area of research. Traditional methods rely heavily on extensive data for Reinforcement Learning from Human Feedback (RLHF), but representation engineering offers a new, training-free approach. This technique leverages semantic features to control the representation of LLM's intermediate hidden states, enabling the model to meet specific requirements such as increased honesty or heightened safety awareness. However, a significant challenge arises when attempting to fulfill multiple requirements simultaneously. It proves difficult to encode various semantic contents, like honesty and safety, into a singular semantic feature, restricting its practicality. In this work, we address this issue through ``Sparse Activation Control''. By delving into the intrinsic mechanisms of LLMs, we manage to identify and pinpoint components that are closely related to specific tasks within the model, i.e., attention heads. These heads display sparse characteristics that allow for near-independent control over different tasks. Our experiments, conducted on the open-source Llama series models, have yielded encouraging results. The models were able to align with human preferences on issues of safety, factuality, and bias concurrently.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑