1
A Persistence-Aware Framework for Age Violation Control in Wireless Status Update Systems
arXiv:2605.13002v1 [cs.NI] 13 May 2026
Haoyuan Pan, Member, IEEE, Chen Chen, Shiyong Zhou, Kun Chen, and Tse-Tin Chan, Member, IEEE
Abstract—Timely and reliable status updates are essential for emerging QoS-sensitive wireless applications. Common age of information (AoI)-based metrics, such as average AoI and age violation rate (AVR), characterize time-averaged freshness or violation frequency but do not explicitly capture the temporal persistence of consecutive age violations, which can be critical in safety-sensitive wireless applications. To address this limitation, we develop a persistence-aware reliability framework based on the consecutive age violation rate (C-AVR) vector, whose components quantify AoI threshold violations over consecutive time windows of different lengths. Through flexible weighting schemes, the proposed framework unifies reliability objectives ranging from average persistence to tail-sensitive performance. Optimizing weighted C-AVR objectives is challenging because consecutive violations are temporally correlated, leading to sparse learning signals. To address this issue, we develop a distributional reinforcement learning approach based on a quantile regression dueling double deep Q-network (QR-D3QN). By modeling a quantile-based return distribution rather than only a scalar expected return, QR-D3QN provides richer valueestimation signals for rare but prolonged violation sequences under stochastic packet arrivals, unreliable channels, and transmission cost constraints. Simulation results demonstrate that QR-D3QN consistently outperforms expectation-based baselines across a wide range of weighting schemes and system settings, with particularly significant gains under tail-sensitive persistence objectives. Component-wise analysis further shows that distributional value learning substantially improves reliability across multiple persistence scales, especially for long consecutive violation sequences. Overall, our results establish the proposed C-AVR framework as an effective foundation for persistenceaware reliability evaluation and demonstrate the advantages of distributional reinforcement learning for wireless status update systems. Index Terms—Age of information, distributional reinforcement learning, reliability, wireless scheduling.
I. I NTRODUCTION Ensuring timely and reliable status updates is a fundamental requirement in wireless communication systems supporting industrial automation, autonomous driving, intelligent transportation, and real-time monitoring. In such applications, outdated information may degrade control accuracy, compromise safety, or destabilize closed-loop operation. To characterize information freshness, the age of information (AoI) has emerged as a widely adopted metric, measuring the time elapsed since H. Pan, C. Chen, S. Zhou, and K. Chen are with the College of Computer Science and Software Engineering, Shenzhen University, Shenzhen, China (e-mails: [email protected]; [email protected]; [email protected]; [email protected]). T.-T. Chan is with the Department of Mathematics and Information Technology, The Education University of Hong Kong, Hong Kong, China (e-mail: [email protected]).
the generation of the most recently received update [1]– [3]. Extensive studies have investigated AoI-aware scheduling, sampling, queue management, and resource allocation in wireless networks [4]–[7]. Most existing works focus on average AoI or related mean-based freshness metrics due to their analytical tractability and intuitive interpretation of longterm system performance [8]–[11]. However, average AoI is often insufficient for quality-ofservice (QoS)-sensitive applications. In many practical systems, occasional but severe freshness degradation may dominate operational risk even when the average AoI remains acceptable. To address this limitation, reliability-oriented freshness metrics have been introduced. In particular, the age violation rate (AVR) measures the probability that AoI exceeds a predefined threshold and therefore provides a thresholdbased characterization of timeliness reliability [12]–[14]. Compared with average AoI, AVR is more directly connected to outage-like QoS guarantees and has been studied in wireless systems with stochastic arrivals, unreliable channels, and constrained scheduling resources [15], [16]. Nevertheless, existing violation-based metrics primarily quantify the frequency of violations, while largely ignoring their temporal persistence. This limitation is particularly important in safety-critical systems. Consider two systems with the same AVR: in one system, violations occur sporadically and are rapidly corrected; in the other, violations occur in long uninterrupted runs. Although both systems exhibit identical violation probabilities, their operational risks can differ substantially. In applications such as autonomous control and industrial monitoring, prolonged information staleness may lead to sustained control errors, persistent sensing failures, or safety hazards [17]. Therefore, reliability-aware freshness control should characterize not only whether AoI violations occur, but also how long they persist consecutively. Motivated by this observation, we develop a persistenceaware reliability framework based on a vectorized characterization of consecutive AoI violations. Specifically, we introduce the consecutive age violation rate (C-AVR) vector, whose k-th component measures the probability that AoI exceeds a threshold over k consecutive time slots. This vector representation provides a persistence profile of age violations across multiple time scales. Scalar reliability metrics arise as special cases under appropriate weightings on this vector; in particular, the scalar C-AVR for a fixed k corresponds to a one-hot weighting. More generally, flexible weighting schemes over the C-AVR vector enable a unified treatment of reliability objectives, ranging from average persistence (uniform weighting) to tailsensitive criteria that emphasize long violation runs (e.g.,
2
exponential weights). We refer to this scalar objective as the weighted C-AVR. To the best of our knowledge, this is the first AoI-based framework that models and optimizes consecutive violation persistence across multiple time scales. We study this framework in a multiuser wireless status update system, where a central scheduler selects users under transmission cost constraints, stochastic packet arrivals, and unreliable channels. Optimizing persistence-aware C-AVR objectives in this setting is challenging. Consecutive violation events are temporally coupled: once AoI exceeds the threshold, the violation tends to persist unless a successful update occurs. As a result, unlike average AoI or standard AVR, system performance depends not only on frequent and short violations, but also on rare and prolonged violation sequences. Moreover, system dynamics such as packet arrivals and channel reliability are often unknown or difficult to model accurately, limiting the applicability of classical model-based optimization methods [18]–[20]. These challenges motivate the use of deep reinforcement learning (DRL), which has recently emerged as an effective framework for wireless scheduling and resource allocation under uncertainty [21], [22]. Existing DRL-based wireless scheduling methods commonly employ expectation-based value-learning architectures, such as deep Q-networks (DQN) and their variants [23], [24]. Such methods learn scheduling policies by optimizing the expected long-term return and have demonstrated strong performance in many communication and networking problems. DRL has also been applied to AoI-aware scheduling, particularly in scenarios where explicit system models are unavailable [22]. However, most existing DRLbased AoI studies focus on average AoI, average cost, or single-threshold reliability objectives, all of which can be adequately represented through scalar expected returns. Persistence-aware reliability objectives fundamentally differ from these formulations. Weighted C-AVR objectives induce heterogeneous return distributions dominated by both frequent short violation runs and rare but prolonged persistence events. Under such objectives, expectation-based value learning may obscure critical tail behavior. Different scheduling actions may produce similar expected AoI performance while exhibiting substantially different risks of generating long consecutive violation sequences. Since expectation-based DRL represents the future return using only a scalar expectation, important persistence-related information in the return distribution may be obscured. This limitation becomes particularly pronounced under exponential weighting schemes, where long violation runs dominate system reliability. To address this issue, we adopt distributional reinforcement learning, which models the full return distribution rather than only its expectation [25]–[27]. By preserving information about return variability during value learning, distributional DRL is well-suited to persistence-aware reliability objectives whose rewards are induced by temporally correlated violation events. In particular, quantile-regression-based methods such as QR-DQN approximate the return distribution using learnable quantiles, enabling accurate modeling of asymmetric and high-variance return structures induced by consecutive violation events [26]. Recent studies have demonstrated
the effectiveness of distributional DRL in uncertainty-aware wireless control, risk-sensitive scheduling, and reliabilityconstrained resource allocation [28]–[35]. Nevertheless, its role in persistence-aware AoI control has remained largely unexplored. Building on these observations, we develop a quantile regression-based dueling double deep Q-network (QR-D3QN) tailored for persistence-aware wireless scheduling. The proposed architecture combines Double Q-learning to mitigate overestimation bias [23], a dueling network structure to improve state-value and advantage estimation [24], and quantile regression to model the full return distribution [26]. These components jointly enable more stable learning under sparse and high-variance persistence-aware feedback, thereby reducing prolonged AoI violations while satisfying transmission cost constraints. Extensive simulations suggest that multi-scale persistence objectives can benefit from distributional value representations, and that distributional DRL provides an effective learning approach. Specifically, the proposed QR-D3QN consistently outperforms expectation-based baselines (i.e., DQN and D3QN) across a wide range of system configurations and weighting schemes, including uniform, exponential, and one-hot weighted C-AVR objectives. The performance gains become increasingly significant as the optimization objective places greater emphasis on long violation windows, confirming the importance of modeling the return distribution for persistence-aware reliability control. Furthermore, componentwise analysis of the C-AVR vector reveals that QR-D3QN achieves particularly substantial improvements at large persistence scales, demonstrating its effectiveness in suppressing prolonged consecutive AoI violations that dominate tail reliability. We further compare the proposed approach with the model-based drift-plus-penalty (DPP) method [36]. The results show that DRL methods achieve performance comparable to DPP under short persistence objectives, while QR-D3QN exhibits clear advantages when the objective emphasizes long violation windows. Moreover, all performance gains are achieved under the same transmission cost constraint, indicating that the improvements arise from more effective scheduling decisions rather than increased resource consumption. Overall, the results validate both the proposed persistence-aware CAVR framework and the effectiveness of distributional DRL for reliability-aware wireless scheduling. The main contributions of this work are summarized as follows: • We introduce the C-AVR vector to characterize AoI violations across multiple persistence scales, thereby capturing the full temporal persistence profile of information staleness. Based on this representation, we further develop the weighted C-AVR framework, which unifies average persistence, tail-sensitive reliability, and single-scale persistence objectives under flexible weighting schemes. • We demonstrate empirically that persistence-aware CAVR objectives induce heterogeneous and high-variance learning signals for which distributional value representations can be more effective than expectation-based DRL methods. To address this issue, we develop a
3
distributional DRL framework based on QR-D3QN for persistence-aware wireless scheduling under stochastic arrivals, unreliable channels, and transmission cost constraints. • We conduct extensive simulations under multiple weighting schemes and heterogeneous system settings. The results demonstrate that distributional value learning consistently outperforms expectation-based baselines, with particularly significant gains for long persistence windows and tail-sensitive reliability objectives. Componentwise analysis further confirms the effectiveness of the proposed method in suppressing prolonged AoI violation sequences across multiple temporal scales. II. S YSTEM M ODEL AND P ROBLEM F ORMULATION A. System Model We consider a time-slotted wireless status update system with stringent information timeliness QoS requirements, consisting of M independent information sources, indexed by {1, 2, ..., M }, and their corresponding receivers coordinated by a central scheduler, as illustrated in Fig. 1. Time is indexed by discrete time slots t = 1, 2, . . . . At the beginning of each slot, every source independently generates at most one fresh status update packet according to a Bernoulli process, i.e., source m generates a new packet with probability pm g . Newly generated packets are instantaneously delivered to the central node through an ideal uplink and overwrite any previously stored packet from the same source. Hence, the central node maintains a buffer of size M , consisting of M per-source entries, with one entry per source storing only the most recent packet. This per-source replacement structure reflects QoS-driven designs where only the freshest information is relevant [3]. Due to shared downlink resources, at most one packet can be transmitted in each time slot. A scheduled transmission to receiver m succeeds with probability pm s , independently across slots.1 Upon a successful transmission, the receiver sends an acknowledgment (ACK) through an error-free feedback link; otherwise, a negative acknowledgment (NACK) is returned. The freshness of information is quantified using the age of information (AoI), defined as the time elapsed since the generation of the most recently successfully received status update. To track information timeliness across the system, AoI is maintained at both the transmitter and the receiver for each source m ∈ {1, 2, ..., M }: • At the central node, the transmitter-side AoI for source m at slot t is the number of slots since the most recent packet from source m arrived at the buffer. The transmitterside AoI for source m at slot t, denoted by ∆s (t, m), is ∆s (t, m) = t − Ts (t, m), where Ts (t, m) denotes the arrival time of the most recent packet from source m. 1 For tractability, this work adopts a fixed transmission success probability
model, which is commonly used in the AoI literature. This abstraction enables clear evaluation of scheduling and decision-making mechanisms without introducing additional physical-layer complexities. The proposed CMDP and DRL framework can be readily extended to more realistic time-varying wireless channels, such as Rayleigh fading or Markov fading channels, due to its model-free nature.
A buffer of size M
1
Source 1
2
�1�
�1�
A CK
�� � Source M
Packet generation from sources to central node
M
/NAC
�� �
A CK Central node Packet transmission form central node to receivers
/NAC
K
K
Receiver 1
Receiver M
ACK/NACK feedback form receivers to central node
Fig. 1. A time-slotted wireless status update system in which a central node schedules update packets to different receivers.
The transmitter-side AoI is reset to zero upon a new packet arrival and increases by one in each subsequent slot otherwise. • At the receiver, the receiver-side AoI ∆r (t, m) measures the time elapsed since the generation of the most recently successfully received packet from source m. If an update packet from source m is successfully delivered at the end of slot t, then ∆r (t + 1, m) = ∆s (t, m) + 1; otherwise, ∆r (t + 1, m) = ∆r (t, m) + 1. The transmitter-side and receiver-side AoI are initialized to zero and one, respectively. Their coupled evolution is illustrated via an example in Fig. 2. The transmitter-side AoI ∆s (t, m) and the receiver-side AoI ∆r (t, m) are depicted by the red curve and the blue curve, respectively. Red arrows indicate packet-generation events at the beginning of slots. Blue arrows represent packet transmission outcomes at the end of slots, where solid arrows correspond to successful transmissions and dashed arrows denote failures. As shown in Fig. 2, the transmitter-side AoI is determined solely by packet generation events. Whenever a new packet is generated, ∆s (t, m) is reset to zero, and otherwise it increases linearly by one in each subsequent slot. In contrast, the receiver-side AoI depends jointly on the transmitter-side AoI and the transmission outcome. When a packet transmission is successful at slot t, the receiver-side AoI at the next slot is ∆r (t + 1, m) = ∆s (t, m) + 1, reflecting that the receiver obtains the most recently generated packet with a one slot delay. When a transmission attempt fails or when source m is not scheduled, the receiver-side AoI simply increases by one. Fig. 2 demonstrates that consecutive transmission failures or scheduling delays can lead to sustained growth of receiverside AoI even when fresh packets are available, highlighting the importance of capturing temporal persistence in reliability metrics. B. The Consecutive Age Violation Rate (C-AVR) Metric Most existing AoI-aware scheduling studies focus on longterm average performance metrics, with system average AoI being the most widely adopted measure of information freshness [10]. While effective for characterizing time-averaged performance, average AoI does not capture system reliability during rare but severe freshness-degradation events. To better
4
AoI 6 5 4 3 2 1 0
1
2
3
4
5
7
6
8
9
10
11
12
t
Fig. 2. Sample AoI evolution for source m. The red curve shows the transmitter-side AoI ∆s (t, m), and the blue curve shows the receiver-side AoI ∆r (t, m). Red arrows indicate packet generation events at the beginning of slots, while blue arrows denote packet transmission outcomes at the end of slots: solid for successful transmissions and dashed for failures.
capture reliability, recent works have introduced the age violation rate (AVR) metric, which measures the fraction of time slots in which the AoI exceeds a predefined threshold ζ, T
Φr = lim sup T →∞
M
1 XX E [1[∆r (t, m) > ζ]] . M T t=1 m=1
(1)
where 1[·] denotes the indicator function that returns 1 when its argument is true and 0 otherwise. Unlike average AoI, AVR captures the frequency of threshold violations rather than their magnitude, thereby providing a threshold-based notion of timeliness reliability. However, AVR fundamentally treats violations as independent events and ignores their temporal structure. In particular, it does not distinguish between isolated threshold crossings and sustained violation periods, despite their markedly different implications for system performance. From a QoS and safety perspective, prolonged sequences of outdated information are significantly more harmful than sporadic and shortlived violations, as they may induce sustained decision errors or trigger system-level failures. This observation highlights a key limitation of AVR and motivates the need for a reliability metric that explicitly captures the temporal persistence of AoI violations. To address this limitation, we put forth the consecutive age violation rate (C-AVR), which quantifies the probability that the AoI exceeds the threshold across multiple consecutive time slots. Let a positive integer k ≥ 1 denote a prescribed violation window length. A window of length k ending at slot t is said to be fully AoI violated for source m if 1[Vtk (m)] = 1, where Vtk (m) =
k \
{∆r (t − k + i, m) > ζ}.
(2)
i=1
That is, ∆r (t − k + i, m) > ζ for all i = 1, · · · , k. From a QoS standpoint, C-AVR characterizes the probability of timeliness outage events whose duration exceeds k consecutive slots, thereby directly capturing persistence in reliability degradation. The system-level C-AVR is defined as Ψkr = lim sup T →∞
T X M X 1 E 1[Vtk (m)] . M (T − k + 1) m=1 t=k
(3)
Optimizing C-AVR therefore enforces temporal QoS guarantees on information freshness, rather than merely improving average performance. C-AVR naturally recovers AVR as a special case when k = 1, while larger values of k place increasing emphasis on severe reliability degradation caused by prolonged violation sequences. Furthermore, since C-AVR depends on the joint temporal distribution of AoI exceedances, it cannot be inferred from AVR alone. This highlights that persistence is an intrinsic dimension of reliability that is not captured by existing metrics. To the best of our knowledge, no such window-based formulation that explicitly characterizes violation persistence has been systematically developed in the AoI scheduling literature. C. C-AVR Vector and Weighted Reliability Objective While the C-AVR defined in (3) captures persistence over a fixed window length k, it remains inherently a single-scale characterization. In practice, however, violations of different durations correspond to different levels of operational risk. Short consecutive violations may be tolerable in certain applications, whereas long violation runs can lead to sustained performance degradation or even system instability. Therefore, a metric defined at a single persistence scale is insufficient to fully characterize system reliability. We introduce a vectorized characterization of persistence, called the C-AVR vector, in which each component corresponds to a distinct violation window length. Specifically, for a maximum window length kmax , the C-AVR vector is defined as Ψ = Ψ1r , Ψ2r , . . . , Ψkr max , (4) This representation provides a multi-scale persistence profile of AoI violations, capturing reliability behavior across a spectrum of temporal horizons rather than at a single scale. As such, it elevates C-AVR from an isolated scalar metric to a structured description of temporal reliability. To enable tractable optimization while preserving this multiscale structure, we introduce a weighted aggregation of the C-AVR vector. Let W = (w1 , w2 , ..., kmax ) denote the Pw kmax weighting vector satisfying wk ≥ 0 and k=1 wk = 1, where wk specifies the relative importance assigned to persistence at scale k, i.e., the C-AVR scalar Ψkr . The resulting scalar objective, referred to as the weighted C-AVR, is defined as Ψ̄ = WΨ⊤ =
kX max
wk Ψkr ,
(5)
k=1
Importantly, this formulation establishes a unified optimization framework over the persistence profile. By adjusting the weights, the scheduler can emphasize different persistence regimes, ranging from average persistence to tail-sensitive criteria that prioritize long violation runs. In this sense, conventional scalar formulations arise as special cases within the proposed framework. In particular, a one-hot weighting reduces the weighted C-AVR to a single-scale C-AVR metric, while more general weightings capture interactions across multiple persistence scales. In this work, we consider the following representative weighting schemes:
5
•
Uniform Weights assign equal importance across all persistence scales, i.e., wk =
1 , kmax
k = 1, 2, . . . , kmax ,
(6)
thus reflecting scenarios where both short and long violation sequences are considered equally relevant. • Exponential Weights assign progressively larger weights to longer violation windows, i.e., βk wk = Pkmax j=1
βj
,
β > 1,
cast problem (9) as a constrained Markov decision process (CMDP), which serves as the foundation for the proposed distributional reinforcement learning approach.
(7)
thereby emphasizing tail behavior. Consequently, the resulting weighted objective is particularly sensitive to long but rare violations, with β controlling the degree of tail emphasis. • One-hot Weights concentrate all weight on a single persistence scale ko , i.e., wko = 1 and wk = 0 for k ̸= ko . In this case, the weighted C-AVR reduces exactly to the scalar C-AVR at window length ko , demonstrating that the scalar formulation is a special case of the proposed vector framework.
III. CMDP F ORMULATION The scheduling problem formulated in Section II-D involves long-term persistence-aware reliability optimization under an average transmission cost constraint. Such a problem can be naturally modeled as a CMDP [37]. In this section, we present a complete CMDP formulation that is explicitly aligned with the weighted C-AVR objective introduced previously. The CMDP is defined as a sextuple (S, A, P, R, C, γ), where S is the state space, A is the action space, P : S × A × S → [0, 1] is the state transition probability, R : S ×A → R is the reward function, C : S × A → R is the cost function, and γ is the discount factor. Each component is specified in detail below. a) State Space: The system state captures both the instantaneous information freshness and its temporal persistence at a given time slot. Since both packet generation and packet delivery affect AoI evolution, the state jointly includes the transmitter-side AoI, the receiver-side AoI, and the consecutive violation status for each source m. At the beginning of slot t, the system state is represented by a 3M -dimensional vector
D. Problem Formulation Let at ∈ {0, 1, ..., M } denote the scheduling action at slot t, where at = m indicates that source m is scheduled for transmission and at = 0 indicates no transmission. The longterm average transmission cost is defined as η = lim sup T →∞
T 1X
T t=1
E[1[at ̸= 0]].
(8)
which must satisfy η ≤ ηmax , where ηmax ∈ (0, 1] reflects the resource constraint. Combining the persistence-aware reliability objective with the resource constraint, we formulate the weighted C-AVR scheduling problem as min Ψ̄, {at }
s.t.
(9) η ≤ ηmax .
By optimizing the weighted C-AVR, the scheduler is explicitly driven to control the temporal persistence of AoI violations, rather than merely their instantaneous occurrence. This distinction is critical because unlike AVR, which primarily reflects frequency-based characteristics, the weighted C-AVR objective is inherently sensitive to the temporal correlation structure of violations and is therefore dominated by rare but high-impact events. Furthermore, the weighting mechanism enables a unified treatment of reliability objectives across multiple persistence scales. In particular, uniform weighting captures average persistence behavior, exponential weighting emphasizes tail risk, and one-hot weighting isolates singlescale reliability requirements. This unified formulation highlights the conceptual contribution of the proposed framework: reliability is not characterized by a single metric, but by a structured persistence profile that can be flexibly aggregated based on system requirements. In the next section, we
st = ∆s (t, 1), ∆r (t, 1), v(t, 1), . . . , ∆s (t, m), ∆r (t, m), v(t, m), . . . , ∆s (t, M ), ∆r (t, M ), v(t, M ) .
(10)
Here, ∆s (t, m) and ∆r (t, m) denote the transmitter-side and receiver-side AoI of source m, respectively. The variable v(t, m) tracks the length of the current consecutive AoI violation segment relative to the threshold ζ, evolving as ( v(t − 1, m) + 1, if ∆r (t, m) > ζ, v(t, m) = (11) 0, if ∆r (t, m) ≤ ζ, with initial condition v(0, m) = 0. v(t, m) is reset whenever the AoI returns to a non-violating level. To ensure a finitestate MDP, the violation counter is truncated at kmax in this work, i.e., v(t, m) ∈ {0, 1, . . . , kmax }. Specifically, whenever the number of consecutive AoI violations exceeds kmax , the counter remains fixed at kmax . This state representation explicitly encodes both instantaneous AoI levels and their temporal persistence, which is essential for evaluating multi-scale violation events underlying the C-AVR vector. To ensure a finite state space, all AoI values are truncated by a predefined upper bound ∆max . Once an AoI value reaches ∆max , it remains at this value. This truncation preserves the relevant persistence structure while improving computational tractability. b) Action Space: At each time slot t, the scheduler selects an action at ∈ A, where A = {0, 1, . . . , M }. Action at = m schedules source m for transmission, while at = 0 indicates no transmission in the current slot. c) State Transition Probability: Given the current state st and action at , the system moves to the next state st+1 according to the transition probability P(st+1 | st , at ). The transition is determined by stochastic packet generation and
6
random transmission outcomes. Specifically, packet generation for source m follows a Bernoulli process with probability m pm g , and successful transmission occurs with probability ps . These system parameters are assumed unknown, making the transition model unavailable in closed form. Consequently, the scheduling problem naturally motivates a model-free reinforcement learning approach. d) Cost Function: To model resource constraints, each transmission incurs a unit cost that captures resource consumption, such as energy or bandwidth. The instantaneous cost at slot t is defined as ct = 1[at ̸= 0]. The long-term PTaverage 1 transmission cost is required to satisfy limT →∞ T t=1 ct ≤ ηmax , where ηmax ∈ (0, 1] specifies the allowable resource budget. e) Reward Function: The objective is to minimize the weighted C-AVR defined in (5), which aggregates violation events across multiple persistence scales. To incorporate this objective into the CMDP framework while satisfying the resource constraint, we adopt a Lagrangian relaxation approach. Specifically, after executing action at during slot t, the reward is defined as kmax M X 1 X k wk 1[Vt+1 (m)] − λ(ct − ηmax ). rt ≜ − M m=1
(12)
k=1
k Here, Vt+1 (m) denotes the event that the receiver-side AoI
of source m exceeds the threshold ζ for k consecutive slots ending at slot t + 1. The index t + 1 appears because the scheduling action taken in slot t affects the AoI state only after the transmission outcome is realized at the end of the slot. Consequently, the persistence-aware violation event induced by action at is evaluated using the updated AoI state at slot t + 1. The second term in (12) enforces the transmission cost constraint through the Lagrange multiplier λ ≥ 0. The first term represents a weighted aggregation of multi-scale violation events, and its time average corresponds to the empirical estimate of the weighted C-AVR defined in (5). It therefore penalizes violations across multiple persistence scales simultaneously, with their relative importance determined by the weight vector W. For example, uniform weights emphasize average persistence behavior, whereas exponentially increasing weights place greater emphasis on rare but prolonged violation runs associated with tail reliability degradation. The empirical average transmission cost is updated as 1 η ← η + ct − η , (13) t and the Lagrange multiplier λ is updated using the projected stochastic subgradient descent λ ← max (0, λ + ξ(η − ηmax )) ,
(14)
where ξ > 0 is the step size. This update dynamically adjusts the penalty to enforce the long-term constraint, i.e., increasing the penalty when the empirical transmission cost exceeds the allowable budget. f) Discount Factor: The discount factor γ ∈ (0, 1] controls the relative importance of immediate and future rewards. When the system performance P∞ is evaluated through the expected discounted return Eπ [ t=0 γ t rt ] under policy
π, a value of γ close to one emphasizes long-term reliability performance, which is critical for persistence-aware objectives where the impact of actions accumulates over consecutive time slots. While the CMDP formulation provides a principled framework for optimizing the weighted C-AVR objective under resource constraints, it also exposes a key challenge. The weighted C-AVR objective is affected by temporally correlated and potentially rare violation events, especially under tailsensitive weighting schemes. Consequently, the induced return distribution can be high-variance and asymmetric. When future performance is summarized solely by its expectation, some persistence-driven variability is compressed into a scalar value estimate. In particular, conventional DRL methods based on expectation-valued value functions may yield less informative learning signals for prolonged violation sequences under persistence-aware objectives. This observation motivates value representations that preserve distributional information during training. In the next section, we develop a distributional DRL framework tailored to weighted C-AVR optimization. IV. D ISTRIBUTIONAL DRL FOR W EIGHTED C-AVR O PTIMIZATION This section develops a distributional DRL approach explicitly aligned with the weighted C-AVR objective. By modeling the full distribution of future returns, rather than only their expectation, the proposed method captures the return variability and tail behavior induced by persistence-aware AoI violation penalties across multiple time scales. This richer representation enables the scheduler to distinguish policies with similar average performance but markedly different reliability profiles. Leveraging quantile-based distributional learning together with a reliability-aware state and reward design introduced in the CMDP formulation, we construct a QR-D3QN framework that is naturally suited to minimizing persistence-driven QoS violations under resource constraints. The overall learning framework is illustrated in Fig. 3. A. From Expected Value Learning to Distributional Value Learning In standard Deep Q-Networks (DQN), the agent learns a scalar action-value function Q(s, a), representing the expected discounted return obtained by taking action a in state s. The optimal action-value function satisfies the Bellman optimality equation h i ′ ′ Q(s, a) = E r + γ max Q(s , a ) | s, a , (15) ′ a
where r is the immediate reward, s′ is the next state, and γ ∈ (0, 1] is the discount factor. In practice, Q(s, a) is approximated by a neural network Q(s, a; w) with parameters w, trained by minimizing the temporal-difference (TD) loss h 2 i L(w) = E(s,a,r,s′ )∼B y − Q(s, a; w) , (16) where the TD target is y = r + γ max Q(s′ , a′ ; w− ), ′ a
(17)
7
Target network
Experience Replay Buffer
... ...
Mini-batch ... ...
Online network ...
K
AC
CK
...
/NA
...
ACK
C-AVR Environment
...
K/N
AC
Quantile Hube Loss
...
Fig. 3. QR-D3QN–based training framework for weighted C-AVR–aware scheduling, where distributional value estimation enables learning under rare but persistent AoI violation events.
and w− denotes the parameters of a periodically updated target network. Here, B represents mini-batches sampled uniformly from the replay buffer M, which mitigates temporal correlations and stabilizes learning. A known limitation of DQN is the overestimation bias introduced by the maximization operator. Dueling Double DQN (D3QN) mitigates this issue through two complementary mechanisms: • Double Q-learning: Action selection and evaluation are decoupled, yielding the TD target ′ ′ − y ∗ = r + γQ s′ , arg max Q(s , a ; w); w . (18) ′
To address this challenge, we adopt a distributional reinforcement learning framework. Instead of approximating the expected return, distributional methods model the entire return distribution. Let Z(s, a) denote the random return obtained by taking action a in state s. The expected action value is then Q(s, a) = E[Z(s, a)].
(20)
By explicitly representing the distribution of returns Z(s, a), distributional DRL preserves information about variability and low-probability, high-impact events, which are precisely the events governing weighted C-AVR performance.
a
Dueling architecture: The Q-function is decomposed B. QR-D3QN for C-AVR-based Scheduling into a state-value component and an advantage compoAmong distributional DRL approaches, Quantile Regression nent DQN (QR-DQN) offers a favorable balance between modeling 1 X Q(s, a; w) = V (s; w)+ A(s, a; w)− A(s, a′ ; w) , flexibility and computational efficiency. Rather than using |A| ′ a fixed categorical supports in C51, QR-DQN approximates the (19) return distribution using a set of learnable quantiles. where |A| is the number of actions. This separation For each state action pair (s, a), the cumulative distribution improves stability and emphasizes meaningful actions function (CDF) is partitioned into N uniform probability relative to the state. intervals {τ1 , τ2 , ..., τi , ..., τN }, where τi = i/N . The midpoint While D3QN improves learning stability and action discrimof the interval (τi−1 , τi ) is τ̂i = (τi−1 + τi )/2. The network ination, it still estimates only the expected return. This limilearns quantile values θi (s, a; w) representing the τ̂i -quantiles, tation is particularly relevant for weighted C-AVR optimizaand the return distribution of Z(s, a; w) is approximated by a tion. Specifically, the reward in (12) aggregates multi-scale uniform mixture of Dirac masses at these learned quantiles violation events and is influenced by temporally correlated and N potentially rare trajectories involving persistent AoI violations, 1 X δθ (s,a;w) , (21) Z(s, a; w) = especially under tail-sensitive weighting schemes. As a result, N i=1 i the induced return distribution can be heterogeneous and highvariance. When learning is based solely on expected returns, where δz denotes a Dirac distribution centered at z. This these persistence-driven variations are compressed into scalar representation captures not only the central tendency of the value targets. In particular, actions or policies with similar return but also its dispersion and tail behavior. In the conexpected returns may still be associated with return distri- text of weighted C-AVR optimization, the reward explicitly butions of different variability, which can make expectation- penalizes persistence of AoI violations across multiple time based value learning less informative under persistence-aware scales. Consequently, trajectories containing prolonged violation sequences accumulate substantially larger penalties and reliability objectives. •
8
therefore induce significantly lower cumulative returns. These rare but high-impact trajectories primarily influence the lower quantiles of the learned return distribution. Consequently, the distributional representation preserves richer training information about persistence-driven reliability degradation, whereas expectation-based methods compress such information into a single scalar target. To further improve learning stability, we integrate quantile regression with the D3QN architecture. For each quantile index i ∈ {1, . . . , N }, the corresponding quantile value θi (s, a; w) is decomposed into a state-value component and an advantage component as 1 X Ai (s, a′ ; w) , θi (s, a; w) = Vi (s; w) + Ai (s, a; w) − |A| ′ a (22) with 1 X θi (s, a′ ; w), (23) Vi (s; w) = |A| ′ a
Ai (s, a; w) = θi (s, a; w) − Vi (s; w). (24) P This decomposition enforces a Ai (s, a; w) = 0 for each quantile i, yielding a consistent separation between statevalue and action-dependent effects at every quantile level. It extends the classical dueling architecture to the distributional setting, where the components Vi (s) capture the system-wide reliability baseline at different return quantiles, while Ai (s, a) quantify the action-specific deviations relative to this baseline. The quantile-wise dueling architecture thus enables the model to distinguish how different actions influence not only the expected performance but also the distributional characteristics of returns, including variability and tail risk induced by persistent violation events. For action selection, the expected action value is computed by averaging the quantile estimates N
Q(s, a; w) =
1 X θi (s, a; w), N i=1
(25)
In QR-D3QN, the mean action value used for scheduling is obtained by averaging the learned quantile returns. The key distinction from expectation-based DRL lies in how this action value is learned. Instead of directly fitting a single scalar TD target, QR-D3QN learns a set of quantile targets through distributional Bellman updates, thereby preserving information about return variability, asymmetry, and low-probability persistence events during training. As a result, the resulting mean action-value estimates are informed by a richer distributional representation, which helps the scheduler identify actions that reduce prolonged AoI violation sequences under high-variance persistence-aware rewards. In addition, actions are selected according to an ϵ-greedy exploration policy, where the exploration rate ϵe at episode e follows the linear decay schedule e−1 ϵ (ϵstart − ϵend ), e ≤ Edecay , start − Edecay − 1 ϵe = ϵend , e > Edecay . (26)
Algorithm 1 QR-D3QN for Weighted C-AVR Scheduling Input: Number of training episodes Emax ; episode length Tmax ; minimum replay buffer size Mmin for training; discount factor γ; learning rate α; target update frequency G; linear decay exploration schedule parameters (ϵstart , ϵend , Edecay ); transmission cost constraint ηmax ; Lagrange multiplier λ; number of quantiles N ; step size ξ for λ update. 1: Initialize online network w and target network w − ← w, replay buffer M, Lagrange multiplier λ ≥ 0, empirical cost estimate η ← 0, and global time slot t ← 0 2: for episode e = 1 to Emax do 3: Reset the environment 4: Update exploration rate ϵe according to the linear decay schedule (26) 5: for n = 1 to Tmax do 6: Update global time slot index t ← t + 1 7: Observe the current state st consisting of AoI states and consecutive violation counters 8: Compute quantile values {θi (st , a; w)}N i=1 for all a ∈ A using the quantile-wise dueling architecture (22) 9: Compute mean action values Q(st , a; w) = PN 1 θ (s i=1 i t , a; w) N 10: Select action at using ϵe -greedy exploration 11: Execute at , observe cost ct , compute the Lagrangian reward rt using the current multiplier λ, and obtain the next state st+1 12: Store transition (st , at , rt , st+1 ) in M 13: Update empirical average transmission cost estimate η by (13) 14: if |M| ≥ Mmin then 15: Sample a mini-batch of transitions B = {(s, a, r, s′ )} uniformly from the replay buffer M 16: Select the greedy next action using the online network by (27) 17: Compute target quantiles by (28) 18: Compute quantile Huber loss L(w) by (29) 19: Gradient update: w ← w − α∇w L(w) 20: if t mod G = 0 then 21: Update target network: w− ← w 22: end if 23: end if 24: Update Lagrange multiplier λ by (14) 25: end for 26: end for 27: return Trained QR-D3QN parameters w Here, ϵstart and ϵend denote the initial and final exploration rates, respectively. Edecay specifies the number of training episodes over which the exploration rate decreases linearly from ϵstart to ϵend . The distributional Bellman update follows the double Qlearning principle. For a sampled transition (s, a, r, s′ ), the next action is selected as N
a∗ = arg max ′ a
1 X θi (s′ , a′ ; w), N i=1
(27)
9
and the target quantiles are computed as ′
∗
−
T θj = r + γθj (s , a ; w ),
TABLE I T RAINING PARAMETERS OF THE QR-D3QN A LGORITHMS
j = 1, . . . , N,
(28) Parameter AoI upper bound Discount factor DRL learning rate Step size for λ update Initial exploration rate Final exploration rate Number of exploration decay episodes Target network update period Replay buffer size Minimum replay buffer size for training Mini-batch size Hidden-layer size Number of quantiles Huber loss threshold Number of training episodes Time slots per episode
where w and w− denote the online and target network parameters, respectively. To train the network parameters, we minimize the quantile Huber loss L(w) =
N N 1 XX κ ρτ̂i T θj − θi (s, a; w) , N i=1 j=1
(29)
with ( ρκτ (u) =
1
1 2 2 u |τ − {u < 0}|, κ(|u| − 21 κ)|τ − {u < 0}|,
1
|u| ≤ κ, otherwise.
(30)
where κ > 0 controls smoothness near zero. This objective ensures accurate learning of the return distribution while maintaining training stability. As shown in Fig. 3, the proposed QR-D3QN framework integrates quantile-based distributional value learning with the D3QN learning architecture to optimize the weighted C-AVR objective in stochastic wireless systems. At each time slot, the agent observes the system state s, which includes transmitterside AoI, receiver-side AoI, and consecutive violation counters, and selects an action a via an ϵ-greedy policy based on the expected Q-values computed by averaging the learned quantiles. After executing the action, the resulting transition (s, a, r, s′ ) is stored in the replay buffer. Mini-batches sampled from this buffer are used to update the online network by minimizing the quantile Huber loss, where target quantiles are constructed through a distributional Bellman update combined with double Q-learning. A target network is periodically synchronized to ensure training stability. Crucially, by modeling a quantile-based return distribution during training, the learned value representation retains information related to both the frequency of AoI violations and their temporal persistence across different scales. This can improve the mean action-value estimates used by the mean-greedy scheduler when returns are high-variance and persistence-aware. Such capability is useful for weighted CAVR optimization, especially under weighting schemes that place larger penalties on prolonged violation events. The complete procedure is summarized in Algorithm 1. V. P ERFORMANCE E VALUATION This section evaluates the proposed persistence-aware scheduling framework from three perspectives: (i) the effectiveness of the weighted C-AVR objective in capturing multiscale reliability, (ii) the benefit of distributional reinforcement learning in reducing persistent AoI violations, and (iii) the robustness of the proposed method under varying system conditions. A. Simulation Setup All algorithms are implemented in Python 3.9 using PyTorch and evaluated under identical software and hardware configurations to ensure a fair and reproducible comparison.
Symbol ∆max γ α ξ ϵstart ϵend Edecay G |M| Mmin |B| – N κ Emax Tmax
Value 100 0.98 2 × 10−3 0.10 1.0 0.05 1000 3 106 1000 256 2 × 128 64 1 3000 100
Unless otherwise specified, we consider a system with M = 10 sources, and a homogeneous setting for different sources with packet generation probability pm g = 0.7, transmission success probability pm s = 0.7, AoI violation threshold ζ = 15, and transmission cost constraint ηmax = 0.75. Heterogeneous settings are considered in Section V-D. The AoI is truncated at ∆max = 100. The training parameters are summarized in Table I: discount factor γ = 0.98, learning rate α = 2 × 10−3 , replay buffer size |M| = 106 , minimum buffer size Mmin = 1000, mini-batch size |B| = 256, target update period G = 3, and number of quantiles N = 64. All results are averaged over 10 independent runs with different random seeds. For weighted C-AVR simulations, we set kmax = 9 unless otherwise stated. Three weighting schemes are considered: (i) Uniform weights; (ii) Exponentially increasing weights (default β = 2); and (iii) One-hot weights, which reduce the objective to a single-window C-AVR. We compare the proposed QR-D3QN method with the following baselines: DQN: Standard Deep Q Network that learns the expected return as a scalar value function. • D3QN: Builds on DQN with Double Q-learning and a dueling architecture to improve stability and action-value estimation, still based on expected returns. • QR-DQN: Distributional DRL algorithm using quantile regression to model the return distribution, without Double Q-learning or dueling structures. • QR-D3QN: The proposed approach combining quantilebased distributional learning with Double Q-learning and dueling architectures, designed to capture rare but persistent violations while maintaining stable learning. • Drift-Plus-Penalty (DPP) Policy: DPP is a wellestablished Lyapunov optimization framework for constrained stochastic optimization in networked systems and has been widely applied in AoI scheduling problems. Unlike DRL methods, DPP exploits system structure and does not rely on value function approximation, thereby providing a complementary benchmark from a classical control and operations research perspective. In this work, the resulting policy serves as a model-based heuristic benchmark derived from Lyapunov optimization prin•
10