jiekeshi commited on
Commit
6ac37a1
·
1 Parent(s): 5068218

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +24 -0
README.md CHANGED
@@ -1,3 +1,27 @@
1
  ---
2
  license: mit
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
  ---
4
+
5
+ This is the 3 MB compressed version of GraphCodeBERT that has been fine-tuned for the Vulnerability Prediction task using [Devign](https://sites.google.com/view/devign) dataset.
6
+
7
+ The compression is based on our ASE 2022 paper named ["**Compressing Pre-trained Models of Code into 3 MB**"](https://arxiv.org/abs/2208.07120).
8
+
9
+ If you are interested in using this model, please check our **GitHub repository: https://github.com/soarsmu/Compressor.git**. If you use the model or any code from our repo in your paper, please kindly cite:
10
+ ```
11
+ @inproceedings{shi2022compressing,
12
+ author = {Shi, Jieke and Yang, Zhou and Xu, Bowen and Kang, Hong Jin and Lo, David},
13
+ title = {Compressing Pre-Trained Models of Code into 3 MB},
14
+ year = {2023},
15
+ isbn = {9781450394758},
16
+ publisher = {Association for Computing Machinery},
17
+ address = {New York, NY, USA},
18
+ url = {https://doi.org/10.1145/3551349.3556964},
19
+ doi = {10.1145/3551349.3556964},
20
+ booktitle = {Proceedings of the 37th IEEE/ACM International Conference on Automated Software Engineering},
21
+ articleno = {24},
22
+ numpages = {12},
23
+ keywords = {Pre-Trained Models, Model Compression, Genetic Algorithm},
24
+ location = {Rochester, MI, USA},
25
+ series = {ASE '22}
26
+ }
27
+ ```