Explainable AI-Guided Efficient Approximate DNN Generation for Multi-Pod Systolic Arrays

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Siddique, Ayesha, Khalil, Khurram, Hoque, Khaza Anuarul
Formato: Preprint
Publicado: 2025
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866917964424413184
author Siddique, Ayesha
Khalil, Khurram
Hoque, Khaza Anuarul
author_facet Siddique, Ayesha
Khalil, Khurram
Hoque, Khaza Anuarul
contents Approximate deep neural networks (AxDNNs) are promising for enhancing energy efficiency in real-world devices. One of the key contributors behind this enhanced energy efficiency in AxDNNs is the use of approximate multipliers. Unfortunately, the simulation of approximate multipliers does not usually scale well on CPUs and GPUs. As a consequence, this slows down the overall simulation of AxDNNs aimed at identifying the appropriate approximate multipliers to achieve high energy efficiency with a minimum accuracy loss. To address this problem, we present a novel XAI-Gen methodology, which leverages the analytical model of the emerging hardware accelerator (e.g., Google TPU v4) and explainable artificial intelligence (XAI) to precisely identify the non-critical layers for approximation and quickly discover the appropriate approximate multipliers for AxDNN layers. Our results show that XAI-Gen achieves up to 7x lower energy consumption with only 1-2% accuracy loss. We also showcase the effectiveness of the XAI-Gen approach through a neural architecture search (XAI-NAS) case study. Interestingly, XAI-NAS achieves 40\% higher energy efficiency with up to 5x less execution time when compared to the state-of-the-art NAS methods for generating AxDNNs.
format Preprint
id arxiv_https___arxiv_org_abs_2503_16583
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Explainable AI-Guided Efficient Approximate DNN Generation for Multi-Pod Systolic Arrays
Siddique, Ayesha
Khalil, Khurram
Hoque, Khaza Anuarul
Machine Learning
Artificial Intelligence
Hardware Architecture
Approximate deep neural networks (AxDNNs) are promising for enhancing energy efficiency in real-world devices. One of the key contributors behind this enhanced energy efficiency in AxDNNs is the use of approximate multipliers. Unfortunately, the simulation of approximate multipliers does not usually scale well on CPUs and GPUs. As a consequence, this slows down the overall simulation of AxDNNs aimed at identifying the appropriate approximate multipliers to achieve high energy efficiency with a minimum accuracy loss. To address this problem, we present a novel XAI-Gen methodology, which leverages the analytical model of the emerging hardware accelerator (e.g., Google TPU v4) and explainable artificial intelligence (XAI) to precisely identify the non-critical layers for approximation and quickly discover the appropriate approximate multipliers for AxDNN layers. Our results show that XAI-Gen achieves up to 7x lower energy consumption with only 1-2% accuracy loss. We also showcase the effectiveness of the XAI-Gen approach through a neural architecture search (XAI-NAS) case study. Interestingly, XAI-NAS achieves 40\% higher energy efficiency with up to 5x less execution time when compared to the state-of-the-art NAS methods for generating AxDNNs.
title Explainable AI-Guided Efficient Approximate DNN Generation for Multi-Pod Systolic Arrays
topic Machine Learning
Artificial Intelligence
Hardware Architecture
url https://arxiv.org/abs/2503.16583