← Search

Safwan Labbi

4 accepted papers

2026

Beyond Softmax and Entropy: Convergence Rates of Policy Gradients with $\boldsymbol{f}$-SoftArgmax Parameterization $\&$ Coupled Regularization

ICLR 2026poster

Policy gradient methods are known to be highly sensitive to the choice of policy parameterization. In particular, the widely used softmax parameterization can induce ill-conditioned optimization landscapes and lead to exponentially slow convergence. Although this can be mitigated by preconditioning,…

Cited by 0SourceScholar
2025

Federated UCBVI: Communication-Efficient Federated Regret Minimization with Heterogeneous Agents

AISTATS 2025poster

In this paper, we present the Federated Upper Confidence Bound Value Iteration algorithm ($\texttt{Fed-UCBVI}$), a novel extension of the $\texttt{UCBVI}$ algorithm (Azar et al., 2017) tailored for the federated learning framework. We prove that the regret of $\texttt{Fed-UCBVI}$ scales as $\tilde O…

Cited by 0SourceScholar
2024

SCAFFLSA: Taming Heterogeneity in Federated Linear Stochastic Approximation and TD Learning

NeurIPS 2024poster

In this paper, we analyze the sample and communication complexity of the federated linear stochastic approximation (FedLSA) algorithm. We explicitly quantify the effects of local training with agent heterogeneity. We show that the communication complexity of FedLSA scales polynomially with the inver…

Cited by 5SourcePDFScholar