Next Article in Journal
A Hybrid Interval Type-2 Fuzzy AHP (IT2F-AHP)–VIKOR–TOPSIS Framework for Environmental Performance Assessment of Helicopter Engines
Previous Article in Journal
A Deep Feature Approach to Visual Similarity Analysis of Ethnic Brocades in Southwest China
 
 
Font Type:
Arial Georgia Verdana
Font Size:
Aa Aa Aa
Line Spacing:
Column Width:
Background:
Article

Attribution-Guided Active Exploration in Deep Reinforcement Learning for Autonomous Driving Decision-Making

1
College of Mechanical and Vehicle Engineering, Hunan University, Changsha 410082, China
2
School of Mechanical and Electrical Engineering, Guilin University of Electronic Technology, Guilin 541004, China
*
Author to whom correspondence should be addressed.
Appl. Sci. 2026, 16(10), 4931; https://doi.org/10.3390/app16104931
Submission received: 16 April 2026 / Revised: 8 May 2026 / Accepted: 12 May 2026 / Published: 15 May 2026
(This article belongs to the Section Computing and Artificial Intelligence)

Abstract

Deep reinforcement learning often suffers from inefficient exploration, which is commonly addressed by introducing an auxiliary model that assigns intrinsic rewards when the agent encounters novel scenarios. However, such approaches increase training complexity and computational overhead. This paper proposes an Attribution-Guided Reinforcement Learning (AGRL) framework that exploits real-time attribution analysis to guide exploration in autonomous driving decision-making. The proposed method is built upon the Kolmogorov–Arnold-Network-based Interpretable Deep Reinforcement Learning (KAN-IDRL) framework. Specifically, action-wise attribution patterns are computed online, and perturbations are applied to the state inputs to measure attribution sensitivity. The resulting attribution-sensitivity signal identifies actions whose decision rationales are more locally responsive to state changes, and these actions are therefore preferentially explored. In addition, local attribution results collected from a pretrained interpretable policy are aggregated into global feature-importance scores, which are then used to initialize a trainable prior attention gate in a Prior-Attention-Enhanced Kolmogorov–Arnold Network (PAE-KAN). This design allows the policy to incorporate attribution-derived prior knowledge while maintaining sufficient adaptability for task-specific learning. Experiments across multiple autonomous driving scenarios demonstrate that the proposed AGRL framework achieves faster convergence and competitive final performance compared with representative baseline methods. These findings indicate that attribution information can be transformed from a post hoc interpretability tool into an effective guidance signal for improving reinforcement learning.

1. Introduction

The decision-making module of an autonomous vehicle is responsible for receiving information from the surrounding environment and generating high-level intentions, making it a critical component for formulating driving strategies. Current decision-making approaches can generally be categorized into knowledge-driven and data-driven methods. Knowledge-driven approaches, such as Hierarchical State Machines (HSM) [1], Expert Systems (ES) [2], and Finite State Machines (FSM) [3], are characterized by rigorous logic and high interpretability [4]. However, these methods are inherently limited by their dependence on prior knowledge, which constrains their ability to handle unexpected scenarios [5]. In contrast, data-driven approaches, including end-to-end (E2E) learning [6], Imitation Learning (IL) [7], and Deep Reinforcement Learning (DRL) [8], have emerged due to their strong self-learning capability and adaptability to dynamic environments. In particular, DRL algorithms optimize driving strategies through real-time interaction with the environment, enhancing their effectiveness in complex scenarios [9].
Despite these advancements, traditional DRL methods rely heavily on extensive random exploration to enable the policy to acquire a comprehensive understanding of the environment [10]. Such random exploration is inefficient, especially in high-dimensional state spaces or complex environments, as it typically requires a large number of interactions to achieve meaningful learning progress. To address this limitation, numerous studies [11,12] have proposed active exploration methods, which aim to enable agents to explore the environment more efficiently by reducing unnecessary trial-and-error and avoiding uninformative actions. Common active exploration strategies include prediction-error-based approaches, which estimate the agent’s uncertainty about a given state to guide exploration. For example, Pathak et al. [13] proposed a curiosity-driven approach that trains a model to predict future states and uses the prediction error to quantify the value of exploring a state. Additionally, some methods exploit model uncertainty to drive exploration. Houthooft et al. [14] proposed an uncertainty-driven exploration framework, where Bayesian inference or ensemble models are employed to quantify uncertainty and guide the agent toward less familiar regions of the environment. Although these approaches improve exploration efficiency and reduce unnecessary interactions, they suffer from two main limitations. First, they rely on external models to guide exploration, which significantly increases training complexity and computational cost. Second, many methods incorporate exploration-related metrics directly into the reward function to incentivize the agent to visit novel states. However, this introduces a discrepancy with the core objective of RL—maximizing environment rewards—which may increase environmental uncertainty and lead to suboptimal behaviors.
To address this limitation, this paper proposes an Attribution-Guided Reinforcement Learning (AGRL) framework that uses attribution analysis to characterize the internal decision rationale of the policy network and guide exploration accordingly. Instead of treating attribution merely as a post hoc explanation tool, AGRL transforms action-wise attribution responses into an online exploration signal. Specifically, we build upon the Kolmogorov–Arnold Network-based Interpretable DRL (KAN-IDRL) framework [15] to compute real-time attributions for each action. By introducing small perturbations to the state inputs, we measure the perturbation-induced sensitivity of action-wise attributions. A larger Euclidean variation in attribution indicates that the corresponding action is more locally sensitive to changes in the driving state. Therefore, AGRL biases exploration toward actions with stronger attribution responses, rather than selecting exploratory actions purely at random. This approach enables more efficient and targeted exploration.
Moreover, to further accelerate training, we incorporate global attribution knowledge extracted from a pretrained policy model into the AGRL models. Specifically, we collect and aggregate local attributions over a large number of driving scenarios to obtain a global explanation, i.e., the global importance of input features. The resulting feature-level global importance is then introduced as a prior attention module immediately after the input layer, such that the input features are enhanced before being fed into the decision-making network. This mechanism is able to suppress less informative features and emphasize decision-relevant ones, thereby accelerating model convergence.
The main contributions of this work are summarized as follows:
  • We propose an intrinsic exploration signal based on attribution sensitivity. By applying small perturbations to the state and measuring the Euclidean variation in action-wise attributions, the proposed method identifies actions whose decision rationales are locally sensitive to state changes. This enables efficient and targeted exploration without requiring additional exploration models, thereby improving sample efficiency.
  • We introduce a global-attribution-based prior attention mechanism that embeds feature-level global attributions from a pretrained policy into the new decision model, accelerating convergence with negligible additional parameters.
  • We validate AGRL on multiple autonomous-driving decision-making tasks and show that it achieves faster convergence and competitive final performance compared with representative baseline methods.
The remainder of this paper is organized as follows: Section 2 reviews related work on active exploration and interpretable reinforcement learning; Section 3 details the proposed AGRL framework, including the attribution-sensitivity-guided exploration strategy and the global-attribution-based prior attention mechanism; Section 5 presents the experimental design and results; Section 6 concludes the paper with a summary of the main findings and directions for future research.

2. Related Work

2.1. Active Exploration in Reinforcement Learning

Traditional DRL algorithms typically rely on random exploration, which is often inefficient in high-dimensional or complex environments. To alleviate this limitation, a variety of active exploration methods have been developed to encourage agents to prioritize more informative states or actions [12]. Existing approaches can be broadly divided into count-based methods and curiosity-driven methods.
Count-based exploration methods estimate state novelty according to visitation frequency and assign larger exploration bonuses to less frequently visited states [16,17]. While such strategies are effective in small or discrete state spaces, they become difficult to apply directly when the state space is large or continuous. To address this issue, a line of work has introduced pseudo-count mechanisms, which use neural networks or other learned models to estimate state visitation frequency in high-dimensional spaces [18,19,20,21].
Curiosity-driven exploration methods, by contrast, use intrinsic motivation signals to encourage the agent to explore unfamiliar regions of the environment. A representative line of work derives intrinsic rewards from prediction errors of learned environment models [13]. For example, Pathak et al. [13] proposed a curiosity-driven framework in which a forward dynamics model is trained to predict future states, and the resulting prediction error is used as an intrinsic exploration signal. Such methods encourage the agent to visit poorly predicted states, but their effectiveness depends heavily on the learnability of the environment dynamics. When the environment is highly stochastic or difficult to model, the prediction model itself may become unreliable, which in turn weakens the quality of the exploration bonus. To mitigate the dependence on explicit dynamics modeling, subsequent studies proposed alternatives such as Random Network Distillation (RND), which measures novelty through the prediction error between a fixed random target network and a trainable predictor network [22].
Overall, existing active exploration methods have improved exploration efficiency to varying degrees. However, many of them still rely on auxiliary models or additional prediction networks. These extra components not only increase computational overhead, but may also become less reliable in complex environments with strong uncertainty. This limitation motivates the present work, in which exploration is guided by attribution sensitivity derived directly from the decision model itself, without introducing an additional external model.

2.2. Attributing Decisions to Inputs

