beyond commited on
Commit
bcd8b31
·
1 Parent(s): b8356e1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -20
README.md CHANGED
@@ -36,21 +36,10 @@ inference:
36
 
37
  **基于草稿的文本生成模型**
38
 
39
- - **Paper: [GENIUS: Sketch-based Language Model Pre-training via Extreme and Selective Masking for Text Generation and Augmentation](https://github.com/beyondguo/genius/blob/master/GENIUS_gby_arxiv.pdf)**
40
-
41
- 💡**GENIUS** is a powerful conditional text generation model using sketches as input, which can fill in the missing contexts for a given **sketch** (key information consisting of textual spans, phrases, or words, concatenated by mask tokens). GENIUS is pre-trained on a large- scale textual corpus with a novel *reconstruction from sketch* objective using an *extreme and selective masking* strategy, enabling it to generate diverse and high-quality texts given sketches.
42
-
43
- **Example 1:**
44
-
45
- - sketch: `__ machine learning __ my research interest __ data science __`
46
- - **GENIUS**: `I am a Ph.D. student in machine learning, and my research interest is in data science. I am interested in understanding how humans and machines interact and how we can improve the quality of life for people around the world.`
47
-
48
- **Example 2:**
49
-
50
- - sketch: `自然语言处理__谷歌__通用人工智能__`
51
- - **GENIUS**: `自然语言处理是谷歌在通用人工智能领域的一个重要研究方向,其目的是为了促进人类智能的发展。 `
52
-
53
 
 
54
 
55
  **GENIUS** can also be used as a general textual **data augmentation tool** for **various NLP tasks** (including sentiment analysis, topic classification, NER, and QA).
56
 
@@ -58,9 +47,6 @@ inference:
58
  ![image-20221119164544165](https://cdn.jsdelivr.net/gh/beyondguo/mdnice_pictures/typora/hi-genius.png)
59
 
60
 
61
-
62
- - Models hosted in 🤗 Huggingface:
63
-
64
  **Model variations:**
65
 
66
  | Model | #params | Language | comment|
@@ -71,9 +57,6 @@ inference:
71
  | [`genius-base-ps`](https://huggingface.co/beyond/genius-base) | 139M | English | pre-trained both in paragraphs and short sentences |
72
  | [`genius-base-chinese`](https://huggingface.co/beyond/genius-base-chinese) | 116M | 中文 | 在一千万纯净中文段落上预训练|
73
 
74
- ![image-20221119191940969](https://cdn.jsdelivr.net/gh/beyondguo/mdnice_pictures/typora/202211191919005.png)
75
-
76
-
77
 
78
  ## Usage
79
 
 
36
 
37
  **基于草稿的文本生成模型**
38
 
39
+ - **Paper: [GENIUS: Sketch-based Language Model Pre-training via Extreme and Selective Masking for Text Generation and Augmentation](https://arxiv.org/abs/2211.10330)**
40
+ - **GitHub: [GENIUS, Pre-training/Data Augmentation Tutorial](https://github.com/beyondguo/genius)**
 
 
 
 
 
 
 
 
 
 
 
 
41
 
42
+ 💡**GENIUS** is a powerful conditional text generation model using sketches as input, which can fill in the missing contexts for a given **sketch** (key information consisting of textual spans, phrases, or words, concatenated by mask tokens). GENIUS is pre-trained on a large-scale textual corpus with a novel *reconstruction from sketch* objective using an *extreme and selective masking* strategy, enabling it to generate diverse and high-quality texts given sketches.
43
 
44
  **GENIUS** can also be used as a general textual **data augmentation tool** for **various NLP tasks** (including sentiment analysis, topic classification, NER, and QA).
45
 
 
47
  ![image-20221119164544165](https://cdn.jsdelivr.net/gh/beyondguo/mdnice_pictures/typora/hi-genius.png)
48
 
49
 
 
 
 
50
  **Model variations:**
51
 
52
  | Model | #params | Language | comment|
 
57
  | [`genius-base-ps`](https://huggingface.co/beyond/genius-base) | 139M | English | pre-trained both in paragraphs and short sentences |
58
  | [`genius-base-chinese`](https://huggingface.co/beyond/genius-base-chinese) | 116M | 中文 | 在一千万纯净中文段落上预训练|
59
 
 
 
 
60
 
61
  ## Usage
62