Table of Contents
Fetching ...

A Timeline and Analysis for Representation Plasticity in Large Language Models

Akshat Kannan

TL;DR

The findings are pivotal, showing that while early steering exhibits high plasticity, later stages have a surprisingly responsive critical window, signaling that there is a general pattern of model plasticity that can be used for effective intervention.

Abstract

The ability to steer AI behavior is crucial to preventing its long term dangerous and catastrophic potential. Representation Engineering (RepE) has emerged as a novel, powerful method to steer internal model behaviors, such as "honesty", at a top-down level. Understanding the steering of representations should thus be placed at the forefront of alignment initiatives. Unfortunately, current efforts to understand plasticity at this level are highly neglected. This paper aims to bridge the knowledge gap and understand how LLM representation stability, specifically for the concept of "honesty", and model plasticity evolve by applying steering vectors extracted at different fine-tuning stages, revealing differing magnitudes of shifts in model behavior. The findings are pivotal, showing that while early steering exhibits high plasticity, later stages have a surprisingly responsive critical window. This pattern is observed across different model architectures, signaling that there is a general pattern of model plasticity that can be used for effective intervention. These insights greatly contribute to the field of AI transparency, addressing a pressing lack of efficiency limiting our ability to effectively steer model behavior.

A Timeline and Analysis for Representation Plasticity in Large Language Models

TL;DR

The findings are pivotal, showing that while early steering exhibits high plasticity, later stages have a surprisingly responsive critical window, signaling that there is a general pattern of model plasticity that can be used for effective intervention.

Abstract

The ability to steer AI behavior is crucial to preventing its long term dangerous and catastrophic potential. Representation Engineering (RepE) has emerged as a novel, powerful method to steer internal model behaviors, such as "honesty", at a top-down level. Understanding the steering of representations should thus be placed at the forefront of alignment initiatives. Unfortunately, current efforts to understand plasticity at this level are highly neglected. This paper aims to bridge the knowledge gap and understand how LLM representation stability, specifically for the concept of "honesty", and model plasticity evolve by applying steering vectors extracted at different fine-tuning stages, revealing differing magnitudes of shifts in model behavior. The findings are pivotal, showing that while early steering exhibits high plasticity, later stages have a surprisingly responsive critical window. This pattern is observed across different model architectures, signaling that there is a general pattern of model plasticity that can be used for effective intervention. These insights greatly contribute to the field of AI transparency, addressing a pressing lack of efficiency limiting our ability to effectively steer model behavior.
Paper Structure (18 sections, 3 equations, 17 figures, 2 tables)

This paper contains 18 sections, 3 equations, 17 figures, 2 tables.

Figures (17)

  • Figure 1: Concept Vector Extraction wehner_2024_an
  • Figure 2: Graph of Evaluation Results for GPT-2 Medium with 95% Confidence Interval Error Bars
  • Figure 3: Graph of Evaluation Results for GPT-2 Small with 95% Confidence Interval Error Bars
  • Figure 4: Histogram w/ KDE of GPT-2 Small Average Evaluation Results
  • Figure 5: Q-Q Plot of GPT-2 Small Average Evaluation Results
  • ...and 12 more figures