Attribution methods aim to analyze trained models by identifying which input features contribute most significantly to a model’s prediction. Representative approaches include SHAP [23] and LIME [24], which estimate feature importance by systematically perturbing input features and observing the resulting changes in model outputs. These methods provide detailed attribution analysis and have been widely used in practical applications. However, their computational cost is typically high, making them unsuitable for use as real-time training signals in online learning scenarios.
Gradient-based attribution methods provide another widely used family of interpretation techniques. Methods such as saliency maps [25] and Integrated Gradients [26,27] estimate feature importance by analyzing the gradient of the model output with respect to input features. These techniques are particularly suitable for differentiable models such as neural networks, where the computational graph naturally supports efficient gradient backpropagation [28,29,30]. However, gradient-based methods may suffer from issues such as gradient vanishing, gradient explosion, and sensitivity to gradient noise, which can reduce the fidelity of the generated explanations. Although several improved techniques have been proposed to alleviate these problems, they often introduce additional computational overhead.
Alternatively, relevance-propagation-based methods such as Layer-wise Relevance Propagation (LRP) [15,31] and DeepLIFT [32] redistribute the prediction score backward through the network according to layer-wise conservation rules. These approaches are computationally more efficient and can generate explanations with relatively low overhead. Nevertheless, many of these methods were originally developed for visual interpretation tasks. When applied to structured or non-visual inputs, they may fail to allocate relevance scores accurately.
In our previous work [15], we demonstrated that combining LRP with Kolmogorov–Arnold Networks (KAN) can produce explanations with high fidelity, strong real-time capability, and low computational cost. Building upon this finding, the present study further utilizes relevance-based attribution as an intrinsic signal for policy learning. Specifically, attribution information derived from the policy network is incorporated to guide exploration and improve the learning efficiency of deep reinforcement learning.

3. Methodology

This section presents the theoretical foundations of the proposed framework. We first formulate the autonomous driving decision-making task as a MDP. We then describe the proposed AGRL framework in detail, including the Prior-Attention-Enhanced KAN (PAE-KAN), the LRP-based real-time attribution method, and the attribution-sensitivity-guided exploration mechanism.

3.1. Problem Formulation

The decision-making process is modeled as a MDP, in which an agent interacts with the environment in discrete time steps. At each time step t, the agent observes the current state s t S , selects an action a t A , and receives a reward r t from the environment. The objective of the agent is to learn an optimal policy that maximizes the expected discounted return R t , defined as
R t = k = 0 γ k r t + k ,
where γ [ 0 , 1 ] is the discount factor that balances the importance between immediate and future rewards.
To learn the action-value function, this study adopts the Dueling Deep Q-Network (Dueling DQN) architecture [33]. Unlike conventional Q-learning methods that directly approximate the Q-value function Q ( s , a ) , the Dueling DQN decomposes the Q-value into two components: the state-value function V ( s ) , which represents the overall value of a state, and the advantage function A ( s , a ) , which reflects the relative importance of each action in that state. The Q-value is therefore computed as
Q ( s , a ) = V ( s ) + A ( s , a ) 1 | A | a A ( s , a ) .
In this work, PAE-KAN is employed to parameterize both the state-value function V ( s ) and the advantage function A ( s , a ) , providing a more interpretable functional representation compared with conventional multi-layer perceptrons (MLPs). The learning objective of the DRL model is to minimize the temporal-difference error between the predicted Q-value and the target Q-value. The loss function is defined as
L RL ( θ ) = E Q target Q eval ( s t , a t ; θ ) 2 ,
where Q eval ( s t , a t ; θ ) denotes the Q-value predicted by the evaluation network parameterized by θ , and Q target is computed according to the Bellman equation:
Q target = r t + γ max a Q target ( s t + 1 , a ; ϕ ) ,
where ϕ denotes the parameters of the target network.

3.2. Attribution-Guided Reinforcement Learning Framework

As illustrated in Figure 1, the proposed Attribution-Guided Reinforcement Learning (AGRL) framework consists of three core modules: (1) a driving decision backbone, (2) an attribution module, and (3) an attribution-sensitivity-guided exploration mechanism. Among them, the attribution module serves as the foundation of the overall framework, as it provides the feature-level attribution information required by both the prior attention mechanism and the active exploration strategy.

3.2.1. Prior-Attention-Enhanced KAN Decision Model

To incorporate global attribution knowledge into policy learning, we augment the original KAN-based decision model with a prior attention layer placed immediately after the input layer, as shown in Figure 2. The resulting model is referred to as a Prior-Attention-Enhanced KAN (PAE-KAN). Instead of directly using global attribution scores as fixed feature weights, the proposed model employs them to initialize a trainable prior attention gate, allowing the network to preserve attribution-based prior knowledge while retaining sufficient flexibility for task-specific adaptation.
Let s = [ s 1 , s 2 , , s n ] denote the input state. The prior attention layer produces a feature-wise gate vector g = [ g 1 , g 2 , , g n ] , and the reweighted input is defined as
s ^ = g s ,
where ⊙ denotes element-wise multiplication. The gate vector is parameterized as
g = λ · σ ( z ) ,
where z = [ z 1 , z 2 , , z n ] denotes the trainable gate logits, σ ( · ) is the sigmoid function, and λ is a scaling factor controlling the overall gate magnitude.
The reweighted input s ^ is then fed into the subsequent decision backbone. Accordingly, the mathematical form of the proposed PAE-KAN can be expressed as
f ( s ) = q = 1 2 n + 1 Φ q p = 1 n ϕ q , p ( s ^ p ) ,
where s ^ p = g p s p , ϕ q , p denotes the univariate transformation applied to the p-th reweighted input feature, and Φ q denotes the higher-level aggregation function. In this way, the proposed PAE-KAN explicitly integrates attribution-derived prior knowledge into the input representation, enabling the model to emphasize globally important decision-related features while suppressing less informative ones.

3.2.2. LRP-Based Real-Time Attribution Method

LRP is a post hoc interpretation method designed to identify which input variables contribute most to a model’s prediction. In our previous work, we demonstrated that combining LRP with KAN can effectively overcome the fidelity limitations encountered when applying LRP to conventional multilayer perceptrons.
In this section, we describe how attribution information can be extracted from the proposed PAE-KAN decision model. Given a scene input S t , the prior attention layer first generates the reweighted state S ^ t , which is then fed into the KAN-based Q-network. For a given action a, the action-specific Q-value is denoted by Q a ( S ^ t ) . Attribution is used to quantify the contribution of each input feature x t i to the predicted Q-value Q a ( S ^ t ) . Each action has an independent LRP propagation process. For example, in a lane-changing decision task, LRP can separately explain the predicted Q-values associated with left lane change, right lane change, and lane keeping. The LRP-based attribution process therefore consists of two main stages.
Step 1: Forward pass in PAE-KAN. For notational simplicity, let X = s ^ = { x ^ 1 , x ^ 2 , , x ^ n } denote the prior-attention-reweighted input. The forward computation of the KAN backbone is given by
Q ( X ) = q = 1 2 n + 1 ϕ q p = 1 n ϕ p , q ( x ^ p ) ,
where ϕ p , q denotes the nonlinear transformation applied to the p-th reweighted input feature, and ϕ q represents the nonlinear transformation at the output layer. This structure can be rewritten as
Q ( X ) = q = 1 2 n + 1 ϕ q ( X q ) ,
X q = p = 1 n ϕ q , p ( x ^ p ) ,
where X = { X 1 , X 2 , , X 2 n + 1 } , and X q represents the activation of node q in the intermediate layer.
For deeper KAN architectures, this recursive structure is preserved. The activation value of node k in layer l + 1 is computed as
a l + 1 , k = i = 1 n l ϕ l , i , k ( a l , i ) ,
where a l , i denotes the activation of node i in layer l, ϕ l , i , k is the activation function between the two nodes, and n l is the number of nodes in layer l.
Step 2: Relevance backward propagation.
Relevance is propagated from the output layer to the input layer. Based on the recursive structure in Equation (11), when node k in layer l + 1 distributes relevance to node i in layer l, the propagated relevance is defined as
R i k ( l + 1 l ) = ϕ l , i , k ( a l , i ) i ϕ l , i , k ( a l , i ) R k ( l + 1 ) ,
which leads to the LRP-KAN propagation rule
R i = k ϕ i , k ( a i ) i ϕ i , k ( a i ) R k .
Here, R i denotes the relevance score of node i in layer l, and R k denotes the relevance score of node k in layer l + 1 .
Although PAE-KAN introduces an additional prior attention layer before the KAN backbone, this layer only rescales the input representation and does not alter the subsequent relevance redistribution structure among features. Therefore, the LRP propagation rule within the KAN backbone remains unchanged, and the same relevance analysis can be directly applied to the PAE-KAN model. Moreover, the LRP-based attribution process only requires neuron activation values and simple arithmetic operations such as addition and division. This results in extremely low computational overhead, making it feasible to use attribution signals as real-time feedback during reinforcement learning.
In addition to real-time local attribution, the proposed framework further derives global attribution priors from a large number of driving scenarios. Specifically, let
r ( m ) = [ r 1 ( m ) , r 2 ( m ) , , r d ( m ) ]
denote the LRP attribution vector obtained from the m-th scenario, where d is the number of input features. Since the purpose of global attribution is to characterize feature importance regardless of contribution sign, the absolute value of each attribution score is first taken. The global attribution importance of the i-th feature is then computed as
u i = 1 M m = 1 M r i ( m ) ,
where M denotes the total number of collected scenarios.
To construct the prior attention gate, the global attribution scores are first normalized by min–max scaling:
u ˜ i = u i u min u max u min + ε ,
where u min and u max denote the minimum and maximum values in the global attribution vector, respectively, and ε is a small constant for numerical stability. Finally, the initial gate logits are defined using the logit transform:
z i ( 0 ) = log u ˜ i 1 u ˜ i ,
which are used to initialize the trainable gate parameters z in Equation (6). In this way, local attribution results are aggregated into global feature-level prior knowledge, which is embedded into the PAE-KAN model as a trainable prior attention gate.

