Mar, 2024
SAM-Lightening: 基于扩张闪烁注意力的轻量级段落任意模型,实现30倍加速
SAM-Lightening: A Lightweight Segment Anything Model with Dilated Flash
Attention to Achieve 30 times Acceleration
TL;DR通过引入SAM-Lightening,一种改进的注意力机制,称为Dilated Flash Attention,本研究在COCO和LVIS数据集上通过渐进蒸馏实现了从原始SAM的知识转移,实现了在运行时间效率和分割准确性方面显著优于最先进方法,每幅图片仅需要7毫秒的推理速度,是原始SAM的30.1倍,且内存占用只有原始SAM的3.5%。