Update README.md
Browse files
README.md
CHANGED
|
@@ -6,15 +6,13 @@ tags:
|
|
| 6 |
---
|
| 7 |
|
| 8 |
# SlimSAM: 0.1% Data Makes Segment Anything Slim
|
| 9 |
-
|
| 10 |
-
<img src="images/paper/intro.PNG" width="66%">
|
| 11 |
-
<img src="images/paper/everything.PNG" width="100%">
|
| 12 |
-
</div>
|
| 13 |
|
| 14 |
> **0.1% Data Makes Segment Anything Slim**
|
| 15 |
> [Zigeng Chen](https://github.com/czg1225), [Gongfan Fang](https://fangggf.github.io/), [Xinyin Ma](https://horseee.github.io/), [Xinchao Wang](https://sites.google.com/site/sitexinchaowang/)
|
| 16 |
> [Learning and Vision Lab](http://lv-nus.org/), National University of Singapore
|
| 17 |
> Paper: [[Arxiv]](https://arxiv.org/abs/2312.05284)
|
|
|
|
| 18 |
|
| 19 |
## Introduction
|
| 20 |
|
|
@@ -24,38 +22,9 @@ tags:
|
|
| 24 |
|
| 25 |
**SlimSAM** is a novel SAM compression method, which efficiently reuses pre-trained SAMs without the necessity for extensive retraining. This is achieved by the efficient reuse of pre-trained SAMs through a unified pruning-distillation framework. To enhance knowledge inheritance from the original SAM, we employ an innovative alternate slimming strategy that partitions the compression process into a progressive procedure. Diverging from prior pruning techniques, we meticulously prune and distill decoupled model structures in an alternating fashion. Furthermore, a novel label-free pruning criterion is also proposed to align the pruning objective with the optimization target, thereby boosting the post-distillation after pruning.
|
| 26 |
|
| 27 |
-

|
| 28 |
-
|
| 29 |
SlimSAM achieves approaching performance while reducing the parameter counts to **0.9\% (5.7M)**, MACs to **0.8\% (21G)**, and requiring mere **0.1\% (10k)** of the training data when compared to the original SAM-H. Extensive experiments demonstrate that our method realize significant superior performance while utilizing over **10 times** less training data when compared to other SAM compression methods.
|
| 30 |
|
| 31 |
-
## Visualization Results
|
| 32 |
-
|
| 33 |
-
Qualitative comparison of results obtained using point prompts, box prompts, and segment everything prompts are shown in the following section.
|
| 34 |
|
| 35 |
-
### Segment Everything Prompts
|
| 36 |
-
<div align="center">
|
| 37 |
-
<img src="images/paper/everything2.PNG" width="100%">
|
| 38 |
-
</div>
|
| 39 |
-
|
| 40 |
-
### Box Prompts and Point Prompts
|
| 41 |
-
<div align="center">
|
| 42 |
-
<img src="images/paper/prompt.PNG" width="100%">
|
| 43 |
-
</div>
|
| 44 |
-
|
| 45 |
-
|
| 46 |
-
## Quantitative Results
|
| 47 |
-
|
| 48 |
-
We conducted a comprehensive comparison encompassing performance, efficiency, and training costs with other SAM compression methods and structural pruning methods.
|
| 49 |
-
|
| 50 |
-
### Comparing with other SAM compression methods.
|
| 51 |
-
<div align="center">
|
| 52 |
-
<img src="images/paper/compare_tab1.PNG" width="100%">
|
| 53 |
-
</div>
|
| 54 |
-
|
| 55 |
-
### Comparing with other structural pruning methods.
|
| 56 |
-
<div align="center">
|
| 57 |
-
<img src="images/paper/compare_tab2.PNG" width="50%">
|
| 58 |
-
</div>
|
| 59 |
|
| 60 |
|
| 61 |
|
|
|
|
| 6 |
---
|
| 7 |
|
| 8 |
# SlimSAM: 0.1% Data Makes Segment Anything Slim
|
| 9 |
+
|
|
|
|
|
|
|
|
|
|
| 10 |
|
| 11 |
> **0.1% Data Makes Segment Anything Slim**
|
| 12 |
> [Zigeng Chen](https://github.com/czg1225), [Gongfan Fang](https://fangggf.github.io/), [Xinyin Ma](https://horseee.github.io/), [Xinchao Wang](https://sites.google.com/site/sitexinchaowang/)
|
| 13 |
> [Learning and Vision Lab](http://lv-nus.org/), National University of Singapore
|
| 14 |
> Paper: [[Arxiv]](https://arxiv.org/abs/2312.05284)
|
| 15 |
+
> Code: [[SlimSAM]](https://github.com/czg1225/SlimSAM)
|
| 16 |
|
| 17 |
## Introduction
|
| 18 |
|
|
|
|
| 22 |
|
| 23 |
**SlimSAM** is a novel SAM compression method, which efficiently reuses pre-trained SAMs without the necessity for extensive retraining. This is achieved by the efficient reuse of pre-trained SAMs through a unified pruning-distillation framework. To enhance knowledge inheritance from the original SAM, we employ an innovative alternate slimming strategy that partitions the compression process into a progressive procedure. Diverging from prior pruning techniques, we meticulously prune and distill decoupled model structures in an alternating fashion. Furthermore, a novel label-free pruning criterion is also proposed to align the pruning objective with the optimization target, thereby boosting the post-distillation after pruning.
|
| 24 |
|
|
|
|
|
|
|
| 25 |
SlimSAM achieves approaching performance while reducing the parameter counts to **0.9\% (5.7M)**, MACs to **0.8\% (21G)**, and requiring mere **0.1\% (10k)** of the training data when compared to the original SAM-H. Extensive experiments demonstrate that our method realize significant superior performance while utilizing over **10 times** less training data when compared to other SAM compression methods.
|
| 26 |
|
|
|
|
|
|
|
|
|
|
| 27 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 28 |
|
| 29 |
|
| 30 |
|