README.md 1.62 KB
Newer Older
xuehui's avatar
xuehui committed
1
# Automaticlly Model Architecture Search for Reading Comprehension
xuehui's avatar
xuehui committed
2
3
This example shows us how to use Genetic Algorithm to find good model architectures for Reading Comprehension task.

xuehui's avatar
xuehui committed
4
## Search Space
xuehui's avatar
xuehui committed
5
6
7
8
9
10
11
12
13
14
15
Since attention and recurrent neural network (RNN) module have been proven effective in Reading Comprehension.
We conclude the search space as follow:

1. IDENTITY (Effectively means keep training).
2. INSERT-RNN-LAYER (Inserts a LSTM. Comparing the performance of GRU and LSTM in our experiment, we decided to use LSTM here.)
3. REMOVE-RNN-LAYER
4. INSERT-ATTENTION-LAYER(Inserts a attention layer.)
5. REMOVE-ATTENTION-LAYER
6. ADD-SKIP (Identity between random layers).
7. REMOVE-SKIP (Removes random skip).

xuehui's avatar
xuehui committed
16
![ga-squad-logo](./ga_squad.png =100x100)
xuehui's avatar
xuehui committed
17

xuehui's avatar
xuehui committed
18
## New version
xuehui's avatar
xuehui committed
19
20
Also we have another version which time cost is less and performance is better. We will release soon.

xuehui's avatar
xuehui committed
21
# How to run this example?
xuehui's avatar
xuehui committed
22

xuehui's avatar
xuehui committed
23
## Use downloading script to download data
xuehui's avatar
xuehui committed
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48

Execute the following command to download needed files
using the downloading script:

```
chmod +x ./download.sh
./download.sh
```

## Download manually

1. download "dev-v1.1.json" and "train-v1.1.json" in https://rajpurkar.github.io/SQuAD-explorer/

```
wget https://rajpurkar.github.io/SQuAD-explorer/dataset/train-v1.1.json
wget https://rajpurkar.github.io/SQuAD-explorer/dataset/dev-v1.1.json
```

2. download "glove.840B.300d.txt" in https://nlp.stanford.edu/projects/glove/

```
wget http://nlp.stanford.edu/data/glove.840B.300d.zip
unzip glove.840B.300d.zip
```

xuehui's avatar
xuehui committed
49
# submit this job
xuehui's avatar
xuehui committed
50
51
52
```
nnictl create --config ~/nni/examples/trials/ga_squad/config.yaml
```