arXiv Machine Learning

SCAFFOLD under Hessian Similarity: Sharp Convergence Rates

arXiv:2610. 35663v1 Announce Type: new Abstract: One of the main challenges of federated learning is heterogeneous data split among the clients, leading to biased local updates called client drift.

arXiv Machine Learning
Oct 16

Personalized Projected Ascent for Federated Stackelberg Optimization

The paper introduces a federated minimax framework where the minimization variable is shared across clients while each client retains a personalized maximization variable. It proposes a projected ascent method for the personalized variables and shows that in nonconvex-concave settings it achieves a sublinear convergence rate, but in nonconvex-PL settings it suffers from a noise floor. To overcome this, the authors develop VR-SGDA, a variance‑reduced algorithm that eliminates the noise floor through a two‑timescale update scheme, and demonstrate its effectiveness on subpopulation‑robustness and NLP adversarial training tasks.

By M Yashwanth, Sharannya Ghosh, Saikiran Bulusu, Anirban Chakraborty
arXiv Machine Learning
Oct 22

Curvature-Scaled Sharpness-Aware Minimization: Convergence Beyond Smoothness

The paper introduces Curvature-Scaled Sharpness-Aware Minimization (CS‑SAM), a variant of SAM that scales both the perturbation and descent step by curvature bounds to uniformly control gradient distortion under generalized smoothness conditions. It proves deterministic ≥1/T convergence rates for nonconvex objectives, last‑iterate ≥1/T rates for convex objectives, and linear convergence under the Polyak‑Łojasiewicz condition, while recovering standard gradient descent rates when the perturbation is zero. Experiments on computer vision tasks show that CS‑SAM, particularly the (H0,H1) variant, achieves the highest mean best‑test accuracy across four settings compared to tuned SAM and USAM.

By Dimitris Oikonomou, Nicolas Loizou
arXiv Machine Learning
Oct 8

Convex-Concave Reinforcement Learning

The paper introduces Convex-Concave Reinforcement Learning (CCRL), showing that the exact per‑iteration objective in policy learning can be expressed as a difference‑of‑convex (DC) program in log‑density‑ratio coordinates. This formulation unifies existing methods such as CPI, NPG, TRPO, and AWR as special cases and enables a multi‑step axis that couples consecutive decisions. Using sequential convex programming, the authors provide convergence guarantees and demonstrate that CCRL outperforms or matches PPO on diagnostic MDPs, classic control tasks, and a stochastic mid‑horizon healthcare domain, achieving faster convergence and higher training‑curve area.

By Shripad V. Deshmukh, Yaswanth Chittepu, Dhawal Gupta, Philip Thomas, Scott Niekum
Hugging Face Trending Papers
Aug 4

On the Implicit Flatness Bias of Sharpness-Aware Minimization: A Linear Stability Analysis with Quantitative Hyperparameter Bounds

Sharpness-Aware Minimization (SAM) improves generalization by seeking parameters whose loss is robust to local adversarial perturbations, but the quantitative mechanism underlying its implicit bias toward flat minima remains unclear. In particular, the perturbation radius $ρ$ is typically treated as an isolated tuning parameter, despite defining the neighborhood in which SAM measures sharpness.