Abstract
Data volume demand has increased dramatically due to huge user device increasement along with the development of cellular networks. And macrocell in 5G networks may encounter sudden traffic due to dense users caused by sports or celebration activities. To resolve such temporal hotspot, additional network access point has become a new solution for it, and unmanned aerial vehicle equipped with base stations is taken as an effective solution for coverage and capacity improvement. How to plan the best three-dimensional location of the aerial base station according to the users’ business needs and service scenarios is a key issue to be solved. In this article, first, aiming at maximizing the spectral efficiency and considering the effects of line-of-sight and non-line-of-sight path loss for 5G mmWave networks, a mathematical optimization model for the location planning of the aerial base station is proposed. For this model, the model definition and training process of deep Q-learning are constructed, and through the large-scale pre-learning experience of different user layouts in the training process to gain experience, finally improve the timeliness of the training process. Through the simulation results, it points out that the optimization model can achieve more than 90% of the theoretical maximum spectral efficiency with acceptable service quality.
Introduction
Along with varieties of services and the Internet-of-Things (IoT) devices data communication requirements for different scenarios in 5G networks, traffic generations take on drastic spatial and temporal variations, which have caused tremendous pressure on the basic macro base stations (BSs). 5G networks integrate different network architectures, such as cloud radio access networks, ultra-dense networks with heterogeneous cells to improve network density and coverage, thus achieving higher transmission rate and network capacity. However, in some hotspots, such as sports or celebration activities will cause periodic bursts of traffic on networks, and the deployed BSs may be not to accommodate all the sudden traffic. And it is difficult to reconstruct existing communication infrastructure immediately to resolve this problem. So additional capacity enhancement schemes are required. According to the previous literature, unmanned aerial vehicle (UAV) equipped with BSs, called as aerial base station (aerial-BS), is an effective complementary method. 1
The mmWave, also called mmW, is the frequency band used in various high-speed applications. The mmWave includes wireless frequencies of 30–300 GHz and the wavelength range of the radio wave in this frequency band is from 1 to 10 mm, so it is called millimeter wave. This frequency has obvious advantages, because it supports higher bandwidth, so it is very suitable for wireless infrastructure applications. Millimeter wave combines height-oriented antenna with beamforming and beam tracking to provide a very safe and reliable link. In addition to the advantages of large bandwidth and high-speed rate, millimeter wave has narrow beam, good directivity, and high spatial resolution, which improves the transmission efficiency, so mmWave transmission suitable for direct communications can be a feasible method here. When aerial-BSs can be used as air access points or disconnect relays and enhanced connections between networks, using aerial-BSs as air support for existing cellular networks can handle sudden traffic situation more economically and enhance network capacity better.
Aerial-BSs for emergency communication or hotspot have attracted much attentions from academic and industry field recently. As aerial-BSs can provide rapid deployment and cost reduction, making communication failure or burst traffic to be absorbed soon. 2 And a few literatures have explored for aerial-BSs compensation for network damage or performance degradation due to abnormal traffic.3,4 However, these methods consider less about multiple BSs and the placement algorithms are complex, making them hard for dynamic variations in 5G mmWave networks.
In this article, for optimizing the three-dimensional (3D) deployment of aerial-BSs for 5G mmWave networks, a classic deep reinforcement learning (DRL) network which named deep Q-network (DQN) algorithm is adopted. Compared with traditional algorithm, it can solve high complexity and big state space and action space, so we choose DQN algorithm to solve our problem. DQN uses Q-network to fit Q-table, which solves the problem of dimension disaster well. First, the model is learned and saved with maximal spectrum efficiency, and then the optimal 3D deployment location is quickly found by applying the model in a simulation scenario.
With above analysis, the main contributions are shown as follows:
Modeling the aerial-BSs’ location as a maximal spectral efficiency (SE) problem with quality of service (QoS) constraints considering line-of-sight (LOS)/non-line-of-sight (NLOS) path loss under mmWave networks. Moreover, a simple decompose mechanism to reduce its complexity is proposed as well.
Applying DQN algorithm to the optimization of 3D deployment of aerial-BSs and sudden traffic accommodation. And the 3D location is used here for different aerial-BS action space. Considering the dynamic changes of the network topology environment, the users’ distribution is trained as part of the state, so that the model can be well adapted to different user distribution.
The remaining content is organized as follows: System model is analyzed in section “System model.” And DQN procedures for aerial-BSs’ 3D deployment are shown in section “DQN-based aerial-BS location optimization framework.” And simulations are taken in section “Simulation results.” Finally, conclusions and recommendations are given in section “Conclusion and future work.”
Related work
In recent years, experts pay more attentions to the usage of aerial-BSs in cellular networks as they can achieve rapid deployment solutions to meet the needs of wireless networks. In Yu et al., 1 a scheme of introducing air BS to enhance the capacity of the data traffic burst area is proposed. However, the location deployment of the aerial-BSs has become one of the key challenges. Unlike the traditional fixed-position BS, the location deployment of the aerial-BSs is flexible and can move flexibly in the air, and ultimately determine by its height and angle. Therefore, the aerial-BSs’ location planning is a 3D deployment problem. In Gomez et al., 2 a heuristic algorithm is proposed to serve multiple users using the least number of aerial-BSs, and to obtain 3D location of aerial-BSs. However, in the dynamic environment where the network topology changes, it is necessary to re-initialize the heuristic algorithm and with new topology. And this process will bring many computational complexities ultimately. In Deaton, 3 the vertical and horizontal dimensions of the aerial-BSs are separated, and a deployment scheme of the aerial-BSs with the minimum transmit power serving the maximum number of users is proposed. In Zong, 4 aiming at maximizing the total logarithm of users, an algorithm to deploy the aerial-BSs in 3D location is proposed, besides considering the user BS association and wireless return bandwidth allocation. In the previous literature, user mobility is not considered. A 3D deployment algorithm of aerial-BSs based on Q-learning is proposed in the research work, 5 which considers user mobility. But when the dimension of state space increases, the Q-table will occupy a lot of memory and bring a lot of time overhead.
Still, much research has designed algorithms for aerial-BSs’ placement according to different scenarios.6–8 A polynomial-time algorithm which aims at optimizing aerial-BSs’ placement is adopted in Lyu et al. 9 And more unmanned aerial vehicle base station (UAV-BS) evaluation framework for user coverage with minimum transmit power is shown in Alzenad. 7 And aerial-BSs’ locations considered with height and user locations from different optimization problems are studied as well.10,11 Several intelligent algorithms are taken for aerial-BS deployment, for instance, grid search algorithm for 3D UAV-BS placing 12 and minimal aerial-BS number. 13 However, the above literature assumes that all users take on the same QoS constraints. In addition, the aerial-BS locations are modeled as an optimal QoS requirement problem. 14 However, the algorithm is still hard to obtain the optimal results due to the heuristic algorithms.
In recent years, both industries and experts pay lots of attentions to the DRL which was derived from DeepMind. As the name suggests, DRL is the combination of deep learning and reinforcement learning, so DRL makes up for the shortcomings of DL and RL. First, the RL is a study of the mapping of environment state to action space. It is based on the Markov decision process (MDP), that is, the current state is only related to the previous state, regardless of the cumulative influence before the previous state. MDP is usually defined as a quad (S, A, R, P); in addition to quads, RL has two important functions, namely, value function and Q function, so is a state-of-the-art method to produce control policies using the action set. With the development of DL, DQN algorithm appears, which is a promising tool to address multi-agent optimization problems such as the UAV navigation. In Mnih et al., 15 the DQN algorithm which can receive appropriate strategies directly from high-dimensional perceptron inputs was proposed. For further applying DRL in continuous action control and mass discrete action control, a set of control tasks to measure procedures in continuous action control were proposed by Duan et al. In Duan et al., 16 as an emerging deep machine learning method, DRL has been widely used in UAV-enabled wireless communication control. In Wu et al., 17 in view of DQN, a 3D aerial-BS location planning algorithm is adopted in capacity enhancement, but just one aerial-BS is considered. Consider of environment learning, a two-step algorithm is applied in the UAVs’ intelligent arrangement in Luo et al. 18 Next, in Liu et al., 19 on account of UAV control in coverage and connectivity, a DRL-driven energy saving algorithm which outperforms baseline methods is introduced. For joint virtual reality (VR) content caching and transmission problem with cellular-connected UAVs, a distributed DL algorithm which integrates liquid state machine and echo state networks is introduced in Chen et al. 20 Moreover, a DRL approach which aims at obtaining the optimum trace of UAV is analyzed in Saxena et al. 21 Above DRL algorithms are all aiming at resolving a specific optimization problem, which show their efficiency. Wang et al. 22 have formularized the UAV movement problem as a constrained Markov decision process (CMDP) problem and employed Q-learning to solve the UAV movement problem. But this study has not considered the different requirements of users and the convergence is decided by the number of UAVs. So it may not be suitable for sudden traffic compensation as well. Huang et al. 23 have presented a DRL-based scheme for UAV navigation through massive multiple-input and multiple-output (MIMO). But the optimal location at the UAVs is obtained based on the received signal strengths without requiring global information, so it may fall into local optimum rather than global optimum.
In our previous work, user QoS requirements for different services with DQN are proposed, 24 but just one aerial-BS is considered. Moreover, the reference signal received power (RSRP) and signal to interference and noise ratio (SINR) which are important indicators related to QoS are not considered as well. In this article, we will extend it to multiple BSs scenario with sudden traffic compensation, and the detailed description is shown below.
System model
It can be seen in Figure 1, a macro BS and a lower power node (LPN) are deployed in the ground. When the gymnasium holds concerts or other activities, it will cause sudden traffic hotspots, which put forward higher requirements for the capacity of mobile data access and exceed the allowed capacity. To compensating additional data requirement temporarily, we can enhance the capacity of local networks by deploying mmWave aerial-BSs with high bandwidth, which can also provide services for users beyond the coverage of macro BSs. However, due to its signal attenuation and bad building penetration feature as shown in the figure, the path loss of mmWave link should be considered as well.