3.2.3. Attribution-Sensitivity-Guided Exploration

AGRL uses attribution variation as an action-wise sensitivity signal to guide exploration toward actions whose decision rationales are more responsive to local traffic changes. During exploratory steps, the action with the largest attribution variation is preferentially selected, as shown in Algorithm 1.
Given the current state s t , the PAE-KAN-based Dueling DQN first produces the Q-values for all actions. The LRP-based attribution module subsequently generates an attribution vector for each action:
r a ( s t ) = [ r a ( 1 ) , r a ( 2 ) , , r a ( d ) ] ,
where d is the dimension of the state representation and r a ( i ) represents the contribution of the i-th input feature to the predicted Q-value of action a.
Algorithm 1 Attribution-Sensitivity-Guided Training in AGRL
Require: 
Environment E , initialized PAE-KAN policy Q θ , target network Q ϕ , replay buffer D , exploration rate ϵ , perturbation coefficient σ
Ensure: 
Trained AGRL policy
  1:
for each episode do
  2:
      for each decision step t do
  3:
            Observe current state s t
  4:
            Compute action values Q ( s t , a ) for all a A
  5:
            Compute attribution vectors r a ( s t ) for all a A
  6:
            Generate perturbed state s ˜ t = s t + δ t , where δ t N ( 0 , σ 2 s t 2 )
  7:
            Compute attribution vectors r a ( s ˜ t ) for all a A
  8:
            for each action a A  do
  9:
                  Compute attribution variation D a ( s t , s ˜ t ) = r a ( s t ) r a ( s ˜ t ) 2
10:
            end for
11:
            Select action a t according to
12:
            if  rand ( ) > ϵ  then
13:
                   a t arg max a A Q ( s t , a )
14:
            else
15:
                   a t arg max a A D a ( s t , s ˜ t )
16:
            end if
17:
            Execute a t in E and observe reward r t and next state s t + 1
18:
            Store transition ( s t , a t , r t , s t + 1 ) in D
19:
            Sample a mini-batch from D
20:
            Update Q θ by minimizing the temporal-difference loss
21:
            Periodically update target network parameters: ϕ θ
22:
      end for
23:
end for
24:
return trained AGRL policy
To evaluate attribution sensitivity, we generate a perturbed state s ˜ t from the original state s t by applying bounded stochastic perturbations in normalized feature space. For each continuous feature i, the perturbation is sampled as
δ t ( i ) N ( 0 , σ i 2 ) ,
where σ i is a feature-specific perturbation scale. The perturbed feature is then clipped to the valid normalized range,
s ˜ t ( i ) = clip s t ( i ) + δ t ( i ) , 1 , 1 .
Binary features are kept unchanged. This feature-aware perturbation scheme ensures that the perturbed state remains bounded and semantically meaningful, while preventing out-of-range values and unrealistic observations. The perturbation scales used for the continuous features are listed in Table 1.
The attribution variation in action a is quantified by the Euclidean distance between the attribution vectors before and after perturbation:
D a ( s t , s ˜ t ) = r a ( s t ) r a ( s ˜ t ) 2 .
Here, D a measures the sensitivity of the attribution pattern of action a to input perturbation.
During training, the action selection mechanism follows an attribution-guided ϵ -exploration strategy. Specifically, with probability 1 ϵ , the agent selects the greedy action according to the predicted Q-values:
a t greedy = arg max a A Q ( s ^ t , a ) .
With probability ϵ , instead of selecting a random action as in conventional ϵ -greedy exploration, the agent selects the action with the largest attribution variation:
a t explore = arg max a A D a ( s t , s ˜ t ) .
Accordingly, the overall action selection rule can be expressed as
a t = arg max a A Q ( s ^ t , a ) , with probability 1 ϵ , arg max a A D a ( s t , s ˜ t ) , with probability ϵ .
This design differs from random exploration. Instead of assigning exploration opportunities uniformly across actions, the proposed strategy prioritizes actions whose attribution patterns show stronger responses to small state perturbations. In other words, exploration is directed toward actions whose internal decision rationales are more locally sensitive to the current traffic situation, enabling the agent to collect more targeted training experiences than purely random exploration.

4. Implementation

This section describes the simulation environments and MDP formulation adopted for training and evaluation. Three representative driving tasks are considered, namely highway lane changing, main-road yielding, and on-ramp merging.

4.1. Scenario Modeling

All experiments are conducted in the Highway-Env simulator [34], which has been widely used in the evaluation of DRL decision-making methods for autonomous driving [35]. To comprehensively assess the proposed framework, three scenarios with increasing behavioral complexity are designed: a highway lane-change scenario, a main-road yielding scenario, and an on-ramp merging scenario.
As shown in Figure 3, the highway lane-change scenario considers an ego vehicle driving on a four-lane highway, where the agent is required to make discrete lane-selection decisions while balancing efficiency and safety. Each training run in this scenario contains 300 episodes, and each episode lasts for at most 200 time steps. Since one step corresponds to one second of simulation time, the maximum episode duration is 200 s. The longitudinal motion of all vehicles is governed by the Intelligent Driver Model (IDM) [36], while the lane-changing behavior of surrounding vehicles follows the MOBIL model [37]. The initial spacing between vehicles is 30 m, corresponding to approximately 133 vehicles/km.
The main-road yielding scenario is more challenging because the ego vehicle travels on the main road while interacting with a ramp vehicle that may merge with limited situational awareness. The ego vehicle is initialized near a location where conflict with the incoming ramp vehicle is likely. In addition, a visual obstruction is introduced, as indicated by the brown blocked area in Figure 3, preventing the ramp vehicle from observing the main-road traffic before reaching the acceleration lane. As a result, the ramp vehicle may initiate a merge without complete knowledge of surrounding traffic. In contrast, the ego vehicle is able to observe the ramp vehicle and must make both longitudinal and lateral decisions to avoid potential conflicts. In this scenario, the ramp length is 80 m, followed by an 80 m acceleration lane. After merging, the vehicle is required to travel safely for another 150 m. Surrounding vehicles are generated randomly with an average initial longitudinal spacing of 40 m, leaving a feasible but nontrivial merging opportunity.
Among the three scenarios, on-ramp merging poses the highest level of difficulty. In this task, the ego vehicle starts from the ramp and must identify an appropriate gap before merging safely into the main-road traffic. Excessively aggressive behavior is penalized because it may violate safe spacing requirements, whereas overly conservative behavior may cause merge failure. The ramp length in this scenario is 80 m, followed by a 150 m acceleration lane. After merging, the ego vehicle must continue safely for an additional 150 m on the main road. For both the on-ramp merging and main-road yielding scenarios, each training run consists of 2000 episodes. Vehicles are generated randomly with an average initial longitudinal spacing of 25 m, which creates denser traffic interactions and increases the difficulty of successful merging.
In all scenarios, inappropriate decisions may lead to hazardous interactions and, in severe cases, collisions. An episode is regarded as successful if the ego vehicle completes the task without collision; otherwise, the episode is recorded as a failure and the environment is reset. Based on this setup, the driving task is formulated as an MDP by defining the observation space, action space, and reward function.

4.2. Observation and Action Space

The agent is assumed to perceive a local traffic region surrounding the ego vehicle. In both the highway lane-change and on-ramp merging scenarios, the observation range covers the current lane and its two adjacent lanes. Specifically, in the current lane, the agent observes the ego vehicle itself together with the nearest leading vehicle. In each adjacent lane, both the nearest leading and trailing vehicles are included in the observation. In the main-road yielding scenario, besides six vehicles on the main road, the agent additionally observes the nearest vehicle on the ramp.
Each vehicle is represented by a state vector S i containing its relative kinematic information. For the ego vehicle, the state representation is defined as
S 1 = ( p 1 , Δ x , Δ y , v x 1 , v y 1 , a 1 ) ,
where Δ x and Δ y denote the distances from the ego vehicle to the left and right road boundaries, respectively. The remaining terms v x 1 , v y 1 , and a 1 represent the longitudinal velocity, lateral velocity, and longitudinal acceleration of the ego vehicle.
For surrounding vehicles, the state vector is written as
S i = ( p i , Δ x i , Δ y i , v x i , v y i , a i ) ,
where Δ x i and Δ y i denote the relative longitudinal and lateral positions with respect to the ego vehicle, and v x i , v y i , and a i are the corresponding kinematic variables. If a surrounding vehicle is outside the perception range, its indicator p i is set to zero. Under this definition, the observation dimension is 36 for the highway lane-change and on-ramp merging scenarios, and 42 for the main-road yielding scenario.

4.3. Action Space

