Implicit Bias of Gradient Descent under Feature-Mediated Spurious Correlations
Abstract
We study how feature-mediated spurious correlations affect the learning dynamics of gradient descent on linearly separable data. Unlike the widely studied label-mediated setting, where the spurious feature depends only on the class label and is conditionally independent of the causal features, feature-mediated correlations entangle spurious and causal signals at the instance level through group-specific operators, making them both harder to detect in practice and more complex to analyse theoretically. We first generalise the implicit-bias result of soudry2018implicit to continuous distributions. Building on this, we derive closed-form group-wise classification error rates. In the general setting, each group's error decays at a rate governed by its hard-margin . In the isotropic regime, we identify a single exponent that captures the competition between the minority's geometric margin advantage and the coupling induced by the spurious alignment operators. A phase transition occurs at : below it, both groups decay at rate and group proportion directly controls learning speed; above it, the minority escapes the -dependence entirely and converges at the faster polynomial rate . These results formalise in closed form the widely observed phenomenon that group imbalance implicitly biases optimisation dynamics toward the majority, while revealing that the effect of imbalance can be entirely superseded by geometric margin advantages. Finally we establish a non-vanishing error at test time, under spurious correlation shift. Synthetic experiments confirm the rates, the prefactors and the phase transition, and last-layer retraining on Waterbirds shows the predicted scaling of each group’s error on five frozen representations.
Then back it, or bet against it.
Related papers
Open the market on this paper to see 7 more related papers.