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

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chae, Byungchul, Kim, Jiae, Heo, Seonyeong
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866911864721506304
author Chae, Byungchul
Kim, Jiae
Heo, Seonyeong
author_facet Chae, Byungchul
Kim, Jiae
Heo, Seonyeong
contents 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. This work proposes TinySeg, a new model optimizing framework that enables memory-efficient image segmentation for tiny embedded systems. TinySeg analyzes the lifetimes of tensors in the target model and identifies long-living tensors. Then, TinySeg optimizes the memory usage of the target model mainly with two methods: (i) tensor spilling into local or remote storage and (ii) fused fetching of spilled tensors. This work implements TinySeg on top of the existing tiny machine learning framework and demonstrates that TinySeg can reduce the peak memory usage of an image segmentation model by 39.3% for tiny embedded systems.
format Preprint
id arxiv_https___arxiv_org_abs_2405_01857
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle TinySeg: Model Optimizing Framework for Image Segmentation on Tiny Embedded Systems
Chae, Byungchul
Kim, Jiae
Heo, Seonyeong
Neural and Evolutionary Computing
Computer Vision and Pattern Recognition
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. This work proposes TinySeg, a new model optimizing framework that enables memory-efficient image segmentation for tiny embedded systems. TinySeg analyzes the lifetimes of tensors in the target model and identifies long-living tensors. Then, TinySeg optimizes the memory usage of the target model mainly with two methods: (i) tensor spilling into local or remote storage and (ii) fused fetching of spilled tensors. This work implements TinySeg on top of the existing tiny machine learning framework and demonstrates that TinySeg can reduce the peak memory usage of an image segmentation model by 39.3% for tiny embedded systems.
title TinySeg: Model Optimizing Framework for Image Segmentation on Tiny Embedded Systems
topic Neural and Evolutionary Computing
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2405.01857