P ERSONALIZED F EDERATED L EARNING WITH F IRST O RDER M ODEL
O PTIMIZATION
Anonymous authors
Paper under double-blind review
A BSTRACT
While federated learning traditionally aims to train a single global model across decentralized local datasets, one model may not always be ideal for all participating clients.
Here we propose an alternative, where each client only federates with other relevant clients to obtain a stronger model per client-speciﬁc objectives.
To achieve this personalization, rather than computing a single model average with constant weights for the entire federation as in traditional FL, we efﬁciently calculate optimal weighted model combinations for each client, based on ﬁguring out how much a client can beneﬁt from another’s model.
We do not assume knowledge of any underlying data distributions or client similarities, and allow each client to optimize for arbitrary target distributions of interest, enabling greater ﬂexibility for personalization.
We evaluate and characterize our method on a variety of federated settings, datasets, and degrees of local data heterogeneity.
Our method outperforms existing alternatives, while also enabling new features for personalized FL such as transfer outside of local data distributions.
I NTRODUCTION
Federated learning (FL) has shown great promise in recent years for training a single global model over decentralized data.
While seminally motivated by effective inference on a general test set similar in distribution to the decentralized data in aggregate (McMahan et al., 2016; Bonawitz et al., 2019), here we focus on federated learning from a client-centric or personalized perspective.
We aim to enable stronger performance on personalized target distributions for each participating client.
Such settings can be motivated by cross-silo FL, where clients are autonomous data vendors (e.g. hospitals managing patient data, or corporations carrying customer information) that wish to collaborate without sharing private data (Kairouz et al., 2019).
Instead of merely being a source of data and model training for the global server, clients can then take on a more active role: their federated participation may be contingent on satisfying client-speciﬁc target tasks and distributions.
A strong FL framework in practice would then ﬂexibly accommodate these objectives, allowing clients to optimize for arbitrary distributions simultaneously in a single federation.
In this setting, FL’s realistic lack of an independent and identically distributed (IID) data assumption across clients may be both a burden and a blessing.
Learning a single global model across non-IID data batches can pose challenges such as non-guaranteed convergence and model parameter divergence (Hsieh et al., 2019;
Zhao et al., 2018; Li et al., 2020).
Furthermore, trying to ﬁne-tune these global models may result in poor adaptation to local client test sets (Jiang et al., 2019).
However, the non-IID nature of each client’s local data can also provide useful signal for distinguishing their underlying local data distributions, without sharing any data.
We leverage this signal to propose a new framework for personalized FL.
Instead of giving all clients the same global model average weighted by a constant local training size as in prior work (McMahan et al., 2016), for each client we compute a weighted combination of the available models to best align with that client’s interests, modeled by evaluation on a personalized target test distribution.
Key here is that after each federating round, we maintain the client-uploaded parameters individually, allowing clients in the next round to download these copies independently of each other.
Each federated update is then a two-step process: given a local objective, clients (1) evaluate how well their received models perform on their target task and (2) use these respective performances to weight each model’s parameters in a personalized update.
We show that this intuitive process can be thought of as a particularly coarse version of popular iterative optimization algorithms such as SGD, where instead of directly accessing other clients’ data points and iteratively training our model with the granularity of gradient decent, we limit ourselves to working with their uploaded models.
We hence propose an efﬁcient method to calculate these optimal combinations for each client, calling it FedFomo , as (1) each client’s federated update is calculated with a simple first-order model optimization approximating a personalized gradient step, and (2) it draws inspiration from the “fear of missing out”, every client no longer necessarily factors in contributions from all active clients during each federation round.
In other words, curiosity can kill the cat.
Each model’s personalized performance can be saved however by restricting unhelpful models from each federated update.
We evaluate our method on federated image classiﬁcation and show that it outperforms other methods in various non-IID scenarios.
Furthermore, we show that because we compute federated updates directly with respect to client-speciﬁed local objectives, our framework can also optimize for outof-distribution performance, where client’s target distributions are different from their local training ones.
In contrast, prior work that personalized based on similarity to a client’s own model parameters (Mansour et al., 2020; Sattler et al., 2020) restricts this optimization to the local data distribution.
We thus enable new features in personalized FL, and empirically demonstrate up to 70% improvement in some settings, with larger gains as the number of clients or level of non-IIDness increases.
Our contributions
We propose a ﬂexible federated learning framework that allows clients to personalize to speciﬁc target data distributions irrespective of their available local training data.
Within this framework, we introduce a method to efﬁciently calculate the optimal weighted combination of uploaded models as a personalized federated update 3.
Our method strongly outperforms other methods in non-IID federated learning settings.
R ELATED W ORK
Federated Learning with Non-IID Data While ﬁne-tuning a global model on a client’s local data is a natural strategy to personalize (Mansour et al., 2020; Wang et al., 2019), prior work has shown that non-IID decentralized data can introduce challenges such as parameter divergence (Zhao et al., 2018), data distribution biases (Hsieh et al., 2019), and unguaranteed convergence Li et al.
Several recent methods then try to improve the robustness of global models under heavily non-IID datasets.
FedProx (Li et al., 2020) adds a proximal term to the local training objective to keep updated parameter close to the original downloaded model.
This serves to reduce potential weight divergence deﬁned in Zhao et al.
(2018), who instead allow clients to share small subsets of their data among each other.
This effectively makes each client’s local training set closer in distribution to the global test set.
More recently, Hsu et al.
(2019) propose to add momentum to the global model update in FedAvgM to reduce the possibly harmful oscillations associated with averaging local models after several rounds of stochastic gradient descent for non-identically distributed data.
While these advances may make a global model more robust across non-IID local data, they do not directly address local-level data distribution performance relevant to individual clients.
Jiang et al.
(2019) argue this latter task may be more important in non-IID FL settings, as local training data differences may suggest that only a subset of all potential features are relevant to each client.
Their target distributions may be fairly different from the global aggregate in highly personalized scenarios, with the resulting dataset shift difﬁcult to handle with a single model.
Personalized Federated Learning Given the challenges above, other approaches train multiple models or personalizing components to tackle multiple target distributions.
Smith et al.
(2017) propose multi-task learning for FL with MOCHA , a distributed MTL framework that frames clients as tasks and learns one model per client.
Mixture methods (Deng et al., 2020; Hanzely & Richt´arik, 2020; Mansour et al., 2020) compute personalized combinations of model parameters from training both local models and the global model, while Peterson et al. (2019) ensure that this is done with local privacy guarantees.
Liang et al.
(2020) apply this mixing across network layers, with lower layers acting as local encoders that map a client’s observed data to input for a globally shared classiﬁer.
Rather than only mix with a shared global model, our work allows for greater control and distinct mixing parameters with multiple local models.
Fallah et al. (2020) instead optimize the global model for fast personalization through meta-learning, while T Dinh et al. (2020) train global and local models under regularization with Moreau envelopes. Alternatively, clustered FL (Sattler et al., 2020; Ghosh et al., 2020; Briggs et al., 2020; Mansour et al., 2020)
assumes that inherent partitions or data distributions exist behind clients’ local data, and aim to cluster these partitions to federate within each cluster.
Our work does not restrict which models are computed together, allowing clients to download suitable models independently.
We also compute client-speciﬁc weighted averages for greater personalization.
Finally, unlike prior work, we allow clients to receive personalized updates for target distributions different from their local training data.
F EDERATED F IRST O RDER M ODEL O PTIMIZATION
We now present FedFomo , a personalized FL framework to efﬁciently compute client-optimizing federated updates.
We adopt the general structure of most FL methods, where we iteratively cycle between downloading model parameters from server to client, training the models locally on each client’s data, and sending back the updated models for future rounds.
However, as we do not compute a single global model, each federated download introduces two new steps: (1) ﬁguring out which models to send to which clients, and (2) computing their personalized weighted combinations.
We deﬁne our problem and describe how we accomplish (1) and (2) in the following sections.
Problem Deﬁnition and Notation
Our work most naturally applies to heterogeneous federated settings where participating clients are critically not restricted to single local training or target test distribution, and apriori we do not know anything about these distributions.
To model this, let C be a population with | C | =
K total clients, where each client c i ∈ C carries local data D i sampled from some distribution D and local model parameters θ (cid:96) ( t ) i during any round t .
Each c i also maintains some personalized objective or task T i motivating their participation in the federation.
We focus on supervised classiﬁcation as a universal task setting.
Each client and task are then associated with a test dataset D test i ∼ D ∗ .
We deﬁne each T i :=
min L ( θ (cid:96) ( t ) i ; D test i ) , where L ( θ ; D ) : Θ (cid:55)→ R is the loss function associated with dataset D , and Θ denotes the space of models possible with our presumed network architecture.
We assume no knowledge regarding clients and their data distributions, nor that test and local data belong to the same distribution.
We aim to obtain the optimal set of model parameters { θ ∗ 1 , . .
, θ ∗ K } = arg min (cid:80) i ∈
[ K ] L T i ( θ i ) .
3. C OMPUTING F EDERATED U PDATES WITH F OMO
Unlike previous work in federated learning, FedFomo learns optimal combinations of the available server models for each participating client.
To do so, we leverage information from clients in two different ways.
First, we aim to directly optimize for each client’s target objective.
We assume that clients can distinguish between good and bad models on their target tasks, through the use of a labeled validation data split D val i ⊂ D i in the client’s local data.
D val i should be similar in distribution to the target test dataset D test i .
The client can then evaluate any arbitrary model θ j on this validation set, and quantify the performance through the computed loss, denoted by L i ( θ j ) .
Second, we directly leverage the potential heterogeneity among client models.
Zhao et al.
(2018) explore this phenomenon as a failure mode for traditional single model FL, where they show that diverging model weights come directly from local data heterogeneity.
However, instead of combining these parameters into a single global model, we maintain the uploaded models individually as a means to preserve a model’s potential contribution to another client.
Critically, these two ideas together not only allow us to compute more personal model updates within non-IID local data distributions, but also enable clients to optimize for data distributions different from their own local data’s.
Federated learning as an iterative local model update The central premise of our work stems from viewing each federated model download − and subsequent changing of local model parameters − as an optimization step towards some objective.
In traditional FL, this objective involves performing well on the global population distribution, similar in representation to the union of all local datasets.
Assuming N federating clients, we compute each global model θ G at time t as: θ G ( t ) =
(cid:80) Nn =1 w n · θ (cid:96) ( t ) n , where w n = | D train n | / (cid:80)
Nj =1 | D train j | .
If client c i downloads this model, we can view this change to their local model as an update: θ (cid:96) ( t +1) i
[Equation]
since (cid:80) n w n = 1 .
This then updates a client’s current local model parameters in directions speciﬁed by the weights w and models { θ n } in the federation.
A natural choice to optimize for the global target distribution sets w n as above and in McMahan et al.
(2017), e.g. as an unbiased estimate of global model parameters.
However, in our personalized scenario, we are more interested in computing the update uniquely with respect to each client’s target task.
We then wish to ﬁnd the optimal weights w = (cid:104) w 1 , . . .
, w N (cid:105) that optimize for the client’s objective, minimizing L i ( θ (cid:96) i ) .
Efﬁcient personalization with FedFomo Intuitively, we wish to ﬁnd models { θ (cid:96) ( t ) m : m ∈ [ N ] \ i } such that moving towards their parameters leads to better performance on our target distribution, and accordingly weight these θ higher in a model average.
If a client carries a satisfying number of local data points associated with their target objective L i , then they could obtain a reasonable model through local training alone, e.g. directly updating their model parameters through SGD:
[Equation]
However, without this data, clients are more motivated to federate.
In doing so they obtain useful updates, albeit in the more restricted form of ﬁxed model parameters { θ n : n ∈ N } .
Then for personalized or non-IID target distributions, we can iteratively solve for the optimal combination of client models w ∗ =
arg min
L V i ( θ ) by computing:
[Equation]
Unfortunately, as the larger federated learning algorithm is already an iterative process with many rounds of communication, computing w ∗ through Eq.
2 may be a cumbersome procedure.
Worse, if the model averages are only computed server-side as in traditional FL, Eq. 2 becomes prohibitively expensive in communication rounds (McMahan et al., 2017).
Following this line of reasoning however, we thus derive an approximation of w ∗ for any client:
Given previous local model parameters θ (cid:96) ( t − 1) i , set of fellow federating models available to download { θ (cid:96) ( t ) n } and local client objective captured by L i , we propose weights of the form:
[Equation]
where the resulting federated update θ (cid:96) ( t ) i
[Equation]
) directly optimizes for client c
i ’s objective up to a ﬁrst-order approximation of the optimal w ∗ .
We default to the original parameters θ (cid:96) ( t − 1) i if w n < 0 above, i.e. w n = max( w n , 0) , and among positive w n normalize with w n = max( w n , 0) (cid:80) n max( w n , 0) to maintain w ∈ [0 , 1] and (cid:80) n =1 w n = 1 or 0 .
We derive Eq.
3 as a ﬁrst-order approximation of w ∗ in Appendix A.1. Here we note that our formulation captures the intuition of federating with client models that perform better than our own model, e.g. have a smaller loss on L i .
Moreso, we weigh models more heavily as this positive loss delta increases, or the distance between our current parameters and theirs decreases, in essence most heavily weighing the models that most efﬁciently improve our performance.
We use local parameters at t - 1 to directly compute how much we should factor in current parameters θ (cid:96) ( t ) i , which also helps prevent overﬁtting as L i ( θ (cid:96) ( t − 1) i ) − L i ( θ (cid:96) ( t ) i ) < 0 causes “early-stopping” at θ (cid:96) ( t − 1) i . Communication and bandwidth overhead Because the server can send multiple requested models in one download to any client, we still maintain one round of communication for model downloads and one round for uploads in between E local training epochs. Furthermore, because w in Eq. 3 is simple to calculate, the actual model update can also happen client-side, keeping the total number of communications with T total training epochs at (cid:98) 2 TE (cid:99) , as in FedAvg .
However FedFomo also needs to consider the additional bandwidth from downloading multiple models.
While quantization and distillation (Chen et al., 2017; Hinton et al., 2015; Xu et al., 2018) can alleviate this, we also avoid worst case N 2 overhead with respect to the number of active clients N by restricting the number of models downloaded M . Whether we can achieve good personalization here involves ﬁguring out which models beneﬁt which clients, and our goal is then to send as many helpful models as possible given limited bandwidth. To do so, we invoke a sampling scheme where the likelihood of sending model
θ j to client c i relies on how well θ j performed regarding client c i ’s target objective in previous rounds. Accordingly, we maintain an afﬁnity matrix P composed of vectors p
i = (cid:104) p i, 1 , . .
, p i,K (cid:105) , where p i,j measures the likelihood of sending θ j to client c i , and at each round send the available uploaded models corresponding to the top M values according to each participating client’s p .
Initially we set P = diag (1 , . .
. , 1) , i.e. each model has an equal chance of being downloaded. Then during each federated update, we update p ← p + w from Eq.
3, where w can now be negative.
If N (cid:28) K , we may beneﬁt from additional exploration, and employ an ε -greedy sampling strategy where instead of picking strictly in order of p , we have ε chance to send a random model to the client.
We investigate the robustness of FedFomo to these parameters through ablations of ε and M in the next section.
E XPERIMENTS
Experimental Setup We consider two different scenarios for simulating non-identical data distributions across federating clients.
First we evaluate with the pathological non-IID data setup presented in McMahan et al.
(2016), where each client is randomly assigned 2 classes amongtotal classes.
We also utilize a latent distribution non-IID setup, where we ﬁrst partition our datasets based on feature and semantic similarity, and then sample from them to setup different local client data distributions.
We use number of distributions ∈ { 2 , 3 , 4 , 5 , 10 } and report the average Earth
Mover’s Distance (EMD) between local client data and the total dataset across all clients to quantify non-IIDness.
We evenly allocate clients among distributions and include further details in Appendix A.3.
We evaluate under both setups with two FL scenarios: 15 and 100 clients with 100% and 10% participation respectively, reporting ﬁnal accuracy after training with E = 5 local epochs per round for 20 communication rounds in the former and 100 rounds in the latter.
Dataset and Model Implementation Details Based on prior work (McMahan et al., 2016; Liang et al., 2020), we use the MNIST (LeCun et al., 1998), CIFAR-10 (Krizhevsky et al., 2009), and
CIFAR-100 datasets.
For all experiments, we use the same CNN model architecture in McMahan et al.
We train with SGD, 0 .
1 lr, 0 momentum, 1e-4 weight decay, and 0 .
99 lr decay for CIFAR-10/100, and 0 .
01 lr for MNIST.
For FedFomo we use n = 5 and n = 10 downloads per client, ε = 0 .
3 with 0 .
05 decay each round, and separate D train and D val with an 80-20 split.
Federated Learning Baselines We compare FedFomo against methods broadly falling under two categories: they (1) propose modiﬁcations to train a single global model more robust to nonIID local datasets, or (2) aim to train more than one model or model component to personalize performance directly to client test sets.
For the former group, we consider FedAvg , FedProx , and the 5% data-sharing strategy with FedAvg , while in the latter we compare our method to MOCHA ,
LG-FedAvg , Per-FedAvg , pFedMe , and a local training baseline.
All accuracy are reported with mean and standard deviation over three runs, with local training epochs E = 5 , the same number of communication rounds (20 for 15 clients, 100% participation; 100 for 100 clients, 10% participation) and learning rate 0 .
01 for MNIST, 0 .
1 for CIFAR-10).
We implemented all results 1 .
Pathological Non-IID
We follow precedent and report accuracy after assigning two classes out of the ten to each client for the pathological setting in Table 1.
Across datasets and client setups, our proposed FedFomo strongly outperforms alternative methods in settings with larger number clients, and achieves competitive accuracy in the 15 client scenario.
In the larger 100 client scenario, each individual client participates less frequently but also carries less local training data.
Such settings motivate a higher demand for efﬁcient federated updates, as there are less training rounds for each client overall.
Meanwhile, methods that try to train a single robust model perform with mixed success over the FedAvg baseline, and notably do not perform better than local training alone.
Despite the competitive performance, we note that this pathological setting is not the most natural scenario to apply FedFomo .
In particular when there are less clients, each client’s target distribution carries only 2 random classes, there is no guarantee that any two clients share the same objective such that they can clearly beneﬁt each other.
With more clients however, we can also expect higher frequencies of target distribution overlap, and accordingly ﬁnd that we outperform all other methods.
Latent Distribution Non-IID
We next report how each FL method performs in the latent distribution setting in Fig.
1, with additional results in Table 4.
Here we study the relative performance of FedFomo across various levels of statistical heterogeneity, and again show that our method is able to strongly outperform others in highly non-IID settings.
We observe this performance gap widen as local datasets become more non-IID, where global FL methods may suffer more from combining increasingly divergent weights while also experiencing high target data distribution shift (quantiﬁed with higher EMD) due to local data heterogeneity.
Sharing a small amount of data among clients uniformly helps, as does actively trying to reduce this divergence through FedProx , but higher [Table] with pathological non-IID splits. Best results in bold.
Across pathological non-IID settings, our method outperforms or achieves competitive personalization accuracy, especially with larger federating populations.
performance most convincingly come from methods that do not rely on a single model.
The opposite trend occurs with local training, as more distributions using the same 10 or 100 classes leads to smaller within-distribution variance.
Critically, FedFomo is competitive with local training in the most extreme non-IID case while strongly outperforming FedAvg , and outperforms both in moderately non-IID settings (EMD ∈ [1 , 2] ), suggesting that we can selectively leverage model updates that best ﬁt client objectives to justify federating.
When data is more IID, it is more likely that any individual client models can beneﬁt each other, and it becomes harder for a selective update to beat a general model average.
FedFomo also outperforms personalizing-component and multi-model approaches ( MOCHA and LG-FedAvg ), where with respect to data heterogeneity we see similar albeit weaker and more stochastic trends in performance.
Personalized model weighting We next investigate FedFomo ’s personalization by learning optimal client to client weights overtime, visualizing P during training in Fig. We depict clients with the same local data distributions next to each other (e.g. clients 0 , 1 , 2 belong to distribution 0 ). Given the initial diagonal P depicting equal weighting for all other clients, we hope FedFomo increases the weights of clients that belong to the same distribution, discovering the underlying par- [Table]
titions without knowledge of client datasets. In Fig 2a we show this for the 15 client 5 non-IID latent distribution setting on CIFAR-10 with 5 clients downloaded and ε = 0 . 3 (lighter = higher weight). These default parameters adjust well to settings with more total clients (Fig 2b), and when we change the number of latent distributions (and IID-ness) in the federation (Fig 2c).
[Figure] Exploration with ε and number of models downloaded M To further understand FedFomo ’s behavior and convergence in non-IID personalized settings with respect to limited download bandwidth capability, we conduct an ablation over ε and M , reporting results on the 15 client CIFAR5-distribution setting in Fig. 3 over 100 training epochs.
We did not ﬁnd consistent correlation between ε and model performance, although this is tied to M inherently (expecting reduced variance with higher M ). With ﬁxed ε , greater M led to higher performance, as we can evaluate more models and identify the “correct” model-client assignments earlier on. [Table]
[Table]
Out-of-local-distribution personalization We now consider the non-IID federated setting where each client optimizes for target distributions not the same as their local data distribution. Here, although a client may sufﬁciently train an adequate model for one domain, it has another target data distribution of interest with hard to access relevant data. For example, in a self-driving scenario, a client may not have enough data for certain classes due to geographical constraints, motivating the need to leverage info from others.
To simulate this scenario, after organizing data into latent distributions, we randomly shufﬂe ( D val , D test ) as a pair among clients.
We test on the CIFARand CIFAR-100 datasets with 15 clients, full participation, and 5 latent distributions, repeating the shufﬂing ﬁve times, and report mean accuracy over all clients.
As shown in Fig.
4, our method consistently strongly outperforms alternatives in both non-IID CIFAR-10 and CIFAR-100 federated settings.
We compare methods using the same train and test splits randomly shufﬂed between clients, such that through shufﬂing we encounter potentially large amounts of data variation between a client’s training data and its test set.
This then supports the validity of the validation split and downloaded model evaluation components in our method to uniquely optimize for arbitrary data distributions different from a client’s local training data.
All methods other than ours are unable to convincingly handle optimizing for a target distribution that is different from the client’s initially assigned local training data.
Sharing data expectedly stands out among other methods that do not directly optimize for a client’s objective, as each client then increases the label representation overlap between its train and test sets.
We note that in the 2-distribution setting, where each client’s training data consists of 5 classes on average, the higher performance of other methods may likely be a result of our simulation, where with only two distributions to shufﬂe between it is more likely that more clients end up with the same test distribution. To shed further light on FedFomo ’s performance, we visualize how client weights evolve over time in this setting (Fig. 4 bottom), where to effectively personalize for one client, FedFomo should speciﬁcally increase the weights for the other clients belonging to the original client’s target distribution. Furthermore, in the optimal scenario we should upweight all clients with this distribution while downweighting the rest. Here we show that this indeed seems to be the case, with color denoting the local training distribution.
We depict clients 12, 13, and 14, which all carry the same local data distribution, but 13 and 14 optimize for different distributions, and see that in all cases, FedFomo facilitates upweighting clients speciﬁcally carrying the same data distribution, suggesting that we can also discover who should federate with whom with out-of-local-distribution personalization.
We include similar plots for the rest of the clients in the appendix. Locally Private FedFomo While we can implement FedFomo such that downloaded model parameters are inaccessible and any identifying connections between clients and their uploaded mod- [Table]
els are removed to subsequently preserve anonymity, unique real world privacy concerns may rise when sharing individual model parameters.
Accordingly, we now address training FedFomo under ( ε, δ ) -differential
privacy (DP).
Dwork et al.
(2014) present further details, but brieﬂy DP ensures that given two near identical datasets, the probability that querying one produces a result is nearly the same as querying the other (under control by ε and δ ).
Particularly useful here are DP’s composability and robustness to post-processing, which ensure that if we train model parameters θ to satisfy
DP, then any function on θ is also DP.
We then perform local training with DP-SGD (Abadi et al., 2016) for a DP variant of FedFomo , which adds a tunable amount of Gaussian noise to each gradient and reduces the connection between a model update and individual samples in the local training data.
More noise makes models more private at the cost of performance, and here we investigate if
FedFomo retains its performance with increased privacy under noisy local updates.
We consider the in-distribution personalization task with 5 latent non-IID distributions from the
CIFAR-10 and CIFAR-100 datasets, with 15 clients and full participation at each round, and compare FedFomo against FedAvg with varying levels of Gaussian noise, speciﬁed by σ .
At ﬁxed δ , we wish to obtain high classiﬁcation accuracy and low ε .
We use the Opacus Pytorch library 2 for
DP-SGD, and as baselines run FedFomo and FedAvg with the library’s provided SGD optimizer with δ = 0 .
For DP runs, we set δ = 1 × 10 − 5 (cid:28) 3 × 10 − 4 the inverse of the average number of local data points available to each client to maintain reasonable privacy.
As shown in Table 4, FedFomo is able to retain a sizeable improvement over FedAvg , even against the non-DP FedAvg , and does so with minimal ε .
As expected, greater δ leads to improved privacy (lower ε ) at the cost of decreased performance.
Additionaly, in Fig.
5 we show that even with noisy gradients to protect individual data point privacy, FedFomo maintains its ability to discover the larger latent distributions among local data (albeit with more noise initially).
Most importantly, despite adding noise that could potentially derail our federated update, we are able to substantially reduce privacy violation risks under ( ε, δ ) -differential privacy while maintaining strong performance.
C ONCLUSION
We present FedFomo , a ﬂexible federated learning framework that achieves strong performance in personalized FL across various non-IID settings, and uniquely enables clients to also optimize for target distributions distinct from their local training data.
A A PPENDIX A. D ERIVING THE F OMO U PDATE
Recall that we can view each federated model download can be viewed as an iterative update,
[Equation]
where given a client’s current parameters θ (cid:96) ( t ) i , the weights w = (cid:104)
w 1 , . .
w N (cid:105) in conjunction with the model deltas θ (cid:96) ( t ) n − θ (cid:96) ( t ) i determine how much each client should move its local model parameters to optimize for some objective.
Unlike more common methods in machine learning such as gradient descent, the paths we can take to get to this objective are restricted by the ﬁxed model parameters { θ (cid:96) ( t ) n } available to us at time t .
While traditional FL methods presume this objective to be global test set performance, from a client-centric perspective we should be able to set this objective with respect to any dataset or target distribution of interest.
We then view this problem as a constrained optimization problem where (cid:80) n ∈ [ N ] = 1 .
As a small discrepancy, if i ∈ [ N ] , then to also calculate w i or how much client c i should weigh its own model in the federated update directly, we reparameterize Eq.
4 as an update from a version of the local model prior to its current state, e.g.
[Equation]
and again we have current budget of 1 to allocate to all weights w .
Additionally, to go along with
Eq. 5, we deviate a bit from the optimal t + 1 term in Eq. 2 and set
[Equation]
There is then a parallel structure between Eq. 5 and Eq. 6, and we proceed by trying to ﬁnd optimal w that closely approximate the gradient ∇ w .
For client c i , we can use a Taylor series expansion to approximate the update
[Equation]
As an approximation, we consider up to the ﬁrst order term and drop the rest.
As θ (cid:96)
[Equation]
i
[Equation]
for each w n we deﬁne a function:
[Equation]
to parameterize the θ ’s as functions of our weights.
For each partial in Eq.
7, we can re-express θ ’s as ϕ ( w n ) ’s, and get
[Equation]
for any given w n .
Finally, to connect this to our estimated ˆ w ∗ , we note the equivalency between
Eq. 5 and Eq. 6, where
[Equation]
And so each individual weight for client c
i considering model θ n can be computed as
[Equation]
We arrive at Eq. 3 by distributing the negative α to capture the right direction in each update, and then dropping the term as a constant because in practice we normalize with
[Equation]
A. A DDITIONAL L ATENT D ISTRIBUTION N ON -IID E XPERIMENTS
CIFAR-100 Here we show results on the latent non-IID in-distribution personalization setup for the CIFAR-100 dataset.
As in the CIFAR-10 setting, we compare FedFomo against various recent alternative methods when personalizing to a target distribution that is the same as the client’s local training data, and report accuracy as an average over all client runs.
We also show results partitioning the CIFAR-100 dataset into increasing number of data distributions for 15 clients total, and report the increasing EMD in parentheses.
As shown in Table A.2, FedFomo consistently outperforms all alternatives with more non-IID data across different clients.
We note similar patterns to that of the CIFAR-10 dataset, where our method is more competitive when client data is more similar (lower EMD, number of distributions), but handily outperforms others as we increase this statistical label heterogeneity.
Client weights in in-local vs out-of-local distribution personalization Following the visuals for client weights in the out-of-local distribution personalization setting (Fig. 4), we include additional visualizations for the remaining clients (Fig. 6).
{+For comparison, we also include the same visualizations for the 15 client 
+}
non-IID latent distribution setup on CIFAR-10, but when clients optimize for a target distribution the same as their local training data
’s (Fig. 7).
In both, we use color to denote
[Table]
the client’s local training data distribution, such that if FedFomo is able to identify the right clients to federated with that client, we should see the weights for those colors increase or remain steady over federation rounds, while all other client weights drop.
As seen in both Fig.
6 and Fig.
7, FedFomo quickly downweights clients with unhelpful data distributions.
For the in-distribution personalization, it is able to increase and maintain higher weights for the clients from the same distribution, and consistently does so for the other two clients that belong to its distribution.
In the out-of-local distribution personalization setting, due to our shufﬂing procedure we have instances where certain clients have in-distribution targets, while others have out-of-distribution targets.
We see that FedFomo is able to accommodate both simultaneously, and learns to separate all clients belonging to the target distributions of each client from the rest.
FedFomo quickly downweights non-relevant clients while upweighting those that are helpful.
FedFomo quickly downweights non-relevant clients while upweighting or keeping steady those that are helpful.
A. L ATENT D ISTRIBUTION N ON -IID M OTIVATION AND S ETUP
In this subsection, we discuss our latent distribution non-IID setting in more detail.
We believe the pathological setup though useful might not represent more realistic or frequent occurring setups.
As an example, a world-wide dataset of road landscapes may vary greatly across different data points, but variance in their feature representations can commonly be explained by their location.
In another scenario, we can imagine that certain combinations of songs, or genres of music altogether are more likely to be liked by the same person than others.
In fact, the very basis and success of popular recommender system algorithms such as collaborative ﬁltering and latent factor models rely on this scenario (Hofmann, 2004).
Accordingly, in this sense statistical heterogeneity and client local data non-IIDnes is more likely to happen in groups.
We thus propose and utilize a latent distribution method to evaluate FedFomo against other more recent proposed FL work.
To use this setting, we ﬁrst compute image representations by training a VGG-11 convolutional neural network to at least 85% classiﬁcation accuracy on a corresponding dataset.
We then run inference on every data point, and treat the 4096-dimensional vector produced in the second fully-connected layer as a semantic embedding for each individual image.
After further reduction to 256 dimensions through PCA, we use K-Means clustering to partition our dataset into
D disjoint distributions.
Given K total clients, we then evenly assign each client to a distribution D .
For each client we ﬁnally obtain its local data by sampling randomly from D without replacement.
For datasets with pre-deﬁned train and test splits, we cluster embeddings from both at the same time such that similar images across splits are assigned the same K-means cluster, and respect these original splits such that all D test images come from the original test split.
(Fig.
VGG-
[Figure]
A. A DDITIONAL DESIGN ABLATIONS
In this section we present additional work on key hyperparameters or aspects of FedFomo to give further insight into our method’s functionality and robustness to parameters.
We consider key design choices related to the size of each client’s validation split.
Size of the validation split To better organize federated uploaded models into personalized federated updates, our method requires a local validation split D val that reﬂects the client’s objective or target test distribution.
Here, given a pre-deﬁned amount of locally available data, we ask the natural question of how a client should best go about dividing its data points between those to train its own local model and those to evaluate others with respect to computing a more informed personalized update through FedFomo .
We use the 15 client 100% participation setup with 5 latent distributions organized over the CIFAR-10 dataset, and consider both the evaluation curve and ﬁnal test accuracy over allocating a fraction ∈ { 0 . 01 , 0 . 05 , 0 . 1 , 0 . 2 , 0 . 4 , 0 . 6 , 0 . 8 , 0 . 9 } of all clients’ local data to D val , and track evaluation over 20 communication rounds with 5 epochs of local training per round.
On average, each client has 3333 local data points.
We denote ﬁnal accuracy and standard deviation over ﬁve runs in Fig 9.
As reported in Fig.
9, we observe faster convergence to a higher accuracy when allocating under half of all local data points to the validation split, with a notable drop-off using more data points.
This is most likely a result of reducing the amount of data available for each client to train their model locally.
Eventually this stagnates, and observe a slight decrease in performance between validation split fraction 0 .
05 and 0 .