Search papers, labs, and topics across Lattice.
This paper analyzes the convergence rate of Gradient Descent (GD) in a simplified binary classification setting with a two-neuron ReLU network and two training instances, focusing on the robustness margin. The authors prove that GD converges to an optimal robustness margin at a slow rate of $Θ(1/\ln(t))$, even in this minimal setting. This result provides the first explicit lower bound on the convergence rate of the robustness margin in a non-linear model, highlighting a potential limitation of GD in achieving robust solutions.
Even in a minimal neural network setting, gradient descent's convergence to a robust solution is provably, and surprisingly, slow, scaling as $Θ(1/\ln(t))$.
We study the convergence dynamics of Gradient Descent (GD) in a minimal binary classification setting, consisting of a two-neuron ReLU network and two training instances. We prove that even under these strong simplifying assumptions, while GD successfully converges to an optimal robustness margin, effectively maximizing the distance between the decision boundary and the training points, this convergence occurs at a prohibitively slow rate, scaling strictly as $Θ(1/\ln(t))$. To the best of our knowledge, this establishes the first explicit lower bound on the convergence rate of the robustness margin in a non-linear model. Through empirical simulations, we further demonstrate that this inherent failure mode is pervasive, exhibiting the exact same tight convergence rate across multiple natural network initializations. Our theoretical guarantees are derived via a rigorous analysis of the GD trajectories across the distinct activation patterns of the model. Specifically, we develop tight control over the system's dynamics to bound the trajectory of the decision boundary, overcoming the primary technical challenge introduced by the non-linear nature of the architecture.
