TinySeg: Model Optimizing Framework for Image Segmentation on Tiny Embedded Systems

Byungchul Chae,Jiae Kim,Seonyeong Heo
DOI: https://doi.org/10.1145/3652032.3657576
2024-05-03
Abstract:Image segmentation is one of the major computer vision tasks, which is applicable in a variety of domains, such as autonomous navigation of an unmanned aerial vehicle. However, image segmentation cannot easily materialize on tiny embedded systems because image segmentation models generally have high peak memory usage due to their architectural characteristics. This work finds that image segmentation models unnecessarily require large memory space with an existing tiny machine learning framework. That is, the existing framework cannot effectively manage the memory space for the image segmentation models.
Neural and Evolutionary Computing,Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
### Problems Addressed by the Paper The paper primarily addresses the issue of high memory consumption when implementing image segmentation on small embedded systems. Specifically: 1. **Existing Problems**: - Image segmentation models typically have high peak memory usage because the architectural characteristics of these models require intermediate results to be stored in memory for extended periods. - The memory of small embedded systems is very limited (less than 1 megabyte), and existing micro machine learning frameworks (such as TensorFlow Lite for Microcontrollers) cannot effectively manage the memory space of these models. 2. **Proposed Methods**: - TinySeg, a new model optimization framework, is proposed to optimize memory usage through the following two methods: - **Tensor Spilling**: Spilling tensors that are not used for a long time to local or remote storage. - **Fused Fetching**: Fusing spilled tensors with input tensors to reduce the memory footprint of intermediate tensors. 3. **Goals**: - By optimizing memory usage, enable image segmentation models to run on memory-constrained small embedded devices. 4. **Experimental Results**: - Experiments show that TinySeg can reduce the peak memory usage of image segmentation models by up to 39.3%, thereby enabling intelligent image segmentation tasks on low-power embedded systems.