Update README.md
Browse files
README.md
CHANGED
@@ -5,15 +5,12 @@ tags:
|
|
5 |
---
|
6 |
|
7 |
# SlimSAM: 0.1% Data Makes Segment Anything Slim
|
8 |
-
<div align="center">
|
9 |
-
<img src="images/paper/intro.PNG" width="66%">
|
10 |
-
<img src="images/paper/everything.PNG" width="100%">
|
11 |
-
</div>
|
12 |
|
13 |
> **0.1% Data Makes Segment Anything Slim**
|
14 |
> [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/)
|
15 |
> [Learning and Vision Lab](http://lv-nus.org/), National University of Singapore
|
16 |
> Paper: [[Arxiv]](https://arxiv.org/abs/2312.05284)
|
|
|
17 |
|
18 |
## Introduction
|
19 |
|
@@ -23,38 +20,9 @@ tags:
|
|
23 |
|
24 |
**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.
|
25 |
|
26 |
-
![Frame](images/paper/frame.PNG?raw=true)
|
27 |
-
|
28 |
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.
|
29 |
|
30 |
-
## Visualization Results
|
31 |
-
|
32 |
-
Qualitative comparison of results obtained using point prompts, box prompts, and segment everything prompts are shown in the following section.
|
33 |
-
|
34 |
-
### Segment Everything Prompts
|
35 |
-
<div align="center">
|
36 |
-
<img src="images/paper/everything2.PNG" width="100%">
|
37 |
-
</div>
|
38 |
-
|
39 |
-
### Box Prompts and Point Prompts
|
40 |
-
<div align="center">
|
41 |
-
<img src="images/paper/prompt.PNG" width="100%">
|
42 |
-
</div>
|
43 |
-
|
44 |
-
|
45 |
-
## Quantitative Results
|
46 |
|
47 |
-
We conducted a comprehensive comparison encompassing performance, efficiency, and training costs with other SAM compression methods and structural pruning methods.
|
48 |
-
|
49 |
-
### Comparing with other SAM compression methods.
|
50 |
-
<div align="center">
|
51 |
-
<img src="images/paper/compare_tab1.PNG" width="100%">
|
52 |
-
</div>
|
53 |
-
|
54 |
-
### Comparing with other structural pruning methods.
|
55 |
-
<div align="center">
|
56 |
-
<img src="images/paper/compare_tab2.PNG" width="50%">
|
57 |
-
</div>
|
58 |
|
59 |
|
60 |
|
|
|
5 |
---
|
6 |
|
7 |
# SlimSAM: 0.1% Data Makes Segment Anything Slim
|
|
|
|
|
|
|
|
|
8 |
|
9 |
> **0.1% Data Makes Segment Anything Slim**
|
10 |
> [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/)
|
11 |
> [Learning and Vision Lab](http://lv-nus.org/), National University of Singapore
|
12 |
> Paper: [[Arxiv]](https://arxiv.org/abs/2312.05284)
|
13 |
+
> Code: [[GitHub]](https://github.com/czg1225/SlimSAM)
|
14 |
|
15 |
## Introduction
|
16 |
|
|
|
20 |
|
21 |
**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.
|
22 |
|
|
|
|
|
23 |
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.
|
24 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
25 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
26 |
|
27 |
|
28 |
|