Scenario description for sudden traffic.
The deployment location of aerial-BSs not only affects the number of users in its coverage area, but also affects the quality of the air-to-ground link. Because of the characteristics of aerial-BSs, the air-to-ground channel suitable for aerial-BSs should consider the 3D location effect; moreover, it has a higher chance of LOS connection.
In 5G mmWave networks, signal transmission will be affected by LOS and NLOS connections. Although aerial-BSs are suitable for direct beamforming, all the loss should be considered as well. To evaluate the received signal properly, in this article, an air-to-ground propagation model for aerial-BSs is constructed first. Next, the correlations from path loss to the maximum coverage radius are analyzed. Finally, average SE is calculated for optimizing the best aerial-BS location next.
Air-to-ground path loss model
As shown in our previous work, 24 the air-to-ground channel should consider the impact on the occurrence of LOS. Based on recommendation from the International Telecommunication Union (ITU) for radio transmission, 25 the important parameters for determining geometric probability of LOS transmission are given as follows:
with
where
Next, the probability of NLOS link from aerial-BS j to ground user i is given by
Moreover, the path loss of the entire link from aerial-BS j to user i is given by
where
For aerial-BS j, its coverage radius
where
Optimization problem
Based on above channel model, we will analyze the aerial-BS requirement scenario for compensating sudden traffic when macro BS cannot satisfy all the data requirement. We assume that K different QoS requirements exist in the network under sudden traffic. Assuming that
To maximize the effectiveness of aerial-BSs, SE should also be considered. Assuming the total bandwidth of aerial-BS j is
where
where
Basic SE is shown as below (consider of Shannon’s theorem)
According to above analysis, the average SE with mmWave path is shown as
And in our article, we want to find best locations for all the aerial-BSs with highest SE with all the users under sudden traffic. And the optimization problem is shown below
when the ith user is connected to the aerial-BS j with service k,
From our previous work in Guo et al., 24 this problem is complex and hard to be resolved with classic algorithms, and present simulated evolutionary algorithms are difficult to get global optimal results as well, so we need an efficient scheme for it. To reduce its complexity, we propose a scheme for how to find sufficient aerial-BSs and determine the connections between them first, and then a DQN-based framework is proposed to get the best location of aerial-BSs, which is shown below.
DQN-based aerial-BS location optimization framework
In this section, in order to maximize the total SE with suitable aerial-BS number and users’ QoS constraints and consider of DQN, we propose an framework for 3D location deployment of aerial-BSs. First, we use K-means to cluster the users according to their geographical location to determine connections and aerial-BS number, and use DQN to seek out the optimal positions of aerial-BSs next. And the framework can be seen in Figures 2–4.

