mlsanityData to 14 Oct 2026

Stream separation improves Bregman conditioning in transformers

arXiv 2603.21317 ↗First posted 22 Mar 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Linear methods for steering transformer representations, including probing, activation engineering, and concept erasure, implicitly assume the geometry of representation space is Euclidean. Park et al. [Park et al., 2026] showed that softmax induces a curved Bregman geometry whose metric tensor is the Hessian of the log-normalizer, H(λ) = Cov[γ | λ]. Ignoring this curvature causes Euclidean steering to leak probability mass to unintended tokens. Their analysis applies at the output layer. We measure this Hessian at intermediate layers in a controlled 2x2 design crossing stream separation with per-layer supervision (vocabulary decoding loss at each layer), all at matched vocabulary and parameter count. In standard single-stream transformers, H is severely degenerate at intermediate layers (effective rank 8 in 516 dimensions). Stream separation improves conditioning by up to 22 in effective rank, even without auxiliary supervision. Per-layer supervision helps, but less. The cosine similarity between primal and dual concept directions predicts per-layer steering effectiveness on downstream tasks, with a threshold near 0.3. These results bear on the reliability of linear safety interventions, which depend on the geometry being well-conditioned at the layer where they are applied.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 7 of its 10 references that are in the map. First posted on arXiv 22 Mar 2026.

Builds on

7 of 10 references.

References · 7

  1. 2016Understanding intermediate layers using linear classifier probescited by 1,353
  2. 2023Steering Language Models With Activation Engineeringcited by 922
  3. 2023LEACE: Perfect linear concept erasure in closed formcited by 267
  4. 2024On the Role of Attention Heads in Large Language Model Safetycited by 72
  5. 2025Superposition Yields Robust Neural Scalingcited by 31
  6. 2026The Information Geometry of Softmax: Probing and Steeringcited by 11
  7. 2026The Dual-Stream Transformer: Channelized Architecture for Interpretable Language Modelingcited by 3

Built on by

Not cited yet.