System disruptions
There are recurring service disruptions on some of the search portals due to high traffic. We are working to resolve the issue, you may temporarily encounter an error message.
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf
Toward Scalable, Robust, and Efficient Reinforcement Learning: Theoretical Foundations and Algorithms for Sample-Efficient, Communication-Aware, and Heterogeneity-Robust RL
Stockholm University, Faculty of Social Sciences, Department of Computer and Systems Sciences.ORCID iD: 0000-0003-4884-4600
2026 (English)Doctoral thesis, comprehensive summary (Other academic)
Abstract [en]

Reinforcement learning (RL) holds the promise of autonomous decision-making in robotics, energy management, and personalized healthcare, yet three practical obstacles curb its impact. A solitary agent typically needs an impractically large number of interactions before useful behavior emerges; distributing the workload across many learners shortens the wall-clock time but burdens the network with costly model exchanges, and once experience is gathered on diverse devices, differences in local data and dynamics erode the guarantees enjoyed by standard federated algorithms.

This thesis offers a unified response that advances sample efficiency, communication awareness, and heterogeneity robustness in tandem. For single-agent settings, it introduces two complementary forms of lightweight guidance: an automatically annealed curriculum of auxiliary goals that accelerates exploration in sparse-or-composite-reward tasks, and a decaying surrogate loss that soft-injects imperfect heuristic controllers during early training. In distributed scenarios, a new convergence analysis proves that momentum-based optimization remains stable even when updates are aggressively compressed, leading to a federated Q-learning framework that preserves linear progress while transmitting an order-of-magnitude fewer bits. Finally, the work establishes convergence bounds for federated optimization without tying step-size choices to explicit data similarity constants and extends them to collaborative value estimation when each learner operates in a perturbed environment, revealing precisely how mismatch and communication frequency shape residual error. Experiments across tabular and high-dimensional decision-making domains confirm that interaction demands plummet, bandwidth usage drops sharply, and learning degrades gracefully under severe non-IID conditions. Collectively, the six papers assembled here chart a coherent route toward sample-efficient, communication-constrained, and heterogeneity-robust RL, advancing the feasibility of real-world deployment on resource-limited, privacy-sensitive platforms.

Abstract [sv]

Förstärkningsinlärning (reinforcement learning, RL) utlovar självgående beslutsfattande inom områden som robotik, energihantering och personcentrerad sjukvård. Tre praktiska hinder begränsar dock teknikens genomslag. En ensam agent behöver ofta ett orimligt stort antal interaktioner innan användbart beteende uppstår. När flera agenter tränas parallellt blir visserligen väggklocktiden kortare, men nätverket belastas i stället av omfattande modellutbyten. Dessutom undergrävs teoretiska garantier när lärandet sker på heterogena enheter med skilda datakällor och dynamik.

Avhandlingen presenterar ett samlat angreppssätt som samtidigt höjer proveffektiviteten, minskar kommunikationsbehovet och ökar robustheten mot heterogenitet. För enskilda agenter introduceras två kompletterande, lättviktiga styrmekanismer: dels ett automatiskt avtonande målkurrikulum som påskyndar utforskningen i glesa eller sammansatta belöningsmiljöer, dels en avtagande surrogatförlust som injicerar ofullkomliga heuristiska kontrollregler under de tidiga träningsfaserna. I distribuerade scenarier visar en ny konvergensanalys att momentumbaserad optimering förblir stabil även när uppdateringar kraftigt komprimeras. Resultatet är ett federerat Q-lärningsramverk som bibehåller linjär inlärningstakt samtidigt som det sänder en storleksordning färre bitar. Vidare härleds konvergensgränser för federerad optimering utan att steglängden behöver kopplas till någon datasimilaritetskonstant, och dessa utvidgas till kollaborativ värdeestimering när varje agent verkar i en perturberad miljö, vilket klargör hur modellskillnader och synkroniseringsfrekvens påverkar kvarstående fel. Experiment på såväl tabellära som högdimensionella beslutsproblem bekräftar analysen: interaktionskraven sjunker drastiskt, bandbreddsanvändningen minskar kraftigt och inlärningen degraderas kontrollerat även under starkt icke-IID-betingade förhållanden. Sammantaget utstakar de sex inkluderade studierna en tydlig väg mot proveffektiv, kommunikationssnål och heterogenitetstålig RL och för därmed tekniken ett steg närmare verklig tillämpning på resursbegränsade och integritetskänsliga plattformar.

