CFU Playground: Full-Stack Open-Source Framework for Tiny Machine Learning (TinyML) Acceleration on FPGAs

• Published in: 2023 IEEE International Symposium on Performance Analysis of Systems and Software (ISPASS) pp. 157 - 167
• Main Authors: Prakash, Shvetank, Callahan, Tim, Bushagour, Joseph, Banbury, Colby, Green, Alan V., Warden, Pete, Ansell, Tim, Reddi, Vijay Janapa
• Format: Conference Proceeding
• Language: English
• Published: IEEE 01.04.2023
• Subjects: FPGAs; Hardware Acceleration; Open Source; TinyML
• DOI: 10.1109/ISPASS57527.2023.00024

Need for the efficient processing of neural networks has given rise to the development of hardware accelerators. The increased adoption of specialized hardware has highlighted the need for more agile design flows for hardware-software co-design and domain-specific optimizations. In this paper, we present CFU Playground- a full-stack open-source framework that enables rapid and iterative design and evaluation of machine learning (ML) accelerators for embedded ML systems. Our tool provides a completely open-source end-to-end flow for hardwaresoftware co-design on FPGAs and future systems research. This full-stack framework gives the users access to explore experimental and bespoke architectures that are customized and co-optimized for embedded ML. Our rapid, deploy-profileoptimization feedback loop lets ML hardware and software developers achieve significant returns out of a relatively small investment in customization. Using CFU Playground's design and evaluation loop, we show substantial speedups between 55 \times and 75 \times. The soft CPU coupled with the accelerator opens up a new, rich design space between the two components that we explore in an automated fashion using Vizier, an open-source black-box optimization service.