SqueezeSAM: User friendly mobile interactive segmentation

Balakrishnan Varadarajan,Bilge Soran,Forrest Iandola,Xiaoyu Xiang,Yunyang Xiong,Lemeng Wu,Chenchen Zhu,Raghuraman Krishnamoorthi,Vikas Chandra
2024-05-21
Abstract:The Segment Anything Model (SAM) has been a cornerstone in the field of interactive segmentation, propelling significant progress in generative AI, computational photography, and medical imaging. Despite its ability to process arbitrary user input and generate corresponding segmentation masks, SAM's 600 million parameter architecture, based on ViT-H, is not compatible with current mobile hardware due to its high computational demands and large model size. Our research aims to adapt SAM for use in mobile photography applications. To this end, we have developed a fully convolutional SqueezeSAM model architecture, which is 62.5 times faster and 31.6 times smaller than the original SAM, making it a viable solution for mobile applications. Furthermore, our tiny model achieves an mIOU within 1% of the original VIT-H architecture.
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?