Reinforcement Learning to Minimize Age of Information with an Energy Harvesting Sensor with HARQ and Sensing Cost
Elif Tu\u{g}\c{c}e Ceran, Deniz G\"und\"uz, and Andr\'as Gy\"orgy

TL;DR
This paper investigates optimal scheduling policies for energy-harvesting sensors to minimize the age of information, using reinforcement learning in unknown environments and analyzing feedback mechanisms.
Contribution
It introduces a reinforcement learning approach for AoI minimization in energy-harvesting sensors with unknown parameters, extending prior work with real-time learning capabilities.
Findings
Reinforcement learning effectively minimizes AoI in unknown environments.
Optimal policies depend on feedback mechanisms and system parameters.
Numerical results validate the proposed methods' effectiveness.
Abstract
The time average expected age of information (AoI) is studied for status updates sent from an energy-harvesting transmitter with a finite-capacity battery. The optimal scheduling policy is first studied under different feedback mechanisms when the channel and energy harvesting statistics are known. For the case of unknown environments, an average-cost reinforcement learning algorithm is proposed that learns the system parameters and the status update policy in real time. The effectiveness of the proposed methods is verified through numerical results.
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.
Reinforcement Learning to Minimize Age of Information with an Energy Harvesting Sensor with HARQ and Sensing Cost
Elif Tuğçe Ceran, Deniz Gündüz, and András György
Department of Electrical and Electronic Engineering, Imperial College London
Email: {e.ceran14, d.gunduz, a.gyorgy}@imperial.ac.uk
Abstract
The time average expected age of information (AoI) is studied for status updates sent from an energy-harvesting transmitter with a finite-capacity battery. The optimal scheduling policy is first studied under different feedback mechanisms when the channel and energy harvesting statistics are known. For the case of unknown environments, an average-cost reinforcement learning algorithm is proposed that learns the system parameters and the status update policy in real time. The effectiveness of the proposed methods is verified through numerical results.
I Introduction
There has been a growing interest in minimizing the age of information (AoI) of energy harvesting (EH) communication systems [1, 2, 3, 4, 5, 6, 7, 8, 9]. The AoI quantifies the staleness of the information at the receiver, and is defined as the time elapsed since the generation time of the most recent status update packet successfully received at the receiver.
Prior works have investigated online [1, 3, 7] and offline [1, 5] methods for different scenarios in order to optimize the timeliness of information under the energy causality constraints in EH systems. It is shown in [3, 7, 9] that the optimal policy is of a threshold type for a finite-size battery when the cost of sensing (monitoring) the status of a process is not considered or assumed to be zero. Until recently, prior literature in the AoI framework assumed that the cost of sensing (monitoring) the status of a process is negligible compared to the cost of transmitting the status update. However, in most practical sensing systems acquiring a new sample of the underlying process of interest also has a considerable energy cost. The sampling/sensing cost has been taken into account in [10], where a status update system with ARQ and an unlimited energy source is considered. Closed form expressions are presented for the energy consumption and AoI, assuming that a packet is re-transmitted until either it is received, or a prescribed maximum number of transmissions is reached.
In this paper, similarly to [10], we study a status update system considering both the sensing and transmission energy costs. We consider an EH transmitter, which uses the energy harvested from the environment to power the sensing and communication operations. Moreover, we consider a hybrid automatic repeat request (HARQ) protocol, where the partial information obtained from previous unsuccessful transmission attempts is combined to increase the decoding probability.
In our previous work, we studied status-update systems with HARQ under a transmission-rate constraint [11, 12, 13]. Here we consider the intermittent availability of energy and find the online status updating policy to minimize the long-term average AoI at the receiver, subject to the energy causality constraints at the transmitter. However, in many practical scenarios the statistical information about either the energy arrival process or the channel conditions are not available or may change over time [14]. Previous work on EH communication systems without a-priori information on random processes governing the system exploited reinforcement learning (RL) methods in order to maximize throughput or minimize delay [15, 16].
To adapt the status-update scheme to the unknown energy arrival process and channel statistics, we propose a learning theoretic approach using RL algorithms. In particular, we consider a value-based RL algorithm, GR-learning [17], and a policy-based RL algorithm, finite-difference policy gradient [18], and compare their performances with the relative value iteration (RVI) algorithm which assumes a-priori knowledge on the system characteristics. We propose a suboptimal threshold policy and demonstrate that policy gradient algorithm exploiting the structural characteristics of a threshold policy outperforms GR-learning algorithm. We investigate the effects of the EH process on the average AoI, and we show by simulations that temporal correlations in EH increase the average AoI significantly. We compare the average AoI with EH with the average AoI under an average transmission constraint [11] and demonstrate that the performance of RH transmitter approximates to the one with average transmission constraint for a battery with unlimited capacity and zero sampling/sensing cost.
II System Model
We consider a time-slotted status update system over an error-prone wireless communication link (see Figure 1). The transmitter (TX) can sense the underlying time-varying process and generate a status update at each time slot at a certain energy cost. Status updates are communicated to the receiver (RX) over a time-varying wireless channel. Each transmission attempt of a status update takes constant time, which is assumed to be equal to the duration of one time slot.
The AoI measures the timeliness of the status information at the receiver, and is defined at any time slot as the number of time slots elapsed since the generation time of the most up-to-date packet successfully decoded at the receiver. Formally, the AoI at the receiver at time is defined as , where a maximum value on the AoI is imposed to limit the impact of the AoI on the performance after some level of staleness is reached.
We assume that the channel changes randomly from one time slot to the next in an independent and identically distributed (i.i.d.) fashion, and the instantaneous channel state information is available only at the receiver. We further assume the availability of an error- and delay-free single-bit feedback from the receiver to the transmitter for each transmission attempt. Successful reception of the status update at the end of time slot is acknowledged by an ACK signal (denoted by ), while a NACK signal is sent in case of a failure (denoted by ).
There are three possible actions the transmitter can take at each time slot : it can either sample and transmit a new status update (), remain idle () or retransmit the last transmitted status update (). If an ACK is received at the transmitter, we can restrict the action space to as retransmitting an already decoded status update is strictly suboptimal.
We consider the HARQ protocol: that is, the received signals from previous transmission attempts for the same packet are combined for decoding. The probability of error using retransmissions, denoted by , depends on and the particular HARQ scheme used for combining multiple transmission attempts (an empirical method to estimate is presented in [19]). As in any reasonable HARQ strategy, we assume that is non-increasing in the number of retransmissions ; that is, for all . Standard HARQ methods only combine information from a finite maximum number of retransmissions [20]. Accordingly, we consider a truncated retransmission count of a status update, denoted by for the status update transmitted at time , where ; that is, the receiver can combine information from the last retransmissions at most. We also assume that so that there is no previously transmitted packet at the transmitter at time .
At the end of each time slot , a random amount of energy is harvested and stored in a rechargeable battery at the transmitter, denoted by , following a first-order discrete-time Markov model, characterized by stationary probabilities , defined as . It is also assumed that , . Harvested energy is first stored in a rechargeable battery with a limited capacity of energy units and the energy harvested when the battery is full is lost. The energy consumption for status sensing is denoted by , while the energy consumption for a transmission attempt is denoted by .
The battery state at time , denoted by , and the energy causality constraints can be written as follows:
[TABLE]
where the indicator function is equal to if event holds, and zero otherwise. Eqn. (1) implies that the battery overflows if energy is harvested when the battery is full, while Eqn. (2) imposes that the energy consumed by sensing or transmission operations at time slot is limited by the energy available in the battery at the beginning of that time slot.
The age of the most recently generated status update at the transmitter at the beginning of time slot resets to if a new status update is generated at time slot , and increases up to otherwise, i.e.,
[TABLE]
The AoI of the most recent successfully decoded packet at the receiver at time , , evolves as follows:
[TABLE]
We note that refers to the number of time slots elapsed since the generation of the most recently sensed status update at the transmitter side, while denotes the AoI of the most recently received status update at the receiver side. The system model also implies that whenever a new status update packet is generated, the previous packet at the transmitter is dropped and can not be retransmitted. The number of retransmissions is zero for a newly sensed and generated status update and increases up to as we keep retransmitting the same packet.
[TABLE]
The state of the system is formed by five components . At each time slot, the transmitter knows the state of the system and the goal is to find a policy which minimizes the expected average AoI at the receiver over an infinite time horizon, which is given by:
[TABLE]
III Markov Decision Process (MDP) and RVI
An average-cost finite-state MDP provides the necessary framework for modeling and solving the AoI minimization problem in (3). An MDP is defined by the quadruple \big{(}\mathcal{S},\mathcal{A},P,c\big{)} [21]: The finite set of states is and the finite set of actions are already defined. refers to the transition probabilities, where is the probability that action in state at time will lead to state at time , which is characterized by the EH statistics and channel error probabilities. The cost function , is the AoI at the receiver, and is defined as for any , , independent of the action .
We note that there exists an optimal stationary deterministic policy, , for this problem111For Markov chains corresponding to every stationary policy, there is only one recurrent class as the state is reachable from all other states (e.g., every transmission is successful but no EH is harvested for a period of time slots) from Theorem 8.4.3 of [21]. [21]. In particular, there exists a function , called the differential cost function for all , satisfying the following Bellman optimality equations for the average-cost finite-state finite-action MDP [21]:
[TABLE]
where is the next state obtained from after taking action , and represents the optimal achievable average AoI under policy . Note that the function satisfying (4) is unique up to an additive factor, and with selecting this additive factor properly, it also satisfies
[TABLE]
We also introduce the state-action cost function:
[TABLE]
Then an optimal policy, for any , takes the action achieving the minimum in (5):
[TABLE]
An optimal policy solving (4), (5) and (6) defined above can be found by relative value iteration (RVI) for finite-state finite-action average-cost MDPs from Section 8.5.5 of [21]:
Starting with an arbitrary initialization of , , and setting an arbitrary but fixed reference state , a single iteration of the RVI algorithm is given as follows:
[TABLE]
where , and denote the state-action value function, value function and differential value function for iteration , respectively. By Theorem 8.5.7 and Section 8.5.5 of [21], converges to , and converges to .
IV A Reinforcement Learning Approach
In most practical scenarios, channel error probabilities for retransmissions and the EH characteristics may not be known at the time of deployment, or may change over time. In this section, we assume that the transmitter does not know the system characteristics a-priori, and has to learn them. We employ two different online learning algorithms. First, we employ a value-based RL algorithm, namely GR-learning, which converges to an optimal policy; then, we consider a structured policy search algorithm, namely finite-difference policy gradient, which does not necessarily find the optimal policy but performs very well in practice, as demonstrated through simulations in Section V. We also note that GR-learning learns from a single trajectory generated during learning steps while policy gradient uses Monte-Carlo roll-outs for each policy update. Thus, GR-learning is more applicable to real-time systems.
IV-A GR-Learning with Softmax
The literature for average-cost RL is quite limited compared to discounted cost problems [22, 23]. For the average AoI minimization problem in (3), we employ a modified version of the GR-learning algorithm proposed in [17], as outlined in Algorithm 1, with Boltzmann (softmax) exploration. The resulting algorithm is called GR-learning with softmax.
Notice that, by only knowing , one can find the optimal policy using (6) without knowing the transition probabilities characterized by and . Thus, GR-learning with softmax starts with an initial estimation of and finds the optimal policy by estimating state-action values in a recursive manner. In the iteration, after taking action , the transmitter observes the next state , and the instantaneous cost value . Based on this, the estimate of is updated by a weighted average of the previous estimate and the estimated expected value of the current policy in the next state . Moreover, we update the gain at every time slot based on the empirical average of AoI.
In each time slot, the learning algorithm
- •
observes the current state ,
- •
selects and performs an action ,
- •
observes the next state and the instantaneous cost ,
- •
updates its estimate of using the current estimate of by
[TABLE]
where is the update parameter (learning rate) in the iteration, and depends on the function , which is the number of times the state–action pair was visited till the iteration.
- •
updates its estimate of based on the empirical average as follows:
[TABLE]
where is the update parameter in the iteration.
The transmitter action selection method should balance the exploration of new actions with the exploitation of actions known to perform well. In particular, the Boltzmann (softmax) action selection method, which chooses each action randomly relative to its expected cost, is used in this paper as follows:
[TABLE]
Parameter in (12) is called the temperature parameter and decays exponentially with decay parameter . High corresponds to more uniform action selection (exploration) whereas low is biased toward the best action (exploitation). According to Theorem 2 of [17], if , satisfy , , , GR-Learning converges to an optimal policy.
IV-B Finite-Difference Policy Gradient
GR-learning in Section IV-A is a value-based RL method, which learns the state-action value function for each state-action pair. In practice, can be large, which might slow down the convergence of GR-learning due to a large state-space.
In this section, we are going to simplify the problem and obtain a structured possibly sub-optimal policy, which can be learned via the policy gradient method [18]. We make two assumptions on the policy space in order to obtain a more efficient learning algorithm:
- •
We assume that a packet is retransmitted until it is successfully decoded, provided that there is enough energy in the battery, that is, the transmitter is not allowed to preempt an undecoded packet and transmit a new one.
- •
The solution to the simplified problem is threshold-type, that is,
[TABLE]
for some .
Note that if () for (); that is, . This ensures that energy causality constraints in (2) hold. Other thresholds will be determined using policy gradient.
In order to employ the policy gradient method, we approximate the policy by a parameterized smooth function with parameters , and convert the discrete policy search problem into estimating the optimal values of some continuous parameters, which can be numerically solved by stochastic approximation algorithms [24].
In particular, with a slight abuse of notation, we let denote the probability of taking action () if (), and consider the parameterized sigmoid function:
[TABLE]
We note that and as . Therefore, in order to converge to a deterministic policy , can be taken as a sufficiently small constant, or can be decreased gradually to zero. The total number of parameters to be estimated is minus the parameters corresponding to () for () due to energy causality constraints as stated previously.
With a slight abuse of notation, we map the parameters to a vector of size . Starting with some initial estimates of , the parameters can be updated in each iteration using the gradients as follows:
[TABLE]
where the step size parameter is a positive decreasing sequence and satisfies the first two convergence properties given at the end of Section IV-A.
Computing the gradient of the average AoI directly is not possible; however, several methods exist in the literature to estimate the gradient [24]. In particular, we employ the finite-difference policy gradient [18] method. In this method, the gradient is estimated by estimating at slightly perturbed parameter values. First, a random perturbation vector of size is generated according to a predefined probability distribution, e.g., each component of is an independent Bernoulli random variable with parameter . The thresholds are perturbed with a small amount in the directions defined by to obtain . Then, empirical estimates of the average AoI corresponding to the perturbed parameters , obtained from Monte-Carlo rollouts, are used to estimate the gradient:
[TABLE]
where denotes the transpose of vector .
V Simulation Results
In this section, we provide numerical results for all the proposed algorithms, and compare the achieved average AoI. Motivated by previous research on HARQ [25], [19], [20], we assume that the decoding error reduces exponentially with the number of retransmissions, that is, for some , where denotes the error probability of the first transmission and is the retransmission count (set to [math] for the first transmission). The exact value of the rate depends on the particular HARQ protocol and the channel model. Following the IEEE 802.16 standard[20], the maximum number of retransmissions used for decoding is set to . In the following experiments, and are set to . and are both assumed to be constant and equal to 1 unit of energy unless otherwise stated. is set to .
We choose the exact step sizes for the learning algorithms by fine-tuning in order to balance the algorithm stability in the early time steps with nonnegligible step sizes in the later time steps. In particular, we use step size parameters of , where and (which satisfy the convergence conditions) and choose and such that the oscillations are low and the convergence rate is high. We have observed that a particular choice of parameters gives similar performance results for scenarios addressed in simulations results.
V-A Uncorrelated EH
We first investigate the average AoI with HARQ when the EH process, , is i.i.d. over time with probability distribution , . The RVI algorithm in Section III is employed, and the effects of the battery capacity , energy consumption of sensing , and on the average AoI are shown in Figure 2. As expected, the average AoI increases with decreasing , decreasing and increasing . We note that, when and , the problem defined in (3) corresponds to minimizing the average AoI under an average transmission rate constraint , studied in [11, 13]. The average AoI under average transmission rate constraint () is also shown in Figure 2.
Figure 3 shows the evolution of the average AoI over time when the average-cost RL algorithms are employed. As a baseline, we have also included the performance of a greedy policy, which sends a new status update whenever there is sufficient energy for both sensing and transmission. It retransmits the last transmitted status update when the energy in the battery is sufficient only for transmission, and it remains idle otherwise; that is, if , if and if . It can be observed that the average AoI achieved by the proposed RL algorithms, converge to values close to the one obtained from the RVI algorithm, which has a priori knowledge of and , while the AoI of the greedy algorithm is significantly higher. Although the policy gradient algorithm based on threshold policy does not allow preemption of an undecoded status update, it performs better than GR-learning since it tries to learn significantly smaller number of threshold values (i.e., ) than GR-learning which learns one value for each state-action pair (i.e., ).
V-B Temporally Correlated EH
Next, we investigate the performance when the EH process has temporal correlations. A symmetric two-state Markovian EH process is assumed, such that and . That is, if the transmitter is in harvesting state, it is more likely to continue harvesting energy, and vice versa for the non-harvesting state.
Figure 4 illustrates the policy obtained by RVI. As it can be seen from the figure, the resulting policy is less likely to transmit if the battery level or the AoI is low. Moreover, the policy tends to retransmit the previous update rather than sensing a new update when the battery level is low and the AoI is high. When the system is in the non-harvesting state (i.e., ), the transmitter is more conservative in transmitting the status updates compared to the case , e.g., it might not transmit even if the battery is full depending on the AoI level.
Figure 5 shows the evolution of the average AoI over time when the average-cost RL algorithms are employed. It can be observed again that the average AoI achieved by the learned threshold parameters in Section IV-B, denoted by policy gradient in the figure, performs very close to the one obtained from the RVI algorithm, which has a priori knowledge of and . GR-learning, on the other hand, outperforms the greedy policy but converges to the optimal policy much more slowly, and the gap between the two RL algorithms is even longer compared to the i.i.d. case. Tabular methods in RL, like GR-learning, need to visit each state-action pair infinitely often for RL to converge [22]. GR-learning in the case of temporally correlated EH does not perform as well as in the i.i.d. case since the state space becomes larger with the addition of the EH state.
Next, we investigate the impact of the burstiness of the EH process, measured by the correlation coefficient between and . Figure 6 illustrates the performance of the proposed RL algorithms for different correlation coefficients, which can be computed easily for the 2-state symmetric Markov chain; that is, . Note that corresponds to the i.i.d. EH with . We note that the average AoI is minimized by transmitting new packets successfully at regular intervals, which has been well investigated in previous works [1, 11, 2]. Intuitively, for highly correlated EH, there are either successive transmissions or successive idle time slots, which increases the average AoI. Hence, the AoI is higher for higher values of . Figure 6 also shows that both RL algorithms result in much lower average AoI than the greedy policy and policy gradient RL outperforms GR-learning since it benefits from the structural characteristics of a threshold policy.
VI Conclusions
We have considered an EH system with a finite size battery and investigated scheduling policies transmitting time-sensitive data over a noisy channel with the average AoI as the performance measure, which quantifies the timeliness of the data available at the receiver. In addition to identifying a RVI solution for the optimal policy when the system characteristics are known, efficient RL algorithms are also presented for practical applications when the system characteristics may not be known in advance. The effects of battery size, EH characteristics and the HARQ structure on the average AoI are investigated through numerical simulations. The algorithms adopted in this paper are relevant to other systems concerning the timeliness of information or those powered by renewable energy sources.
The reference list from the paper itself. Each links out to its DOI / PubMed record.
- 1[1] B. T. Bacinoglu, E. T. Ceran, and E. Uysal-Biyikoglu, “Age of information under energy replenishment constraints,” in Inf. Theory and Applications Workshop (ITA) , Feb 2015, pp. 25–31.
- 2[2] R. D. Yates, “Lazy is timely: Status updates by an energy harvesting source,” in IEEE Int’l Symposium on Information Theory (ISIT) , 2015, pp. 3008–3012.
- 3[3] B. T. Bacinoglu and E. Uysal-Biyikoglu, “Scheduling status updates to minimize age of information with an energy harvesting sensor,” Co RR , vol. abs/1701.08354, 2017.
- 4[4] A. Arafa, J. Yang, and S. Ulukus, “Age-minimal online policies for energy harvesting sensors with random battery recharges,” Co RR , vol. abs/1802.01563, 2018.
- 5[5] A. Arafa and S. Ulukus, “Age minimization in energy harvesting communications: Energy-controlled delays,” Co RR , vol. abs/1712.03945, 2017.
- 6[6] X. Wu, J. Yang, and J. Wu, “Optimal status update for age of information minimization with an energy harvesting source,” IEEE Tran. on Green Comms. and Networking , vol. 2, no. 1, pp. 193–204, March 2018.
- 7[7] B. T. Bacinoglu, Y. Sun, E. Uysal-Biyikoglu, and V. Mutlu, “Achieving the age-energy tradeoff with a finite-battery energy harvesting source,” Co RR , vol. abs/1802.04724, 2018.
- 8[8] S. Feng and J. Yang, “Age of information minimization for an energy harvesting source with updating erasures: With and without feedback,” Co RR , 2018.
