Taming Preconditioner Drift: Unlocking the Potential of Second-Order Optimizers for Federated Learning on Non-IID Data
Second-order optimizers can significantly accelerate large-scale training, yet their naive federated variants are often unstable or even diverge on non-IID data. We show that a key culprit is emph{preconditioner drift}: client-side second-order training induces heterogeneous emph{curvature-defined geometries} (i.e., preconditioner coordinate systems), and server-side model averaging updates computed under incompatible metrics, corrupting the global descent direction. To address this geometric mismatch, we propose texttt{FedPAC}, a emph{preconditioner alignment and correction} framework for reliable federated second-order optimization. texttt{FedPAC} explicitly decouples parameter […]