The highway lane-change task is formulated as a discrete decision problem with three candidate actions:
a t = ( a 0 , a 1 , a 2 ) ,
where a 0 , a 1 , and a 2 denote left lane change, lane keeping, and right lane change, respectively.
For the main-road yielding and on-ramp merging tasks, the action set is extended to include longitudinal control, resulting in a five-dimensional discrete action space:
a t = ( a 0 , a 1 , a 2 , a 3 , a 4 ) ,
where a 0 , a 1 , and a 2 retain the same meanings as above, while a 3 and a 4 correspond to acceleration and deceleration, respectively.

4.4. Reward Function

The reward function is designed to jointly account for efficiency, safety, comfort, and energy-related objectives, so that the learned policy better matches practical autonomous driving requirements. The overall reward at time step t is defined as
r t = R speed C comfort C distance C collision C fuel C merge .
The individual terms are defined as follows.
(1)
Speed reward. To encourage efficient travel, the speed reward is defined as
R speed = α 1 · v t v min v max v min ,
where v t is the current speed of the ego vehicle, and v min and v max are the lower and upper bounds of the desired speed interval.
(2)
Comfort cost. To discourage excessive longitudinal acceleration and unnecessary lane changes, the comfort cost is given by
C comfort = α 2 | acc t | + α 3 δ lc ,
where acc t denotes the longitudinal acceleration and δ lc is a binary indicator for lane-change execution.
(3)
Safe-distance cost. To penalize unsafe following behavior and near-collision situations, the safe-distance cost is defined as
C distance = α 5 , if 0 < d safe < 5 m , α 5 1 d safe 5 5 , if 5 m d safe < 10 m , 0 , otherwise ,
where d safe represents the distance to the nearest leading vehicle in the current lane, and α 5 controls the penalty strength.
(4)
Collision penalty. To explicitly discourage unsafe behaviors that lead to crashes, the collision penalty is defined as
C collision = 1 , if a collision occurs , 0 , otherwise .
(5)
Merging speed penalty. For the main-road yielding scenario, an additional penalty is imposed when the merging vehicle moves too slowly, encouraging the ego vehicle to create sufficient space for safe merging:
C merge = α 6 · v max v t m v max ,
where v t m denotes the instantaneous speed of the merging vehicle and v max is the upper speed bound.

5. Experimental Results

In this section, the proposed AGRL framework is evaluated through comprehensive empirical studies. First, AGRL is compared with representative baseline methods in three autonomous driving decision-making scenarios, namely highway lane changing, main-road yielding, and on-ramp merging. Then, two groups of ablation studies are conducted in the highway lane-change scenario. The first group separately verifies the effectiveness of the global-attribution-based prior attention mechanism and the attribution-sensitivity-guided exploration module. The second group compares different attribution-change metrics as exploration signals to justify the use of Euclidean attribution variation in AGRL. Finally, an interpretability analysis is provided to illustrate how the proposed attribution-sensitivity signal reflects action-wise local responses to perturbations.

5.1. Baselines

To evaluate the effectiveness of the proposed AGRL framework, comprehensive comparisons are conducted against several representative baseline methods. The selected baselines are divided into two categories: standard value-based reinforcement learning methods and active-exploration-based methods.
  • Standard value-based RL baselines: Deep Q-Network (DQN) [38], Double DQN (DDQN) [39], and Dueling DQN (DuDQN) [33]. These methods are adopted as conventional value-based baselines that rely on standard random exploration strategies.
  • Interpretable value-based RL baseline: KAN-IDRL is included to distinguish the contribution of the KAN-based interpretable policy representation from that of the proposed attribution-guided exploration mechanism. Similar to the above value-based baselines, KAN-IDRL also relies on random exploration.
  • Active-exploration baselines: Random Network Distillation (RND) is included as a representative intrinsic-motivation-based exploration method, Intrinsic Curiosity Module (ICM) [13] is added as a representative curiosity-driven exploration method, while SimHash [40] is employed as a representative count-based exploration method. For a fair comparison, RND, ICM, and SimHash are all re-implemented on the same Dueling DQN backbone, with only the exploration mechanism changed. These three methods are selected as baselines to compare the proposed AGRL framework against classical active-exploration strategies.
All compared methods are implemented within a value-based reinforcement learning framework. To ensure a fair comparison, the same network training settings and optimization hyperparameters are used across all methods within each scenario, as summarized in Table 2. In addition, each algorithm is trained with five different random seeds to improve statistical reliability.

5.2. Evaluation Metrics

During training, the learning process in the highway lane-change scenario is evaluated using five metrics: average cumulative reward, success rate, average speed, comfort cost, and safety cost. For the main-road yielding and on-ramp merging scenarios, comfort cost is not considered, while an additional metric, namely merging speed cost, is introduced. Among these metrics, comfort cost, safety cost, and merging speed cost are computed according to Equation (29), Equation (30), and Equation (32), respectively. The cumulative reward is defined as the total reward accumulated over an episode, where the reward at each time step is computed according to Equation (27). As a comprehensive performance indicator, cumulative reward reflects the overall trade-off among efficiency, safety, comfort, and task-related objectives. The success rate is defined as the proportion of episodes in which the autonomous agent completes the driving task without collision or failure. Average speed is used to characterize driving efficiency more directly.
In addition to the above driving-performance metrics, we further evaluate training-stage safety and convergence efficiency. Training-stage safety is measured by the number of collisions during training, where a collision refers to the ego vehicle colliding with surrounding vehicles or road boundaries. Convergence efficiency is quantified by the number of episodes required to reach a predefined reward threshold. Let R e denote the episodic return at episode e. We first compute the moving-average reward with a window size of W:
R ¯ e = 1 W i = e W + 1 e R i .
The number of episodes required to reach the reward threshold is then defined as:
E reward = min { e R ¯ e R th } ,
where R th denotes the predefined reward threshold. This metric is computed independently for each random seed.
The moving-average window size W, and reward threshold R th used for convergence evaluation are summarized in Table 3. The threshold in each scenario is selected as approximately 90% of the final moving-average reward achieved by AGRL, so that the convergence metric reflects the episode at which each method reaches a near-converged performance level.

5.3. Policy Evaluation

5.3.1. Decision Model Training

Figure 4 illustrates the learning curves in the highway lane-change scenario. Compared with all baseline methods, AGRL exhibits substantially faster convergence in terms of cumulative reward, success rate, safety cost, and comfort cost. In particular, the proposed method reaches a stable and favorable performance level considerably earlier than the baseline methods. By contrast, AGRL yields a relatively lower average speed, indicating that the learned policy tends to adopt a more conservative driving style in exchange for improved safety and higher overall return.
It is also observed that RND, SimHash, and ICM achieve performance close to that of conventional value-based baselines such as DQN, DDQN, and DuDQN. One important reason is that these methods rely on external novelty estimation mechanisms, such as state prediction or state counting, to guide exploration. However, the driving environments considered in this work exhibit strong uncertainty. Surrounding vehicles change lanes autonomously and also respond to the behavior of the ego vehicle, making the environment highly dynamic and partially non-stationary. Under such conditions, predicting future states or constructing reliable novelty estimates becomes inherently difficult. In fact, modeling the environment well enough to support effective exploration is nearly as challenging as learning the driving policy itself. As a result, the additional exploration signals provided by RND, SimHash, and ICM become less informative, which limits their advantage over conventional random-exploration baselines. In contrast, the proposed AGRL framework does not rely on an auxiliary model to predict environmental novelty. Instead, it depends only on the internal attribution signals of the decision model itself, making the exploration signal more directly coupled to the policy’s local decision response and therefore more effective in highly complex driving environments.
Table 4 summarizes the total number of training-stage collisions and the convergence episodes in the highway lane-change scenario. Compared with the baseline methods, AGRL requires fewer episodes to reach convergence, with an average convergence episode of 65.8 , while the other methods require between 77.0 and 90.8 episodes. Moreover, AGRL substantially reduces the number of collisions during training. Specifically, AGRL records only 23.4 ± 7.57 collisions, which is substantially lower than all baseline methods. Therefore, in the highway lane-change scenario, AGRL provides advantages in both learning efficiency and training-stage safety.
In addition to convergence efficiency, we further analyze the computational overhead of different methods in the highway lane-change scenario. All simulation experiments were conducted on a computer equipped with an Intel i9-10900K CPU, an NVIDIA GeForce RTX 3090 GPU, and 32 GB of RAM. The comparison includes the number of trainable parameters, runtime per training step, total training time, and single-step inference latency during deployment, as reported in Table 5.
In terms of trainable parameters, AGRL remains lightweight. The KAN-IDRL backbone contains only 2400 trainable parameters, and the prior-attention layer introduced in AGRL adds only a small number of additional parameters. As a result, AGRL has 2436 trainable parameters, which is substantially fewer than Dueling DQN, RND, and ICM.
However, AGRL introduces additional computational cost during training. Its runtime per training step and total training time are higher than those of the DQN-based baselines. This overhead mainly comes from two sources: the spline-based computation in KAN and the additional forward and attribution-computation operations required to obtain the attribution-sensitivity signal. The spline-based computation is inherent to the KAN-based policy and remains during deployment, whereas the exploration-specific operations are mainly incurred during training. During deployment, the additional perturbed-state forward pass and attribution computation can be disabled, resulting in an inference latency close to that of KAN-IDRL.
Figure 5 presents the learning curves in the main-road yielding scenario. AGRL again demonstrates a clear advantage in convergence speed over the baseline methods. Although its cumulative reward is lower than that of some baselines during the early training stage, its success rate is higher at the same time, suggesting that the learned policy initially favors safer and more conservative actions. As training proceeds, the policy gradually learns to balance safety and efficiency more effectively, which is reflected in the rapid improvement of both cumulative reward and average speed. Similar to the observations in the highway lane-change scenario, the performance of RND, SimHash, and ICM remains close to that of the conventional value-based baselines, indicating that their exploration advantage is limited in the current driving task.
Table 6 reports the training-stage collision count and convergence episodes in the main-road yielding scenario. AGRL achieves the fastest convergence, requiring only 415.2 episodes on average, whereas the baseline methods require between 438.6 and 471.4 episodes. AGRL also records the fewest training-stage collisions, with 185.4 ± 34.1 collisions, which is substantially lower than all baseline methods.
Figure 6 shows the learning curves in the on-ramp merging scenario. This scenario is more challenging because purely conservative behavior cannot successfully complete the task; if the ego vehicle fails to merge within the designated road segment, the episode is terminated as a failure. Despite this increased complexity, AGRL still achieves faster convergence than the baseline methods. This result indicates that the proposed framework is able to guide the agent more efficiently toward effective decision policies even in tasks that require a delicate balance between safety and assertiveness. The superior training performance in this scenario further demonstrates the robustness and adaptability of AGRL under more demanding decision-making conditions.
Table 7 reports the training-stage collision count and convergence episodes in the on-ramp merging scenario, AGRL achieves the fastest convergence in the on-ramp merging scenario, requiring only 617.3 episodes on average, while the baseline methods require between 652.6 and 773.2 episodes. For training-stage collisions, AGRL obtains 246.0 ± 20.8 , which is comparable to most baseline methods and lower than DDQN, RND, SimHash, and ICM. Although it does not produce the lowest collision count, AGRL achieves a clear convergence advantage while maintaining a comparable level of training safety in this challenging merging task.

