A Bias-Free Revenue-Maximizing Bidding Strategy for Data Consumers in Auction-based Federated Learning
Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence. Main Track. Pages 4991-4999.
https://doi.org/10.24963/ijcai.2024/552
Keywords
- Machine Learning: ML: Federated learning
Abstract
Auction-based Federated Learning (AFL) is a burgeoning research area. However, existing bidding strategies for AFL data consumers (DCs) primarily focus on maximizing expected accumulated utility, disregarding the more complex goal of revenue maximization. They also only consider winning bids, leading to biased estimates by overlooking information from losing bids. To address these issues, we propose a Bias-free Revenue-maximizing Federated bidding strategy for DCs in AFL (BR-FEDBIDDER). Our theoretical exploration of the relationships between Return on Investment (ROI), bid costs, and utility, and their impact on overall revenue underscores the complexity of maximizing revenue solely by prioritizing ROI enhancement. Leveraging these insights, BR-FEDBIDDER optimizes bid costs with any given ROI constraint. In addition, we incorporate an auxiliary task of winning probability estimation into the framework to achieve bias-free learning by leveraging bid records from historical bid requests, including both winning and losing ones. Extensive experiments on six widely used benchmark datasets show that BR-FEDBIDDER outperforms eight state-of-the-art methods, surpassing the best-performing baseline by 5.66%, 6.08% and 2.44% in terms of the total revenue, ROI, and test accuracy of the resulting FL models, respectively.
- [Cai et al., 2017] Han Cai, Kan Ren, Weinan Zhang, Klean- this Malialis, Jun Wang, Yong Yu, and Defeng Guo. Real- time bidding by reinforcement learning in display advertis- ing. In Proceedings of the 10th ACM International Confer- ence on Web Search and Data Mining (WSDM'17), pages 661-670, 2017.
- [Clanuwat et al., 2018] Tarin Clanuwat, Mikel Bober-Irizar, Asanobu Kitamoto, Alex Lamb, Kazuaki Yamamoto, and David Ha. Deep learning for classical japanese literature. arXiv preprint, page 1812.01718, 2018.
- [Cohen et al., 2017] Gregory Cohen, Saeed Afshar, Jonathan Tapson, and Andre Van Schaik. EMNIST: Extending MNIST to handwritten letters. In Proceedings of the 2017 International Joint Conference on Neural Networks (IJCNN'17), pages 2921-2926, 2017.
- [Khan et al., 2020] Latif U Khan, Shashi Raj Pandey, Nguyen H Tran, Walid Saad, Zhu Han, Minh NH Nguyen, and Choong Seon Hong. Federated learning for edge net- works: Resource optimization and incentive mechanism. IEEE Communications Magazine, 58(10):88-93, 2020.
- [Liu et al., 2020] Yang Liu, Anbu Huang, Yun Luo, He Huang, Youzhi Liu, Yuanyuan Chen, Lican Feng, Tian- jian Chen, Han Yu, and Qiang Yang. Fedvision: An on- line visual object detection platform powered by federated learning. In Proceedings of the 32nd Annual Conference on Innovative Applications of Artificial Intelligence (IAAI- 20), pages 13172-13179, 2020.
- [Liu et al., 2022] Zelei Liu, Yuanyuan Chen, Yansong Zhao, Han Yu, Yang Liu, Renyi Bao, Jinpeng Jiang, Zaiqing Nie, Qian Xu, and Qiang Yang. Contribution-aware federated learning for smart healthcare. In Proceedings of the 34th Annual Conference on Innovative Applications of Artificial Intelligence (IAAI-22), pages 12396-12404, 2022.
- [Perlich et al., 2012] Claudia Perlich, Brian Dalessandro, Rod Hook, Ori Stitelman, Troy Raeder, and Foster Provost. Bid optimizing and inventory scoring in tar- geted online advertising. In Proceedings of the 18th ACM SIGKDD International Conference on Knowledge Discov- ery and Data Mining (KDD'12), pages 804-812, 2012.
- [Ren et al., 2017] Kan Ren, Weinan Zhang, Ke Chang, Yifei Rong, Yong Yu, and Jun Wang. Bidding machine: Learn- ing to bid for directly optimizing profits in display adver- tising. IEEE Transactions on Knowledge and Data Engi- neering, 30(4):645-659, 2017.
- [Shi and Yu, 2023] Yuxin Shi and Han Yu. Fairness-aware client selection for federated learning. In Proceedings of IEEE International Conference on Multimedia and Expo 2023 (ICME'23), 2023.
- [Tang and Yu, 2023a] Xiaoli Tang and Han Yu. Competitive- cooperative multi-agent reinforcement learning for auction-based federated learning. In IJCAI, 2023.
- [Tang and Yu, 2023b] Xiaoli Tang and Han Yu. Multi- session budget optimization for forward auction-based federated learning. arXiv preprint arXiv:2311.12548, 2023.
- [Tang and Yu, 2023c] Xiaoli Tang and Han Yu. Utility- maximizing bidding strategy for data consumers in auction-based federated learning. In Proceedings of the 2023 IEEE International Conference on Multimedia and Expo (ICME'23), pages 330-335, 2023.
- [Tang et al., 2024a] Xiaoli Tang, Han Yu, and Xiaoxiao Li. Agent-oriented joint decision support for data owners in auction-based federated learning. In ICME, 2024.
- [Tang et al., 2024b] Xiaoli Tang, Han Yu, Xiaoxiao Li, and Sarit Kraus. Intelligent agents for auction-based feder- ated learning: A survey. arXiv preprint arXiv:2404.13244, 2024. IJCAI
- [Xiao et al., 2017] Han Xiao, Kashif Rasul, and Roland Vollgraf. Fashion-MNIST: A novel image dataset for benchmarking machine learning algorithms. arXiv preprint, page 1708.07747, 2017.
- [Yang et al., 2019] Qiang Yang, Yang Liu, Tianjian Chen, and Yongxin Tong. Federated machine learning: Concept and applications. ACM Transactions on Intelligent Systems and Technology, 10(2):1-19, 2019.
- [Yang et al., 2021] Haizhi Yang, Tengyun Wang, Xiaoli Tang, Qianyu Li, Yueyue Shi, Siyu Jiang, Han Yu, and Hengjie Song. Multi-task learning for bias-free joint ctr prediction and market price modeling in online adver- tising. In Proceedings of the 30th ACM International Conference on Information and Knowledge Management (CIKM'21), pages 2291-2300, 2021.
- [Zhan et al., 2021] Yufeng Zhan, Jie Zhang, Zicong Hong, Leijie Wu, Peng Li, and Song Guo. A survey of incentive mechanism design for federated learning. IEEE Trans- actions on Emerging Topics in Computing, 10(2):1035- 1044, 2021.
- [Zhang and Wang, 2015] Weinan Zhang and Jun Wang. Sta- tistical arbitrage mining for display advertising. In Proceedings of the 21th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (KDD'15), pages 1465-1474, 2015.
- [Zhang et al., 2014] Weinan Zhang, Shuai Yuan, and Jun Wang. Optimal real-time bidding for display advertising. In Proceedings of the 20th ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (KDD'14), pages 1077-1086, 2014.
- [Zhang et al., 2021] Jingwen Zhang, Yuezhou Wu, and Rong Pan. Incentive mechanism for horizontal federated learn- ing based on reputation and reverse auction. In Proceed- ings of the Web Conference 2021 (WWW'21), pages 947- 956, 2021.
- [Zhang et al., 2022] Jingwen Zhang, Yuezhou Wu, and Rong Pan. Online auction-based incentive mechanism design for horizontal federated learning with budget constraint. arXiv preprint, page 2201.09047, 2022.
References were extracted automatically from the PDF and may contain errors.
Full text extracted automatically from the PDF; figures, tables and equations may differ. The PDF is the version of record.
1 Introduction
Driven by stringent user privacy and data confidentiality requirements, Federated Learning (FL) has recently garnered substantial attention from both academic and industrial domains [Liu et al., 2022; Liu et al., 2020; Yang et al., 2019]. With data owners (DOs), a.k.a. FL clients, being selfinterested entities that weigh a myriad of factors, ranging from costs to potential utility gains, when deciding which FL data consumer (DC) to collaborate with, the design of FL incentive mechanisms [Zhan et al., 2021; Khan et al., 2020] has taken center stage. These mechanisms aim to incentivize DOs to participate in FL through various rewarding strategies.
Auction-based federated learning (AFL) is a pivotal domain in the field of FL incentive mechanism design, primarily due to its potential to achieve both efficiency and fairness. AFL methods can be classified into three categories based on their primary emphasis: 1) DC-side methods, 2) auctioneerside methods, and 3) DO-side methods [Tang et al., 2024b; Tang et al., 2024a]. The DC-side methods focus on how DCs select and bid for DOs, with the aim of optimizing key performance indicators (KPIs) while adhering to budgetary constraints. The auctioneer-side methods explore the optimization of DC-DO matching and pricing strategies, along with the design of effective auction mechanisms. The overarching goal is to achieve specific operational objectives, such as maximizing social welfare or minimizing social costs for the entire AFL ecosystem. Issues pertaining to DOs revolve around determining the allocation of resources and setting reserve prices to maximize their profits.
Recently, increasing research attention [Tang and Yu, 2023c; Tang and Yu, 2023a; Tang and Yu, 2023b] has been paid to DC-side issues, with optimal bidding methods emerging. Existing bidding strategies for DCs mainly focus on utility maximization, while revenue maximization remains an open research problem. This poses a challenge for DCs aiming to maximize their revenues when bidding for DOs in a competitive AFL marketplace.
In addition, optimal bidding in AFL involves a number of subtasks, including utility estimation, bid cost estimation, and winning price prediction. However, due to systemic differences in data distributions between the inference and training spaces, existing AFL bidding methods suffer from sample selection bias (SSB) [Yang et al., 2021]. They require labeled data for supervised learning, but in the widely adopted generalized second-price (GSP) auction mechanism, only the winning DC can observe the actual utility of bid requests and market price. As a result, existing methods are only trained on winning records, while making inferences on the full-volume space with all bid records, leading to issues of SSB that significantly weaken model performance and generalization.
To address this challenge, we propose the Bias-free Revenue-maximizing Federated bidding strategy for DCs in AFL (BR-FEDBIDDER). We start by exploring the theoretical relationships between return on investment (ROI), bid costs and utility, and how they collectively impact overall revenue. Our analysis reveals that solely focusing on increasing ROI without considering bid costs might lead to sub-optimal revenue. Based on these findings, we design a novel revenuemaximizing bidding strategy that focuses on bid cost maximization with an acceptable ROI constraint. In addition, we incorporate winning probability estimation as an auxiliary task into BR-FEDBIDDER to achieve bias-free learning by leveraging the abundant unsuccessful bid records in the full-volume historical bid request data, including both winning and losing ones.
Extensive experiments on six benchmark datasets show that BR-FEDBIDDER outperforms eight state-of-the-art approaches. On average, it achieves 5.66% higher total revenue, 6.08% higher ROI, and 2.44% higher FL model accuracy compared to the best baseline.
2 RelatedWorks
Our research primarily aligns with the domain of bidding strategy design for forward auction-based FL DCs. [Tang and Yu, 2023c] proposed the first AFL DC bidding strategy, FedBidder. It considers the constraints of DCs' budgets, the relevance of DOs, and incorporates prior auction-related knowledge (e.g., DO distribution, probability of the DC winning the ongoing auction) into the design of bidding functions. It demonstrates the critical roles played by accurate estimation of DO utility, and the selection of an appropriate winning function in determining the optimal bidding strategy. To deal with the complex relationships among DCs during the auction process, [Tang and Yu, 2023a] frames the AFL ecosystem as a multi-agent system designed to nudge DCs to strategically bid, steering the AFL ecosystem towards equilibria with desirable overall system characteristics.
Nevertheless, existing methods focus on the utility, ignoring how to optimize revenue. In addition, these methods are faced with SSB issues. This paper bridges these important research gaps.
3 Preliminaries
A typical AFL ecosystem involves three primary stakeholders: 1) data owners (DOs), possessing potentially sensitive but valuable data; 2) data consumers (DCs), seeking data for collaboratively training AI models; and 3) an FL auctioneer who is responsible for matching DOs with DCs. The auctioneer initiates an auction, prompting interested DCs to bid for a specific DO's resources. Each DC determines its bid value b based on its objectives, limitations and the perceived utility of the DO's resources. Subsequently, by employing an auction mechanism (e.g., the generalised second-price auction (GSP)), the auctioneer determines the winning DC and the amount it needs to pay. Upon depletion of eligible DOs, the process concludes. Then, each DC commences FL model training with the recruited DOs.
Let N denote the number of qualified bid requests in an AFL ecosystem. Each bid request is a high-dimensional feature vector q following the identical independent distribution (IID) from the prior distribution pQ(q), with entries comprising features related to the DO (e.g., the quantity of the auctioned data samples, the identity of the DO). Then, each DC estimates the potential utility of bid request q if it wins the corresponding auction, upon receiving the transmitted bid request from the auctioneer. Specifically, we use s(q) to denote the estimated utility of bid request q to the target DC. Similar to q ∼ pQ(q), we denote the prior probability distribution of s(q) as pS(s). Then, based on the estimated utility s(q), the target DC calculates its bid price bi for bid request q according to its adopted bidding strategy (i.e., bidding function b(s(q))). After collecting all the bid prices from DCs, the auctioneer decides the corresponding market price z following the GSP auction mechanism, which follows a prior distribution pZ(z). Given the market price z and the corresponding bid price b(s(q)), the probability of the target DC winning the auction can be formulated as WZ(b(s(q)))). The estimated cost of bid request q with bid price b(s(q)) is denoted as c(b(s(q))).
To ensure the equations in the upcoming sections are solvable, we make the following assumptions: 1) The estimated utility of bid request q to the target DC is solely determined by q. 2) The bidding strategy of the target DC is solely determined by the estimated utility s(q). 3) The winning probability function is monotonically increasing with respect to (w.r.t.) the bidding price, and is only influenced by the bidding price. Then, we can conclude that bid request q influences the winning probability through the following chain: q → s(q) → b(s(q)) → WZ (b(s(q)))).
Under the GSP setting, only winning DCs can observe the utility of q and the corresponding market price, while other DCs can observe their own respective bid prices. Therefore, in the winning set Hw of each DC, each bid request record is denoted as a quadruple (q, y, z, b) ∈ Hw, where y is the actual utility of bid request q. In the losing set Hl, each record is represented as a tuple (q, b) ∈ Hl.
4 The Proposed Approach
In this section, we begin by formulating the overall revenue, ROI and bid cost, and analysing their relationships. Building on the analysis, we propose BR-FEDBIDDER. It is based on the multi-task learning framework with winning probability estimation as an auxiliary task to jointly model the utility estimation function and market price function, while avoiding bias by employing both the winning set Hw and losing set Hl.
4.1 Relationship Analysis
The goal of the target DC is to maximize its overall revenue, which can be reformulated as:
which means that the overall revenue of the target DC depends on both the overall ROI and the overall bid cost. The bid cost and estimated ROI of the target DC on bid request q can be denoted as:
respectively. Here, ζ represents the value of marginal utility, signifying the unit price of auctioned data resources.
As shown in Eq. (2), the bid cost Cost(q) is affected by the bid request q through the corresponding utility s(q), which means q → s(q) → Cost(q). As a result, Eq. (2) can be rewritten as:
As shown in Eq. (3), both the bid cost and ROI are affected by the estimated utility s. Therefore, it is essential to analyse the relationship between the ROI and estimated utility, and the bid cost and estimated utility.
1) Relationship between ROI and Utility: Suppose the bid price is fixed and does not change with the estimated utility, we obtain ∂b(s) ∂s = 0, ∂WZ(b(s(q))) ∂s = 0, and ∂E[Cost(b(s(q)))] ∂s = 0. Taking the derivation of ROI w.r.t. the estimated utility based on Eq. (3), we obtain:
Taking each part on the right side of Eq. (4) into consideration, we obtain ∂ROI(s) ∂s ≥ 0, which means that the ROI of the bid request increases monotonically w.r.t. the corresponding estimated utility. Thus, the bid request with higher utility increases the ROI of the DC.
- Relationship between Bid Cost and Utility: Let ¯ s be the minimum acceptable estimated utility of bid requests. Then, we formulate the abridged bidding strategy as
where b(s) is the bid price for bid requests with the estimated utility s ≥ ¯ s; 0 for the case s < ¯ s. This implies the bidding function opts not to bid for bid requests below the specified utility threshold. Given N available bid requests, substituting this abridged bidding function into Eq. (3) yields:
I[·] is an indicator function. Based on Eq. (6), we obtain:
According to the definitions of each part on the right-hand side of Eq. (7), we obtain ∂Overall Cost(¯ s) ∂¯ s ≤ 0. Thus, the overall bid cost monotonically decreases with the increase of the estimated utility of bid requests, implying a cost reduction with bid requests of higher estimated utility.
As shown above, ROI monotonically increases w.r.t. the utility of the corresponding bid request when the bid price is fixed. In this sense, higher ROI will generate better overall revenue of the corresponding DC according to Eq. (1) when the overall cost is fixed. However, as shown in Section 4.1, the overall bid cost reduces with the increase of estimated utility. The major reason for this is the number of higherutility bid requests tends to be limited in general, leading to severe competition among the DCs. Such severe competition tends to limit the total cost of bid requests with higher utility.
Thus, as the utility of bid requests increases, the overall bid cost and overall ROI exhibit opposing trends. Consequently, further examination is necessary to understand the relationship between estimated utility and overall revenue, which is the product of overall bid cost and overall ROI.
- Relationship between Revenue and Utility: Substituting the babr (s) in Eq. (5) into Eq. (3) yields:
Substituting Overall ROI(¯ s) defined in Eq. (8) and Overall Cost(¯ s) defined in Eq. (6) into Eq. (1) yields:
The derivation of Overall Revenue(¯ s) w.r.t. ¯ s is:
According to the definition of each part on the right-hand side of Eq. (10), we can get ∂Overall Revenue(¯ s) ∂¯ s ≤ 0. This means that the overall revenue decreases monotonically with ¯ s. Therefore, a focus on higher-utility bid requests may restrict revenue improvement.
The above analysis shows that it is difficult to boost revenue by focusing just on higher-utility bid requests and ignoring the balance between the bid cost and ROI. To address this issue, we propose the following revenue-maximizing bidding strategy for the target DC.
4.2 Revenue-maximizing Bidding Strategy
The revenue-maximizing bidding strategy for the target DC is formulated as:
where bO () is the bidding function, which transforms the original revenue maximization problem into overall bid cost maximization under a given ROI R. To obtain the optimal bidding function bO () for the target DC, we suppose that if each bid request q's ROI is greater than R, the target DC's overall ROI is no less than R. Based on this assumption, substituting the overall bid cost defined in Eq. (6) and ROI defined in Eq. (3) into Eq. (11) yields:
where s is the estimated utility of q. However, the ROI condition defined in Eq. (12) is too stringent and cannot be met in practice. Thus, we adopt ϵs = max(0, 1- Rs R ) to relax this requirement, where Rs = WZ(b(s))s E[Cost(b(s))] is the ROI associated with s. Then, Eq. (12) is reformulated as:
Let C denote the hyperparameter for regularization. Then, the loss function of bO () defined in Eq. (13) is formulated as:
Eq. (14) and Eq. (15) indicate that the expected bid cost E[Cost(b(s))], corresponding winning function WZ(b(s)) and estimated utility s of the bid request by the target DC determine the optimal bidding function bO (). We discuss how to obtain these three functions in the following sections.
Bid Cost & Winning Probability. Under GSP, the target DC wins only if its bid price b(s) exceeds the corresponding market price z. Moreover, if it wins, it pays z as the actual cost; otherwise, it pays nothing. Therefore, the bid cost c(b(s)) can be expressed as:
Then, as z ∼ pZ(z), we formulate c(b(s)) as:
Given q, the probability of the target DC winning the auction (i.e., WZ(b(s))) is defined as the probability that its bid price b(s) exceeds the corresponding market price z:
Then, the cost estimation and winning function modeling are transformed into the market price modeling. Therefore, the optimal bidding function depends on the utility estimation and market price modeling. It is intuitive to address these two issues independently. However, they are interdependent. On the one hand, as shown in [Tang and Yu, 2023c], estimated utility and bid price are positively related. On the other hand, under the GSP auction mechanism, the market price is the highest bid price submitted by competing DCs. Thus, the market price has a significant positive correlation with the estimated utility of data resources in theory. This makes it sub-optimal to perform these estimation tasks separately.
To deal with this issue, we design a multi-task learningbased framework to jointly optimize the utility estimation and market price modeling simultaneously, which is explained in detail in the following section.
4.3 Multi-Task Learning-based Utility Estimation & Market Price Modeling
The framework for jointly optimizing the utility estimation and market price modeling consists of four main modules: 1) the interaction modeling module captures the high-order interactions among features of bid requests for better-shared bid request representations for the following modules; 2) the utility estimation module performs the utility estimation task based on the shared feature representation; 3) the market price modeling module executes the task of modeling market prices by leveraging the shared feature; and 4) the winning probability prediction module predicts the winning probability of a specific bid price based on the output of the market price modeling module.
1) Interaction Modeling Module: To capture the nonlinear interaction patterns among features of bid requests, we then utilize the hard parameter sharing method widely adopted in multi-task learning approaches to further map the representation q of each bid request into the shared representation vector q′ , i.e.:
where fsrm(·) represents the mapping function (i.e., hardparameter sharing layers), and consists of a number of hidden layers, each of whose parameters are jointly trained by the three tasks that come after.
2) Utility Estimation Module: Based on the shared feature representation q′ output by the upstream module, this module formulates the utility estimation as a regression task, which is defined as s(q) = fue(q′ ) = s. fue(·) represents the mapping function of the utility estimation module. Then, the widely used squared error (SE) loss is adopted to train the model:
where y is the actual utility of the bid request q.
3) Market Price Modeling Module: As bid prices are discrete integers within a specific range under AFL, we incorporate a Softmax layer into this module to take the predicted probability density of each possible market price value as the market price distribution output result [Yang et al., 2021]. Specifically, given a market price z = k, the mapping function f kmp (·) first maps the representation q′ generated by the interaction module to a score. Then, the normalized probability density p(z = k|q) is given by: p(z = k|q) = exp(f kmp (q′ )) ∑z=bmax z=0 exp(f zmp (q′ )) , where bmax represents the maximum potential bid price. To train this module, the following loss function is adopted:
4) Winning Probability Prediction Module: Eq. (20) and Eq. (21) show that only the winning cases are used for the training of both the utility estimation task and the market price modeling task. This might result in sample selection biases. To deal with this problem, we take advantage of the bid information generated under both the winning cases and the losing cases in the multi-task learning framework, and incorporate the winning probability prediction as the auxiliary task. As described above, the winning probability is defined as the likelihood of the bid price b exceeding the corresponding market price z. Given each market price probability density function p(z|(q))) obtained above, a specific bid price b's winning probability WZ(b(s)) is derived as:
Then, we can directly obtain the winning probability of any bid price b on the basis of the output of market price distribution p(z|q) in the market price modeling module according to Eq. (22) without additional parameter.
However, under the GSP auction mechanism, the market price is observable only when the DC successfully wins the corresponding bid request by exceeding the market price. Conversely, if the DC loses the auction, it is only aware that the market price is equal to or greater than its bid price, represented as z ≥ b. To ensure unbiased estimation, the proposed winning probability prediction module is designed to maximize the likelihood of both winning and losing scenarios, effectively utilizing the complete bid information available in the entire inference space. Consequently, the loss function for this module is formulated as:
where w = 1 if the DC wins q; otherwise, w = 0. Ht = Hw ∩ Hl. Eq. (23) shows that Lwp is used to predict all training samples' winning probability.
The market price modeling module is trained across the entire inference space to directly rectify the bias by introducing the winning probability prediction as an auxiliary task into the multi-task learning framework. Moreover, the utility estimation module is also able to execute bias-free training indirectly via the shared representation, benefiting from multitask learning. In this sense, the sample selection bias issue can be adequately mitigated by winning probability prediction (i.e., the auxiliary task).
Overall TrainingObjective. The objective of the proposed multi-task learning-based framework is the joint optimization of both the utility estimation and market price modeling, taking advantage of the auxiliary winning probability prediction. The overall training goal to be optimized is formulated as the combination of all these losses described above: L(θ) = wue · Lue +wmp · Lmp +wwp · Lwp, where θ represents all the parameters involved in the multi-task learning framework. wue, wmp, and wwp are hyperparameters used to control the importance of different tasks.
4.4 Optimal Bidding Strategy for Data Consumers
After estimating the utility and modeling the market price based on the multi-task learning framework, we can get the predicted utility of each bid request, denoted as H = {s1, · · · , sm, · · · , }, where sm denotes the estimated utility of bid request qm. Then, the loss function defined in Eq. (14) and (15) can be re-expressed as: L(b(s)) = - ∑s∈H (1 - Cϵs)E[Cost(b(s))], where ϵs is denoted as: ϵs = max(0, 1 - ζR × WZ(b(s))s E[Cost(b(s))] ). Based on Eq. (17),
As shown in Eq. (24), we need to define the detailed formulation of the bidding strategy b(s). Following [Zhang et al., 2014; Perlich et al., 2012; Zhang and Wang, 2015; Ren et al., 2017], we adopt the following bidding function: 1) The linear bidding function b1(s) (BR-FEDBIDDER1):
2) The concave shape bidding function b2(s) (BRFEDBIDDER2):
Alg. 1 provides the learning process for optimizing ω in the bidding function. The complete learning procedure of BRFEDBIDDER is as follows: 1) Utilize the multi-task learningbased framework to model the utility and market price of the bid request being auctioned. 2) Estimate the bid cost and winning probability using Eqs. (17) and (18). 3) Optimize the parameter of the bidding function following Alg. 1.
Algorithm 1 The learning procedure of parameter ω in the bidding function of BR-FEDBIDDER.
INPUT: Training Data H, the multi-task learning framework, Learning rate ηω
OUTPUT: Optimal bidding function bO ()
- 1: Select the bidding function defined in Eq. (25) or (26)
- 2: Initialize ω of the selected bidding function
- 3: for training rounds do
- 4: for s ∈ H do
- 5: Get ∂L(b(s)) ∂b(s) according to Eq. (24)
- 6: Get ∂b(s) ∂ω according to the selected bidding function
- 7: ω ← ω - ηω ∂L(b(s)) ∂b(s) ∂b(s) ∂ω
- 8: end for
- 9: end for
- 10: return bO ()
5 Experimental Evaluation
5.1 Experiment Setup
Datasets. We conduct experimental evaluation using commonly adopted datasets in FL studies, including MNIST (http://yann.lecun.com/exdb/mnist/), CIFAR-10 (https://www.cs.toronto.edu/kriz/cifar.html), FashionMNIST (i.e., FMNIST) [Xiao et al., 2017], EMNIST-digits (i.e., EMNIST-D), EMNIST-letters (i.e., EMNIST-L) [Cohen et al., 2017] and Kuzushiji-MNIST (i.e., KMNIST) [Clanuwat et al., 2018]. The FL models used are the same as those employed in [Tang and Yu, 2023c].
| Method | MNIST | CIFAR | FMNIST | EMNISTD | EMNISTL | KMNIST | |||||||||||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| Rev | ROI | Acc | Rev | ROI | Acc | ROI | Acc | Rev | Acc | ||||||||||||
| Const Rand Bmub Lin BM FBs | Rev 9.92 11.03 10.40 11.96 14.12 16.42 16.33 17.31 | ROI 0.22 0.23 0.21 0.24 0.29 0.33 0.33 0.35 | Acc 84.77 84.65 85.06 85.05 85.01 85.32 84.98 85.83 | 9.53 12.10 11.04 13.34 14.63 16.87 16.47 16.97 | 0.21 0.25 0.22 0.27 0.30 0.35 0.35 0.34 | 45.00 44.95 45.79 46.66 46.58 47.05 47.62 47.86 | 11.38 11.60 11.94 14.02 14.92 16.80 17.01 17.62 | ROI 0.25 0.23 0.25 0.29 0.31 0.34 0.35 | Acc 73.57 74.78 74.74 75.29 76.27 77.50 77.17 | Rev 11.44 12.15 11.61 12.93 13.95 15.41 15.99 16.42 | 0.25 0.25 0.24 0.27 0.29 0.32 0.33 0.34 | 82.72 82.34 82.78 83.03 82.78 83.55 83.37 83.60 | Rev 7.73 10.50 9.06 11.37 14.62 16.42 16.26 16.80 | 0.17 0.22 0.19 0.24 0.30 0.34 0.33 | 74.32 74.62 74.57 74.87 74.83 75.17 74.54 | 7.98 10.57 9.63 13.24 13.19 16.19 16.24 17.81 | ROI 0.18 0.22 0.20 0.27 0.27 0.34 0.34 0.36 0.38 | 73.11 71.66 75.61 73.77 76.77 76.37 76.43 77.38 | |||
| FMRB1 FMRB2 | 17.61 | 0.37 | 87.49 | 17.26 | 0.36 | 49.98 45.53 | 17.67 14.07 | 0.36 | 78.91 | 16.91 | 0.35 | 84.78 | 17.12 12.05 | 0.35 0.08 | 75.66 | 17.87 | 78.68 | ||||
| 17.65 | 0.36 | 87.86 85.27 | 17.20 14.51 13.41 | 0.35 0.10 0.09 0.09 0.12 | 45.50 46.84 47.01 47.82 48.17 | 15.54 16.18 18.46 | 0.37 0.10 0.11 | 79.60 | 16.77 | 0.36 | 82.80 | 10.23 | 0.07 0.10 0.12 0.16 0.24 | 74.60 75.11 75.23 75.33 75.22 | 18.08 13.87 16.36 | 0.38 | 78.78 73.56 | ||||
| FMRB1 FMRB2 Const | 50.31 50.06 | 0.34 0.34 | 87.81 88.71 85.38 | 47.33 16.70 | 0.06 0.10 | 51.85 46.11 45.88 | 51.17 17.74 | 0.35 0.06 0.10 | 78.97 74.50 | 49.75 | 0.32 0.32 | 85.22 82.96 | 48.58 14.66 27.06 | 0.33 0.05 0.09 | 74.58 74.95 | 51.00 17.13 | 0.06 0.09 | 78.36 74.13 72.02 | |||
| RLB | 0.09 0.10 0.12 | 47.44 47.69 48.63 | 30.39 28.47 30.47 37.28 67.31 | 0.10 0.10 0.12 0.23 0.25 | 75.10 75.13 | 25.73 29.29 36.61 64.91 | 0.09 0.10 0.12 | 74.79 75.11 | |||||||||||||
| FBc | 0.36 | ||||||||||||||||||||
| 17.63 | 77.77 | 17.13 | 0.34 0.35 | 74.91 | |||||||||||||||||
| 49.51 | 84.86 | 75.42 | |||||||||||||||||||
| Const Rand Bmub | 14.13 | 0.10 | 74.27 | 12.20 | 0.08 | 74.42 | 0.10 | ||||||||||||||
| 0.10 | 85.04 | 12.82 | 0.09 | 74.86 | 0.11 | 72.68 | |||||||||||||||
| 13.88 | 85.53 | 74.89 | 82.61 | ||||||||||||||||||
| Lin BM | 17.05 18.99 | 0.11 0.13 | 85.58 | 14.02 18.33 | 0.11 0.12 | 75.14 76.73 | 14.97 17.67 | 0.10 | 83.26 83.33 | 15.00 18.46 | 15.09 18.60 | 0.10 0.12 0.17 | 75.25 74.38 | ||||||||
| FBs | 24.35 | 0.16 | 85.59 | 23.08 | 0.16 | 24.81 | 0.17 | 77.37 | 22.86 | 0.12 | 83.26 | 23.32 | 24.65 | 0.25 | 77.27 | ||||||
| FBc | 37.81 | 0.25 | 85.88 | 35.73 | 0.24 | 37.75 | 0.25 | 77.83 | 35.28 | 0.15 | 36.63 | 77.29 | |||||||||
| 0.28 | 85.76 | 39.94 | 0.27 | 48.86 | 42.89 | 0.29 | 77.46 | 39.07 | 0.24 | 83.93 | 35.80 | 40.19 | 0.27 | 76.55 | |||||||
| RLB | 41.78 | 0.33 | 0.26 | 83.72 | 40.00 | 0.27 | |||||||||||||||
| 46.08 | 49.09 | 49.25 | 77.48 | 45.98 | 0.30 | 48.47 | 0.33 | 78.01 | |||||||||||||
| 0.31 | 0.31 | 44.28 | 0.33 | 75.58 | |||||||||||||||||
| 45.37 | 0.35 | 0.31 | 83.94 | 49.37 | 78.97 | ||||||||||||||||
| 86.55 | 76.30 | ||||||||||||||||||||
| 0.32 | 78.92 | 84.71 | 50.88 | 0.34 0.34 | |||||||||||||||||
| Rand | 51.10 | 0.09 | 82.69 83.13 | 0.09 | 74.69 | ||||||||||||||||
| Bmub | 48.42 | 0.09 | 0.10 | 75.30 | |||||||||||||||||
| 50.50 | 17.87 | 27.18 | |||||||||||||||||||
| Lin | 0.32 | 26.03 26.12 | 0.10 | 83.36 | 75.97 | 26.60 | 0.12 | 0.06 | |||||||||||||
| BM | 15.69 25.20 25.22 | 29.92 | 29.90 | ||||||||||||||||||
| 29.99 39.03 | 0.05 0.09 0.08 | 47.64 | 0.12 | 75.41 | 0.22 | 77.80 | |||||||||||||||
| 69.76 | 0.13 0.23 | 85.12 | 30.62 27.61 | 83.13 84.06 | 0.22 | 75.46 | 36.61 | ||||||||||||||
| FBs | 0.10 | 85.73 85.82 86.25 86.32 | 29.43 36.77 64.73 | 36.61 | 0.22 | 83.98 | 67.00 | 0.22 | 75.67 | 64.81 | 0.23 | 76.93 | |||||||||
| 0.24 | 86.19 | 68.07 | 72.88 | 76.91 78.19 | 79.53 | 0.27 | 75.60 | 67.31 | 0.27 | 76.18 78.31 | |||||||||||
| FBc | 72.08 | 0.22 | 85.64 | 0.29 0.31 | 78.30 | 64.56 68.23 | 0.23 | 84.13 | 87.44 | 76.73 | 79.53 | 0.30 | 79.96 | ||||||||
| RLB FMRB1 | 86.08 | 0.29 | 86.86 | 0.23 | 49.02 48.91 | 92.61 | 78.07 78.58 | 79.53 85.95 | 0.27 | 76.76 | 0.31 | 85.39 | 0.30 | ||||||||
| 93.05 | 0.31 | 88.08 | 78.83 | 0.26 | 49.69 51.75 | 79.99 | 0.29 | 0.29 | |||||||||||||
| 0.29 | 92.82 | 0.29 | |||||||||||||||||||
| 85.18 | 52.05 | ||||||||||||||||||||
| FMRB2 | 92.87 | 86.06 | 87.69 | 88.17 | 80.57 | ||||||||||||||||
| 85.54 | 0.29 | ||||||||||||||||||||
| 0.31 | |||||||||||||||||||||
| 87.69 | |||||||||||||||||||||
| 79.74 | |||||||||||||||||||||
| 85.30 | |||||||||||||||||||||
| 0.29 | |||||||||||||||||||||
| 89.98 | |||||||||||||||||||||
Comparison Methods. We compare the two variants of BR-FEDBIDDER: fmrb1 based on Eq. (25) and fmrb2 based on Eq. (26) with eight established bidding approaches: Constant Bid (Const) [Zhang et al., 2014], Randomly Generated Bid (Rand) [Zhang et al., 2021; Zhang et al., 2022], Below Max Utility Bid (Bmub), Linear-Form Bid (Lin) [Perlich et al., 2012], Bidding Machine (BM) [Ren et al., 2017], and Reinforcement Learning-based Bid (RLB) [Cai et al., 2017]. Details of these methods can be found in [Tang and Yu, 2023a]. Moreover, we also include Fed-Bidder [Tang and Yu, 2023c] for comparison. Tailored for data consumers in AFL settings, Fed-Bidder aims to competitively bid for data owners to maximize the utility. It offers two variants: one with a simple winning function, FedBidder-sim (FBs), and the other with a complex winning function, Fed-Biddercom (FBc).
Experiment Scenarios. We compare BR-FEDBIDDER with baselines in two scenarios, each containing 10,000 DOs: 1) IID data: Following [Tang and Yu, 2023c], we assign sequential IDs to DOs and introduce Gaussian noise to the data resources of the first half of the population, creating lowquality resources, while the second half remains unchanged with high-quality data. Dataset sizes of various DOs vary randomly from 500 to 5,000 samples. 2) Non-IID data: Following [Shi and Yu, 2023], one class (for datasets other than EMNISTL) or six classes (for EMNISTL) is designated as the minority class, assigned to the first 100 DOs. These DOs possess images for all classes, while others exclusively have images for the remaining classes. Each DO has 3,000 images.
Implementation Details. To gather historical AFL bidding data, we employed a data accumulation approach, observing the behaviors of 160 DCs and 10,000 DOs in four distinct settings. In the first, each baseline bidding method was used by one-eighth of the DCs. In the second, four strategies, including BM, FBs, FBc, and RLB with AI techniques like BRFEDBIDDER, were adopted by three-sixteenths of the population each, while the remaining baselines were chosen by one-sixteenth each. In tailored AFL settings, FBs and FBc were favored. In the third setting, 50 DCs opted for FBs and FBc, while 10 DCs selected each of the other six baselines. In the fourth, 65 DCs chose FBs and FBc, with 5 DCs each for the other six baselines. Using the GSP forward auction mechanism, we continuously monitored DC behaviors to gradually accumulate data, compensating for the absence of a dedicated AFL dataset.
| Bud | Method | MNIST | CIFAR | FMNIST | EMNISTD | EMNISTL | KMNIST |
|---|---|---|---|---|---|---|---|
| Const | 75.04 | 21.86 | 65.51 | 78.10 | 66.44 | 69.31 | |
| Rand | 74.80 | 20.15 | 65.17 | 77.95 | 65.77 | 68.32 | |
| Bmub | 74.96 | 22.15 | 65.18 | 78.05 | 67.04 | 68.50 | |
| Lin | 75.39 | 24.45 | 65.72 | 78.29 | 68.35 | 68.78 | |
| 50 | BM | 75.79 | 27.00 | 65.98 | 79.09 | 69.55 | 69.70 |
| FBs | 76.51 | 27.51 | 66.14 | 79.35 | 69.43 | 70.06 | |
| FBc | 75.48 | 26.56 | 65.79 | 79.14 | 69.28 | 69.67 | |
| RLB | 76.20 | 28.02 | 66.43 | 79.66 | 70.15 | 70.11 | |
| FMRB1 | 77.92 | 29.95 | 67.19 | 80.62 | 70.80 | 71.83 | |
| FMRB2 | 77.78 | 29.70 | 67.59 | 79.96 | 71.11 | 72.90 | |
| Const | 75.03 | 23.24 | 65.16 | 78.23 | 66.59 | 69.38 | |
| Rand | 75.01 | 22.80 | 65.29 | 78.12 | 65.79 | 68.55 | |
| Bmub | 75.49 | 23.75 | 65.68 | 78.34 | 67.30 | 69.16 | |
| Lin | 75.70 | 27.08 | 65.89 | 78.65 | 68.63 | 69.05 | |
| 150 | BM | 76.23 | 28.33 | 66.44 | 79.23 | 70.16 | 70.22 |
| FBs | 76.52 | 29.48 | 66.46 | 79.64 | 70.49 | 70.78 | |
| FBc | 75.94 | 28.34 | 66.57 | 79.06 | 69.96 | 69.74 | |
| RLB | 76.73 | 30.42 | 66.78 | 80.07 | 70.98 | 70.82 | |
| FMRB1 | 78.20 | 32.22 | 68.24 | 81.24 | 71.25 | 72.18 | |
| FMRB2 | 78.37 | 32.31 | 68.07 | 81.46 | 71.90 | 73.32 | |
| Const | 75.02 | 24.67 | 65.68 | 78.28 | 67.32 | 69.21 | |
| Rand | 74.95 | 24.66 | 65.40 | 78.10 | 65.95 | 68.69 | |
| Bmub | 75.47 | 25.99 | 65.83 | 78.50 | 67.79 | 69.85 | |
| Lin | 76.03 | 28.13 | 66.26 | 78.99 | 68.69 | 69.32 | |
| 300 | BM | 76.20 | 28.22 | 66.51 | 79.48 | 70.43 | 70.51 |
| FBs | 76.57 | 30.03 | 66.93 | 79.68 | 70.29 | 70.86 | |
| FBc | 76.25 | 28.95 | 66.67 | 79.36 | 69.98 | 70.05 | |
| RLB | 76.43 | 30.41 | 66.82 | 80.21 | 71.05 | 70.96 | |
| FMRB1 | 79.50 | 32.89 | 68.34 | 81.73 | 71.68 | 73.59 | |
| FMRB2 | 80.27 | 32.51 | 69.40 | 81.99 | 72.93 | 72.52 |
| Method | MNIST | CIFAR-10 | FMNIST | EMNISTD | EMNISTL | KMNIST | ||||||
|---|---|---|---|---|---|---|---|---|---|---|---|---|
| IID | NIID | IID | NIID | IID | NIID | IID | NIID | IID | NIID | IID | NIID | |
| BR-FEDBIDDER | 87.08 | 77.50 | 50.75 | 32.89 | 79.98 | 67.34 | 84.39 | 80.26 | 75.72 | 71.68 | 79.14 | 71.59 |
| BR-FEDBIDDER-W | 85.98 | 77.27 | 49.14 | 32.53 | 79.74 | 65.40 | 82.54 | 77.27 | 73.76 | 70.93 | 77.56 | 71.52 |
To evaluate the effectiveness of BR-FEDBIDDER, we create a group of ten FL DCs, each utilizing one of the aforementioned bidding approaches to join the auction for each bid request from the available FL DOs. Following [Tang and Yu, 2023c], bid requests are delivered in chronological order. Upon receiving a bid request, each DC derives its bid price based on its adopted bidding strategy. Then, the auctioneer gathers the bid prices, identifies the winner, and determines the market price using the GSP auction mechanism. The winning DC remits the market price to the DO, and the data aligns with the winning DC for FL model training. The process concludes when there are no more bid requests or when the budget is depleted.
Evaluation Metrics. Our study aims to optimize total revenue while maintaining a satisfactory ROI for the data consumer with the objective of achieving satisfactory FL model performance. Therefore, we adopt the revenue (Rev), ROI and the accuracy (Acc) of the FL models as the evaluation metrics.
5.2 Results and Discussion
To evaluate the performance of the proposed BRFEDBIDDER, we perform experiments on six datasets by varying the budget among {50, 150, 300} under the IID and Non-IID scenarios. The results are shown in Tables 1 and 2.
Table 1 shows the comparison results for revenue, ROI, and FL model accuracy under the IID scenario. Across all six datasets and three budget settings, our proposed BR-FEDBIDDER consistently outperforms baseline methods. Specifically, compared to the best-performing baseline, BRFEDBIDDER achieves a 5.66% improvement in revenue. In addition, BR-FEDBIDDER consistently achieves satisfactory ROI, with a 6.08% improvement over the best-performing baseline. These results demonstrate the efficacy of our approach in optimizing the revenue of data consumers while ensuring a satisfactory ROI. The accuracy results in Table 1 align with the performance in revenue and ROI, with BRFEDBIDDER achieving a 2.01% improvement under the IID scenario.
The comparative results under the Non-IID scenario can be found in Table 2. It can be observed that under the Non-IID scenario, the proposed method BR-FEDBIDDER consistently outperforms existing methods in terms of FL model accuracy. In particular, on average, BR-FEDBIDDER achieves 2.87% higher FL model accuracy compared to the best performance achieved by baselines under the Non-IID data scenario. All these results demonstrate the effectiveness of our approach BR-FEDBIDDER in helping DCs optimize their key performance indicators and bidding strategies for DOs under the emerging AFL scenarios.
Lin and Bmub typically outperform Const and Rand due to their use of utility in the bidding process. However, Bmub is less effective than Lin due to its inclusion of randomness. Meanwhile, the more advanced methods BM, FBs, FBc, RLB, and BR-FEDBIDDER perform significantly better than the simpler approaches. This is largely due to the inclusion of winning probability estimation and the use of machine learning and reinforcement learning frameworks.
RLB and BR-FEDBIDDER both outperform BM, FBs, and FBc, likely due to their direct consideration of bid cost and available budget. While BM does consider market price distribution, it derives this distribution by marginalizing the prediction of the market price density of each bid request, which may lead to overfitting. In contrast, BR-FEDBIDDER obtains the market price distribution by the predefined winning function, which helps predict the expectation of the bid cost more accurately. While RLB uses dynamic programming to optimize its bidding process, its immediate reward setting can lead to blindly bidding for data samples without considering cost, which is not an issue with BR-FEDBIDDER.
Ablation Study. To compare the impact of using fullvolume bidding records versus only winning records, we compare the performance of the BR-FEDBIDDER approach with its variant, BR-FEDBIDDER-W. The key difference lies in the loss function used in the winning probability prediction module (Eq. (23)). BR-FEDBIDDER-W employs a modified version of the loss function considering only winning records, while BR-FEDBIDDER uses the entire bidding data. Table 3 displays the results of our experiments. We note that BRFEDBIDDER outperforms BR-FEDBIDDER-W consistently across all six datasets, both under IID and NONIID settings. This indicates that the auxiliary task allows training the model on the entire inference space, thus effectively mitigating the SSB issue, leading to significantly improved test accuracy.
6 Conclusions
This paper focuses on providing the optimal bidding strategies for data consumers in the AFL ecosystem to guide them in bidding for data owners in the competitive market so as to maximize their revenue and improve FL model accuracy by proposing a novel revenue-maximizing bidding strategy (BRFEDBIDDER). This paper first theoretically explores the relationships between Return on Investment (ROI) and utility, as well as bid cost and utility, and their impact on overall revenue, which highlights the challenge of revenue maximization when solely prioritizing ROI enhancement without taking into account the impact of bid cost. Based on these findings, BR-FEDBIDDER maximizes bid costs with a targeted ROI constraint. In this process, it incorporates an ancillary task of winning probability estimation into the framework to achieve bias-free learning by leveraging bid records from the entire volume of bid requests.
Acknowledgments
This research is supported, in part, by the National Research Foundation Singapore and DSO National Laboratories under the AI Singapore Programme (No. AISG2-RP-2020-019); Joint NTU-WeBank Research Centre on Fintech; and the RIE 2020 Advanced Manufacturing and Engineering (AME) Programmatic Fund (No. A20G8b0102), Singapore.
@inproceedings{ijcai2024p552,
title = {A Bias-Free Revenue-Maximizing Bidding Strategy for Data Consumers in Auction-based Federated Learning},
author = {Tang, Xiaoli and Yu, Han and Li, Zengxiang and Li, Xiaoxiao},
booktitle = {Proceedings of the Thirty-Third International Joint Conference on
Artificial Intelligence, {IJCAI-24}},
publisher = {International Joint Conferences on Artificial Intelligence Organization},
editor = {Kate Larson},
pages = {4991--4999},
year = {2024},
month = {8},
note = {Main Track},
doi = {10.24963/ijcai.2024/552},
url = {https://doi.org/10.24963/ijcai.2024/552},
}
TY - CPAPER TI - A Bias-Free Revenue-Maximizing Bidding Strategy for Data Consumers in Auction-based Federated Learning AU - Tang, Xiaoli AU - Yu, Han AU - Li, Zengxiang AU - Li, Xiaoxiao T2 - Proceedings of the Thirty-Third International Joint Conference on Artificial Intelligence, IJCAI-24 PB - International Joint Conferences on Artificial Intelligence Organization PY - 2024 SP - 4991 EP - 4999 DO - 10.24963/ijcai.2024/552 UR - https://doi.org/10.24963/ijcai.2024/552 ER -Copied
Machine-readable versions of this paper, for tools and AI agents: