paper-with-me

Papers

Interpreting Learned Feedback Patterns in Large Language Models

2023-10-12 · Luke Marks, Amir Abdullah, Clement Neo, Rauno Arike, David Krueger, Philip Torr, Fazl Barez

Reinforcement learning from human feedback (RLHF) is widely used to train large language models (LLMs). However, it is unclear whether LLMs accurately learn the underlying preferences in human feedback data. We coin the term \textit{Learned Feedback Pattern} (LFP) for patterns in an LLM's activations learned during RLHF that improve its performance on the fine-tuning task. We hypothesize that LLMs with LFPs accurately aligned to the fine-tuning feedback exhibit consistent activation patterns for outputs that would have received similar feedback during RLHF. To test this, we train probes to estimate the feedback signal implicit in the activations of a fine-tuned LLM. We then compare these estimates to the true feedback, measuring how accurate the LFPs are to the fine-tuning feedback. Our probes are trained on a condensed, sparse and interpretable representation of LLM activations, making it easier to correlate features of the input with our probe's predictions. We validate our probes by comparing the neural features they correlate with positive feedback inputs against the features GPT-4 describes and classifies as related to LFPs. Understanding LFPs can help minimize discrepancies between LLM behavior and training objectives, which is essential for the safety of LLMs.

📄 PDF Abstract BibTeX arXiv:2310.08164

Code (1)

apartresearch/interpreting-reward-models 공식 구현 pytorch

Methods 이 논문이 사용한 방법론

Attention 설명 없음
Linear Layer A Linear Layer is a projection $\mathbf{XW + b}$.
Residual Connection 설명 없음
Layer Normalization Unlike batch normalization, Layer Normalization directly estimates the normalization statistics from the summed inputs…
Multi-Head Attention 설명 없음
Position-Wise Feed-Forward Layer 설명 없음
Adam 설명 없음
BPE Byte Pair Encoding, or BPE, is a subword segmentation algorithm that encodes rare and unknown words as sequences of subword units. The intuition is that various word…

Similar Papers 제목 키워드 기반

KnowledgeVIS: Interpreting Language Models by Comparing Fill-in-the-Blank Prompts

2024-03-07 · Adam Coscia, Alex Endert

Recent growth in the popularity of large language models has led to their increased usage for summarizing, predicting, and generating text, making it vital to help researchers and engineers understand how and why they wo…

Interpreting Training Aspects of Deep-Learned Error-Correcting Codes

2023-05-07 · N. Devroye, A. Mulgund, R. Shekhar, Gy. Turán 외

As new deep-learned error-correcting codes continue to be introduced, it is important to develop tools to interpret the designed codes and understand the training process. Prior work focusing on the deep-learned TurboAE …

Decoder

From Black Box to Bijection: Interpreting Machine Learning to Build a Zeta Map Algorithm

2025-11-16 · Xiaoyu Huang, Blake Jackson, Kyu-Hwan Lee arxiv

There is a large class of problems in algebraic combinatorics which can be distilled into the same challenge: construct an explicit combinatorial bijection. Traditionally, researchers have solved challenges like these by…

Interpreting and Steering LLMs with Mutual Information-based Explanations on Sparse Autoencoders

2025-02-21 · Xuansheng Wu, Jiayi Yuan, Wenlin Yao, Xiaoming Zhai 외

Large language models (LLMs) excel at handling human queries, but they can occasionally generate flawed or unexpected responses. Understanding their internal states is crucial for understanding their successes, diagnosin…

Prototypical Reward Network for Data-Efficient RLHF

2024-06-06 · Jinghan Zhang, Xiting Wang, Yiqiao Jin, Changyu Chen 외

The reward model for Reinforcement Learning from Human Feedback (RLHF) has proven effective in fine-tuning Large Language Models (LLMs). Notably, collecting human feedback for RLHF can be resource-intensive and lead to s…