Skip to content

Commit d5ac087

Browse files
Update README.md
1 parent 61c7a29 commit d5ac087

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

README.md

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -8,17 +8,6 @@ In our paper, We introduce a novel spectrally relaxed k -means regularization, t
88

99
Our proposed pipeline, termed Deep k -Means, has well-aligned goals between re-training and compression stages. We further propose an improved set of metrics to estimate energy consumption of CNN hardware implementations, whose estimation results are verified to be consistent with previously proposed energy estimation tool extrapolated from actual hardware measurements. We have evaluated Deep k -Means in compressing several CNN models in terms of both compression ratio and energy consumption reduction, observing promising results without incurring accuracy loss.
1010

11-
### Citation
12-
13-
If you find this code useful, please cite the following paper:
14-
15-
@article{deepkmeans,
16-
title={Deep k-Means: Re-Training and Parameter Sharing with Harder Cluster Assignments for Compressing Deep Convolutions},
17-
author={Junru Wu, Yue Wang, Zhenyu Wu, Zhangyang Wang, Ashok Veeraraghavan, Yingyan Lin},
18-
journal={ICML},
19-
year={2018}
20-
}
21-
2211
### To do items
2312

2413
#### PyTorch Model
@@ -48,3 +37,14 @@ If you find this code useful, please cite the following paper:
4837

4938

5039
### Testing Deep k-Means
40+
41+
### Citation
42+
43+
If you find this code useful, please cite the following paper:
44+
45+
@article{deepkmeans,
46+
title={Deep k-Means: Re-Training and Parameter Sharing with Harder Cluster Assignments for Compressing Deep Convolutions},
47+
author={Junru Wu, Yue Wang, Zhenyu Wu, Zhangyang Wang, Ashok Veeraraghavan, Yingyan Lin},
48+
journal={ICML},
49+
year={2018}
50+
}

0 commit comments

Comments
 (0)