Incentivize Contribution and Learn Parameters Too: Federated Learning with Strategic Data Owners
Drashthi Doshi, Aditya Vema Reddy Kesari, Avishek Ghosh, Swaprava Nath, Suhas S Kowshik
TL;DR
This work tackles incentive design in federated learning by treating data contributors as strategic agents who decide how much data to sample. It introduces Updated Parameter Best Response Dynamics (UPBReD), which jointly updates contributions and the global model, and proves convergence to a pure Nash equilibrium. To achieve truthfulness, it adds ex-post incentive-compatible payments via T-UPBReD, and to reach the socially optimal outcome it proposes 2P-UPBReD, a two-phase, budget-balanced mechanism that drives full data contribution and attains $w^{OPT}$. Empirical results on CIFAR-10, FEMNIST, and Twitter show 2P-UPBReD closely matches FedAvg performance while overcoming strategic manipulation, with robustness enhancements such as trimmed-mean aggregation to resist adversarial behavior. The framework provides a practical path to incentive-aware FL with provable convergence and welfare guarantees, applicable to real-world, heterogeneous data settings.
Abstract
Classical federated learning (FL) assumes that the clients have a limited amount of noisy data with which they voluntarily participate and contribute towards learning a global, more accurate model in a principled manner. The learning happens in a distributed fashion without sharing the data with the center. However, these methods do not consider the incentive of an agent for participating and contributing to the process, given that data collection and running a distributed algorithm is costly for the clients. The question of rationality of contribution has been asked recently in the literature and some results exist that consider this problem. This paper addresses the question of simultaneous parameter learning and incentivizing contribution in a truthful manner, which distinguishes it from the extant literature. Our first mechanism incentivizes each client to contribute to the FL process at a Nash equilibrium and simultaneously learn the model parameters. We also ensure that agents are incentivized to truthfully reveal information in the intermediate stages of the algorithm. However, this equilibrium outcome can be away from the optimal, where clients contribute with their full data and the algorithm learns the optimal parameters. We propose a second mechanism that enables the full data contribution along with optimal parameter learning. Large scale experiments with real (federated) datasets (CIFAR-10, FEMNIST, and Twitter) show that these algorithms converge quite fast in practice, yield good welfare guarantees and better model performance for all agents.