5.3.2. Decision Model Testing

This subsection presents a quantitative comparison of all methods across the three driving scenarios.
As shown in Table 8, all methods perform well in the highway lane-change scenario, with success rates above 95%. Among them, AGRL achieves the highest average cumulative reward of 26.57, the highest success rate of 98.45%, and the lowest safety and comfort costs, i.e., 0.0023 and 0.0027 , respectively. Although AGRL yields a slightly lower average speed than the Dueling DQN baseline, its success rate is 2.73% higher. These results indicate that AGRL achieves the best overall driving performance in this scenario and provides a more favorable balance between safety and efficiency.
In the more complex main-road yielding scenario, AGRL outperforms all baseline methods in terms of cumulative reward, success rate, and average speed. As shown in Table 9, AGRL attains a success rate of 97.67% while simultaneously achieving the highest average speed of 29.96 m/s and the highest cumulative reward of 10.32. These results suggest that AGRL is able to promote more effective learning of complex yielding behaviors, enabling the agent to maintain both traffic efficiency and collision avoidance performance under interactive driving conditions.
In the most challenging on-ramp merging scenario, which requires real-time gap selection and fine-grained speed control, AGRL also demonstrates strong performance. As shown in Table 10, AGRL achieves the highest success rate of 98.32%, the highest cumulative reward of 8.64, and the lowest safety and comfort costs. Although its average speed is slightly lower than that of DDQN, its success rate remains 1.97% higher. This result indicates that AGRL is better able to balance aggressiveness and safety in difficult merging tasks, thereby improving the overall quality of the learned policy.
Overall, AGRL achieves the strongest driving performance across all scenarios, consistently obtaining the highest average cumulative reward and the highest success rate.

5.4. Ablation Study

To further analyze the contribution of different components in AGRL, we conduct two groups of ablation studies. The first group evaluates the effectiveness of the main algorithmic components, including the prior-attention mechanism and the attribution-sensitivity-guided exploration module. The second group compares different attribution-change metrics as exploration signals to verify the choice of Euclidean attribution variation.

5.4.1. Component Ablation

This experiment evaluates the individual contributions of the global-attribution-based prior attention mechanism and the attribution-sensitivity-guided exploration module. Specifically, we compare AGRL with four ablated variants: KAN-IDRL, AGRL-w/o-PA, AGRL-UniformPA, and AGRL-w/o-ASE. KAN-IDRL denotes the backbone model without prior attention or attribution-sensitivity-guided exploration. AGRL-w/o-PA removes the prior-attention module while retaining attribution-sensitivity-guided exploration. AGRL-UniformPA keeps the trainable prior-attention gate but initializes all attention coefficients to 1, so that no attribution-derived feature preference is introduced at initialization. AGRL-w/o-ASE retains the attribution-derived prior attention but removes the attribution-sensitivity-guided exploration module and uses random exploration instead.
As shown in Figure 7, the variants without the attribution-sensitivity-guided exploration module, including KAN-IDRL and AGRL-w/o-ASE, show clear performance degradation in cumulative reward, success rate, and convergence speed. This indicates that the attribution-sensitivity-guided exploration strategy plays an important role in improving exploration efficiency. AGRL-w/o-PA and AGRL-UniformPA achieve performance closer to full AGRL, but both still show degradation compared with AGRL. This suggests that the prior-attention module contributes to learning efficiency by providing attribution-derived feature-level prior knowledge, especially in the early training stage.
Moreover, the similar performance of AGRL-w/o-PA and AGRL-UniformPA indicates that merely adding a trainable attention gate is insufficient to reproduce the full benefit of AGRL. The improvement mainly comes from the attribution-derived prior initialization rather than from the additional gate structure itself. Overall, the strengthened ablation results demonstrate that both the prior-attention module and the attribution-sensitivity-guided exploration module contribute to AGRL, with the exploration module having a more pronounced effect on convergence acceleration.

5.4.2. Ablation of Attribution-Based Exploration Signals

This experiment examines whether Euclidean attribution variation is more suitable for guiding exploration than other attribution-change metrics. Specifically, we compare five variants: AGRL, AGRL-CosDir, AGRL-SignFlip, AGRL-TopK, and AGRL-w/o-ASE. AGRL corresponds to the main exploration strategy used in the proposed method, where the Euclidean distance between the original and perturbed action-wise attribution vectors is used as the exploration signal. AGRL-CosDir uses the cosine-direction change between the two attribution vectors, AGRL-SignFlip uses the sign-flip ratio of attribution values, and AGRL-TopK uses the change in the top-k most important attribution features. AGRL-w/o-ASE removes the attribution-guided active exploration module and uses random exploration instead.
The ablation experiment is conducted in the highway lane-change scenario, and the results are shown in Figure 8. The results indicate that AGRL, which uses Euclidean attribution variation as the exploration signal, achieves the fastest convergence among all variants. In contrast, AGRL-w/o-ASE, which removes the attribution-guided active exploration module and relies on random exploration, shows the slowest convergence. This confirms that the attribution-guided exploration module contributes directly to improving learning efficiency.
Among the alternative attribution-change metrics, AGRL-CosDir achieves the closest performance to AGRL. This suggests that direction-related attribution information also has potential for guiding exploration. AGRL-SignFlip and AGRL-TopK also improve over AGRL-w/o-ASE to some extent, but their effects are weaker than those of AGRL and AGRL-CosDir. A possible reason is that the sign-flip ratio and top-k feature-overlap change are relatively discrete metrics. They may fail to capture fine-grained continuous changes in attribution magnitude, making the resulting exploration signal less stable and less informative. Overall, these results support the use of Euclidean attribution variation as the primary exploration signal in AGRL.

5.5. Interpretability Analysis in a Continuous Semantic Driving Scenario

To verify whether attribution variation reflects meaningful learned decision responses rather than random attribution noise or irrelevant feature interactions, we conduct a diagnostic interpretability analysis in a continuous semantic driving scenario.
As shown in Figure 9, we select a continuous semantic driving scenario to compare the attribution changes of a well-trained policy and an untrained policy. In this scenario, the ego vehicle drives in the leftmost lane, while the longitudinal distance to the front-right vehicle, denoted as Δ x 3 , gradually decreases. We analyze the attribution heatmaps of the right-lane-change action.
The six rows in the heatmap represent the observations of the DRL agent: row 1 corresponds to the ego vehicle state, while rows 2 to 6 correspond to the states of the surrounding vehicles. Red indicates a positive contribution, blue indicates a negative contribution, and white represents a neutral or negligible contribution. These values indicate whether each feature supports or opposes the predicted advantage of the queried action.
For the well-trained policy, the attribution score associated with Δ x 3 exhibits the most significant magnitude change, decreasing from 0.19 to 1.63 . This is consistent with the driving semantics of the scenario: as the front-right vehicle becomes closer, the risk of executing a right-lane-change action increases, and the policy assigns a stronger negative attribution to the corresponding safety-critical feature. Therefore, the large Euclidean attribution variation in this case mainly reflects a meaningful magnitude response to increasing traffic risk. In contrast, the untrained policy does not exhibit such a semantically consistent attribution response.
To further quantify this phenomenon, we compute four attribution-change metrics among consecutive attribution vectors: Euclidean distance, cosine similarity, sign-flip rate, and top-5 feature overlap. The Euclidean distance measures the magnitude of attribution changes, while cosine similarity, sign-flip rate, and top-5 overlap characterize the consistency of attribution direction, sign pattern, and key feature selection, respectively. We report the average values over all three vector pairs in Table 11.
The well-trained policy exhibits a larger mean Euclidean distance than the untrained policy, indicating stronger attribution-magnitude responses to the continuous change in the driving scenario. However, it also achieves higher cosine similarity, a lower sign-flip rate, and higher top-5 overlap. This means that the trained policy does not produce disordered attribution changes. Instead, it maintains a more consistent attribution direction and a more stable key-feature structure, while the attribution magnitudes of safety-critical features change more strongly. This finding supports that Euclidean attribution variation is a meaningful signal, as it reflects the response strength of the learned policy to continuous environmental changes rather than random or disordered attribution fluctuations.

