Zigeng commited on
Commit
79c09c1
1 Parent(s): 4251774

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -33
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