arXivDaily arXiv每日学术速递 周一至周五更新
看起来今天可以划水了
arXiv 2009.01431cs.LGcs.DC
收藏

基于FPGA的高效可扩展的在线决策树学习加速

Towards Efficient and Scalable Acceleration of Online Decision Tree Learning on FPGA

Hong Kong University of Science and Technology(香港科技大学);Indian Institute of Technology (IIT) Goa(印度理工学院果阿分校)

Zhe Lin, Sharad Sinha, Wei Zhang

摘要:

决策树是各种应用场景中常用的机器学习模型。在大数据时代,传统的决策树归纳算法由于对数据存储的严格要求而不适合学习大规模数据集。在线决策树学习算法已被设计来解决这个问题,同时训练与传入的样本,并提供推理结果。然而,即使是最新的在线树学习算法仍然遭受高内存使用或高计算强度与依赖性和长延迟,使它们具有挑战性的硬件实现。为了克服这些困难,我们引入了一个新的基于分位数的算法,以改善归纳的Hoeffding树,一个国家的最先进的在线学习模型。所提出的算法是轻量级的内存和计算需求,同时仍然保持较高的泛化能力。从硬件的角度研究了一系列针对该算法的优化技术,包括粗粒度和细粒度并行、动态和基于内存的资源共享、带数据转发的流水线等。我们进一步提出了一个高性能,硬件高效和可扩展的在线决策树学习系统的现场可编程门阵列(FPGA)与系统级优化技术。实验结果表明,我们提出的算法优于国家的最先进的Hoeffding树学习方法,导致0.05%至12.3%的提高推理精度。完整学习系统在FPGA上的实际实现表明,与最先进的设计相比,执行时间加快了384倍至1581倍。

英文摘要:

Decision trees are machine learning models commonly used in various application scenarios. In the era of big data, traditional decision tree induction algorithms are not suitable for learning large-scale datasets due to their stringent data storage requirement. Online decision tree learning algorithms have been devised to tackle this problem by concurrently training with incoming samples and providing inference results. However, even the most up-to-date online tree learning algorithms still suffer from either high memory usage or high computational intensity with dependency and long latency, making them challenging to implement in hardware. To overcome these difficulties, we introduce a new quantile-based algorithm to improve the induction of the Hoeffding tree, one of the state-of-the-art online learning models. The proposed algorithm is light-weight in terms of both memory and computational demand, while still maintaining high generalization ability. A series of optimization techniques dedicated to the proposed algorithm have been investigated from the hardware perspective, including coarse-grained and fine-grained parallelism, dynamic and memory-based resource sharing, pipelining with data forwarding. We further present a high-performance, hardware-efficient and scalable online decision tree learning system on a field-programmable gate array (FPGA) with system-level optimization techniques. Experimental results show that our proposed algorithm outperforms the state-of-the-art Hoeffding tree learning method, leading to 0.05% to 12.3% improvement in inference accuracy. Real implementation of the complete learning system on the FPGA demonstrates a 384x to 1581x speedup in execution time over the state-of-the-art design.

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

↑