6. Conclusions

In this work, we proposed an Attribution-Guided Reinforcement Learning (AGRL) framework for autonomous driving decision-making. The proposed method performs active exploration by evaluating perturbation-induced attribution sensitivity, allowing the agent to identify actions whose decision rationales are more locally responsive to state changes and to explore them in a targeted manner. Compared with conventional exploration-enhanced methods that rely on auxiliary predictive models or external novelty estimation, AGRL exploits the internal attribution signals of the decision model itself, making the exploration process more directly coupled to the policy’s local decision response. In addition, a global-attribution-based prior attention module is introduced to transfer attribution-derived prior knowledge from a pretrained interpretable policy to a new decision model, thereby accelerating learning in the early training stage.
This work highlights the potential of transforming attribution-based explainability from a post hoc analysis tool into an effective guidance signal for policy learning. Future work will focus on real-vehicle experiments to further validate the effectiveness and practical applicability of the proposed framework.

Author Contributions

Conceptualization, J.H.; formal analysis, J.H. and R.Z.; investigation, J.H. and X.S.; data curation, Y.W., J.H. and X.S.; writing—original draft preparation, J.H. and R.Z.; writing—review and editing, R.Z. All authors have read and agreed to the published version of the manuscript.

Funding

This work was supported by the National Natural Science Foundation of China (NSFC) under Grant No. 51975194.

Institutional Review Board Statement

Not applicable.

Informed Consent Statement

Not applicable.

Data Availability Statement

The data presented in this study are available on request from the corresponding author. The data are not publicly available due to research privacy restrictions.

Conflicts of Interest

The authors declare no conflicts of interest.

References

  1. Wang, X.; Qi, X.; Wang, P.; Yang, J. Decision making framework for autonomous vehicles driving behavior in complex scenarios via hierarchical state machine. Auton. Intell. Syst. 2021, 1, 10. [Google Scholar] [CrossRef] [Scilit]
  2. Fu, Y.; Li, C.; Yu, F.R.; Luan, T.H.; Zhang, Y. Hybrid autonomous driving guidance strategy combining deep reinforcement learning and expert system. IEEE Trans. Intell. Transp. Syst. 2021, 23, 11273–11286. [Google Scholar] [CrossRef] [Scilit]
  3. Bae, S.H.; Joo, S.H.; Pyo, J.W.; Yoon, J.S.; Lee, K.; Kuc, T.Y. Finite state machine based vehicle system for autonomous driving in urban environments. In Proceedings of the 2020 20th International Conference on Control, Automation and Systems (ICCAS); IEEE: New York, NY, USA, 2020; pp. 1181–1186. [Google Scholar]
  4. Omeiza, D.; Webb, H.; Jirotka, M.; Kunze, L. Explanations in autonomous driving: A survey. IEEE Trans. Intell. Transp. Syst. 2021, 23, 10142–10162. [Google Scholar] [CrossRef] [Scilit]
  5. Li, X.; Bai, Y.; Cai, P.; Wen, L.; Fu, D.; Zhang, B.; Yang, X.; Cai, X.; Ma, T.; Guo, J.; et al. Towards knowledge-driven autonomous driving. arXiv 2023, arXiv:2312.04316. [Google Scholar] [CrossRef] [Scilit]
  6. Jia, X.; Yang, Z.; Li, Q.; Zhang, Z.; Yan, J. Bench2drive: Towards multi-ability benchmarking of closed-loop end-to-end autonomous driving. Adv. Neural Inf. Process. Syst. 2024, 37, 819–844. [Google Scholar]
  7. Sun, J.; Kim, J. Modelling two-dimensional driving behaviours at unsignalised intersection using multi-agent imitation learning. Transp. Res. Part C Emerg. Technol. 2024, 165, 104702. [Google Scholar] [CrossRef] [Scilit]
  8. Huang, Z.; Sheng, Z.; Chen, S. PE-RLHF: Reinforcement Learning with Human Feedback and physics knowledge for safe and trustworthy autonomous driving. Transp. Res. Part C Emerg. Technol. 2025, 179, 105262. [Google Scholar] [CrossRef] [Scilit]
  9. Huang, X.; Jing, P.; Li, Y.; Wang, X.; Wang, Y. Joint optimization of vehicle platoon and traffic signal with mixed traffic flow at intersections: Deep reinforcement learning approach. Transp. Res. Part C Emerg. Technol. 2025, 177, 105184. [Google Scholar] [CrossRef] [Scilit]
  10. Zhou, R.; Huang, J.; Li, M.; Li, H.; Cao, H.; Song, X. Knowledge transfer from simple to complex: A safe and efficient reinforcement learning framework for autonomous driving decision-making. Adv. Eng. Inform. 2025, 65, 103188. [Google Scholar] [CrossRef] [Scilit]
  11. Yang, K.; Tao, J.; Lyu, J.; Li, X. Exploration and anti-exploration with distributional random network distillation. arXiv 2024, arXiv:2401.09750. [Google Scholar] [CrossRef] [Scilit]
  12. Shyam, P.; Jaśkowski, W.; Gomez, F. Model-based active exploration. In Proceedings of the International Conference on Machine Learning; PMLR: Tokyo, Japan, 2019; pp. 5779–5788. [Google Scholar]
  13. Pathak, D.; Agrawal, P.; Efros, A.A.; Darrell, T. Curiosity-driven exploration by self-supervised prediction. In Proceedings of the International Conference on Machine Learning; PMLR: Tokyo, Japan, 2017; pp. 2778–2787. [Google Scholar]
  14. Houthooft, R.; Chen, X.; Duan, Y.; Schulman, J.; De Turck, F.; Abbeel, P. Vime: Variational information maximizing exploration. Adv. Neural Inf. Process. Syst. 2016, 29. [Google Scholar]
  15. Huang, J.; Zhou, R.; Li, M.; Li, H.; Liu, Y.; Song, X. From black-box to white-box: Interpretable deep reinforcement learning with Kolmogorov-Arnold networks for autonomous driving. Transp. Res. Part C Emerg. Technol. 2026, 182, 105386. [Google Scholar] [CrossRef] [Scilit]
  16. Strehl, A.L.; Littman, M.L. An analysis of model-based interval estimation for Markov decision processes. J. Comput. Syst. Sci. 2008, 74, 1309–1331. [Google Scholar] [CrossRef] [Scilit]
  17. Azar, M.G.; Osband, I.; Munos, R. Minimax regret bounds for reinforcement learning. In Proceedings of the International Conference on Machine Learning; PMLR: Tokyo, Japan, 2017; pp. 263–272. [Google Scholar]
  18. Bellemare, M.; Srinivasan, S.; Ostrovski, G.; Schaul, T.; Saxton, D.; Munos, R. Unifying count-based exploration and intrinsic motivation. Adv. Neural Inf. Process. Syst. 2016, 29, 1479–1487. [Google Scholar]
  19. Ostrovski, G.; Bellemare, M.G.; Oord, A.; Munos, R. Count-based exploration with neural density models. In Proceedings of the International Conference on Machine Learning; PMLR: Tokyo, Japan, 2017; pp. 2721–2730. [Google Scholar]
  20. Machado, M.C.; Bellemare, M.G.; Bowling, M. Count-based exploration with the successor representation. In Proceedings of the AAAI Conference on Artificial Intelligence; IEEE: New York, NY, USA, 2020; Volume 34, pp. 5125–5133. [Google Scholar]
  21. Lobel, S.; Bagaria, A.; Konidaris, G. Flipping coins to estimate pseudocounts for exploration in reinforcement learning. In Proceedings of the International Conference on Machine Learning; PMLR: Tokyo, Japan, 2023; pp. 22594–22613. [Google Scholar]
  22. Burda, Y.; Edwards, H.; Storkey, A.; Klimov, O. Exploration by random network distillation. arXiv 2018, arXiv:1810.12894. [Google Scholar] [CrossRef] [Scilit]
  23. Lundberg, S.M.; Lee, S.I. A Unified Approach to Interpreting Model Predictions. In Advances in Neural Information Processing Systems 30; Guyon, I., Luxburg, U.V., Bengio, S., Wallach, H., Fergus, R., Vishwanathan, S., Garnett, R., Eds.; Curran Associates, Inc.: New York, NY, USA, 2017; pp. 4765–4774. [Google Scholar]
  24. Ribeiro, M.T.; Singh, S.; Guestrin, C. “Why should i trust you?” Explaining the predictions of any classifier. In Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining; IEEE: New York, NY, USA, 2016; pp. 1135–1144. [Google Scholar]
  25. Baehrens, D.; Schroeter, T.; Harmeling, S.; Kawanabe, M.; Hansen, K.; Müller, K.R. How to explain individual classification decisions. J. Mach. Learn. Res. 2010, 11, 1803–1831. [Google Scholar]
  26. Khorram, S.; Lawson, T.; Fuxin, L. iGOS++ integrated gradient optimized saliency by bilateral perturbations. In Proceedings of the Conference on Health, Inference, and Learning; IEEE: New York, NY, USA, 2021; pp. 174–182. [Google Scholar]
  27. Sundararajan, M.; Taly, A.; Yan, Q. Axiomatic attribution for deep networks. In Proceedings of the International Conference on Machine Learning; PMLR: Tokyo, Japan, 2017; pp. 3319–3328. [Google Scholar]
  28. Smilkov, D.; Thorat, N.; Kim, B.; Viégas, F.; Wattenberg, M. Smoothgrad: Removing noise by adding noise. arXiv 2017, arXiv:1706.03825. [Google Scholar] [CrossRef] [Scilit]
  29. Selvaraju, R.R.; Cogswell, M.; Das, A.; Vedantam, R.; Parikh, D.; Batra, D. Grad-cam: Visual explanations from deep networks via gradient-based localization. In Proceedings of the IEEE International Conference on Computer Vision; IEEE: New York, NY, USA, 2017; pp. 618–626. [Google Scholar]
  30. Chattopadhay, A.; Sarkar, A.; Howlader, P.; Balasubramanian, V.N. Grad-cam++: Generalized gradient-based visual explanations for deep convolutional networks. In Proceedings of the 2018 IEEE Winter Conference on Applications of Computer Vision (WACV); IEEE: New York, NY, USA, 2018; pp. 839–847. [Google Scholar]
  31. Montavon, G.; Binder, A.; Lapuschkin, S.; Samek, W.; Müller, K.R. Layer-wise relevance propagation: An overview. In Explainable AI: Interpreting, Explaining and Visualizing Deep Learning; Springer: Berlin/Heidelberg, Germany, 2019; pp. 193–209. [Google Scholar]
  32. Li, J.; Zhang, C.; Zhou, J.T.; Fu, H.; Xia, S.; Hu, Q. Deep-LIFT: Deep label-specific feature learning for image annotation. IEEE Trans. Cybern. 2021, 52, 7732–7741. [Google Scholar] [CrossRef] [Scilit]
  33. Wang, Z.; Schaul, T.; Hessel, M.; Hasselt, H.; Lanctot, M.; Freitas, N. Dueling network architectures for deep reinforcement learning. In Proceedings of the International Conference on Machine Learning; PMLR: Tokyo, Japan, 2016; pp. 1995–2003. [Google Scholar]
  34. Leurent, E. An Environment for Autonomous Driving Decision-Making. 2018. Available online: https://github.com/eleurent/highway-env (accessed on 1 March 2026).
  35. Bellotti, F.; Lazzaroni, L.; Capello, A.; Cossu, M.; De Gloria, A.; Berta, R. Explaining a deep reinforcement learning (DRL)-based automated driving agent in highway simulations. IEEE Access 2023, 11, 28522–28550. [Google Scholar] [CrossRef] [Scilit]
  36. Treiber, M.; Hennecke, A.; Helbing, D. Congested traffic states in empirical observations and microscopic simulations. Phys. Rev. E 2000, 62, 1805. [Google Scholar] [CrossRef] [Scilit]
  37. Kesting, A.; Treiber, M.; Helbing, D. General lane-changing model MOBIL for car-following models. Transp. Res. Rec. 2007, 1999, 86–94. [Google Scholar] [CrossRef] [Scilit]
  38. Mnih, V.; Kavukcuoglu, K.; Silver, D.; Rusu, A.A.; Veness, J.; Bellemare, M.G.; Graves, A.; Riedmiller, M.; Fidjeland, A.K.; Ostrovski, G.; et al. Human-level control through deep reinforcement learning. Nature 2015, 518, 529–533. [Google Scholar] [CrossRef] [Scilit] [PubMed]
  39. Van Hasselt, H.; Guez, A.; Silver, D. Deep reinforcement learning with double q-learning. In Proceedings of the AAAI Conference on Artificial Intelligence; IEEE: New York, NY, USA, 2016; Volume 30. [Google Scholar]
  40. Tang, H.; Houthooft, R.; Foote, D.; Stooke, A.; Xi Chen, O.; Duan, Y.; Schulman, J.; DeTurck, F.; Abbeel, P. #Exploration: A study of count-based exploration for deep reinforcement learning. Adv. Neural Inf. Process. Syst. 2017, 30, 2750–2759. [Google Scholar]
