Update README.md
Browse files
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://
|
| 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 |

|
| 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 |
-

|
| 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 |

|
| 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 |
|