Aligned but Apart: Why Cosine Conflict Diagnostics Misread Auxiliary-Loss Interference at the Output Projection

Cosine similarity between task gradients is the standard instrument for detecting conflict in auxiliary-loss training: a negative cosine reads as interference, and gradient surgery acts on it. At a language model's shared, tied output projection it is misleading. We measure the per-row gradients a next-token (CE) and an auxiliary multi-token (MTP) loss deposit there, and find them aligned in direction. On the ≈8% of rows that receive a target the median per-row cosine is ≈0.53 and only ≈3–4% are opposed; the remaining ≈92% are scalar multiples, cosine +1 by construction. The tensor is tied, so an embedding-path contribution to those cosines is unbounded here (Sections 3.2, 6). Yet the aggregate cosine runs from −0.08 to −0.19 under Muon, shallow enough at one end to read as benign, and −0.28 to −0.34 under AdamW. What comes apart is the two readings, not the supports: per-row gradient norms (three seeds, both optimizers) put both losses on the same rows, at a norm-profile cosine of 0.98, but 60–69% of that mass sits on ≈0.3% of rows, the most frequent function words and punctuation. This is norm-weighted cancellation, not disjoint support. An exact rewrite shows the shared-head objective is cross-entropy toward a next-and-future-token mixture, whose zero-parameter KL cost reproduces 62–91% of the matched within-sweep gap across estimator variants. That makes the shifted optimum, not gradient conflict, the leading account of the +0.39-nat degradation we measure at 124M (p≈8×10⁻⁵). Here the aggregate cosine is uninformative about where interference lives. Preprint; not peer reviewed. Code, committed results, and figure-regeneration scripts: https://github.com/Y1ssh/perrow-gradient-interference

Authors

Publication Details

Journal
Zenodo (CERN European Organization for Nuclear Research)
Published
2026-08-20
DOI
https://doi.org/10.5281/zenodo.22032374
Primary Topic
Stochastic Gradient Optimization Techniques
Type
preprint
Controls
|||
ALL TIME
JAN
FEB
MAR
APR
MAY
JUN
JUL
AUG
SEP
OCT
preprint

Aligned but Apart: Why Cosine Conflict Diagnostics Misread Auxiliary-Loss Interference at the Output Projection

Yash Madelwar
Zenodo (CERN European Organization for Nuclear Research)
Stochastic Gradient Optimization Techniques
preprint

Aligned but Apart: Why Cosine Conflict Diagnostics Misread Auxiliary-Loss Interference at the Output Projection

Yash Madelwar
preprint en

Abstract

Cosine similarity between task gradients is the standard instrument for detecting conflict in auxiliary-loss training: a negative cosine reads as interference, and gradient surgery acts on it. At a language model's shared, tied output projection it is misleading. We measure the per-row gradients a next-token (CE) and an auxiliary multi-token (MTP) loss deposit there, and find them aligned in direction. On the ≈8% of rows that receive a target the median per-row cosine is ≈0.53 and only ≈3–4% are opposed; the remaining ≈92% are scalar multiples, cosine +1 by construction. The tensor is tied, so an embedding-path contribution to those cosines is unbounded here (Sections 3.2, 6). Yet the aggregate cosine runs from −0.08 to −0.19 under Muon, shallow enough at one end to read as benign, and −0.28 to −0.34 under AdamW. What comes apart is the two readings, not the supports: per-row gradient norms (three seeds, both optimizers) put both losses on the same rows, at a norm-profile cosine of 0.98, but 60–69% of that mass sits on ≈0.3% of rows, the most frequent function words and punctuation. This is norm-weighted cancellation, not disjoint support. An exact rewrite shows the shared-head objective is cross-entropy toward a next-and-future-token mixture, whose zero-parameter KL cost reproduces 62–91% of the matched within-sweep gap across estimator variants. That makes the shifted optimum, not gradient conflict, the leading account of the +0.39-nat degradation we measure at 124M (p≈8×10⁻⁵). Here the aggregate cosine is uninformative about where interference lives. Preprint; not peer reviewed. Code, committed results, and figure-regeneration scripts: https://github.com/Y1ssh/perrow-gradient-interference

Zenodo (CERN European Organization for Nuclear Research)
Peace, Justice and strong institutions
Stochastic Gradient Optimization Techniques
AI Navigator

Ask Laika to Summarize, Analyze, and Connect papers live on the map.

Summarize Papers & Methodologies

Extract key findings, datasets, and comparative methods across publications.

Benchmark Rankings & Visual Analytics

Rank top research institutions, authors, funders, topics, and journals by Field-Weighted Citation Impact (FWCI) and paper volume with instant charts.

Connect Distant Disciplines

Bridge topological clusters on the map to find hidden collaborative intersections.