Figure 1. Overview of the proposed AGRL framework.
Figure 1. Overview of the proposed AGRL framework.
Applsci 16 04931 g001
Figure 2. Overview of the proposed PAE-KAN. Local attribution explanations are first extracted from a well-trained KAN-based model over a large number of driving scenarios and then aggregated into a global explanation. The resulting global feature importance is subsequently embedded into a new PAE-KAN decision model as a global-attribution-based prior attention layer.
Figure 2. Overview of the proposed PAE-KAN. Local attribution explanations are first extracted from a well-trained KAN-based model over a large number of driving scenarios and then aggregated into a global explanation. The resulting global feature importance is subsequently embedded into a new PAE-KAN decision model as a global-attribution-based prior attention layer.
Applsci 16 04931 g002
Figure 3. Experimental driving scenarios. (a) Four-lane highway lane-change. (b) Main-road yielding. (c) On-ramp merging.
Figure 3. Experimental driving scenarios. (a) Four-lane highway lane-change. (b) Main-road yielding. (c) On-ramp merging.
Applsci 16 04931 g003
Figure 4. Learning curves of autonomous driving agents in the highway lane-change scenario. (a) Training cumulative reward. (b) Success rate. (c) Average speed. (d) Safety cost. (e) Comfort cost.
Figure 4. Learning curves of autonomous driving agents in the highway lane-change scenario. (a) Training cumulative reward. (b) Success rate. (c) Average speed. (d) Safety cost. (e) Comfort cost.
Applsci 16 04931 g004
Figure 5. Learning curves of autonomous driving agents in the main-road yielding scenario. (a) Training cumulative reward. (b) Success rate. (c) Merge speed cost. (d) Safety cost. (e) Comfort cost. (f) Average speed.
Figure 5. Learning curves of autonomous driving agents in the main-road yielding scenario. (a) Training cumulative reward. (b) Success rate. (c) Merge speed cost. (d) Safety cost. (e) Comfort cost. (f) Average speed.
Applsci 16 04931 g005
Figure 6. Learning curves of autonomous driving agents in the on-ramp merging scenario. (a) Training cumulative reward. (b) Success rate. (c) Average speed. (d) Safety cost. (e) Comfort cost.
Figure 6. Learning curves of autonomous driving agents in the on-ramp merging scenario. (a) Training cumulative reward. (b) Success rate. (c) Average speed. (d) Safety cost. (e) Comfort cost.
Applsci 16 04931 g006
Figure 7. Training performance of different ablated versions of AGRL in the highway lane-change scenario. (a) Training cumulative reward. (b) Success rate. (c) Average speed. (d) Safety cost. (e) Comfort cost.
Figure 7. Training performance of different ablated versions of AGRL in the highway lane-change scenario. (a) Training cumulative reward. (b) Success rate. (c) Average speed. (d) Safety cost. (e) Comfort cost.
Applsci 16 04931 g007
Figure 8. Ablation study of different attribution-based exploration signals in the highway lane-change scenario. (a) Training cumulative reward. (b) Success rate. (c) Average speed. (d) Safety cost. (e) Comfort cost.
Figure 8. Ablation study of different attribution-based exploration signals in the highway lane-change scenario. (a) Training cumulative reward. (b) Success rate. (c) Average speed. (d) Safety cost. (e) Comfort cost.
Applsci 16 04931 g008
Figure 9. LRP-KAN attribution responses in a continuous semantic driving scenario. The red arrows indicate changes in attribution responses.
Figure 9. LRP-KAN attribution responses in a continuous semantic driving scenario. The red arrows indicate changes in attribution responses.
Applsci 16 04931 g009
Table 1. Standard deviations of Gaussian perturbations in the normalized feature space.
Table 1. Standard deviations of Gaussian perturbations in the normalized feature space.
Feature TypeVariableStandard Deviation σ i
Binary/discretepresence0
Continuousx0.03
Continuousy0.03
Continuous v x 0.02
Continuous v y 0.02
Continuous a x 0.01
Table 2. Hyperparameters for different driving scenarios.
Table 2. Hyperparameters for different driving scenarios.
ParametersHighway Lane-ChangeMain-Road YieldingOn-Ramp Merging
Total Episodes30010002000
Discount Factor0.80.80.95
Soft Update Rate0.0050.0050.005
Exploration Decay Rate 5 × 10 4 5 × 10 5 5 × 10 5
Replay Buffer Capacity 1 × 10 6 1 × 10 6 1 × 10 6
Batch Size256256256
Table 3. Settings for convergence-threshold evaluation in different scenarios.
Table 3. Settings for convergence-threshold evaluation in different scenarios.
ScenarioWindow Size WReward Threshold R th
Highway lane-change3024.0
Main-road yielding1009.0
On-ramp merging2007.5
Table 4. Training-stage collision count and convergence episodes in the highway lane-change scenario. Bold values indicate the best performance among all compared methods.
Table 4. Training-stage collision count and convergence episodes in the highway lane-change scenario. Bold values indicate the best performance among all compared methods.
MethodTraining CollisionsConvergence Episodes
DQN 57.6 ± 5.18 77.0 ± 8.15
DDQN 75.8 ± 4.27 80.8 ± 14.4
Dueling DQN 63.5 ± 6.74 84.2 ± 11.4
KAN-IDRL 39.6 ± 8.45 78.2 ± 19.7
RND 53.6 ± 4.74 78.8 ± 13.4
SimHash 58.3 ± 4.72 90.8 ± 17.25
ICM 67.7 ± 6.35 95.9 ± 14.2
AGRL 23.4 ± 7.57 65.8 ± 14.5
Table 5. Computational overhead comparison in the highway lane-change scenario.
Table 5. Computational overhead comparison in the highway lane-change scenario.
MethodTrainable Params.Runtime/StepTotal Training TimeInference Latency
Dueling DQN76,2920.06003099.96 s0.0589
KAN-IDRL24000.07544086.24 s0.0625
RND184,4520.06163148.73 s0.0587
SimHash76,2920.06063175.52 s0.0589
ICM251,9110.06203130.67 s0.0588
AGRL24360.08304759.39 s0.0624
Table 6. Training-stage collision count and convergence episodes in the main-road yielding scenario. Bold values indicate the best performance among all compared methods.
Table 6. Training-stage collision count and convergence episodes in the main-road yielding scenario. Bold values indicate the best performance among all compared methods.
MethodTraining CollisionsConvergence Episodes
DQN 253.2 ± 16.6 471.4 ± 14.7
DDQN 239.2 ± 19.3 447.4 ± 13.9
Dueling DQN 245.6 ± 21.8 459.8 ± 16.5
KAN-IDRL 221.4 ± 24.6 438.6 ± 18.7
RND 248.2 ± 15.7 457.6 ± 8.6
SimHash 250.8 ± 18.7 463.6 ± 11.0
ICM 281.0 ± 42.3 453.0 ± 13.5
AGRL 185.4 ± 34.1 415.2 ± 21.4
Table 7. Training-stage collision count and convergence episodes in the on-ramp merging scenario. Bold values indicate the best performance among all compared methods.
Table 7. Training-stage collision count and convergence episodes in the on-ramp merging scenario. Bold values indicate the best performance among all compared methods.
MethodTraining CollisionsConvergence Episodes
DQN 238.6 ± 25.6 743.4 ± 165.4
DDQN 251.6 ± 9.4 773.2 ± 229.1
Dueling DQN 241.0 ± 23.0 673.8 ± 109.7
KAN-IDRL 232.8 ± 27.4 652.6 ± 96.5
RND 249.4 ± 22.1 721.6 ± 138.2
SimHash 255.2 ± 24.7 734.8 ± 151.6
ICM 263.8 ± 31.5 706.4 ± 127.3
AGRL 246.0 ± 20.8 617.3 ± 53.0
Table 8. Performance comparison of all algorithms in the highway lane-change scenario. Bold values indicate the best performance among all compared methods.
Table 8. Performance comparison of all algorithms in the highway lane-change scenario. Bold values indicate the best performance among all compared methods.
MethodAvg. Cumulative RewardSuccess Rate (%)Avg. Speed (m/s)Safety Cost (× 10 3 )Comfort Cost (× 10 3 )
DQN26.04 ± 1.8496.35 ± 2.5423.89 ± 1.12−8.5 ± 8.21−6.1 ± 0.81
DDQN26.13 ± 1.7095.11 ± 2.8923.84 ± 1.08−7.4 ± 7.68−5.3 ± 0.57
Dueling DQN25.86 ± 1.7695.72 ± 2.6724.30 ± 1.15−9.3 ± 8.46−4.6 ± 0.74
KAN25.49 ± 1.5895.67 ± 2.3123.40 ± 0.96−7.8 ± 7.24−4.3 ± 0.62
RND26.25 ± 1.6995.34 ± 2.7623.76 ± 1.04−9.6 ± 8.73−5.4 ± 0.79
SimHash25.59 ± 1.6395.76 ± 2.4823.84 ± 1.01−2.4 ± 2.16−6.3 ± 0.86
ICM26.18 ± 1.7295.58 ± 2.6223.91 ± 1.06−8.8 ± 8.05−5.7 ± 0.77
AGRL26.57 ± 1.2198.45 ± 2.1823.21 ± 0.88−2.3 ± 1.74−2.7 ± 0.61
Table 9. Performance comparison of all algorithms in the main-road yielding scenario. Bold values indicate the best performance among all compared methods.
Table 9. Performance comparison of all algorithms in the main-road yielding scenario. Bold values indicate the best performance among all compared methods.
MethodAvg. Cumulative RewardSuccess Rate (%)Avg. Speed (m/s)Merging Speed Cost (× 10 3 )Safety Cost (× 10 3 )
DQN10.07 ± 0.4696.45 ± 2.1829.95 ± 0.42−1.58 ± 0.61−0.68 ± 0.29
DDQN10.03 ± 0.5195.40 ± 2.4729.92 ± 0.45−2.01 ± 0.74−0.79 ± 0.33
Dueling DQN10.29 ± 0.4395.53 ± 2.3629.81 ± 0.48−1.51 ± 0.58−0.84 ± 0.36
KAN9.649 ± 0.5795.11 ± 2.6929.03 ± 0.63−2.29 ± 0.82−1.24 ± 0.45
RND10.11 ± 0.4996.24 ± 2.2129.94 ± 0.44−2.07 ± 0.76−0.87 ± 0.37
SimHash10.19 ± 0.4795.97 ± 2.3329.75 ± 0.50−1.89 ± 0.69−0.90 ± 0.39
ICM10.16 ± 0.4896.08 ± 2.2929.88 ± 0.46−1.96 ± 0.72−0.86 ± 0.35
AGRL10.32 ± 0.3197.67 ± 1.4229.96 ± 0.28−2.09 ± 0.64−0.69 ± 0.27
Table 10. Performance comparison of all algorithms in the on-ramp merging scenario. Bold values indicate the best performance among all compared methods.
Table 10. Performance comparison of all algorithms in the on-ramp merging scenario. Bold values indicate the best performance among all compared methods.
MethodAvg. Cumulative RewardSuccess Rate (%)Avg. Speed (m/s)Safety Cost (× 10 3 )Comfort Cost (× 10 3 )
DQN8.32 ± 0.5497.13 ± 1.9629.71 ± 0.39−2.15 ± 0.82−9.62 ± 1.74
DDQN8.51 ± 0.4996.35 ± 2.2129.84 ± 0.35−1.65 ± 0.71−5.84 ± 1.21
Dueling DQN8.12 ± 0.6193.51 ± 2.8429.74 ± 0.42−1.86 ± 0.76−8.59 ± 1.58
KAN8.16 ± 0.5896.87 ± 2.0529.65 ± 0.44−2.31 ± 0.89−6.48 ± 1.36
RND8.04 ± 0.6397.05 ± 1.9829.55 ± 0.47−2.19 ± 0.85−6.74 ± 1.42
SimHash8.42 ± 0.5296.75 ± 2.1329.21 ± 0.51−1.98 ± 0.79−7.51 ± 1.49
ICM8.39 ± 0.5596.92 ± 2.0729.68 ± 0.43−1.91 ± 0.77−6.21 ± 1.30
AGRL8.64 ± 0.3498.32 ± 1.1729.57 ± 0.31−1.24 ± 0.52−4.34 ± 0.96
Table 11. Quantitative comparison of attribution-change metrics between the well-trained and untrained models.
Table 11. Quantitative comparison of attribution-change metrics between the well-trained and untrained models.
ModelMean Euclidean DistanceMean Cosine SimilarityMean Sign-Flip RateMean Top-5 Overlap
Well-trained1.830.7918.5%73.3%
Untrained1.440.4835.2%53.3%
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content.

