Refusal Lives Downstream of Persona in Chat Models

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning · Depth: Expert, quick

Summary

A study on instruction-tuned chat models, Qwen2.5-7B-Instruct and Llama-3.1-8B-Instruct, reveals that refusal behavior is significantly influenced by the model's persona. Researchers identified linear directions in activation space for both refusal and persona traits, demonstrating their interaction rather than isolation. By extracting and intervening on a compliant model-persona direction and a refusal direction, they found that compliant persona steering effectively suppresses refusal. For instance, in Llama-3.1-8B-Instruct, the refusal rate decreased from 97% to 2%. Reintroducing the refusal direction partially restored refusal at late layers but not at early ones, indicating that refusal is gated at the late-layer expression stage, downstream of where it is computed. This research highlights that treating refusal as a single, isolated direction overlooks its critical dependence on the model's persona.

Key takeaway

For Machine Learning Engineers developing instruction-tuned chat models, understanding the persona-refusal interaction is crucial for safety and alignment. You should consider persona steering as a primary mechanism to control refusal rates, rather than solely focusing on isolated refusal directions. This insight suggests that interventions targeting late-layer expression stages, influenced by persona, will be more effective in managing model compliance and preventing unwanted responses.

Key insights

Compliant model persona directly gates refusal behavior in instruction-tuned chat models at late-layer expression stages.

Principles

Method

Researchers extracted compliant model-persona and refusal directions in activation space. They intervened on both directions to observe their interaction and impact on refusal rates in Qwen2.5-7B-Instruct and Llama-3.1-8B-Instruct.

In practice

Topics

Best for: Research Scientist, AI Engineer, NLP Engineer, AI Scientist, Machine Learning Engineer

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.