通过稀疏激活控制,让大模型同时提升安全、真实与无偏性。
Enhancing Multiple Dimensions of Trustworthiness in LLMs via Sparse Activation Control
- 识别模型中负责特定任务的注意力头,实现独立调控。
- 在Llama系列模型上同步达成安全、事实性与去偏三重目标。
- 无需训练即可调整模型行为,适合需要多维度可信增强的场景。
随着大语言模型(LLMs)的发展与应用不断推进,提升其可信度并使其符合人类偏好已成为关键研究方向。传统方法依赖大量人类反馈强化学习(RLHF)数据,而表示工程提供了一种无需训练的新路径:通过语义特征调控模型中间隐藏状态的表征,使模型满足如提高诚实性或增强安全性等特定需求。然而,同时满足多个需求时面临挑战——难以将诚实、安全等不同语义内容编码至单一语义特征中,限制了实际应用。本文提出「稀疏激活控制」方法,深入分析大模型内在机制,识别出与特定任务密切相关的组件(即注意力头)。这些注意力头具有稀疏特性,可实现对不同任务的近乎独立控制。在开源Llama系列模型上的实验表明,该方法能有效协同提升模型在安全性、事实性和偏见方面的表现。
原文摘要 · Abstract (English)
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.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。