Place, publisher, year, edition, pages
Stockholm: Department of Computer and Systems Sciences, Stockholm University , 2026. , p. 104
Series
Report Series / Department of Computer & Systems Sciences, ISSN 1101-8526 ; 26-008
Keywords [en]
Artificial Intelligence, Machine Learning, Reinforcement Learning, Federated Learning, Federated Reinforcement Learning
National Category
Computer Systems Computer Vision and Learning Systems Communication Systems Robotics and automation Control Engineering
Research subject
Computer and Systems Sciences
Identifiers
URN: urn:nbn:se:su:diva-257712ISBN: 978-91-8107-680-6 (print)ISBN: 978-91-8107-681-3 (electronic)OAI: oai:DiVA.org:su-257712DiVA, id: diva2:2086838
Public defence
2026-10-16, L50, Borgarfjordsgatan 12, Kista, Stockholm, 13:00 (English)
Opponent
Supervisors
Available from: 2026-09-23 Created: 2026-07-16 Last updated: 2026-08-27Bibliographically approved
List of papers
1. TA-Explore: Teacher-Assisted Exploration for Facilitating Fast Reinforcement Learning: Extended Abstract
Open this publication in new window or tab >>TA-Explore: Teacher-Assisted Exploration for Facilitating Fast Reinforcement Learning: Extended Abstract
2023 (English)In: AAMAS '23: Proceedings of the 2023 International Conference on Autonomous Agents and Multiagent Systems / [ed] Noa Agmon; Bo An; Alessandro Ricci; William Yeoh, The International Foundation for Autonomous Agents and Multiagent Systems (IFAAMAS) , 2023, p. 2412-2414Conference paper, Published paper (Refereed)
Abstract [en]

Reinforcement Learning (RL) is crucial for data-driven decision-making but suffers from sample inefficiency. This poses a risk to system safety and can be costly in real-world environments with physical interactions. This paper proposes a human-inspired framework to improve the sample efficiency of RL algorithms, which gradually provides the learning agent with simpler but similar tasks that progress toward the main task. The proposed method does not require pre-training and can be applied to any goal, environment, and RL algorithm, including value-based and policy-based methods, as well as tabular and deep-RL methods. The framework is evaluated on a Random Walk and optimal control problem with constraint, showing good performance in improving the sample efficiency of RL-learning algorithms.

Place, publisher, year, edition, pages
The International Foundation for Autonomous Agents and Multiagent Systems (IFAAMAS), 2023
Keywords
PPO, policy optimization, exploration, deep RL, sample efficiency
National Category
Computer Sciences
Research subject
Computer and Systems Sciences
Identifiers
urn:nbn:se:su:diva-218297 (URN)9781450394321 (ISBN)
Conference
AAMAS '23: International Conference on Autonomous Agents and Multiagent Systems, London, United Kingdom, 29 May-2 June, 2023.
Available from: 2023-06-19 Created: 2023-06-19 Last updated: 2026-07-16Bibliographically approved
2. Accelerating actor-critic-based algorithms via pseudo-labels derived from prior knowledge
Open this publication in new window or tab >>Accelerating actor-critic-based algorithms via pseudo-labels derived from prior knowledge
2024 (English)In: Information Sciences, ISSN 0020-0255, E-ISSN 1872-6291, Vol. 661, article id 120182Article in journal (Refereed) Published
Abstract [en]

Despite the huge success of reinforcement learning (RL) in solving many difficult problems, its Achilles heel has always been sample inefficiency. On the other hand, in RL, taking advantage of prior knowledge, intentionally or unintentionally, has usually been avoided, so that, training an agent from scratch is common. This not only causes sample inefficiency but also endangers safety –especially during exploration. In this paper, we help the agent learn from the environment by using the pre-existing (but not necessarily exact or complete) solution for a task. Our proposed method can be integrated with any RL algorithm developed based on policy gradient and actor-critic methods. The results on five tasks with different difficulty levels by using two well-known actor-critic-based methods as the backbone of our proposed method (SAC and TD3) show our success in greatly improving sample efficiency and final performance. We have gained these results alongside robustness to noisy environments at the cost of just a slight computational overhead, which is negligible.

Keywords
Reinforcement learning, Deep RL, Actor-critic methods, Policy optimization, Sample efficiency, Exploration
National Category
Computer Sciences
Research subject
Computer and Systems Sciences
Identifiers
urn:nbn:se:su:diva-226601 (URN)10.1016/j.ins.2024.120182 (DOI)001173851600001 ()2-s2.0-85183589956 (Scopus ID)
Available from: 2024-02-14 Created: 2024-02-14 Last updated: 2026-07-16Bibliographically approved
3. Parallel Momentum Methods Under Biased Gradient Estimations
Open this publication in new window or tab >>Parallel Momentum Methods Under Biased Gradient Estimations
2025 (English)In: IEEE Transactions on Control of Network Systems, E-ISSN 2325-5870, Vol. 12, no 2, p. 1721-1732Article in journal (Refereed) Published
Abstract [en]

