p2pfl.learning.aggregators.fedmedian moduleΒΆ

Federated Median (FedMedian) Aggregator.

class p2pfl.learning.aggregators.fedmedian.FedMedian(disable_partial_aggregation=False)[source]ΒΆ

Bases: Aggregator

Federated Median (FedMedian) [Yin et al., 2018].

Paper: https://arxiv.org/pdf/1803.01498v1.pdf

Parameters:

disable_partial_aggregation (bool)

SUPPORTS_PARTIAL_AGGREGATION: bool = FalseΒΆ
aggregate(models)[source]ΒΆ

Compute the median of the models.

Parameters:

models (list[P2PFLModel]) – Dict with the models (node: model, num_samples).

Return type:

P2PFLModel

Returns:

A P2PFLModel with the aggregated