FedFormer: Contextual Federation with Attention in Reinforcement Learning
A core issue in federated reinforcement learning is defining how to aggregate
insights from multiple agents into one. This is commonly done by taking the
average of each participating agent's model weights into one common model
(FedAvg). We instead propose FedFormer, a novel federation strategy that
utilizes Transformer Attention to contextually aggregate embeddings from models
originating from different learner agents. In so doing, we attentively weigh
contributions of other agents with respect to the current agent's environment
and learned relationships, thus providing more effective and efficient
federation. We evaluate our methods on the Meta-World environment and find that
our approach yields significant improvements over FedAvg and non-federated Soft
Actor Critique single agent methods. Our results compared to Soft Actor
Critique show that FedFormer performs better while still abiding by the privacy
constraints of federated learning. In addition, we demonstrate nearly linear
improvements in effectiveness with increased agent pools in certain tasks. This
is contrasted by FedAvg, which fails to make noticeable improvements when
scaled.