English

First-Order Predictable but Pairwise Fragile: Local Task Adaptation in Trained Transformers

Machine Learning 2026-07-18 v1 Artificial Intelligence

Abstract

Task arithmetic, sequential fine-tuning, activation steering, and first-order random search all operate through relatively small perturbations around an already trained checkpoint, and they rely on different local approximations: individual perturbations should be first-order predictable, task updates should compose with controlled interference, useful tangent structure should be stable and possible to estimate, and weight edits should have counterparts in representation space. We measure 8 such properties with the same harness around a multitask LoRA operating point, on 9 transformers (82M-7B), with a prospectively registered property list, thresholds, and test split. We find a shared one-direction validity window up to the tested scale 10210^{-2}, but no universal radius for pairwise composition or update ordering. Along individual directions, changes of the probe loss remain first-order predictable throughout the grid: a perturbation's effect on the loss is essentially its projection onto the gradient, which is also what makes local random search work. Pairwise structure, however, proves to be far more fragile: on over a third of the measured (model, task pair) combinations, two-update order sensitivity sets in strictly inside that window; task-gradient subspaces rotate within tens of steps; additivity under our fixed activation probe fails at full task-vector scale on several models, including both held-out 7B models; and no model median passes the registered global mean-vector weight-to-steering correspondence bar. For two sequential task-gradient steps, the leading order-dependent term is the Lie bracket HBgAHAgBH_B\textbf{g}_A-H_A\textbf{g}_B; its normalized prediction c(η)=ηκ+O(η2)c(\eta)=\eta\kappa+O(\eta^2) tracks the measured defect at median ratio 1.002, while the onset scale η0.10/κ\eta^\dagger\approx0.10/\kappa spans three orders of magnitude across models and task pairs.

Cite

@article{arxiv.2607.16821,
  title  = {First-Order Predictable but Pairwise Fragile: Local Task Adaptation in Trained Transformers},
  author = {Irina Piontkovskaia and Sergey Nikolenko},
  journal= {arXiv preprint arXiv:2607.16821},
  year   = {2026}
}

Comments

36 pages, 11 figures, 10 tables