🤖 AI Summary
This work addresses the limitations of the classical Lipschitz smoothness assumption in scenarios involving rapidly varying gradients and highly heterogeneous data by proposing a decentralized optimization algorithm tailored for directed communication graphs. The method integrates gradient tracking with adaptive gradient clipping and, within the generalized $(L_0, L_1)$-smoothness framework, establishes the first convergence guarantee that does not rely on the bounded gradient dissimilarity assumption. Experimental evaluations on LIBSVM and CIFAR-10 datasets—using regularized logistic regression and convolutional neural networks, respectively—demonstrate that the proposed algorithm consistently outperforms existing approaches in both convergence speed and stability.
📝 Abstract
Decentralized optimization has become a fundamental tool for large-scale learning systems; however, most existing methods rely on the classical Lipschitz smoothness assumption, which is often violated in problems with rapidly varying gradients. Motivated by this limitation, we study decentralized optimization under the generalized $(L_0, L_1)$-smoothness framework, in which the Hessian norm is allowed to grow linearly with the gradient norm, thereby accommodating rapidly varying gradients beyond classical Lipschitz smoothness. We integrate gradient-tracking techniques with gradient clipping and carefully design the clipping threshold to ensure accurate convergence over directed communication graphs under generalized smoothness. In contrast to existing distributed optimization results under generalized smoothness that require a bounded gradient dissimilarity assumption, our results remain valid even when the gradient dissimilarity is unbounded, making the proposed framework more applicable to realistic heterogeneous data environments. We validate our approach via numerical experiments on standard benchmark datasets, including LIBSVM and CIFAR-10, using regularized logistic regression and convolutional neural networks, demonstrating superior stability and faster convergence over existing methods.