Explainable AI-aided Feature Selection and Model Reduction for DRL-based V2X Resource Allocation

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Khan, Nasir, Abdallah, Asmaa, Celik, Abdulkadir, Eltawil, Ahmed M., Coleri, Sinem
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866912201195913216
author Khan, Nasir
Abdallah, Asmaa
Celik, Abdulkadir
Eltawil, Ahmed M.
Coleri, Sinem
author_facet Khan, Nasir
Abdallah, Asmaa
Celik, Abdulkadir
Eltawil, Ahmed M.
Coleri, Sinem
contents Artificial intelligence (AI) is expected to significantly enhance radio resource management (RRM) in sixth-generation (6G) networks. However, the lack of explainability in complex deep learning (DL) models poses a challenge for practical implementation. This paper proposes a novel explainable AI (XAI)- based framework for feature selection and model complexity reduction in a model-agnostic manner. Applied to a multi-agent deep reinforcement learning (MADRL) setting, our approach addresses the joint sub-band assignment and power allocation problem in cellular vehicle-to-everything (V2X) communications. We propose a novel two-stage systematic explainability framework leveraging feature relevance-oriented XAI to simplify the DRL agents. While the former stage generates a state feature importance ranking of the trained models using Shapley additive explanations (SHAP)-based importance scores, the latter stage exploits these importance-based rankings to simplify the state space of the agents by removing the least important features from the model input. Simulation results demonstrate that the XAI-assisted methodology achieves 97% of the original MADRL sum-rate performance while reducing optimal state features by 28%, average training time by 11%, and trainable weight parameters by 46% in a network with eight vehicular pairs.
format Preprint
id arxiv_https___arxiv_org_abs_2501_13552
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Explainable AI-aided Feature Selection and Model Reduction for DRL-based V2X Resource Allocation
Khan, Nasir
Abdallah, Asmaa
Celik, Abdulkadir
Eltawil, Ahmed M.
Coleri, Sinem
Signal Processing
Artificial Intelligence
Machine Learning
Multiagent Systems
Artificial intelligence (AI) is expected to significantly enhance radio resource management (RRM) in sixth-generation (6G) networks. However, the lack of explainability in complex deep learning (DL) models poses a challenge for practical implementation. This paper proposes a novel explainable AI (XAI)- based framework for feature selection and model complexity reduction in a model-agnostic manner. Applied to a multi-agent deep reinforcement learning (MADRL) setting, our approach addresses the joint sub-band assignment and power allocation problem in cellular vehicle-to-everything (V2X) communications. We propose a novel two-stage systematic explainability framework leveraging feature relevance-oriented XAI to simplify the DRL agents. While the former stage generates a state feature importance ranking of the trained models using Shapley additive explanations (SHAP)-based importance scores, the latter stage exploits these importance-based rankings to simplify the state space of the agents by removing the least important features from the model input. Simulation results demonstrate that the XAI-assisted methodology achieves 97% of the original MADRL sum-rate performance while reducing optimal state features by 28%, average training time by 11%, and trainable weight parameters by 46% in a network with eight vehicular pairs.
title Explainable AI-aided Feature Selection and Model Reduction for DRL-based V2X Resource Allocation
topic Signal Processing
Artificial Intelligence
Machine Learning
Multiagent Systems
url https://arxiv.org/abs/2501.13552