← Search

Wei-Ming Chen

4 accepted papers

2022

Lite Pose: Efficient Architecture Design for 2D Human Pose Estimation

CVPR 2022poster

Pose estimation plays a critical role in human-centered vision applications. However, it is difficult to deploy state-of-the-art HRNet-based pose estimation models on resource-constrained edge devices due to the high computational cost (more than 150 GMACs per frame). In this paper, we study efficie…

Cited by 139PDFcodeScholar
2022

On-Device Training Under 256KB Memory

NeurIPS 2022accept

On-device training enables the model to adapt to new data collected from the sensors by fine-tuning a pre-trained model. Users can benefit from customized AI models without having to transfer the data to the cloud, protecting the privacy. However, the training memory consumption is prohibitive for I…

Cited by 258SourcePDFScholar
2021

Memory-efficient Patch-based Inference for Tiny Deep Learning

NeurIPS 2021poster

Tiny deep learning on microcontroller units (MCUs) is challenging due to the limited memory size. We find that the memory bottleneck is due to the imbalanced memory distribution in convolutional neural network (CNN) designs: the first several blocks have an order of magnitude larger memory usage th…

Cited by 209SourcePDFScholar
2020

MCUNet: Tiny Deep Learning on IoT Devices

NeurIPS 2020spotlight

Machine learning on tiny IoT devices based on microcontroller units (MCU) is appealing but challenging: the memory of microcontrollers is 2-3 orders of magnitude smaller even than mobile phones. We propose MCUNet, a framework that jointly designs the efficient neural architecture (TinyNAS) and the l…

Cited by 670SourcePDFScholar