arXiv:2603.24186cs.LGcs.AR2026-03中稿 · the 63rd Design Au…

用可编程芯片实现快速自适应的无权重神经网络训练

TsetlinWiSARD: On-Chip Training of Weightless Neural Networks using Tsetlin Automata on FPGAs

  • 基于强化学习的自动机机制实现迭代式在线训练
  • 训练速度提升千倍,资源占用减少22%
  • 适合边缘设备上需低延迟、低功耗的实时学习场景

边缘计算对自适应性、隐私与安全的需求持续推动新型机器学习算法的发展,要求模型具备在芯片上完成训练与推理的能力。权值无关神经网络(WNN)通过查表式简单神经元结构,相较依赖乘加运算的深度神经网络,具有低延迟、低复杂度的优势。然而传统WNN采用记忆式单次训练,易导致过拟合或需繁琐后期调整,限制了其在芯片上的高效训练能力。本文提出TsetlinWiSARD,利用泰斯林自动机(TAs)实现概率性、反馈驱动的学习机制,通过迭代优化克服了单次训练的过拟合问题,同时保持连续二进制反馈以保障硬件效率。核心是基于现场可编程门阵列(FPGA)的训练架构,在实现顶尖准确率的同时显著提升硬件效率:相比传统WNN实现,训练速度提升超过1000倍;相较于其他基于FPGA的机器学习加速器,资源使用减少22%,延迟降低93.3%,功耗降低64.2%。

原文摘要 · Abstract (English)

Increasing demands for adaptability, privacy, and security at the edge have persistently pushed the frontiers for a new generation of machine learning (ML) algorithms with training and inference capabilities on-chip. Weightless Neural Network (WNN) is such an algorithm that is principled on lookup table based simple neuron structures. As a result, it offers architectural benefits, such as low-latency, low-complexity inference, compared to deep neural networks that depend heavily on multiply-accumulate operations. However, traditional WNNs rely on memorization-based one-shot training, which either leads to overfitting and reduced accuracy or requires tedious post-training adjustments, limiting their effectiveness for efficient on chip training. In this work, we propose TsetlinWiSARD, a training approach for WNNs that leverages Tsetlin Automata (TAs) to enable probabilistic, feedback-driven learning. It overcomes the overfitting of WiSARD's one-shot training with iterative optimization, while maintaining simple, continuous binary feedback for efficient on-chip training. Central to our approach is a field programmable gate array (FPGA)-based training architecture that delivers state-of-the-art accuracy while significantly improving hardware efficiency. Our approach provides over 1000x faster training when compared with the traditional WiSARD implementation of WNNs. Further, we demonstrate 22% reduced resource usage, 93.3% lower latency, and 64.2% lower power consumption compared to FPGA-based training accelerators implementing other ML algorithms.

边缘计算无权重网络FPGA加速在线学习

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