heroding77 commited on
Commit
3bb9ab2
·
verified ·
1 Parent(s): b85b151

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +70 -3
README.md CHANGED
@@ -1,3 +1,70 @@
1
- ---
2
- license: apache-2.0
3
- ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ tags:
4
+ - Automated Peer Reviewing
5
+ - SFT
6
+ ---
7
+
8
+ ## Automated Peer Reviewing in Paper SEA: Standardization, Evaluation, and Analysis
9
+
10
+ Paper Link: https://arxiv.org/abs/2311.09278
11
+
12
+ Project Page: https://ecnu-sea.github.io/
13
+
14
+
15
+ ## 🔥 News
16
+
17
+ - 🔥🔥🔥 We have made SEA series models (7B) public !
18
+
19
+ ## Model Description
20
+ The SEA-E model utilizes [Mistral-7B-Instruct-v0.2](https://huggingface.co/mistralai/Mistral-7B-Instruct-v0.2) as its backbone. It is derived by performing supervised fine-tuning (SFT) on a high-quality peer review instruction dataset, standardized through the SEA-S model. This model can provide comprehensive and insightful review feedback for submitted papers.
21
+
22
+ ## Review Paper With SEA-E
23
+
24
+ ```python
25
+ from transformers import AutoModelForCausalLM, AutoTokenizer
26
+
27
+ instruction = system_prompt_dict['instruction_e']
28
+ paper = read_txt_file(mmd_file_path)
29
+ idx = paper.find("## References")
30
+ paper = paper[:idx].strip()
31
+
32
+ messages = [
33
+ {"role": "system", "content": instruction},
34
+ {"role": "user", "content": paper},
35
+ ]
36
+
37
+ encodes = tokenizer.apply_chat_template(messages, return_tensors="pt")
38
+ encodes = encodes.to("cuda:0")
39
+ len_input = encodes.shape[1]
40
+ generated_ids = chat_model.generate(encodes,max_new_tokens=8192,do_sample=True)
41
+ # response = chat_model.chat(messages)[0].response_text
42
+ response = tokenizer.batch_decode(generated_ids[: , len_input:])[0]
43
+
44
+ ```
45
+ The code provided above is an example. For detailed usage instructions, please refer to https://github.com/ecnu-sea/sea.
46
+
47
+ ## Additional Clauses
48
+
49
+ The additional clauses for this project are as follows:
50
+
51
+ - The SEA-E model is intended solely to provide informative reviews for authors to polish their papers instead of directly recommending acceptance/rejection on papers.
52
+ - Currently, the SEA-E model is only applicable within the field of machine learning and does not guarantee insightful comments for other disciplines.
53
+
54
+
55
+ ## Citation
56
+
57
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
58
+
59
+ If you find our paper or models helpful, please consider cite as follows:
60
+
61
+ ```bibtex
62
+ @misc{yu2024sea,
63
+ title={Automated Peer Reviewing in Paper SEA: Standardization, Evaluation, and Analysis},
64
+ author={Jianxiang Yu and Zichen Ding and Jiaqi Tan and Kangyang Luo and Zhenmin Weng and Chegnhua Gong and Long Zeng and Renjing Cui and Chengcheng Han and Qiushi Sun and Zhiyong Wu and Yunshi Lan and Xiang Li},
65
+ year={2024},
66
+ eprint={2406.26456},
67
+ archivePrefix={arXiv},
68
+ primaryClass={cs.AI}
69
+ }
70
+ ```