openfl.interface.aggregation_functions.fedcurv_weighted_average.FedCurvWeightedAverage

class openfl.interface.aggregation_functions.fedcurv_weighted_average.FedCurvWeightedAverage(*args, **kwargs)

Bases: WeightedAverage

Aggregation function of FedCurv algorithm.

Applies weighted average aggregation to all tensors except Fisher matrices variables (u_t, v_t). These variables are summed without weights.

FedCurv paper: https://arxiv.org/pdf/1910.07796.pdf

Methods

call

Apply aggregation.

__call__(local_tensors, db_iterator, tensor_name, fl_round, tags)

Use magic function for ease.

call(local_tensors, tensor_db, tensor_name, fl_round, tags)

Apply aggregation.