A Feature-Based Bayesian Method for Content Popularity Prediction in Edge-Caching Networks
Sajad Mehrizi, Anestis Tsakmalis, Symeon Chatzinotas, Bjorn Ottersten

TL;DR
This paper presents a Bayesian Poisson regression model utilizing content features for predicting content popularity in edge-caching networks, aiming to improve caching efficiency amid limited memory and unpredictable requests.
Contribution
It introduces a novel feature-based Bayesian Poisson regression model with MCMC inference for content popularity prediction in edge-caching systems.
Findings
Model accurately predicts content requests in simulations.
Incorporating content features improves prediction performance.
Bayesian approach offers robustness against overfitting.
Abstract
Edge-caching is recognized as an efficient technique for future wireless cellular networks to improve network capacity and user-perceived quality of experience. Due to the random content requests and the limited cache memory, designing an efficient caching policy is a challenge. To enhance the performance of caching systems, an accurate content request prediction algorithm is essential. Here, we introduce a flexible model, a Poisson regressor based on a Gaussian process, for the content request distribution in stationary environments. Our proposed model can incorporate the content features as side information for prediction enhancement. In order to learn the model parameters, which yield the Poisson rates or alternatively content popularities, we invoke the Bayesian approach which is very robust against over-fitting. However, the posterior distribution in the Bayes formula is…
Click any figure to enlarge with its caption.
Figure 1
Figure 2
Figure 3| True value | N=25 | N=50 | N=100 | N=200 | N=400 | |
| 0.0001 | 0.0081 | 0.0032 | 0.0031 | 0.0016 | 0.0009 | |
| 0.1 | 0.1187 | 0.1443 | 0.1268 | 0.1313 | 0.1269 | |
| 0.25 | 0.1633 | 0.1553 | 0.1879 | 0.1848 | 0.2083 | |
| 0 | 0.0676 | 0.0484 | 0.0383 | 0.0146 | 0.0192 | |
| 0.1 | 0.0755 | 0.0535 | 0.0542 | 0.0871 | 0.0843 | |
| 0.5 | 0.3354 | 0.3441 | 0.3904 | 0.4180 | 0.4495 |
| True value | N=25 | N=50 | N=100 | N=200 | N=400 | |
| 0.0001 | 0.0035 | 0.0014 | 0.0010 | 0.0006 | 0.0002 | |
| 0.1 | 0.1179 | 0.1225 | 0.1141 | 0.1117 | 0.1129 | |
| 0.25 | 0.2187 | 0.2296 | 0.2232 | 0.2451 | 0.2428 | |
| 0 | 0.0466 | 0.0179 | 0.0072 | 0.0077 | 0.0045 | |
| 0.1 | 0.0736 | 0.0762 | 0.0902 | 0.0969 | 0.1043 | |
| 0.5 | 0.3732 | 0.4504 | 0.4649 | 0.4536 | 0.4753 |
Peer Reviews
No public reviews on file for this paper yet. If you reviewed it on a platform where reviews are public (OpenReview, ICLR, NeurIPS, ICML), you can paste yours below so the community can read it here.
Videos
No videos yet. Explain this paper in a talk, walkthrough, or lecture? Add one.
A Feature-Based Bayesian Method for Content Popularity Prediction in Edge-Caching Networks
Sajad Mehrizi, Anestis Tsakmalis, Symeon Chatzinotas, Björn Ottersten
\left\{sajad.mehrizi,anestis.tsakmalis,symeon.chatzinotas,bjorn.ottersten\right\}[email protected]
Interdisciplinary Centre for Security, Reliability and Trust (SnT), University of Luxembourg
Abstract
Edge-caching is recognized as an efficient technique for future wireless cellular networks to improve network capacity and user-perceived quality of experience. Due to the random content requests and the limited cache memory, designing an efficient caching policy is a challenge. To enhance the performance of caching systems, an accurate content request prediction algorithm is essential. Here, we introduce a flexible model, a Poisson regressor based on a Gaussian process, for the content request distribution in stationary environments. Our proposed model can incorporate the content features as side information for prediction enhancement. In order to learn the model parameters, which yield the Poisson rates or alternatively content popularities, we invoke the Bayesian approach which is very robust against over-fitting. However, the posterior distribution in the Bayes formula is analytically intractable to compute. To tackle this issue, we apply a Monte Carlo Markov Chain (MCMC) method to approximate the posterior distribution. Two types of predictive distributions are formulated for the requests of existing contents and for the requests of a newly-added content. Finally, simulation results are provided to confirm the accuracy of the developed content popularity learning approach.
Index Terms:
Popularity prediction, Stationary environment, Content features, Poisson distribution, Gaussian process, Bayesian Learning
I Introduction
Mobile data traffic is forecast to increase at a compound annual growth rate (CAGR) from 2016 to 2021, two times faster than the growth of global IP fixed traffic during the same period. [1]. This is largely due to the growth in both the number of mobile devices and the user interest towards high-rate multimedia applications. Nevertheless, supporting such a huge data traffic turns to be a big challenge which indicates the need for developing new architectures. To mitigate this issue, edge-caching is recognized as one of the leading technologies [2, 3]. It can bring the requested content from the core network close to the end mobile user, instead of downloading the same content multiple times through the backhaul links. Therefore, by serving the mobile users locally, edge-caching can jointly offload traffic burden on the backhaul links, reduce system costs and improve quality of service (QoS) of the mobile users.
Over the past few years, extensive research has been carried out on edge-caching networks, which has mainly focused on the performance analysis of caching, cache placement optimization and transmission strategies. A cache placement algorithm has been proposed to minimize the excepted downloading time for contents in [2]. In [4], physical layer features are used in the cache placement problem to minimize network cost while to satisfy users’ QoS requirements. The authors in [5] investigated energy efficiency and time delivery of an edge-caching network. In addition, various coding schemes, intra and inter sessions, have been proposed to enhance caching performance [2, 6, 7].
The main assumption of the aforementioned papers is that the content popularity is known in advance. However, in practice, the popularity is unknown and has to be estimated and predicted. In this respect, the popularity learning problem can be categorized in two general approaches: model-free and model-based. In the model-free approach, there is no assumption on the content request distribution. The popularity learning is then performed within the process of optimizing a reward function (e.g cache hit ratio) by the so-called exploration-exploitation procedure. Multi-armed-bandit (MAB) and reinforcement learning algorithms are mostly based on this approach which also have been adapted to edge-caching applications [8, 9, 10, 11]. On the other hand, in the model-based approach, it is assumed that the content requests are generated by a parametric distribution. The Poisson stochastic process is a popular model adopted in the content delivery networks [12] and also has been used in edge-caching [13]. Once the request is modeled, the next step is to estimate the popularity. A simple way is to take the average of instantaneous requests, which is equivalent to the maximum likelihood estimation (MLE) from the estimation theory perspective. However, the MLE suffers from overfitting especially in edge caching systems where only a few request observations are available. For example, as it is reported in [14], a base station cache typically may receive 0.1 requests/content/day which is too small in contrast with a typical content delivery network cache which normally receives 50 requests/content/day.
To improve the popularity estimation accuracy, side information (user profile and content features) can also be incorporated in learning algorithms. In [15, 13], user profiles are leveraged to speed up the learning convergence rate. One important issue with this kind of side information is that users may not be willing to share their personal profiles to the edge-cache entity. On the other hand, content features (e.g topic categories) can be easily and cheaply obtained from the content server without jeopardizing users’ privacy. In addition, knowing the most important content features can be useful to design advanced cache-placement algorithms. For example, the authors of [16] observed that there is a traffic pattern under different topic categories of contents by doing experimental validation on the dataset of a real mobile network. Therefore, besides learning the popularities of the contents, in order to have a better understanding about the hidden request pattern, it is advantageous to also learn the importance of content features.
In this paper, we take the content features into account and introduce a new probabilistic model for the content requests. The learning process is performed in the Bayesian paradigm which is robust against overfitting and provides a way to quantify our uncertainty about the estimation. The model allows us to define different types of predictive distributions by which we can effectively model the uncertainty of future requests. The statistical information of these posterior predictive distributions can be used to design a sophisticated caching policy. Here, we should also mention that the central contribution of this paper is not to devise a caching policy but rather to propose a more accurate and reasonable probabilistic model for content requests. Overall, the main contributions of the paper are summarized as:
- •
We provide a probabilistic model, a Poisson regressor based on a Gaussian process, for stationary content requests which captures the similarity between contents. The Gaussian process is a very flexible and powerful statistical model that can model nonlinear relationships between the popularities and the features.
- •
The parameters of the model are learnt in the Bayesian framework. Due to few request samples in the local cache, Bayesian learning provides a powerful framework to mitigate overfitting.
- •
For prediction, two types of predictive distributions are specified. One is used to predict the future requests for the existing contents and the other to predict the popularity of a new content that may come to the system.
The rest of the paper is organized as follows: the system model and problem statement are described in Section II. In Section III, we apply the Bayesian approach for popularity learning. Finally, Section IV shows the simulation results and Section V concludes the paper.
II System Model and problem formulation
In this paper, we consider a cellular network consisting of a base station (BS) serving its mobile users. Users can make random requests from a library of contents , where is the total number of contents. Each content is assumed to have a set of features. For instance, a video content may have a specific topic (e.g education, entertainment, science-technology,.. ) and some other features such as release year. We use to be the feature vector of content with dimensions whose values can be either binary or continuous.
The BS is equipped with a limited capacity cache memory, and is connected to the remote content server through the backhaul links. Additionally, the remote server has access to the whole content library . At each time slot111The time slots can be hours, days, etc., each user independently requests a content (or contents)222There is no limitation on the number of requests by a user at a time slot from the library . To alleviate the traffic burden on the backhaul links and increase the users’ QoS, some contents are stored in the cache depending on the caching policy. The requested contents by the users will be served directly if they are already cached; otherwise they are fetched from the content server. We suppose that the cache module of the BS can only monitor the number of user requests towards contents of the library and cannot perform any user profiling. In addition, it is assumed that the content popularity is fixed (we can assume it does not change over short time intervals, e.g. a few days) and the requests are samples generated from a stationary distribution.
We define to be the request vector where is the total number of requests for content during time slot with duration . For simplicity, we assume that , . Therefore, we can drop and show the request vector by . Also, the requests for are presumed to be statistically independent random variables. A common parametric model for the requests is the Poison stochastic process and the MLE approach to estimate the rate request, or the popularity (we use the terms rate and popularity interchangeably) [13] as:
[TABLE]
where is the popularity of content and is the total number of request observations during the training period. Although this approach is simple, it is not very accurate for popularity estimation. Firstly, MLE suffers from severe overfitting especially when the training set has only a few request observations. Secondly, it cannot incorporate any kind of side information. For example, users commonly request contents based on their features. Therefore, we expect content popularities to be correlated in the feature space. By appropriately using this underlying prior knowledge about requests, the accuracy of popularity estimation can be significantly improved. In the next sections, we present our probabilistic model in order to deal with these issues. Before introducing the model, we summarize the basic concepts of Gaussian processes which are essential for the subsequent sections.
II-A Gaussian Process in a Nutshell
A Gaussian process is a collection of random variables, any finite number of which have a joint Gaussian distribution. Using a Gaussian process, we can define a distribution over functions :
[TABLE]
where is an arbitrary input333Here, input is a very general concept that can be time, location, … . In our problem, it is content features. variable with dimensions, and the mean function, , and the Kernel function, , are respectively defined as:
[TABLE]
This means that any finite collection of function values has a joint Gaussian distribution:
[TABLE]
where and the covariance matrix has the entities . The kernel function specifies the main characteristics of the function that we wish to model and the basic assumption is that variables with inputs which are close are likely to be correlated. Choosing a good kernel function for a learning task depends on intuition and experience. A popular and simple kernel is the squared exponential kernel (SEK):
[TABLE]
where is the vertical scale variation and is the horizontal scale variation on dimension of the function. By using different scales for each input dimension, we let them to have different importance. If is close to zero, dimension will have little influence on the covariance of variables. Covariance function (6) is infinitely differentiable and is thus very smooth. More details about the Gaussian process and the kernel functions can be found in [17].
II-B The proposed model
In this subsection, we introduce our probabilistic model for content requests. The following regression-based hierarchical (multilevel) probabilistic model is proposed:
[TABLE]
The first level of the model, (7a), is the Poisson observation distribution for content requests. At this level, the request for content is assumed to follow a Poisson distribution with natural parameter which is a function of its features. We note that the request rate is an exponential function of the natural parameter, . As we previously mentioned, it is expected that there is a similar request pattern between contents with similar features. This prior information is employed at the higher levels. In (7b), follows a normal distribution with mean and variance . By this assumption, we allow contents with exactly the same features to have different popularities which is possible in practice. At the higher level of the model, (7c), we assume that are realizations of function drawn from a Gaussian process with zero mean and kernel function . By this assumption, contents with similar features are encouraged to be correlated in the feature space.
III Bayesian Learning
III-A Inference
In this section, we exploit the Bayesian framework to learn the probabilistic model in (7). In other words, given the content request observations , we aim to update our belief about the model’s parameters . However, we cannot estimate the infinite-dimensional function and hence the focus is only on the realizations at . Moreover, to simplify the inference, we can integrate out from the model. By doing this, we have:
[TABLE]
where . Additionally, in practice, the available prior knowledge may not be enough to fix the parameters . A common approach to estimate these parameters is cross validation. However, this trial and error experiment may be tedious and computationally extensive. A very systematic way to learn these parameters is to model their uncertainty by a prior distribution. Since the values of must be positive, a natural choice would be Gamma priors:
[TABLE]
where and are respectively the shape and the rate of each Gamma distribution.
Fig.1 shows the graphical representation of the Bayesian model. The shaded node represents the observed requests and the plates represent multiple samples of random variables. The unshaded circle nodes indicate unknown quantities and the squares show the deterministic parameters of the model.
The inference of all unknown variables of the model is given by the Bayes rule as:
[TABLE]
where is the posterior distribution and the denominator is a normalization constant. Unfortunately, the normalization constant is intractable to compute and there is no closed-form expression for the posterior distribution. So, instead, we use a Monte Carlo Markov Chain (MCMC) method to approximate the posterior distribution. Specifically, we use the Hamiltonian Monte Carlo (HMC) method which has been one of the most successful MCMC methods to sample from an unnormalized distribution. Now, we give an overview of the HMC. The complete description can be found in [18].
HMC is based on the simulation of Hamiltonian dynamics as a method to generate a sequence of samples from a desired -variate distribution by exploring its sample space. It combines gradient information of and auxiliary variables, , with density . The Hamiltonian function is then defined as:
[TABLE]
where is the negative log of the unnormalized and is usually assumed to be the identity matrix. The physical analogy of (11) is the Hamiltonian dynamics which describe the sum of the potential energy (the first term) and the kinetic energy (the last two terms).
Hamiltonian dynamics are simulated by discretizing their continuous analogue equations using the leapfrog method. This discretization has two parameters, number of leapfrog steps and step-size . The full description of a movement in HMC which is from a current state (sample) to a new state is depicted in Alg.1. HMC is only applicable for differentiable and unconstrained variables. However, in (10), there are some variables, , that must be positive. To handle this issue, we exploit the exponential-transformation where instead of , we use with serving as an unconstrained auxiliary variable. Note that to use these transformations, we also need to compute the Jacobian determinant as a result of the change of random variables.
By defining and as the posterior distribution (10), the negative log of unnormalized (after the exponential-transformation) is given by:
[TABLE]
Also, the gradient of (III-A), which is required in Alg.1, can be easily computed by using matrix derivatives [19]:
[TABLE]
Once, we collect enough samples from the HMC, any function of the posterior distribution moments can be computed. The initial MCMC samples are usually discarded because they may be far away from the true distribution. These samples are called burn-in samples.
Nevertheless, our goal is not just to learn the parameters of the model based on the training set but is to make prediction about the possible content request values in future. The next subsection explains how this can be performed.
III-B Prediction
Here, we aim to perform prediction in two ways. The first one is to predict the requests for the existing contents. This can be performed using the posterior predictive distribution (distribution of a new request) given in (13):
[TABLE]
where is a Poisson distribution and is the marginal posterior distribution of . However, we would like to make a point prediction rather than dealing with the whole predictive distribution. The best guess for a point estimation in the Bayesian context is based on risk (or loss) minimization [20, Chapter 2]. In other words, a loss function is defined which specifies the loss incurred by guessing the value when the actual value is . The most common loss evaluation metric is the quadratic loss. The value of that minimizes this risk function is the mean of the predictive distribution which can be approximated as:
[TABLE]
The second prediction task is to predict the popularity of a newly-added content that may enter the system. This can be calculated by a second type of posterior predictive distribution defined as:
[TABLE]
where is the feature vector of the new content. To compute , we note that the joint distribution of is a Normal distribution with zero mean and covariance matrix:
[TABLE]
where . Based on the properties of Normal distributions, the conditional distribution is a Normal distribution with mean and variance:
[TABLE]
Again, the optimal predictive value for (15) considering the quadratic loss is its mean. It should be noted that (15) is the distribution of the natural parameter of a new content. The point estimation of the request rate can be approximated as:
[TABLE]
IV Simulation Results
In this section, we present our simulation results to show the performance of the proposed probabilistic content request model denoted by ”Bayesian Poisson-GP”. To compare our results, we use the independent Poisson model with MLE in (1) denoted by ”MLE Poisson” as a benchmark. As far as the HMC technique is concerned, we set and and ran it for 5000 samples where the first 2500 samples were considered as the burn-in samples. The number of features is and specifically features , , are binary whose values are randomly generated from Bernoulli distributions with parameters , and for all , respectively. Feature is continuous and generated from a Normal distribution with zero mean and unit variance for all . Moreover, the parameters of the Kernel function (6) are and .
Fig.2 shows the root mean square error (RMSE) of the popularity predictive type 1 in (14) versus the number of observations in the training set, . It can be seen that the Bayesian Poisson-GP significantly performs better than the MLE Poisson. We can also observe that as the number of contents increases, the Bayesian Poisson-GP performance is improved. This is because as increases, the Gaussian process can learn better the relationship between the popularities and the features.
Now, we investigate the performance of our model in terms of how well it can predict the popularity of a new content (popularity predictive type 2 in (16)). The feature of the new content is randomly generated with the same process as for the existing contents. Fig. 3 shows the RMSE of the predicted popularity of the new content versus . As we see the performance of the model improves when the size of the training set or the number of contents increase. In this scenario, there is no explicit way to use the content features in the MLE Poisson to make prediction about the popularity of a new content, therefore we are unable to compare our model with it.
Next, we show the accuracy of Kernel parameter learning efficiency of our model. Tables I and II show the estimated mean values of the kernel function parameters. As we expected, it is observed that as the number of observations increases we get closer to the true values. However, from the tables, the accuracy improvement of the parameters is largely affected by the number of contents. For example, for feature , which does not affect the outcome of the model, the value of its scale variation, , has a better estimation at for in comparison with . These results confirm our previous simulations that as increases the Gaussian process gets more accurate and consequently shows a better prediction performance. The reason for this behavior is that by increasing the number of observations in the feature space increases which results in a better prediction accuracy.
V Conclusions
In this paper, we proposed a flexible model for modeling the content requests and predicting their popularity. We proposed a multilevel probabilistic model, the Poisson regressor based on Gaussian process, that can capture the similarity between contents in terms of their features. We utilized Bayesian learning to obtain the parameters of the model because it is robust against overfitting and therefore efficient in edge-caching system where overfitting is a big challenge due to small number of request observations. Then, two posterior predictive distributions were specified for prediction purposes. In the simulation results, we showed that the Bayesian Poisson-Gaussian process structure significantly outperforms the MLE independent Poisson in terms of content popularity prediction.
Acknowledgment
This work was funded by the National Research Fund (FNR), Luxembourg under the projects ”LISTEN” and ”PROCAST”. This work was also supported by the European Research Council (ERC) under the project ”AGNOSTIC”.
The reference list from the paper itself. Each links out to its DOI / PubMed record.
- 1[1] C. V. N. Index, “Global mobile data traffic forecast update, 2016–2021 white paper, accessed on may 2, 2017.”
- 2[2] K. Shanmugam, N. Golrezaei, A. G. Dimakis, A. F. Molisch, and G. Caire, “Femtocaching: Wireless content delivery through distributed caching helpers,” IEEE Transactions on Information Theory , vol. 59, no. 12, pp. 8402–8413, 2013.
- 3[3] X. Ge, H. Cheng, M. Guizani, and T. Han, “5G wireless backhaul networks: challenges and research advances,” IEEE Network , vol. 28, no. 6, pp. 6–11, 2014.
- 4[4] X. Peng, J.-C. Shen, J. Zhang, and K. B. Letaief, “Joint data assignment and beamforming for backhaul limited caching networks,” in Personal, Indoor, and Mobile Radio Communication (PIMRC), 2014 IEEE 25th Annual International Symposium on . IEEE, 2014, pp. 1370–1374.
- 5[5] T. X. Vu, S. Chatzinotas, and B. Ottersten, “Edge-caching wireless networks: Performance analysis and optimization,” IEEE Transactions on Wireless Communications , 2018.
- 6[6] M. A. Maddah-Ali and U. Niesen, “Fundamental limits of caching,” IEEE Transactions on Information Theory , vol. 60, no. 5, pp. 2856–2867, 2014.
- 7[7] W. Han, A. Liu, and V. K. Lau, “PHY-caching in 5G wireless networks: Design and analysis,” IEEE Communications Magazine , vol. 54, no. 8, pp. 30–36, 2016.
- 8[8] S. Müller, O. Atan, M. van der Schaar, and A. Klein, “Context-aware proactive content caching with service differentiation in wireless networks,” IEEE Transactions on Wireless Communications , vol. 16, no. 2, pp. 1024–1036, 2017.
