Wireless Federated Learning over Resource-Constrained Networks: Digital versus Analog Transmissions
Abstract
To enable wireless federated learning (FL) in communication resource-constrained networks, two communication schemes, i.e., digital and analog ones, are effective solutions. In this paper, we quantitatively compare these two techniques, highlighting their essential differences as well as respectively suitable scenarios. We first examine both digital and analog transmission schemes, together with a unified and fair comparison framework under imbalanced device sampling, strict latency targets, and transmit power constraints. A universal convergence analysis under various imperfections is established for evaluating the performance of FL over wireless networks. These analytical results reveal that the fundamental difference between the digital and analog communications lies in whether communication and computation are jointly designed or not. The digital scheme decouples the communication design from FL computing tasks, making it difficult to support uplink transmission from massive devices with limited bandwidth and hence the performance is mainly communication-limited. In contrast, the analog communication allows over-the-air computation (AirComp) and achieves better spectrum utilization. However, the computation-oriented analog transmission reduces power efficiency, and its performance is sensitive to computation errors from imperfect channel state information (CSI). Furthermore, device sampling for both schemes are optimized and differences in sampling optimization are analyzed. Numerical results verify the theoretical analysis and affirm the superior performance of the sampling optimization.
Index Terms:
Federated learning (FL), digital communication, over-the-air computation (AirComp), convergence analysis.I Introduction
The dramatic development of data science has catalyzed significant advances in artificial intelligence (AI), which is driving innovation for anticipated sixth-generation (6G) mobile networks. The integration of AI and communication is envisioned to drive the shift from connected things to ubiquitous connected intelligence in wireless networks, supporting a large number of emerging intelligent applications [2, 3, 4, 5]. Nonetheless, traditional centralized learning paradigms depend on extensive data transmission and considerable computational resources at cloud servers, which is challenging to implement in wireless networks. To better embrace AI, edge learning (EL) is viewed as a promising distributed learning technique that harnesses massive data and computational capacity available in edge devices distributed across wireless networks [6, 7, 8]. Distinguishing it from the traditional separate design for computation and communication, EL integrates the two and achieves efficient utilization of resources and improves performance through learning task-oriented communication design.
In particular, a key EL paradigm, namely federated learning (FL), has garnered significant attention from both academic and industrial circles, primarily due to its communication-efficient and privacy-enhancing characteristics [9, 10]. In FL, distributed edge devices utilize local datasets to collaboratively train a shared learning model with the assistance of a central parameter server (PS). By exchanging model parameters instead of raw data, the PS iteratively updates the global model until convergence. FL scheme minimizes the amount of transmitted data, as well as hel** safeguard privacy and security. Recent studies have explored implementation of FL algorithms at wireless edge to support emerging AI applications [11, 12, 13, 14]. However, limited communication resources has posed a significant bottleneck to the performance of wireless FL [15, 16]. One particular concern regards the uplink transmission process, where numerous participating devices need to transmit local updates to the PS, leading to a substantial increase in communication overhead and transmission latency [17]. Hence, the development of efficient uplink transmission is crucial to enable wireless FL.
To support data transmission in wireless FL, digital communication schemes have been widely considered in recent works, where local updates are quantized into finite bits and then transmitted to the PS via traditional frequency division multiple access (FDMA) and time division multiple access (TDMA) schemes. At the receiver, the PS relies on channel coding for error detection and correction, before model aggregation using the received local updates. In [12] and [18], the authors characterized the impact of packet errors on the convergence of FL, which enabled a task-oriented communication resource allocation scheme. The influence of various finite-precision quantization schemes in uplink and downlink communications was considered in [19]. Building upon convergence analysis of the quantized FL, the quantization bits allocation was optimized in [20] and [21] to adapt channel diversity and requirements of the FL tasks. To further alleviate the communication bottleneck, one-bit quantization technique and reconfigurable intelligent surface (RIS) were used in [22] to reduce communication overhead and enhance communication reliability, respectively. Apart from resource allocation methods, modifications from the algorithmic perspective have been considered to combat unreliable transmissions. In [23], the authors proposed a user datagram protocol (UDP)-based robust training algorithm, which asymptotically achieved the same convergence rate as that with error-free communications. Moreover in [24], for replacing erroneous local updates, a global model reusing scheme, namely the GoMORE scheme, was devised to successfully mitigate the negative impacts of packet loss. Alternatively, another solution is to further squeeze the communication overhead, thus improving the convergence over resource-constrained networks. The model pruning in [25] was seen to be an effective way to compress the large-scale model into a smaller size, facilitating communication-efficient FL design.
In addition to these digital communication schemes, analog communication is an alternative communication-efficient way for deploying wireless FL. In particular, the local updates are amplitude-modulated and then simultaneously transmitted by reusing the available radio resource. Due to the superposition property of radio channels, the global model can be computed automatically over-the-air, which is therefore referred to as over-the-air computation (AirComp) [26]. Unlike the digital paradigm, analog communication pushes model aggregation from the PS to the air, which not only functionally but physically integrates the computation and communication. Benefiting from the over-the-air aggregation, the communication latency is substantially reduced and the spectrum utilization is much more efficient, leading to fast-convergent and communication-efficient FL. It was shown in [27] that the convergence rate of centralized learning remains approachable with this analog approach without power control and beamforming. Furthermore in [28], to combat deep fading, a novel truncated channel inversion scheme was proposed to exclude devices experiencing deep fades from the training process avoiding excessive energy consumption. Further insights into analog aggregation schemes were also discussed in the context of fundamental trade-offs between communication and learning. Besides, the impact of over-the-air aggregation errors on optimality gap was analyzed in [29] and [30] with power control optimization. Furthermore, the authors in [31] proposed an AirComp-based adaptive reweighing scheme for the aggregation, and jointly considered the power control and device selection deign based on the derived optimality gap. To combat the additive noise, robust FL training methods were proposed in [32] for both the expectation-based and the worst-case noise models. Considering multi-antenna scenarios, the beamforming design at the receiver was optimized by solving a sparse and low-rank optimization problem in [33]. In practice, considering the lack of perfect channel state information (CSI) for accurate power control, the work [34] investigated the impact of CSI uncertainty at the transmitter on FL convergence and revealed that CSI imperfection plays an key factor affecting the AirComp performance and convergence.
As mentioned above, by incorporating learning task-oriented resource allocation, both digital and analog transmissions are effective ways to fulfill the communication requirements of wireless FL [35, 36, 37]. In traditional communication for data transmission, digital communication schemes have been proven not only in theory but also in practice as dominantly outperforming analog communication techniques in almost all cases of interest. In communications for computation tasks, however, analog communication has shown to be exceptionally effective in some cases of resource-constrained networks [38]. Hence, it is of interest to comprehensively compare digital and analog transmissions for wireless FL. Several recent studies have compared the two communication paradigms from some specific perspectives, including communication latency [28, 39] and convergence performance [40, 41]. However, to the best of our knowledge, there is a lack of literature that presents a comprehensive and quantitative comparison between the two fundamental communication paradigms, especially under practical constraints. Also, there have been few attempts to elucidate the fundamental differences between digital and analog transmissions in the context of FL, which is crucial for its deployment and design.
Against this background, in this paper, we conduct a theoretical comparison between the digital and analog transmission schemes under practical constraints. The main contributions of this paper are summarized as follows.
-
•
We propose a unified framework for digital and analog transmissions in wireless FL, and characterize the model aggregation distortion caused by wireless transmission schemes. Using this framework, a fair comparison is conducted under the consideration of a stringent transmission delay target and two types of transmit power budgets. We exploit optimality gap, defined by the gap between the optimal and actually achieved loss function value, to characterize the convergence behavior and establish a stringent upper bound of the optimality gap for precise analysis and optimization in the digital/analog transmission enabled wireless FL. It offers a precise characterization of the influence of wireless transmission imperfections on convergence in closed-form.
-
•
Analytical results reveal that the digital transmission is hard to achieve satisfactory performance especially with limited radio resources due to orthogonal access and decoupled design. In contrast, the analog scheme exhibits a performance gain in terms of the optimality gap of the order of with the increasing number of participating devices, , and thereby achieving a higher level of efficiency in spectrum utilization. However, the introduction of computation goals in the analog communication process results in less efficient transmit power utilization, and the presence of CSI uncertainties inevitably comes with computational distortion, thus enlarging the optimality gap by the order of with a decreasing level of channel estimation accuracy .
-
•
Based on the derived optimality gap, we formulate an inclusion probability optimization problem for effective device sampling in wirless FL. The optimization problems for both digital and analog cases are optimally solved by checking the Karush-Kuhn-Tucker (KKT) conditions and exploiting the Dinkelbach algorithm, respectively. Through the examination of optimal solutions, we identify the essential differences underlying the device sampling optimization for digital and analog transmissions.
Extensive numerical simulations are conducted to validate the derived analytical observations and the proposed sampling optimization. In particular, it is observed that the digital scheme has better power utilization, while the analog transmission is more spectrum-efficient.
The rest of this paper is organized as follows. In Section \@slowromancapii@, we describe the typical FL algorithm, with details of digital and analog transmissions, and propose a fair comparison framework. Section \@slowromancapiii@ provides some preliminaries for the convergence analysis. In Section \@slowromancapiv@, we analyze the convergence performance under different transmission schemes and offer engineering insights. Then, in Section \@slowromancapv@, we optimize the inclusion probabilities for both the digital and analog schemes. Simulation results and conclusions are given in Sections \@slowromancapvi@ and \@slowromancapvii@, respectively.
Notation: Boldface lowercase (uppercase) letters represent vectors (matrices). The set of all real numbers is denoted by . Superscripts and stand for the transpose and conjugate operations, respectively. The operator returns the real part of the input complex number. The operator takes the Euclidean norm of vectors. A circularly symmetric complex Gaussian distribution is denoted by , and is the expectation operation.
II System Model and Communication Framework
We consider a typical wireless FL system as shown in Fig. 1, where distributed devices are coordinated by a central PS to perform FL. The training procedure and transmission model are elaborated in the sequel.
![Refer to caption](x1.png)
II-A Federated Learning Model
In FL, the distributed devices collaboratively train a shared machine learning model via local computing based on their local datasets and information exchange with the PS. Let denote the local dataset owned by the -th device, which contains training samples. The goal of the FL algorithm is to find the optimal -dimensional model parameter vector, denoted by , to minimize the global loss function , i.e.,
(1) |
where , represents the aggregation weight for the -th user, and is the local loss function at device defined as
(2) |
where denotes a training sample selected from , and represents the sample-wise loss function with respect to . Due to the heterogeneity of the system, we note that local datasets at distinct devices are usually non-independent and non-identically distributed (non-IID), and the optimal model parameters in (II-A) are not necessarily the optimal for local datasets. Let denote the locally optimal model at device , i.e., . It is usually different from the globally optimal unless the local dataset experiences the same distribution as the whole data population.
To effectively handle the optimization problem in (II-A), an FL algorithm performs the model training in an iterative manner. Specifically, the -th round of the FL algorithm consists of the following steps.
-
1)
Model Broadcasting: The PS broadcasts the latest global model to al devices.
-
2)
Local Computing: After receiving , each device exploits its local dataset to compute the local gradient as
(3) -
3)
Local Update Uploading: Each device reports its local gradient to the PS.
-
4)
Model Aggregation: Upon receiving all local gradients, the PS updates the global model according to
(4) where is the learning rate and is given by
(5)
The above steps iterate until a convergence condition is met.
Considering the potentially massive number of devices and limited resources in practice, only a subset of devices can participate in each round of the training. Let denote the set of activated devices selected in the -th communication round and be the number of participating devices per round. Due to imbalanced dataset sizes and data heterogeneity, we assume that the PS performs non-uniform device sampling without replacement to select the participating devices per round. Specifically, the devices are randomly selected one by one from the remaining unselected device set. Once the number of selected devices reaches , the sampling process terminates. Denote the inclusion probability of the device as , which represents the probability of device being sampled per round and satisfies , , and . Due to the non-IID nature of the data, misaligned inclusion probability may bias the global model away from the local optimum, thereby decelerating the convergence and causing performance loss. Hence, in the following sections, we focus on the performance evaluation under fixed inclusion probabilities and characterize the impact of device sampling for wireless FL.
Also, in wireless FL, the parameter transmission in Steps 1) and 3) relies on wireless communication between the PS and devices, which comes with additional imperfection in the model training procedure. Considering a sufficient power budget at the PS, the downlink transmission is usually assumed error-free [12]. Otherwise, for uplink transmission with limited communication resources, additional errors are inevitable. Efficient transmission and resource allocation schemes need to be designed to alleviate this impact of wireless environment.
II-B Uplink Transmission Method
We rely on the wireless uplink transmission to provide an estimation of the actual gradient in (5). Assume that the total uplink bandwidth can be divided into up to subbands, which supports orthogonal access for devices. Without loss of generality, a frequency non-selective block fading channel model is adopted, where the wireless channels remain unchanged within a communication round. Let be the channel between the -th device and the PS, where denotes the distance between the PS and device , represents the large-scale path loss exponent, and represents the small-scale fading of the channel. Assume that the channels are independent Rayleigh fadings, i.e., . In practice, perfect estimation of the small-scale fading of the channel is usually not available. Let denote the estimated channel at device . Then, we model the CSI imperfection of the small-scale fading as
(6) |
where is the correlation coefficient between and to reflect the level of channel estimation accuracy, and is the channel estimation error independent of . In the following, we introduce two typical uplink transmission schemes, i.e., digital and analog transmissions.
II-B1 Digital Transmission Model
In the digital transmission, the selected devices first quantize their local updates into a finite number of bits and then simultaneously transmit the quantized local updates to the PS. Specifically, we assume that the local update is quantized by the stochastic quantization method in [20]. Denote the maximum and the minimum values of the modulus among all parameters in by and , respectively. Then, the interval is divided evenly into quantization intervals. The uniformly distributed knobs are denoted by for . Given , the quantization function is expressed as
(9) |
where represents the signum function and “w.p.” represents “with probability.” Exploiting the quantization function in (9), the local update is quantized as , which is transmitted to the PS. Note that the exact value of and need to be transmitted to the PS with sufficient precision to support effective recovery. Hence, the total number of bits needed for transmitting amounts to
(10) |
where is the number of bits used to represent and , and the additional one bit is the sign bit.
During the uplink FL parameter report, transmission errors are inevitable due to the channel dynamics and limited communication resources. Without loss of generality, we adopt the typical FDMA technique as an example. Assume that and hence each device can occupy different subbands equally to avoid interference with each other.111We generally assume orthogonal access between different devices and refrain from specifying the particular multiple access design. Hence, the following analysis can be safely extended to orthogonal access scenarios like TDMA and orthogonal frequency division multiple access (OFDMA). Then, the channel capacity of device can be evaluated as
(11) |
where is the bandwidth allocated to device and it is set to , is the transmit power at device , and is the noise power density.
The transmission delay under the digital transmission is primarily influenced by stragglers, which refer to devices with poor channel conditions. To avoid the uncontrolled severe delay brought by stragglers, we assume that all the devices transmit the local updates at a fixed rate rather than a dynamic one based on instantaneous signal-to-noise ratio (SNR) levels. Hence, the use of a fixed-rate transmission acts as a truncation mechanism for stragglers. Additionally, for devices experiencing favorable channel conditions, it is more beneficial to transmit at a lower rate with enhanced transmission reliability. The target transmission rate is denoted by , where is a chosen constant. According to [12], the transmission is assumed error-free if the transmission rate is no larger than the channel capacity. Hence, the probability of successful transmission at device is calculated as
(12) |
At the PS, a cyclic redundancy check (CRC) mechanism is applied to check the detected data such that erroneous local updates can be excluded from the model aggregation. Finally, the obtained estimate of the desired gradient in (5) is given by
(13) |
where is an indicator variable for the device selection, and represents distortion brought by packet loss. To be concrete, is 1 if and otherwise is 0. Considering the definition of the inclusion probability, we have , which decreases the desired expected aggregation coefficient for unbiased gradient estimation. In order to compensate for the impact of partial participation, we multiply the coefficient in (13), such that . Analogously, the distortion is characterized by the probability in (12) as
(16) |
to ensure . With the gradient estimate in (13), the global model updated at the -th round equals to
(17) |
where denotes the model obtained at the previous round.
II-B2 Analog Transmission Model
In the analog transmission with AirComp, selected devices simultaneously upload the uncoded analog signals of local gradients to the PS by fully reusing the time-frequency resource. A weighted summation of the local updates in (5) can be achieved by exploiting channel pre-equalization and the waveform superposition nature of the wireless channel. In this study, we consider that the total bandwidth is constrained for fair comparison and all subbands are utilized for the transmission of identical parameters. This is because the uncoded nature of the analog transmission diminishes its robustness, rendering it more vulnerable to interference and even the malicious attacks.222The derived results directly extend to the case of dividing bandwidth for distinct parameter transmission in broadband scenarios[28]. Specifically, the received signal at the PS is expressed as
(18) |
where is the pre-processing factor at device , and is additive white Gaussian noise following . To accurately estimate the desired gradient in (5), the pre-processing factor should be adapted to the channel coefficient . Unlike the digital transmission, CSI is needed at the transmitter for the analog transmission. Channel pre-equalization is performed based on the CSI available at each device. For simplicity, we adopt the typical truncated channel inversion scheme to combat deep fades[28]. It is expressed as
(21) |
where is a predetermined power-cutoff threshold, is a scaling factor for ensuring the transmit power constraint, and compensation coefficient is selected to alleviate the impact of imperfect CSI [34]. Through the pre-processing in (21), we aim to eliminate the influence of the uneven channel fading , and the inclusion probability , thereby ensuring the unbiased gradient estimation.
At the receiver, the PS scales the real part of in (18) with and obtain an estimate of the actual gradient in (5). It yields
(22) |
where is the equivalent noise, and denotes the distortion brought by the analog transmission with imperfect CSI. It follows
(25) |
Similarly, the global model at the -th round under the analog transmission is updated as
(26) |
II-C A Unified Framework for Wireless FL Comparison
To minimize the optimality gap brought by imperfect uplink transmission, the overall FL task-oriented optimization over the wireless networks can be formulated as
minimize | ||||
subject to | ||||
(27) |
where the expectation is taken over channel dynamics, represents uplink transmission delay per round, and denotes the maximum transmission delay target and the transmit power unit, respectively. Constraint and respectively represent the maximum transmission delay and maximum transmit power constraint in practice. Apart from the maximum power budget, another typical transmit power constraint is the average power budget [28], i.e.,
(28) |
where denotes the average power budget and limits the energy consumption during the uplink transmission process.
Paradigms | Gradient estimation | Transmission delay | Power budget |
---|---|---|---|
Digital | (13) | (29) | (31) |
Analog | (22) | (32) | (33), (34) |
For fair comparison between the two transmission paradigms, we measure the achievable objective value of the problem in (II-C) under the same transmission delay target and transmit power budget. Specific constraints for the two transmission paradigms are listed as follows, summarized in Table I.
For digital transmission, the transmission delay per communication round is calculated as
(29) |
where the evaluation holds with a sufficiently large model size . Hence, constraint is reformulated as
(30) |
For constraint , due to its interference-free characteristic, full power transmission is optimal and hence the constraint is reformulated by
(31) |
Also, with the average transmit power budget, we assume invariant transmit power over different communication rounds and have .
For analog transmission, according to [39, Eq. (16)], the per-round delay follows
(32) |
which is a constant. For feasibility, we assume that the target cannot be smaller than . The maximum power constraint is rewritten as
(33) |
for the analog transmission. Unlike the digital transmission, it is impossible to fully utilize the maximum power in analog transmission due to the need for channel pre-equalization. On the other hand, the average power constraint follows
(34) |
where the expectation is taken over the wireless channel dynamics and different communication rounds.
III Preliminaries
To pave the way for performance analysis, this section provides necessary assumptions and lemmas about the learning algorithms and the transmission paradigms, which will be useful in the next section.
III-A Assumptions for Learning Algorithms
To begin with, we make several common assumptions on the loss functions, which are widely used in FL studies like [12, 29, 42].
Assumption 1: The local loss functions are -strongly convex for all devices, that is
(35) |
Assumption 2: The local loss functions are differentiable and have -Lipschitz gradients, which follows
(36) |
and it is equivalent to
(37) |
Assumption 3: In most practical applications, it is safe to assume that the sample-wise gradient is always upper bounded by a finite constant , i.e.,
(38) |
Assumption 4: The distance between the locally optimal model, , and the globally optimal model, , is uniformly bounded by a finite constant , i.e.,
(39) |
III-B Preliminary Lemmas
We present lemmas regarding the strong convexity and Lipschitz smooth properties of the global loss function.
Lemma 1:
With -strongly convex and -smooth local loss functions, the global loss function is also -strongly convex and -smooth.
Proof:
We then provide the following lemma regarding the imperfection in digital and analog transmission paradigms.
Lemma 2:
Proof:
Please refer to Appendix A.
Although both the digital and analog transmissions achieve unbiased gradient estimations, there are fundamental differences in the distortion between the two paradigms. For the digital transmission, the distortion mainly lies in the gradients themselves, i.e., gradient quantization errors. On the other hand, due to the integration of communication and computation in AirComp, the analog transmission additionally suffers from distortion in coefficient aggregation, i.e., computation error, which is due to the CSI imperfection. This essential difference further discriminates the performances of digital and analog transmissions, which are elaborated in the next section.
IV Comparison with Convergence Analysis
In this section, we analyze the convergence performance under the digital and analog transmissions with the practical constraints for wireless FL. Based on the derived results, we further conduct quantitative comparisons between the two paradigms from various perspectives of view.
IV-A Convergence under the Maximum Power Budget
We characterize the convergence performance under different transmission paradigms in the following theorems.
(40) | ||||
(41) |
Theorem 1 (Digital Transmission):
Proof:
Please refer to Appendix B.
Theorem 2 (Analog Transmission):
For a fixed learning rate satisfying , the optimality gap of the distributed gradient update in the -th iteration of the analog transmission is equal to (41) at the top of the next page, where , , and .
Proof:
Please refer to Appendix C.
From Theorems 1-2, we find that the convergence rate mainly depends on the choice of the learning rate , while the imperfections in transmission also have a certain impact. We conclude the following immediate observations on the convergence rate.
Remark 1:
As observed in (40) and (41), the convergence performace of an FL algorithm is negatively related to for digital transmission and to for analog transmission. We refer to and as the virtual sum weight for the digital and analog transmissions, respectively, which reflects the degree of hindrance to the convergence imposed by unequal sampling and vulnerable wireless communication. Under the ideal case, with full device participation and no transmission outage, the virtual sum weight equals to 1, otherwise it is amplified by the imperfect characteristics. It is interesting to note that, for devices with more data samples, i.e., larger , the impact of imperfections is exaggerated.
Remark 2:
Comparing and , it can be seen that the vulnerability of digital transmission introduces additional heterogeneity, i.e., varying , which does not exist in the analog paradigm. This is because outage probability in the digital case is determined by channel conditions and varying across different devices. On the other hand, due to the uniform truncation threshold, all participating devices enjoy the same truncation probability in the analog transmission. Hence, in design of inclusion probabilities for the digital case, we need to adapt the inclusion probabilities to both dataset size and channel condition. By contrast, in the case of analog transmission, only the heterogeneity of the dataset size needs to be considered.
According to Theorems 1-2, we are ready to derive the optimality gap after convergence for further evaluation in the following corollary, which reflects the ultimately achievable performance of the wireless FL.
Corollary 1:
With sufficient iterations, the optimality gap achieved by digital and analog transmissions, respectively, converges to
(42) | ||||
(43) |
Proof:
Consider the digital transmission scenario with a sufficient number of iterations. We have
(44) |
where the inequality is obtained through Theorem 1 and the equality in (a) is due to the fact that , i.e., . Hence, the achieved optimality gap at convergence is bounded by . As for the analog transmission, the proof is almost the same and is omitted here for simplity.
From Corollary 1, we further compare the two typical paradigms from the following perspectives and conclude insightful remarks that are instructive for the deployment of FL in wireless networks. As a summary, we list main comparison results in Table II. For the sake of simplicity in analysis, without loss of generality, we drop the unbalance of the datasets and assume uniform inclusion probabilities, i.e., , and , , which does not cause any essential changes. Also we set that . Note that the learning rate is assumed to be sufficiently small and hence the convergence rate remains the same for all cases.
Paradigms | Transmit power budget, | Device number, | Imperfect CSI, | |
Low SNR | High SNR | |||
Digital | / | |||
Analog |
-
•
* The upward arrow indicates amplification at a certain order, while the downward arrow has the opposite meaning. The horizontal arrow indicates that it ultimately tends towards a fixed value.
IV-A1 Impact of Transmit Power
At low SNR levels, the achievable optimality gap under the digital transmission, , vanishes as with the maximum transmit power budget , where . At high SNR regime, i.e., , the successful transmission probability , and tends to
(45) |
On the other hand, the decay rate for is equal to with low SNR values and the high SNR-limiting value is
(46) |
where .
Remark 3:
As SNR increases, the optimality gap for the analog case mainly comes from the non-IID datasets while the impact of the noise asymptotically diminishes. For the digital case, however, quantization errors additionally impose an impact. Under the analog transmission, the negative impact of non-IID datasets is enlarged due to imperfect AirComp. Imperfect CSI results in mismatched channel inversion in AirComp, rendering perfect computation of weighted sum impossible. Moreover, the performance degradation brought by imperfect CSI in the analog transmission cannot be mitigated by occupying more resources. Conversely, in the digital transmission, the convergence performance can be improved by occupying additional resources for increasing the number of quantization bits.
IV-A2 Impact of Device Number
With the increasing number of participating devices, , the virtual sum rate for the analog transmission, , decreases at a rate of , i.e., a faster convergence rate is achieved. As for the optimality gap, the impact of non-IID datasets asymptotically dominates and the decay rate is equal to . Due to the involvement of more devices, a more accurate global gradient is obtained at the PS, which in turn facilitates the FL convergence and leads to better performance. Meanwhile, since different devices involved in the AirComp share the same time-frequency resource, an increase in access devices causes no deterioration of the AirComp performance, fully capturing the performance gain from more participating devices.
On the other hand, for the digital case, convergence performance does not necessarily monotonically change with . Although more participating devices do bring performance gains, it also leads to a significant deterioration of the transmission performance considering that limited communication resources are divided among additional users. Thus the convergence is compromised between communication reliability and the computation accuracy for wireless FL. Specifically, the optimality gap, , enlarges with a rate of with sufficiently large , where and .
Remark 4:
Benefiting from the characteristics of AirComp, more participating devices in the analog transmission always lead to performance improvement regardless of other parameters. Hence, allowing all active devices to participate in the FL training is the best choice for analog transmission. By contrast, in the digital transmission, it is necessary to seek a balance between the transmission performance and diversity gain through an optimization of .
IV-A3 Impact of Imperfect CSI
The imperfect CSI at the transmitter only affects the performance of analog transmission, which deteriorates at the order of . Due to imperfect CSI, the aggregation computation and the truncation decision in AirComp are contaminated, thus leading to a mismatch in the model aggregation and the impact of noise amplification.
Remark 5:
After incorporating computation capabilities into the analog case, the emergence of computation error as a new source of error has positioned computational accuracy as a crucial factor affecting the convergence performance. It is concluded that CSI is a key factor affecting the performance gain brought by AirComp. Moreover, the truncation threshold should be optimized to adapt different levels of channel estimation accuracy. It can be effectively solved via bisection search in [34].
IV-A4 Impact of the Number of Quantization Bits
In the digital transmission, the number of quantization bits, , also influences the FL performance in the following implicit ways. By selecting the minimum feasible in (30), the achievable optimality gap is rewritten as
(47) |
where the approximation is obtained in region of . It is found that as increases, tends to first decrease and then increase. This is due to the diminishing quantization error term with an increasing quantization accuracy and finally is dominated by the impact of packet loss. Therefore, it is necessary to optimize of the integer variable to pursue better convergence performance, which can be solved by a low-complexity exhaustive search method.
IV-B Convergence Analysis under the Average Power Budget
We consider the convergence with the average transmit power budget. For the digital transmission, by replacing with , we derive the similar results as Theorem 1 and is omitted here due to page limit. As for the analog transmission, we have the following corollary.
Corollary 2:
For a fixed learning rate satisfying , the optimality gap of the distributed gradient update in the -th iteration under the analog transmission follows
(48) |
where . The optimality gap with sufficient iterations follows
(49) |
Proof:
please refer to Appendix D.
Remark 6:
It is worth noting that when . Compared with the maximum transmit power budget, a smaller optimality gap for the analog transmission is achieved with the average power budget. Due to the need for channel alignment in AirComp, the performance is dominantly limited by the device with the worst channel condition. Furthermore, the strict peak power constraint amplifies the impact of worst-case channel conditions, resulting in looser convergence performance compared to the long-term constraint.
To summarize, while the analog AirComp improves the spectrum utilization compared to the digital paradigm, it faces challenges in fully utilizing the power resource, particularly with strict peak power constraints. Conversely, orthogonal access in digital transmission is not suitable for scenarios with massive access due to the limitations in spectrum resources.
IV-C Discussions on Scenarios with Advanced System Designs
To facilitate performance analysis, we introduce assumptions regarding the system design, including multiple access, parameter quantization, and power control methods. Subsequently, we delve into the implications of advanced system designs on the FL performance and comparison.
In the digital transmission, the FL performance can primarily be improved from two aspects, namely enhancing transmission reliability and optimizing resource utilization. Specifically, advanced transmissions strategies help minimize transmission errors and packet losses due to channel fading. Furthermore, if other resource allocation methods, such as the model compression design and device scheduling strategies, are exploited toward the FL tasks, they prioritize crucial parameter/device transmissions and thus lifting the resource utilization. On the other hand, in the analog transmission, the FL performance through AirComp is primarily influenced by the over-the-air computational accuracy. Optimized transceiver and power control designs help mitigate the negative impact of channel fading on the FL performance.
While further optimization of system designs enhances performance, it is essential to note that the performance limits for the digital and analog transmissions remains unchanged. As observed in the above analytical results, in the digital transmission paradigm, due to the decoupling of the communication and computation processes, the number of bits that can be accurately transmitted with the limited resources is determined, which places an upper bound of the FL performance. In contrast, within the analog transmissions, the receiver does not aim to recover information from individual sources but instead prioritizes the precision of computation results derived from the over-the-air superimposed signals, thereby making computational accuracy a decisive role. Hence, the performance limit of the analog transmission is contingent upon the channel estimation accuracy and additive noise level.
V Device Sampling Optimization
Based on the derived results in Section IV, we are able to further establish an optimization design of the device sampling for the wireless FL to improve the convergence.
V-A Digital Transmission
By direct inspection of (42), the optimality gap monotonically decreases with a decreasing virtual sum weight. Hence, the device sampling optimization problem with the digital transmission is formulated as
subject to | (50) |
which is a convex problem. By exploiting the KKT conditions, we obtain the optimal inclusion probability as
(51) |
where is the Lagrangian multiplier and it is selected to satisfy . Note that the value of varies monotonically with and thus we can rely on a bisection-based search method [13] to get the optimal solution of problem (V-A).
Remark 7:
The optimal inclusion probability is positively correlated with the local dataset size while it behaves conversely correlated with the successful transmission probability. In other words, a device with a larger dataset is deemed more important for model training, thereby deserving a sampling bias. Conversely, devices with lower successful transmission probabilities contribute less to the model training process, requiring more frequent sampling to compensate. Thus, the goal of our inclusion probability optimization is to address the imbalances in the dataset size, and the heterogeneity introduced by uneven channel fading. It ensures fair and effective participation among diverse devices.
Moreover, note that the influence of quantization error and data heterogeneity are equally amplified by . It indicates that the optimization of inclusion probabilities cannot adequately adapt to varying local data distributions.
![Refer to caption](x2.png)
![Refer to caption](x3.png)
![Refer to caption](x4.png)
![Refer to caption](x5.png)
V-B Analog Transmission
As for the analog transmission, the device sampling optimization is expressed as
subject to | (52) |
Note that under the average transmit power budget, (49) only differs from the objective value in the constant term, and hence we will not discuss it separately. Considering the intractable fractional form of the objective function in (V-B), we rely on the well-known Dinkelbach algorithm for reformulation [43, 44]. According to the definition of and in (43), it is easy to check that the denominator of the objective function in (V-B) is concave and the numerator is convex. Hence, the iterative Dinkelbach algorithm guarantees to converge to the global optimum of (V-B). Concretely, in the -th iteration, we reformulate the problem in (V-B) as
subject to | (53) |
where is a constant determined in the previous round. Note that the problem in (V-B) is convex and thus can be solved by numerical convex program solvers, e.g., CVX tools [45]. After obtaining the optimal of the -th subproblem in (V-B), the auxiliary constant is updated as
(54) |
Iterating the above steps until convergence, we obtain the optimal of the problem in (V-B).
Remark 8:
Unlike the digital transmission case, the device sampling optimization is committed to seeking a trade-off between the equivalent noise power and virtual sum weight , and the parameter functions as a weighting factor to facilitate the optimal trade-off. At high SNR regimes or with extremely uneven local data distributions, the noise term is comparably ignorable and hence the optimality gap is dominated by . Hence, the optimization of is isolated from specific channel conditions and only needs to match the size of local datasets.
VI Numerical Results
In this section, we provide simulation results to verify the performance analysis and the inclusion probability optimization. We deploy edge devices uniformly distributed in a square area with radius m and a PS at the center of the square area. The most popular MNIST dataset and CIFAR-10 dataset are exploited for the FL performance evaluation. The MNIST dataset contains 10 classes of handwritten digits ranging from 0 to 9 and we train a multi-layer perceptron (MLP) with parameters via the wireless FL algorithm for classification purposes. Moreover, the CIFAR-10 dataset includes 10 classes with labels 0-9 and we train a convolutional neural network (CNN) with parameters. The trained CNN contains two convolutional layers and three fully connected layers. Max pooling operation is conducted following each convolutional layer and the activation function is ReLU. Different edge devices own different data samples, and each local dataset has up to two types of data samples to capture the non-IID characteristic.
Unless otherwise specified, the other parameters are set as: the number of participating devices , the bandwidth, MHz, the path loss exponent, , the noise power dBm/Hz, the maximum transmit power budget, dB, the number of quantization bits, , the truncation threshold, , the delay target is equal to in (32), and the learning rate . We set and , which fall within the existing typical range of values in [46, 47]. Additionally, the parameter , serving as an upper bound of , is estimated through simulation tests.
VI-A Convergence Performance
![Refer to caption](x6.png)
![Refer to caption](x7.png)
In Figs. 2 and 3, we depict the convergence performance for the digital and analog transmission. As shown in Fig. 2, we observe that the convergence rate and optimality gap under digital transmission exhibit a negative correlation with the virtual sum weight, aligning with our theoretical analysis. Moreover, the convergence behavior remains consistent with the analytical results despite the complexity of the classification task, thereby validating the accuracy of the theoretical analysis.
![Refer to caption](x8.png)
For the analog case depicted in Fig. 3, consistent with the analytical findings, we notice that the convergence rate is negatively correlated with the virtual sum weight , which is determined by and . On the other hand, transmit power only affects the achievable optimality gap after convergence. This is because changes in transmit power only affect the equivalent power of the additive noise. Additionally, modifications in and affect the distortion of the aggregation coefficient, which in turn influences the computation error. Furthermore, the increased complexity of FL tasks renders fluctuations in the performance curve more sensitive to noise. Consequently, in the analog communication, the superimposed white Gaussian noise is significantly severer than quantization errors observed in the digital transmission, thus leading to more pronounced fluctuations in convergence performance. It implies that for more complex learning tasks, it becomes imperative to further reduce the variance of gradient estimation to mitigate excessive fluctuations and their adverse impacts on convergence.
VI-B Impact of Transmit Power Budget
In Fig. 4, we show the test accuracy versus different transmit power budgets. It is observed that the digital transmission scheme outperforms the analog scheme, particularly with high SNR levels. In such cases, employing more quantization bits yields the best performance. Conversely, for low SNR levels, reducing the quantization bits leads to marginal performance loss, highlighting the flexibility of the digital schemes by selecting different quantization accuracies. On the other hand, the analog scheme faces significant performance limitations, particularly with the maximum transmit power budget and less CSI, due to the stringent requirements of channel inversion. Therefore, in terms of power utilization, the digital scheme is more efficient than the analog counterpart.
![Refer to caption](x9.png)
![Refer to caption](x10.png)
![Refer to caption](x11.png)
![Refer to caption](x12.png)
VI-C Impact of Participating Device Numbers
Fig. 5 illustrates the test accuracy versus the number of participating devices. We note that for the analog transmission, the test accuracy gradually increases as increases. In contrast, although the performance in digital case may be improved initially, it eventually decline rapidly as each device can only occupy a limited amount of resources, making it unable to support high-rate transmission. Consequently, the results suggest that for digital transmission, the selection of requires further optimization according to the actual conditions, with a preference for fewer devices.
VI-D Impact of Channel Estimation Accuracy
In Fig. 6, we present the impact of channel estimation accuracy on the analog case. It is evident that better performance can be achieved with more accurate CSI. Additionally, we observe that smaller truncation thresholds are more suitable for larger , while larger truncation thresholds are preferred for smaller . This is because higher CSI uncertainties have a significant impact on truncation choices, necessitating looser truncation conditions to reduce incorrect choices.
VI-E Impact of Different Inclusion Probabilities
In Figs. 7 and 8, we depict the convergence performance with different inclusion probabilities. For comparison, we consider the following baselines for comparison. For the sake of fairness, all schemes refrain from utilizing specific information on instantaneous CSI and gradients.
-
•
Uniform [49]: The inclusion probabilities are uniformly assigned the same value, i.e., .
-
•
Learning-oriented [51]: From the perspective of learning algorithms, the probability is set to be proportional to the size of the local datasets, i.e., .
-
•
Channel-aware: From the perspective of wireless channels, the probability is set to be proportional to the large-scale path loss, i.e., .
-
•
Min-distortion [52]: To minimize the communication distortion in the analog transmission, the probability is set to be proportional to by considering both the local datasets and channel conditions.
As shown in Fig. 7, the proposed method consistently outperforms the aforementioned baseline methods across all levels. The first two baselines neglect the influence of the wireless transmission process, resulting in performance degradation. The sampling method based on channel conditions tends to select devices with better channels, effectively reducing packet loss rates and yielding significant performance improvements. However, due to its oversight of imbalanced size of local datasets, its final performance remains inferior to our proposed method. The fourth baseline, tailored for the analog transmission scenarios, partially accounts for the impact of local datasets and wireless channels but lacks optimality, leading to limited performance gains.
As for the analog transmission case in Fig. 8, we note that although the performance of the optimized probability is superior, the performance gain compared to the other baselines is not significant. This limit arises from the reliance on constants , , and in the optimization problem (V-B), which are challenging to determine accurately in practice, thus affecting the final performance. Similarly, akin to the digital transmission, the sampling method based on channel conditions effectively mitigates the negative impact of the imperfect wireless transmission. However, its disregard for data characteristics results in suboptimal performance, particularly in the complex classification tasks on CIFAR-10 dataset, leading to significant performance fluctuations. Furthermore, the baseline method of minimizing computational distortion overlooks the impact of data heterogeneity, thus impeding its ability to achieve satisfactory performance.
VII Conclusion
In this paper, we have provided a detailed comparison between digital and analog transmission enabled wireless FL. To this end, we considered general transmission designs for both schemes and conducted a fair comparison between them. Then, we analyzed the convergence behavior of wireless FL in terms of the convergence rate and optimality gap under digital and analog cases, and compared the convergence performance from multiple perspectives. It was found that digital transmission is more suitable for scenarios with sufficient radio resources and CSI uncertainties. On the other hand, analog transmission is suitable when their are massive numbers of participating devices. Next, we addressed sampling optimization for both cases, and further developed insights for optimization, which ars useful for practical deployment. Finally, experimental results illuminated the analytical results and the sampling strategies. Additionally, an explicit and precise characterization of data heterogeneity and targeted system designs with theoretical guarantees should be of our interest in the future work.
Appendix A Proof of Lemma 2
For the digital case, according to [19, Lemma 5], we first conclude that the quantized gradients is unbiased, i.e.,
(55) |
Combining with the fact that in (16), we have
(56) |
where (a) comes from the definition of and the independence among device sampling, small-scale fadings and stochastic quantization.
Appendix B Proof of Theorem 1
To begin with, we define an auxiliary variable as
(57) |
which represents the model obtained at -th round via ideal communication and full participation. Then, by exploiting Assumption 2 and the fact that , we have
(58) |
where (a) is due to the fact that is an unbiased estimate of . For the term , it is bounded by
(59) |
where (a) is because , (b) exploits the convexity of , and (c) is due to the fact that . According to [20], the variance of quantization error is bounded as
(60) |
where is defined as a uniform upper bound of , . Then, is bounded by
(61) |
where (a) uses and . Next, by expanding the square term, we reformulate as
(62) |
Then for , we have
(63) |
where the inequality in (a) is due to Assumption 1, and (b) is due to the fact that for .
Combining all the results in (B)-(B), it yields
(64) |
We further rewrite the second term in the right hand side (RHS) of (B) as
(65) |
where (a) comes from , (b) exploits Assumption 2, and (c) uses Assumption 3 and the inequality . By defining , we conclude that
(66) |
Plugging (B) into (B), we obtain the convergence result and complete the proof.
Appendix C Proof of Theorem 2
As for the analog transmission, the main difference from the digital transmission lies in the term in (B). With the analog case, is expressed as
(67) |
For the equivalent noise term, recalling that , we first derive the scaling factor . Constrained by the transmit power budget in (33), the scaling factor must satisfy
(68) |
Based on Assumption 3 and the definition in (3), we can conclude that
(69) |
Note that for all the activated devices, we have . Hence, we select the feasible as
(70) |
Then, we have
(71) |
Next, the variance of the coefficient distortion, , is calculated as
(72) |
where (a) exploits the independence of and , (b) is due to [34, Eq. (25)]. Substituting (C) into (C) and combining the results in (B) and (B), we complete the proof.
Appendix D Proof of Corollary 2
References
- [1] J. Yao, W. Xu, Z. Yang, X. You, M. Bennis, and H. V. Poor, “Digital versus analog transmissions for federated learning over wireless networks,” accepted by Proc. IEEE Int. Conf. Commun. (ICC), 2024. Available: https://arxiv.longhoe.net/abs/2402.09657.
- [2] Y. C. Eldar, A. Goldsmith, D. Gundüz, and H. V. Poor, Machine Learning and Wireless Communications. Cambridge, UK: Cambridge University Press, 2022.
- [3] W. Saad, M. Bennis, and M. Chen, “A vision of 6G wireless systems: Applications, trends, technologies, and open research problems,” IEEE Netw., vol. 34, no. 3, pp. 134–142, May/Jun. 2020.
- [4] W. Shi et al., “Intelligent reflection enabling technologies for integrated and green Internet-of-Everything beyond 5G: Communication, sensing, and security,” IEEE Wireless Commun., vol. 30, no. 2, pp. 147–154, Apr. 2023.
- [5] Z. He et al., “Unlocking potentials of near-field propagation: ELAA-empowered integrated sensing and communication,” 2024, arXiv:2404.18587.
- [6] W. Xu et al., “Edge learning for B5G networks with distributed signal processing: Semantic communication, edge computing, and wireless sensing,” IEEE J. Sel. Topics Signal Process., vol. 17, no. 1, pp. 9–39, Jan. 2023.
- [7] G. Zhu et al., “Pushing AI to wireless network edge: An overview on integrated sensing, communication, and computation towards 6G,” Sci. China Inf. Sci., vol. 66, no. pp. 130301:1–19, Feb. 2023.
- [8] M. Chen et al., “Distributed learning in wireless networks: Recent progress and future challenges,” IEEE J. Sel. Areas Commun., vol. 39, no. 12, pp. 3579–3605, Dec. 2021.
- [9] B. McMahan, E. Moore, D. Ramage, S. Hampson, and B. A. Arcas, “Communication-efficient learning of deep networks from decentralized data,” in Proc. Int. Conf. Artif. Intell. Statist., 2017, pp. 1273–1282.
- [10] Y. Yang, Z. Zhang, and Q. Yang, “Communication-efficient federated learning with binary neural networks,” IEEE J. Sel. Areas Commun., vol. 39, no. 12, pp. 3836–3850, Dec. 2021.
- [11] Y. Guo, R. Zhao, S. Lai, L. Fan, X. Lei, and G. K. Karagiannidis, “Distributed machine learning for multiuser mobile edge computing systems,” IEEE J. Sel. Topics Signal Process., vol. 16, no. 3, pp. 460–473, Apr. 2022.
- [12] M. Chen, Z. Yang, W. Saad, C. Yin, H. V. Poor, and S. Cui, “A joint learning and communications framework for federated learning over wireless networks,” IEEE Trans. Wireless Commun., vol. 20, no. 1, pp. 269–283, Jan. 2021.
- [13] Z. Yang, M. Chen, W. Saad, C. S. Hong, and M. Shikh-Bahaei, “Energy efficient federated learning over wireless communication networks,” IEEE Trans. Wireless Commun., vol. 20, no. 3, pp. 1935–1949, Mar. 2021.
- [14] M. M. Amiri and D. Gündüz, “Federated learning over wireless fading channels,” IEEE Trans. Wireless Commun., vol. 19, no. 5, pp. 3546–3557, May 2020.
- [15] T. Li, A. K. Sahu, A. Talwalkar, and V. Smith, “Federated learning: Challenges, methods, and future directions,” IEEE Signal Process. Mag., vol. 37, no. 3, pp. 50–60, May 2020.
- [16] M. Chen, N. Shlezinger, H. V. Poor, Y. C. Eldar, and S. Cui , “Communication-efficient federated learning,” Proc. Nat. Acad. Sci., vol. 118, no. 17, 2021.
- [17] N. Shlezinger, M. Chen, Y. C. Eldar, H. V. Poor, and S. Cui, “UVeQFed: Universal vector quantization for federated learning,” IEEE Trans. Signal Process., vol. 69, pp. 500–514, 2021.
- [18] M. Salehi and E. Hossain, “Federated learning in unreliable and resource-constrained cellular wireless networks,” IEEE Trans. Commun., vol. 69, no. 8, pp. 5136–5151, Aug. 2021.
- [19] S. Zheng, C. Shen and X. Chen, “Design and analysis of uplink and downlink communications for federated learning,” IEEE J. Sel. Areas Commun., vol. 39, no. 7, pp. 2150–2167, Jul. 2021.
- [20] Y. Wang, Y. Xu, Q. Shi, and T.-H. Chang, “Quantized federated learning under transmission delay and outage constraints,” IEEE J. Sel. Areas Commun., vol. 40, no. 1, pp. 323–341, Jan. 2022.
- [21] M. Lan, Q. Ling, S. Xiao, and W. Zhang, “Quantization bits allocation for wireless federated learning,” IEEE Trans. Wireless Commun., vol. 22, no. 11, pp. 8336–8351, Nov. 2023.
- [22] H. Li, R. Wang, W. Zhang, and J. Wu, “One bit agregation for federated edge learning with reconfigurable intelligent surface: Analysis and optimization,” IEEE Trans. Wireless Commun., vol. 22, no. 2, pp. 872–888, Feb. 2023.
- [23] H. Ye, L. Liang, and G. Y. Li, “Decentralized federated learning with unreliable communications,” IEEE J. Sel. Topics Signal Process., vol. 16, no. 3, pp. 487–500, Apr. 2022.
- [24] J. Yao, Z. Yang, W. Xu, M. Chen, and D. Niyato, “GoMORE: Global model reuse for rescource-constrained wireless federated learning,” IEEE Wireless Lett., vol. 12, no. 9, pp. 1543–1547, Sept. 2023.
- [25] S. Liu, G. Yu, R. Yin, J. Yuan, L. Shen, and C. Liu, “Joint model pruning and device selection for communication-efficient federated edge learning,” IEEE Trans. Commun., vol. 70, no. 1, pp. 231–244, Jan. 2022.
- [26] H. H. Yang, Z. Chen, T. Q. S. Quek, and H. V. Poor, “Revisiting analog over-the-air machine learning: The blessing and curse of interference,” IEEE J. Sel. Topics Signal Process., vol. 16, no. 3, pp. 406–419, Apr. 2022.
- [27] T. Sery and K. Cohen, “On analog gradient descent learning over multiple access fading channels,” IEEE Trans. Signal Process., vol. 68, pp. 2897–2911, Apr. 2020.
- [28] G. Zhu, Y. Wang, and K. Huang, “Broadband analog aggregation for low-latency federated edge learning,” IEEE Trans. Wireless Commun., vol. 19, no. 1, pp. 491–506, Jan. 2020.
- [29] X. Cao, G. Zhu, J. Xu, and S. Cui, “Transmission power control for over-the-air federated averaging at network edge,” IEEE J. Sel. Areas Commun., vol. 40, no. 5, pp. 1571–1586, May 2022.
- [30] X. Yu, B. Xiao, W. Ni, and X. Wang, “Optimal adaptive power control for over-the-air federated edge learning under fading channels,” IEEE Trans. Commun., vol. 71, no. 9, pp. 5199–5213, Sept. 2023.
- [31] W. Guo et al., “Joint device selection and power control for wireless federated learning,” IEEE J. Sel. Areas Commun., vol. 40, no. 8, pp. 2395–2410, Aug. 2022.
- [32] F. Ang et al., “Robust federated learning with noisy communication,” IEEE Trans. Commun., vol. 68, no. 6, pp. 3452–3464, Jun. 2020.
- [33] K. Yang, T. Jiang, Y. Shi, and Z. Ding, “Federated learning via over-the-air computation,” IEEE Trans. Wireless Commun., vol. 19, no. 3, pp. 2022–2035, Mar. 2020.
- [34] J. Yao, Z. Yang, W. Xu, D. Niyato, and X. You, “Imperfect CSI: A key factor of uncertainty to over-the-air federated learning,” IEEE Wireless Lett., vol. 12, no. 12, pp. 2273–2277, Dec. 2023.
- [35] K. Guo, Z. Chen, H. H. Yang, and T. Q. S. Quek, “Dynamic scheduling for heterogeneous federated learning in private 5G edge networks,” IEEE J. Sel. Topics Signal Process., vol. 16, no. 1, pp. 26–40, Jan. 2022.
- [36] X. Wei and C. Shen, “Federated learning over noisy channels: Convergence analysis and design examples,” IEEE Trans. Cogn. Commun. Netw., vol. 8, no. 2, pp. 1253–1268, Jun. 2022.
- [37] D. Liu and O. Simeone, “Privacy for free: Wireless federated learning via uncoded transmission with adaptive power control,” IEEE J. Sel. Areas Commun., vol. 39, no. 1, pp. 170–185, Jan. 2021.
- [38] X. Li, G. Zhu, Y. Gong, and K. Huang, “Wirelessly powered data aggregation for IoT via over-the-air function computation: Beamforming and power control,” IEEE Trans. Wireless Commun., vol. 18, no. 7, pp. 3437–3452, Jul. 2019.
- [39] Z. Lin, X. Li, V. K. N. Lau, Y. Gong, and K. Huang, “Deploying federated learning in large-scale cellular networks: Spatial convergence analysis,” IEEE Trans. Wireless Commun., vol. 21, no. 3, pp. 1542–1556, Mar. 2022.
- [40] M. Mohammadi Amiri and D. Gündüz, “Machine learning at the wireless edge: Distributed stochastic gradient descent over-the-air,” IEEE Trans. Signal Process., vol. 68, pp. 2155–2169, 2020.
- [41] H. Xing, O. Simeone, and S. Bi, “Federated learning over wireless device-to-device networks: Algorithms and convergence analysis,” IEEE J. Sel. Areas Commun., vol. 39, no. 12, pp. 3723–3741, Dec. 2021.
- [42] E. Rizk, S. Vlaski, and A. H. Sayed, , “Federated learning under importance sampling,” IEEE Trans. Signal Process., vol. 70, pp. 5381–5396, 2022.
- [43] W. Dinkelbach, “On nonlinear fractional programming,” Manage. Sci., vol. 133, no. 7, pp. 492–498, Mar. 1967.
- [44] Z. He et al., “Energy efficient beamforming optimization for integrated sensing and communication,” IEEE Wireless Commun. Lett., vol. 11, no. 7, pp. 1374–1378, Jul. 2022.
- [45] M. Grant and S. Boyd. (2016). CVX: MATLAB Software for Disciplined Convex Programming. [Online]. Available: http://cvxr.com/cvx
- [46] J. Wang, Y. Mao, T. Wang, and Y. Shi, “Green federated learning over cloud-RAN with limited fronthaul capacity and quantized neural networks,” IEEE Trans. Wireless Commun., vol. 23, no. 5, pp. 4300–4314, May 2024.
- [47] M. M. Amiri, D. Gündüz, S. R. Kulkarni, and H. V. Poor, “Convergence of federated learning over a noisy downlink,” IEEE Trans. Wireless Commun., vol. 21, no. 3, pp. 1422–1437, Mar. 2022.
- [48] J. Yao et al., “Superimposed RIS-phase modulation for MIMO communications: A novel paradigm of information transfer,” IEEE Trans. Wireless Commun., vol. 23, no. 4, pp. 2978–2993, Apr. 2024.
- [49] X. Li, K. Huang, W. Yang, S. Wang, and Z. Zhang, “On the convergence of FedAvg on non-iid data.” in Proc. Int. Conf. Learn. Representations (ICLR), 2019.
- [50] W. Shi et al., “On secrecy performance of RIS-assisted MISO systems over Rician channels with spatially random eavesdroppers,” IEEE Trans. Wireless Commun., early access. 2024.
- [51] T. Li et al., “Federated optimization in heterogeneous networks,” in Proc. Mach. Learn. Syst., 2020, pp. 429–450.
- [52] Y. Sun, Z. Lin, Y. Mao, S. **, and J. Zhang, “Channel and gradient-importance aware device scheduling for over-the-air federated learning,” IEEE Trans. Wireless Commun. (Early Access). 2023.