{"id":"arxiv/2610.02659","url":"https://arxiv.org/abs/2610.02659","tags":["cs.LG","cs.AI","math.OC"],"score":0,"title":"Distributed Learning with Selective State Space Models: Architecture-Aware Convergence Analysis","domain":"cs.LG","source":"arxiv","authors":["Adam Piaseczny","Md Kamran Chowdhury Shisher","Shiqiang Wang","Christopher G. Brinton"],"reviews":0,"subject":"cs.LG","summary":"Modern state space models (SSMs), such as Mamba2, provide a compelling alternative to transformers by combining linear-time sequence modeling with recurrent state-space dynamics. However, the behavior of SSMs in distributed learning settings remains poorly understood. In particular, the existing standard federated learning methods are largely architecture-agnostic, and do not account for the stability, selectivity, and state-space parameterization that characterize modern selective SSMs. To address this, we derive architecture-aware gradient and smoothness bounds for single- and multi-layer…","category":"research","comments":0,"disputed":false,"division":"cs","mentions":0,"discussed":0,"published":"2026-10-02","reactions":{}}