arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2103.11927cs.LGcs.AI
收藏

使用张量处理单元的可解释机器学习的硬件加速

Hardware Acceleration of Explainable Machine Learning using Tensor Processing Units

University of Florida(佛罗里达大学)

Zhixin Pan, Prabhat Mishra

摘要:

机器学习(ML)在各个领域成功地实现了人类水平的性能。然而,由于其黑箱性质,它缺乏解释结果的能力。虽然现有的可解释机器学习是有前途的,但几乎所有这些方法都专注于将可解释性格式化为优化问题。这样的映射导致大量耗时的复杂计算的迭代,这限制了它们在实时应用中的适用性。在本文中,我们提出了一个新的框架,用于使用张量处理单元(TPU)加速可解释的ML。该框架利用了矩阵卷积和傅里叶变换之间的协同作用,并充分利用了TPU加速矩阵计算的自然能力。具体而言,本文做出了三个重要贡献。(1)据我们所知,我们提出的工作是首次尝试使用TPU实现可解释ML的硬件加速。(2)我们提出的方法适用于各种ML算法,有效利用基于TPU的加速可以实现实时结果解释。(3)大量的实验结果表明,我们提出的方法可以提供一个数量级的加速在分类时间(平均25倍)和解释时间(平均13倍)相比,国家的最先进的技术。

英文摘要:

Machine learning (ML) is successful in achieving human-level performance in various fields. However, it lacks the ability to explain an outcome due to its black-box nature. While existing explainable ML is promising, almost all of these methods focus on formatting interpretability as an optimization problem. Such a mapping leads to numerous iterations of time-consuming complex computations, which limits their applicability in real-time applications. In this paper, we propose a novel framework for accelerating explainable ML using Tensor Processing Units (TPUs). The proposed framework exploits the synergy between matrix convolution and Fourier transform, and takes full advantage of TPU's natural ability in accelerating matrix computations. Specifically, this paper makes three important contributions. (1) To the best of our knowledge, our proposed work is the first attempt in enabling hardware acceleration of explainable ML using TPUs. (2) Our proposed approach is applicable across a wide variety of ML algorithms, and effective utilization of TPU-based acceleration can lead to real-time outcome interpretation. (3) Extensive experimental results demonstrate that our proposed approach can provide an order-of-magnitude speedup in both classification time (25x on average) and interpretation time (13x on average) compared to state-of-the-art techniques.

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

↑