Scenario description for sudden traffic.

Scenario description for sudden traffic.

Scenario description for sudden traffic.
To resolve the aerial-BS location problem, we need a network entity which is responsible for the data collecting and solution planning, and pushing corresponding control information to the aerial-BSs. From our previous work, self-organized network (SON) architecture is a reasonable scheme for it. In this article, distributed SON architecture will be adopted here. We assume that macro BS can be set as the relay for aerial-BSs’ backhaul in this scenario. And then, the SON agent deployed in the macro BS is the entity responsible for the collection, analysis, planning, and evaluation.
K-means algorithm for partitioning region with aerial-BSs
K-means algorithm can find out the solutions of clustering problems and belongs to unsupervised learning algorithms. Based on a fixed prior, the given data set can be classified through some clusters. In which the important matter is to set K centroids using K cluster. These centers of mass should be put skillfully, which can lead to various results. Therefore, it is preferably to keep the centers of mass as far apart as possible. And then classify each point to the given data set and link it to the nearest centroid. If all points have been processed, complete the first step and the early grouping. At this point, we need to recalculate K new centroids as the focus of clustering generated in the previous procedure. New bindings must be made between the same data set and the latest centroids after obtaining K new centroids. So the algorithm can be executed in a loop. Consequently, it can be easily found that K centroids gradually change their positions until they are no longer changed. In other words, the center of mass is no longer moving.
For example, in Figure 2, if there are emergencies in the gymnasium. At this time, the users can be classified according to their geographical location because they have different environments and QoS. Like the users who are in the gym and the connection between the macro BS is NLOS, they may often have communication interruption to compete for communication resources owing to many people, so the demand of these users is to provide adequate communication resource to keep the communication process connected. Like the users who are out of the gym, they may be affected by the emergencies in the gymnasium although they are within the communication range of the BS but the load of the macro BS is over. So the demand of these users is to eliminate the interference of the gymnasium and ensure continuous communication. Like the users who are out of the gym and at the edge of the coverage area, so the demand of these users is to attain ample communication resource to finish the process of the communication.
Here, the K-means algorithm aims at minimizing the square error function given by
where K is the cluster number, and
As users’ call admission at wireless link is complex,13,29 in this article, we can just assume that bandwidth for each aerial-BS
where
where
DQN algorithm for location planning
In this section, consider of DQN an algorithm of 3D deployment of aerial-BSs is proposed with definite aerial-BS number for the sake of finding the maximum total SE shown in Alzenad et al.
14
The DQN model used here can be found in Guo et al.
24
as well. The DQN model used here integrates convolutional neural networks with RL model named Q-learning. Through the process of DQN, in view of the rewards received from the interaction procedure with the environment, the agent can learn continuously for the aim of getting the target status. In this algorithm,
In this algorithm, the agent is the candidate aerial-BS sets whose state space, action space, and rewards are defined by as follows:
The state space:
The action space:
The reward: the system SE shown in Alzenad et al. 14 getting the present status of aerial-BSs.
The classic expression of the Q-learning is
where
This will find the best action for each state; however, the Q matrix has limited ability to store information, when the state is too much or discrete, this algorithm will naturally cause dimensional disasters, so Google’s DeepMind team will combine DL with RL, the DQN model is proposed, which uses the value function f trained from DL model to approximate the Q value
From the above value function, the functional relationship refers to learning through a neural network to obtain the Q values and the functional mapping relationship between states and actions. The neural network uses two fully connected neural networks which has same structure but has difference in parameters: main network and target network. During the training process, the first cycle randomly generates different user distribution environments, and the second cycle iterates aim at finding the 3D positions of the aerial-BSs which have maximum SE. First, initialize a random state

