ArXiv

The Geometry of Updates: Fisher Alignment at Vocabulary Scale

Authors
John Sweeney
Categories
cs.LG, cs.CL, stat.ML
arXiv
https://arxiv.org/abs/2606.27242v1
PDF
https://arxiv.org/pdf/2606.27242v1

Brief

The paper addresses source selection for LLM families with shared vocabularies by characterizing update geometry without materializing Fisher matrices. It derives that head Fisher alignment is the cosine between kernel mean embeddings in the joint activation–error space, and introduces FisherSketch to estimate this in one streaming pass (16 KB signature, 192 KB state). Validations including Llama‑3.1‑8B show FisherSketch is informative where representation metrics fail.

Why it matters

Representation metrics (e.g., CKA) are non-identifiable for transfer in shared-output-head settings: models can share identical activations yet have orthogonal head updates. The paper proves an identity that head Fisher alignment equals the cosine between kernel mean embeddings in the joint activation–error space.

Key details

  • FisherSketch computes that cosine in a single streaming pass, making vocabulary-scale head Fisher alignment practical (K=128 or 256) with a 16 KB task signature (m=4096) and a 192 KB per-task streaming state. Experiments and diagnostics, including Llama-3.1-8B verbalizer-shift tests, show FisherSketch distinguishes tasks when activation similarity cannot; paper accepted to ICML 2026.
Source evidence

Abstract

Training-free source selection for LLM families with shared vocabularies arises in scientific string domains such as SMILES, protein, and genomic sequences, where candidate corpora share a tokenizer but differ in prediction targets. This creates an activation-dark regime: representation-similarity metrics can be uninformative without assumptions about label-conditioned error geometry, while classical update-geometry metrics are computationally prohibitive at vocabulary scale. We show that, in a shared-output head setting, representation metrics (e.g., CKA) are non-identifiable for transfer; models can share identical representations yet have orthogonal head updates. The key identity is that head Fisher alignment is exactly a cosine between kernel mean embeddings in the joint activation-error space, exposing activation, error, and coupling factors rather than requiring a materialized Fisher matrix. FisherSketch estimates this cosine directly in a single streaming pass, making K=128,256 head Fisher alignment practical with a 16 KB task signature (m=4096) and a 192 KB per-task streaming state, small enough to store next to a model hash, but encoding transfer-relevant update structure. Beyond source selection, the same signatures and marginals provide a diagnostic instrument for studying whether LLM task similarity is driven by activations, errors, or their coupling; shared-parameter and internal-layer validations, together with Llama-3.1-8B verbalizer-shift experiments, show that FisherSketch remains informative when activation similarity cannot distinguish tasks.

Comment: Accepted at the 43rd International Conference on Machine Learning (ICML 2026), PMLR 306. 64 pages total (main paper plus appendix), 4 figures, 29 tables