Share and Cite

MDPI and ACS Style

Huang, J.; Zhou, R.; Wang, Y.; Song, X. Attribution-Guided Active Exploration in Deep Reinforcement Learning for Autonomous Driving Decision-Making. Appl. Sci. 2026, 16, 4931. https://doi.org/10.3390/app16104931

AMA Style

Huang J, Zhou R, Wang Y, Song X. Attribution-Guided Active Exploration in Deep Reinforcement Learning for Autonomous Driving Decision-Making. Applied Sciences. 2026; 16(10):4931. https://doi.org/10.3390/app16104931

Chicago/Turabian Style

Huang, Jiakun, Rongliang Zhou, Yanlong Wang, and Xiaolin Song. 2026. "Attribution-Guided Active Exploration in Deep Reinforcement Learning for Autonomous Driving Decision-Making" Applied Sciences 16, no. 10: 4931. https://doi.org/10.3390/app16104931

APA Style

Huang, J., Zhou, R., Wang, Y., & Song, X. (2026). Attribution-Guided Active Exploration in Deep Reinforcement Learning for Autonomous Driving Decision-Making. Applied Sciences, 16(10), 4931. https://doi.org/10.3390/app16104931

Note that from the first issue of 2016, this journal uses article numbers instead of page numbers. See further details here.

Article Metrics

Back to TopTop