Parallel stochastic gradient methods are gaining prominence in solving large-scale machine learning problems that involve data distributed across multiple nodes. However, obtaining unbiased stochastic gradients, which have been the focus of most theoretical research, is challenging in many distributed machine learning applications. The gradient estimations easily become biased, for example, when gradients are compressed or clipped, when data is shuffled, and in meta-learning and reinforcement learning. In this work, we establish worst-case bounds on parallel momentum methods under biased gradient estimation on both general non-convex and μ-PL non-convex problems. Our analysis covers general distributed optimization problems, and we work out the implications for special cases where gradient estimates are biased, i.e. in meta-learning and when the gradients are compressed or clipped. Our numerical experiments verify our theoretical findings and show faster convergence performance of momentum methods than traditional biased gradient descent.

Keywords
Stochastic Gradient Descent, Parallel Momentum Methods, Biased Gradient Estimation, Compressed Gradients, Composite Gradients
National Category
Computer Sciences
Research subject
Computer and Systems Sciences
Identifiers
urn:nbn:se:su:diva-237626 (URN)10.1109/TCNS.2025.3527255 (DOI)001512536600040 ()2-s2.0-85214682075 (Scopus ID)
Available from: 2025-01-09 Created: 2025-01-09 Last updated: 2026-07-16Bibliographically approved
4. Compressed Federated Reinforcement Learning with a Generative Model
Open this publication in new window or tab >>Compressed Federated Reinforcement Learning with a Generative Model
2024 (English)In: Machine Learning and Knowledge Discovery in Databases. Research Track: European Conference, ECML PKDD 2024, Vilnius, Lithuania, September 9–13, 2024, Proceedings, Part IV / [ed] Albert Bifet; Jesse Davis; Tomas Krilavičius; Meelis Kull; Eirini Ntoutsi; Indrė Žliobaitė, Springer , 2024, p. 20-37Conference paper, Published paper (Refereed)
Abstract [en]

Reinforcement learning has recently gained unprecedented popularity, yet it still grapples with sample inefficiency. Addressing this challenge, federated reinforcement learning (FedRL) has emerged, wherein agents collaboratively learn a single policy by aggregating local estimations. However, this aggregation step incurs significant communication costs. In this paper, we propose CompFedRL, a communication-efficient FedRL approach incorporating both \textit{periodic aggregation} and (direct/error-feedback) compression mechanisms. Specifically, we consider compressed federated Q-learning with a generative model setup, where a central server learns an optimal Q-function by periodically aggregating compressed Q-estimates from local agents. For the first time, we characterize the impact of these two mechanisms (which have remained elusive) by providing a finite-time analysis of our algorithm, demonstrating strong convergence behaviors when utilizing either direct or error-feedback compression. Our bounds indicate improved solution accuracy concerning the number of agents and other federated hyperparameters while simultaneously reducing communication costs. To corroborate our theory, we also conduct in-depth numerical experiments to verify our findings, considering Top-K and Sparsified-K sparsification operators.

Place, publisher, year, edition, pages
Springer, 2024
Series
Lecture Notes in Computer Science (LNCS), ISSN 0302-9743, E-ISSN 1611-3349
Keywords
Federated Reinforcement Learning, Communication Efficiency, Direct Compression, Error-feedback Compression
National Category
Computer Sciences
Research subject
Computer and Systems Sciences
Identifiers
urn:nbn:se:su:diva-233206 (URN)10.1007/978-3-031-70359-1_2 (DOI)978-3-031-70359-1 (ISBN)978-3-031-70358-4 (ISBN)
Conference
European Conference, ECML PKDD 2024, 9-13 September, 2024, Vilnius, Lithuania.
Available from: 2024-09-04 Created: 2024-09-04 Last updated: 2026-07-16Bibliographically approved
5. On the Convergence of Federated Learning Algorithms Without Data Similarity
Open this publication in new window or tab >>On the Convergence of Federated Learning Algorithms Without Data Similarity
2025 (English)In: IEEE Transactions on Big Data, E-ISSN 2332-7790, Vol. 11, no 2, p. 659-668Article in journal (Refereed) Published
Abstract [en]

Data similarity assumptions have traditionally been relied upon to understand the convergence behaviors of federated learning methods. Unfortunately, this approach often demands fine-tuning step sizes based on the level of data similarity. When data similarity is low, these small step sizes result in an unacceptably slow convergence speed for federated methods. In this paper, we present a novel and unified framework for analyzing the convergence of federated learning algorithms without the need for data similarity conditions. Our analysis centers on an inequality that captures the influence of step sizes on algorithmic convergence performance. By applying our theorems to well-known federated algorithms, we derive precise expressions for three widely used step size schedules: fixed, diminishing, and step-decay step sizes, which are independent of data similarity conditions. Finally, we conduct comprehensive evaluations of the performance of these federated learning algorithms, employing the proposed step size strategies to train deep neural network models on benchmark datasets under varying data similarity conditions. Our findings demonstrate significant improvements in convergence speed and overall performance, marking a substantial advancement in federated learning research.

