English

Stream separation improves Bregman conditioning in transformers

Machine Learning 2026-03-24 v1

Abstract

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[γλ]H({\lambda}) = Cov[{\gamma} | {\lambda}]. 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.

Keywords

Cite

@article{arxiv.2603.21317,
  title  = {Stream separation improves Bregman conditioning in transformers},
  author = {James Clayton Kerce},
  journal= {arXiv preprint arXiv:2603.21317},
  year   = {2026}
}
R2 v1 2026-07-01T11:32:20.104Z