Semantic segmentation models are widely used in various computer vision tasks. However, these models often suffer from biases and inefficiencies, which can limit their performance. In this paper, we propose a novel approach of model pruning using explainable AI (XAI) techniques. The proposed approach aims at identifying and eliminating non-pertinent channels in convolutional layers. It is applied for segmentation-based models, where we use XAI as the criterion for pruning techniques specifically tailored for segmentation tasks. By guiding the pruning process by means of explainability metrics, thorough experiments were conducted on the DIVA-HisDB benchmark dataset to correct model biases and enhance overall efficiency and performance.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

DocXAI-Pruner: Optimizing Semantic Segmentation Models for Document Layout Analysis Via Explainable AI-Driven Pruning

  • Iheb Brini,
  • Najoua Rahal,
  • Maroua Mehri,
  • Rolf Ingold,
  • Najoua Essoukri Ben Amara

摘要

Semantic segmentation models are widely used in various computer vision tasks. However, these models often suffer from biases and inefficiencies, which can limit their performance. In this paper, we propose a novel approach of model pruning using explainable AI (XAI) techniques. The proposed approach aims at identifying and eliminating non-pertinent channels in convolutional layers. It is applied for segmentation-based models, where we use XAI as the criterion for pruning techniques specifically tailored for segmentation tasks. By guiding the pruning process by means of explainability metrics, thorough experiments were conducted on the DIVA-HisDB benchmark dataset to correct model biases and enhance overall efficiency and performance.