DQN framework for optimization.
For high-dimensional state space, the DQN algorithm inputs state S and outputs a matrix,
In this algorithm, it has two key technologies, which are as follows:
Experience reply: first, all samples are placed in the sample pool, next in order to train network, choose a sample randomly from the sample pool. This process makes samples have no relation with each other.
Fixed Q-target network: as shown in Figure 5, the calculation of network target value requires the existing Q value. So the Q value is generated by a slower network. With the training steps, we can get the best locations for each aerial-BS.
Simulation results
Using an area of 3.0 km × 3.0 km urban environment in simulation process. In this area, we assume that 1000 users distributed randomly under three subregions, which is shown in Figure 6. And a ground BS with same parameters is deployed in this region as well. Still, we assume that basic parameters and spectrum efficiency calculation method of all the BSs are the same. Moreover, users choose serving BSs with the higher spectrum efficiency BS (ground BS or aerial-BS). These assumptions make our method able to meet the sudden traffic demands in the region. This scenario can be extended to software-defined networking (SDN)/network function virtualization (NFV)-enabled 5G networks as well. 30

Users generated by simulation.
Next, Table 1 shows the parameters appeared in the algorithm. First, users are clustered by K-means algorithm and divided into small areas. And the result shows that three aerial-BSs are required here. Aim at making the scenario meets the real requirements, put the ground BS in the position of region center. The clustering result is shown in Figure 7 with different colors. As the users are distributed randomly under separate regions, so they can be easily be clustered. Next, we will find the best locations for different aerial-BSs with DQN algorithm.
The value of parameter used in algorithm.
BS: base station; RSRP: reference signal received power; SINR: signal to interference and noise ratio.

