Residual Stream Analysis of Overfitting And Structural Disruptions

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Liu, Quan, Zhou, Han, Wu, Wenquan, Wu, Hua, Su, Sen
Natura: Preprint
Pubblicazione: 2026
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866911514545356800
author Liu, Quan
Zhou, Han
Wu, Wenquan
Wu, Hua
Su, Sen
author_facet Liu, Quan
Zhou, Han
Wu, Wenquan
Wu, Hua
Su, Sen
contents Ensuring that large language models (LLMs) remain both helpful and harmless poses a significant challenge: fine-tuning on repetitive safety datasets, where unsafe prompts are paired with standard refusal templates, often leads to false refusals, in which benign queries are declined. We first quantify this effect, showing that safety data exhibits substantially lower token entropy and 2-gram diversity (0.048) compared to general instruction data. To uncover the root cause, we introduce FlowLens, a stable PCA-based tool for residual-stream geometry analysis, and reveal that higher proportions of safety examples concentrate variance along a few components, reducing representational smoothness and driving false refusals (false refusal rate rises from 63 percent to 84 percent as safety data increases from 0 percent to 40 percent). Guided by these insights, we propose Variance Concentration Loss (VCL), an auxiliary regularizer that penalizes excessive variance concentration in mid-layer residuals. Empirical results demonstrate that VCL reduces false refusals by over 35 percentage points while maintaining or improving performance on general benchmarks such as MMLU and GSM8K.
format Preprint
id arxiv_https___arxiv_org_abs_2603_13318
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Residual Stream Analysis of Overfitting And Structural Disruptions
Liu, Quan
Zhou, Han
Wu, Wenquan
Wu, Hua
Su, Sen
Machine Learning
Artificial Intelligence
Ensuring that large language models (LLMs) remain both helpful and harmless poses a significant challenge: fine-tuning on repetitive safety datasets, where unsafe prompts are paired with standard refusal templates, often leads to false refusals, in which benign queries are declined. We first quantify this effect, showing that safety data exhibits substantially lower token entropy and 2-gram diversity (0.048) compared to general instruction data. To uncover the root cause, we introduce FlowLens, a stable PCA-based tool for residual-stream geometry analysis, and reveal that higher proportions of safety examples concentrate variance along a few components, reducing representational smoothness and driving false refusals (false refusal rate rises from 63 percent to 84 percent as safety data increases from 0 percent to 40 percent). Guided by these insights, we propose Variance Concentration Loss (VCL), an auxiliary regularizer that penalizes excessive variance concentration in mid-layer residuals. Empirical results demonstrate that VCL reduces false refusals by over 35 percentage points while maintaining or improving performance on general benchmarks such as MMLU and GSM8K.
title Residual Stream Analysis of Overfitting And Structural Disruptions
topic Machine Learning
Artificial Intelligence
url https://arxiv.org/abs/2603.13318