用稀疏自编码器让奖励模型可解释,直接看懂为何给分。
Interpretable Reward Model via Sparse Autoencoder
- 将大模型隐层激活映射到可解释的稀疏特征空间
- 实现特征级打分归因,且能动态适应偏好变化
- 适合关注模型对齐透明度的研究者与工程师
大型语言模型(LLMs)已广泛应用于多个领域。基于人类反馈的强化学习(RLHF)利用奖励模型(RMs)作为人类偏好的代理,以对齐模型行为与人类价值观,因此RMs的准确性、可靠性与可解释性对有效对齐至关重要。然而,传统RMs缺乏可解释性,难以揭示评分背后的推理逻辑,且对用户偏好变化不敏感。尽管近期多维RMs试图提升可解释性,但仍无法提供特征级归因,且需高成本标注。为此,我们提出稀疏自编码器增强型奖励模型(SARM),将预训练稀疏自编码器(SAE)融入奖励模型架构。SARM将基于LLM的RM隐层激活映射至可解释、稀疏且单义的特征空间,再通过标量头聚合特征激活,生成透明且概念清晰的奖励分数。实证评估表明,SARM实现了直接的特征级归因,支持偏好动态调整,并在对齐性能上优于传统奖励模型。代码已开源:https://github.com/schrieffer-z/sarm。
原文摘要 · Abstract (English)
Large language models (LLMs) have been widely deployed across numerous fields. Reinforcement Learning from Human Feedback (RLHF) leverages reward models (RMs) as proxies for human preferences to align LLM behaviors with human values, making the accuracy, reliability, and interpretability of RMs critical for effective alignment. However, traditional RMs lack interpretability, offer limited insight into the reasoning behind reward assignments, and are inflexible toward user preference shifts. While recent multidimensional RMs aim for improved interpretability, they often fail to provide feature-level attribution and require costly annotations. To overcome these limitations, we introduce the Sparse Autoencoder-enhanced Reward Model (SARM), a novel architecture that integrates a pretrained Sparse Autoencoder (SAE) into a reward model. SARM maps the hidden activations of LLM-based RM into an interpretable, sparse, and monosemantic feature space, from which a scalar head aggregates feature activations to produce transparent and conceptually meaningful reward scores. Empirical evaluations demonstrate that SARM facilitates direct feature-level attribution of reward assignments, allows dynamic adjustment to preference shifts, and achieves superior alignment performance compared to conventional reward models. Our code is available at https://github.com/schrieffer-z/sarm.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。