arXiv:2412.06795cs.NEcs.AI2024-12被引 9

为脉冲神经网络设计故障注入框架,提升硬件可靠性测试效率。

SpikeFI: A Fault Injection Framework for Spiking Neural Networks

论文配图:SpikeFI: A Fault Injection Framework for Spiking Neural Networks
图 1 · 摘自论文原文
  • 基于SLAYER框架构建,支持单/多故障、永久/瞬时故障注入
  • 可实现训练前、中、后任意阶段的故障测试,覆盖全层次故障位置
  • 开源工具包,支持GPU加速与结果可视化,适合可靠性研究者使用

类脑计算与脉冲神经网络(SNNs)因其在能效和计算速度上的潜力,在各类人工智能任务中日益受到关注。其优势源于对生物大脑结构、功能与效率的模拟,而大脑无疑是迄今最高效且环保的计算系统。当SNN最终部署于硬件处理器时,其在硬件故障下的可靠性成为关键问题,尤其对安全与任务关键型应用而言。本文提出SpikeFI,一个面向SNN的故障注入框架,可自动化完成可靠性分析与测试用例生成。SpikeFI基于SLAYER PyTorch框架,支持在单/多GPU上加速故障实验,内置全面的神经元与突触故障模型库,符合领域文献标准,且可扩展。支持单个或多个故障、永久或瞬时故障、指定、层级随机或网络级随机故障位置,以及训练前、中、后阶段的故障注入。同时提供多种优化加速与结果可视化功能。SpikeFI为开源项目,可通过GitHub获取:https://github.com/SpikeFI。

原文摘要 · Abstract (English)

Neuromorphic computing and spiking neural networks (SNNs) are gaining traction across various artificial intelligence (AI) tasks thanks to their potential for efficient energy usage and faster computation speed. This comparative advantage comes from mimicking the structure, function, and efficiency of the biological brain, which arguably is the most brilliant and green computing machine. As SNNs are eventually deployed on a hardware processor, the reliability of the application in light of hardware-level faults becomes a concern, especially for safety- and mission-critical applications. In this work, we propose SpikeFI, a fault injection framework for SNNs that can be used for automating the reliability analysis and test generation. SpikeFI is built upon the SLAYER PyTorch framework with fault injection experiments accelerated on a single or multiple GPUs. It has a comprehensive integrated neuron and synapse fault model library, in accordance to the literature in the domain, which is extendable by the user if needed. It supports: single and multiple faults; permanent and transient faults; specified, random layer-wise, and random network-wise fault locations; and pre-, during, and post-training fault injection. It also offers several optimization speedups and built-in functions for results visualization. SpikeFI is open-source and available for download via GitHub at https://github.com/SpikeFI.

脉冲神经网络故障注入可靠性测试类脑计算

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。