HEEPstor: an Open-Hardware Co-design Framework for Quantized Machine Learning at the Edge

Palacios Almendros, PedroMedina Morillas, RafaelAnsaloni, GiovanniAtienza Alonso, David2025-04-042025-04-042025-04-04202510.1145/3706594.3726967https://infoscience.epfl.ch/handle/20.500.14299/248681Edge-AI applications necessitate the joint application of hardware acceleration and software optimization to meet energy and area constraints. However, the co-design of these systems is hindered by the lack of integration options for novel hardware prototypes offered by commonly employed Machine Learning frameworks. Bridging this gap, we present HEEPstor, an open-hardware co-design framework that enables the deployment of quantized, PyTorch-defined models on heterogeneous RISC-V systems interfacing custom accelerators. We demonstrate its application by executing quantized ML models for image classification on a X-HEEP platform integrating tailored systolic arrays, showcasing its flexibility for hardware-software explorations.enMachine learning frameworkHardware-software co-designOpen HardwareRISC-VEdge AIHEEPstor: an Open-Hardware Co-design Framework for Quantized Machine Learning at the Edgetext::conference output::conference proceedings::conference paper