arXiv Machine Learning

Do Emergent Misalignment Models Share a Circuit?

The paper investigates whether language models fine‑tuned on different harmful tasks share common computational components. Using direction‑based activation tracing on three LoRA fine‑tunes of Qwen2.5‑3B‑Instruct, the authors identify three attention‑head locations that are shared among the top five heads of all models. Replacing these shared heads with base contributions significantly reduces emergent misalignment across medical, financial, and sports advice tasks, and cross‑organism activation transfer shows that the shared heads have partially interchangeable effects on fixed‑response preference.

arXiv AI
Sep 21

Decoupling Internal Representational Changes and Causal Importance in Fine-Tuned Large Language Models

Fine‑tuning reshapes internal representations of large language models, affecting attention patterns and layer‑wise activations. The study shows that components identified by EAP as important for task performance cluster in specific layers, yet these layers do not align with those undergoing the largest representational changes. Additionally, overlapping EAP components across different tasks do not guarantee cross‑task transfer and can even degrade performance when tasks differ in nature.

By Lingfang Li, Procheta Sen, Shubham Das, Danushka Bollegala
arXiv AI
Sep 16

TAME: Token Attribution and Masking for Emergent misalignment

TAME (Token Attribution and Masking for Emergent misalignment) is a three‑stage framework that identifies which training tokens drive harmful behavior in fine‑tuned language models. It first scores tokens by how much fine‑tuning increases their likelihood, then characterizes patterns among high‑attribution tokens, and finally validates them by masking during training. Experiments on Llama and Qwen show that masking the top‑attribution tokens reduces emergent misalignment by 23‑ to 36‑fold, while random masking has no effect.

By Md Rayhanul Masud, Md Rizwan Parvez
arXiv AI
Sep 30

Which Attention Heads are like the Human Head? Not the Ones that Compute

The study investigates whether attention heads in large language models that align with human EEG signals are causally involved in model computation. By ablating these brain‑aligned heads during a pattern‑completion task, the authors find that while such heads contribute to performance, their removal is less disruptive than removing heads selected by attribution patching. The research also distinguishes two families of brain‑aligned heads—novelty and repetition heads—highlighting that novelty heads track human attention but are less critical than random ablation, whereas repetition heads modestly aid performance and align with abstract‑pattern representations.

By Christopher Pinier, Gustaw Opie{\l}ka, Hannes Rosenbusch, Taylor Webb, Michael D. Nunez, Claire E. Stevenson
arXiv AI
Oct 19

Many Biases, One Circuit: Post-Training Tunes a Pretrained Cue-Routing Mechanism

The paper investigates how language models can be misled by cue words that hint at incorrect answers, a phenomenon linked to sycophancy and other biases. By tracking attention heads across two post‑training pipelines, the authors find that a small shared circuit of heads underlies multiple cue types and that fine‑tuning and reinforcement learning amplify cue‑following. Swapping attention patterns between checkpoints demonstrates that post‑training changes the timing and strength of this circuit’s influence, suggesting a targeted mitigation strategy.

By Prakhar Gupta, Terry Jingchen Zhang, Zhijing Jin
arXiv Machine Learning
Sep 21

Gradient-Stable Attention Heads Signal LLM Correctness

The paper introduces HeadEntropy, a training‑free method that predicts the correctness of large language model (LLM) answers by measuring how stable each attention head’s pattern is to further gradient updates. By linking the trace of the softmax Jacobian to 2‑Renyi entropy, the authors show that attention spread correlates with gradient stability, enabling accurate hallucination detection without reference annotations. Across five instruction‑tuned LLMs and five diverse datasets—including medicine, multi‑hop reasoning, and mathematics—HeadEntropy achieves a 0.736 AUROC, outperforming other training‑free baselines and matching hidden‑state probes while incurring less than 1% of inference cost.

By Sophie Ostmeier, Brian Axelrod, Maya Varma, Asad Aali, Yabin Zhang, Magdalini Paschali, Sanmi Koyejo, Curtis Langlotz, Akshay Chaudhari