arXiv Machine Learning

The Standard Deviation in VICReg Buys a Safety Margin

arXiv Machine Learning
Oct 22

ExGrad: Beyond Suppression - Explicit Clean-Gradient Subtraction in Adversarial Training

ExGrad introduces explicit clean‑gradient subtraction in adversarial training by constructing clean‑input, adversarial‑input, and adversarial‑weight‑perturbed gradients and solving a globally solvable three‑variable problem to select coefficients. Across CIFAR‑10, CIFAR‑100, and ImageNet‑100, ExGrad consistently outperforms AWP and PGD‑AT, with the strongest gains at larger affine radii. The method demonstrates that negative clean coefficients, when properly tuned, improve robust accuracy without sacrificing clean performance.

By Zhiyuan Ye, Xiangyu Zhou, Ji Qi, Hao Zhang, Yi Zhou
arXiv Machine Learning
Sep 18

Stiefel Attention: When the Geometry of Transformer Projection Matrices Dominates Optimizer Choice---and When It Does Not

The paper introduces Stiefel Attention, which constrains the query and key projection matrices of transformers to the Stiefel manifold and optimizes them with a Riemannian Adam variant. It demonstrates that this approach yields steepest‑descent updates, is well‑conditioned, and preserves learned attention geometry during weight decay. Empirical results show significant accuracy gains on modular arithmetic grokking and CIFAR‑10 patches, with the improvement attributed to a step‑scale‑free update rule rather than equivariance or projector changes.

By Rub\'en Dar\'io Guerrero
arXiv Machine Learning
Oct 3

Why Does Train-Validation Separation Emerge? Update-Pressure Density Dynamics in Pretrained Backbones

The paper investigates why the train‑validation performance gap widens during fine‑tuning of pretrained models. It proposes a dynamic structural explanation: as training proceeds, updates shift from broadly reusable features to more example‑specific ones, increasing gradient heterogeneity and the gap. Experiments on synthetic ResMLP hierarchies, NLP models (RoBERTa, DeBERTa, Qwen) across six datasets, and vision models (ResNet‑18) confirm that higher reliance on private features correlates with larger accuracy gaps, supporting the proposed account.

By Yuchen Li, Mingyu Du, Zongqi Fan, Ken-Tye Yong, Nguyen H. Tran