Keywords
Compression algorithms, federated learning, gradient methods, machine learning
National Category
Computer Sciences
Research subject
Computer and Systems Sciences
Identifiers
urn:nbn:se:su:diva-232103 (URN)10.1109/TBDATA.2024.3423693 (DOI)001445067800004 ()2-s2.0-105001081071 (Scopus ID)
Available from: 2024-07-24 Created: 2024-07-24 Last updated: 2026-07-16Bibliographically approved
6. Collaborative Value Function Estimation Under Model Mismatch: A Federated Temporal Difference Analysis
Open this publication in new window or tab >>Collaborative Value Function Estimation Under Model Mismatch: A Federated Temporal Difference Analysis
2025 (English)In: Machine Learning and Knowledge Discovery in Databases. Research Track: European Conference, ECML PKDD 2025, Porto, Portugal, September 15–19, 2025, Proceedings, Part VI / [ed] Rita P. Ribeiro; Bernhard Pfahringer; Nathalie Japkowicz; Pedro Larrañaga; Alípio M. Jorge; Carlos Soares; Pedro H. Abreu; João Gama, Springer , 2025, p. 41-58Conference paper, Published paper (Refereed)
Abstract [en]

Federated reinforcement learning (FedRL) enables collaborative learning while preserving data privacy by preventing direct data exchange between agents. However, many existing FedRL algorithms assume that all agents operate in identical environments, which is often unrealistic. In real-world applications, such as multi-robot teams, crowdsourced systems, and large-scale sensor networks, each agent may experience slightly different transition dynamics, leading to inherent model mismatches. In this paper, we first establish linear convergence guarantees for single-agent temporal difference learning (TD(0)) in policy evaluation and demonstrate that under a perturbed environment, the agent suffers a systematic bias that prevents accurate estimation of the true value function. This result holds under both i.i.d. and Markovian sampling regimes. We then extend our analysis to the federated TD(0) (FedTD(0)) setting, where multiple agents, each interacting with its own perturbed environment, periodically share value estimates to collaboratively approximate the true value function of a common underlying model. Our theoretical results indicate the impact of model mismatch, network connectivity, and mixing behavior on the convergence of FedTD(0). Empirical experiments corroborate our theoretical gains, highlighting that even moderate levels of information sharing significantly mitigate environment-specific errors.

Place, publisher, year, edition, pages
Springer, 2025
Series
Lecture Notes in Computer Science (LNCS), ISSN 0302-9743, E-ISSN 1611-3349 ; 16018
Keywords
Federated Reinforcement Learning, Model Mismatch in Reinforcement Learning, Temporal Difference Learning, Policy Evaluation
National Category
Computer Sciences
Research subject
Computer and Systems Sciences
Identifiers
urn:nbn:se:su:diva-248239 (URN)10.1007/978-3-032-06106-5_3 (DOI)2-s2.0-105020022174 (Scopus ID)978-3-032-06106-5 (ISBN)978-3-032-06105-8 (ISBN)
Conference
European Conference, ECML PKDD 2025, Porto, Portugal, September 15–19, 2025.
Available from: 2025-10-20 Created: 2025-10-20 Last updated: 2026-07-16Bibliographically approved

Open Access in DiVA

Toward Scalable, Robust, and Efficient Reinforcement Learning(3304 kB)21 downloads
File information
File name FULLTEXT01.pdfFile size 3304 kBChecksum SHA-512
96f6d2295f2bcf4b45996c25cff0166e1d4e7e240224661f65807b7e6bac7fd696a11649a095aecbd40f5fb71fa86efb239da02ad35c8893113625e2ff6d045d
Type fulltextMimetype application/pdf

Authority records

Beikmohammadi, Ali

Search in DiVA

By author/editor
Beikmohammadi, Ali
By organisation
Department of Computer and Systems Sciences
Computer SystemsComputer Vision and Learning SystemsCommunication SystemsRobotics and automationControl Engineering

Search outside of DiVA

GoogleGoogle Scholar
The number of downloads is the sum of all downloads of full texts. It may include eg previous versions that are now no longer available

isbn
urn-nbn

Altmetric score

isbn
urn-nbn
Total: 356 hits
CiteExportLink to record
Permanent link

Direct link
Cite
Citation style
  • apa
  • ieee
  • modern-language-association-8th-edition
  • vancouver
  • Other style
More styles
Language
  • de-DE
  • en-GB
  • en-US
  • fi-FI
  • nn-NO
  • nn-NB
  • sv-SE
  • Other locale
More languages
Output format
  • html
  • text
  • asciidoc
  • rtf