Design-Space Exploration of Distributed Neural Networks in Low-Power Wearable Nodes
Wearable devices increasingly support intelligent applications such as augmented reality, continuous health monitoring, and vision-based interaction. However, the neural networks (NNs) required by these applications often exceed the memory and energy budgets of wearable nodes. Cloud offloading reduces local computation but increases communication demand and can introduce latency and privacy concerns. TinyML reduces model complexity, potentially sacrificing accuracy without resolving the memory demands of vision workloads. This paper presents DistNN, a hardware-software framework for distributed NN inference that partitions execution between an ultra-low-power (ULP) wearable node and a nearby resource-rich hub, such as a smartphone. The node processes early NN layers and sends intermediate feature maps to the hub, which executes the remaining layers. To choose the split point, we propose a platform-independent figure of merit (FoM) based on data volume (DV) and cumulative computation cost (CC). The FoM captures the central trade-off: earlier splits reduce node computation but increase communication, whereas later splits reduce transmitted data but increase node computation. We also derive conditions for a net node-side energy benefit from distribution and demonstrate that the target autoencoder (AE) workloads cannot run monolithically on representative wearable microcontrollers. To enable ULP execution, we synthesize a custom node using low-precision fixed-point arithmetic. Post-synthesis estimates place DistNN below the 10 mW node-power target, with average node-compute power approximately 11x lower than that of the recent ML ASIC baselines evaluated. Across convolutional NNs and AEs, DistNN preserves task quality, achieving a structural similarity index (SSIM) of approximately 0.90 for image reconstruction with low-precision node features and a resource-rich hub decoder.