arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2305.04887cs.LGcs.AI
收藏

可解释人工智能的硬件加速

Hardware Acceleration of Explainable Artificial Intelligence

University of Florida(佛罗里达大学)

Zhixin Pan, Prabhat Mishra

AI总结:

提出利用TPU硬件加速多种可解释AI算法,通过矩阵计算与卷积-傅里叶协同,实现平均39倍加速和69倍能效提升,支持实时解释。

摘要:

机器学习(ML)在各个领域成功实现了人类水平的人工智能。然而,由于其黑箱特性,它缺乏解释结果的能力。尽管近期在可解释人工智能(XAI)方面的努力受到了广泛关注,但大多数现有解决方案不适用于实时系统,因为它们将可解释性映射为优化问题,这导致了大量耗时的复杂计算的多次迭代。尽管已有基于硬件的XAI加速框架,但它们通过FPGA实现且针对特定任务设计,导致成本高昂且缺乏灵活性。在本文中,我们提出一个简单而高效的框架,利用现有硬件加速器加速各种XAI算法。具体而言,本文做出了三个重要贡献:(1)所提出的方法是首次尝试探索张量处理单元(TPU)在加速XAI方面的有效性。(2)我们提出的解决方案探索了若干现有XAI算法与矩阵计算之间的紧密联系,并利用了卷积与傅里叶变换之间的协同作用,充分发挥了TPU在加速矩阵计算方面的固有优势。(3)我们提出的方法能够实现实时结果解释。大量实验评估表明,与现有加速技术相比,部署在TPU上的所提方法在解释时间(平均提升39倍)和能源效率(平均提升69倍)方面均能提供显著改进。

英文摘要:

Machine learning (ML) is successful in achieving human-level artificial intelligence in various fields. However, it lacks the ability to explain an outcome due to its black-box nature. While recent efforts on explainable AI (XAI) has received significant attention, most of the existing solutions are not applicable in real-time systems since they map interpretability as an optimization problem, which leads to numerous iterations of time-consuming complex computations. Although there are existing hardware-based acceleration framework for XAI, they are implemented through FPGA and designed for specific tasks, leading to expensive cost and lack of flexibility. In this paper, we propose a simple yet efficient framework to accelerate various XAI algorithms with existing hardware accelerators. Specifically, this paper makes three important contributions. (1) The proposed method is the first attempt in exploring the effectiveness of Tensor Processing Unit (TPU) to accelerate XAI. (2) Our proposed solution explores the close relationship between several existing XAI algorithms with matrix computations, and exploits the synergy between convolution and Fourier transform, which takes full advantage of TPU's inherent ability in accelerating matrix computations. (3) Our proposed approach can lead to real-time outcome interpretation. Extensive experimental evaluation demonstrates that proposed approach deployed on TPU can provide drastic improvement in interpretation time (39x on average) as well as energy efficiency (69x on average) compared to existing acceleration techniques.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