Users generated by simulation.
Then, position planning is carried out for each subregion with corresponding aerial-BSs. Next, choose a 3D position randomly as the initial status of the aerial-BSs. Through training, it is shown in Figure 8 that as the learning process progresses, the aerial-BSs gradually move to the positions where the total SE of the system is the highest. As shown in the figure, the red, green, and blue dots at the bottom represent the users in the three hotspot areas, the triangle of the red, green, and blue dots at the top represents the mobile process of the aerial-BSs related in each subregion, and the yellow pentagrams represent the optimized locations of the aerial-BSs.

Location moving process of air base stations using DQN algorithms.
After a certain scale of learning, the network structure parameters of DQN algorithm are obtained and saved as a model. When the model is directly applied in this scenario, the aerial-BSs will stay in the optimal position of system spectrum efficiency. Figure 9 shows the results of three different clusters comparing to the average SE. It can be seen that, the spectrum efficiency is converged along with training steps, and the lowest spectrum efficiency of the learned aerial-BS can reach 92.77% of the maximum spectrum efficiency of the system under ideal conditions. Moreover, the spectrum efficiency of different aerial-BSs is very near, which denotes the balance of our proposed algorithm. The advantage of using DQN algorithm is that after learning the model, it can be applied directly with near global optimal result, and the application time is very short, so the efficiency is very high.

Average spectral efficiency of different systems in three scenarios.
Besides the spectrum efficiency, it is important to evaluate the users’ quality from signal strength and interference perspective. So we analyze the cumulative probability distribution function of RSRP and SINR for each aerial-BS here, which are shown in Figures 10 and 11 in the three regions, respectively. As shown in Figure 10, different aerial-BSs take on different effect. However, all the users’ RSRP values are higher than −120 dBm and lower than −50 dBm in our scenario, which means that the RSRP constraints are all satisfied. And 95% users’ RSRP is lower than −70 dBm.

Cumulative probability distribution of RSRP.

Cumulative probability distribution of SINR.
Next, as shown in Figure 11, different aerial-BSs take on different variations as well, but all of them are higher than −10 dB, and their values range from −10 to 23 dB. As an important indicator, it means that all the users can be served perfectly. Although the SINR distributions for different aerial-BSs at the beginning have some discrepancies among each other, they will be very close to each other at last, and more than 10% users’ SINR values are higher than 20 dB, which still take on acceptable performance. From above analysis, it is easy to see that our proposed mechanism can receive optimum spectrum efficiency with mmWave aerial-BSs, and make users’ quality above acceptable levels.
Conclusion and future work
Through the article, we have studied multiple aerial-BSs’ optimal locations to cope with sudden traffic with optimal spectrum efficiency. First, in view of probabilistic LOS/NLOS mmWave wireless connections, we can obtain the downlink coverage probability and path loss. And then, consider of DQN, we present an effective location method to satisfy the data requirements of users with different QoS. It can be seen from the simulation results that under certain constraints, if the learning entity has a sufficiently long learning time, after enough iterations, the learning entity will learn the environmental characteristics and save the learning model, and find the best deployment locations in a very short time when applying the model. The learned SE of the system can reach 70.76% of the maximum SE under ideal conditions.
As communication failure in sudden disaster or such emergency scenario has similar traffic problem, so we will extend our solutions for such applications in our future work. Moreover, we will try to find more efficient learning model for this problem, and explore its applications to new network such as WiMAX or use new technologies such as SDN/NFV as well. Still, how to model the interference and UAV formation with dense regions should be considered as well.
Footnotes
Handling Editor: Bo Rong
Declaration of conflicting interests
The author(s) declared no potential conflicts of interest with respect to the research, authorship, and/or publication of this article.
Funding
The author(s) disclosed receipt of the following financial support for the research, authorship, and/or publication of this article: This work was supported by the open fund project of Science and Technology on Communication Networks Laboratory (grant no. SXX18641X024) and the National Natural Science Foundation of China (grant no. 61971053).
