Skip to content

Commit 7ee348d

Browse files
committed
Update lightweight model results
1 parent aa6807c commit 7ee348d

1 file changed

Lines changed: 23 additions & 10 deletions

File tree

light-weight.md

Lines changed: 23 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,24 +1,37 @@
11
MP-CNN Lite Results
22
===================
33

4+
Note, the lite model in this doc just means MP-CNN without per-dimensional convolution and multiple types of pooling. It uses >8x fewer parameters as the full model!
5+
6+
## SICK
7+
8+
```
9+
python main.py sick.mpcnn.lite.model --arch mpcnn_no_per_dim_no_multi_pooling --dataset sick --epochs 19 --lr 0.00086 --regularization 0.0002672 --wide-conv --attention basic --dropout 0.5
10+
```
11+
12+
| Split | Pearson's r | Spearman's p |
13+
| ------------------ |:---------------:| ---------------:|
14+
| Full model (paper) | 0.8686 | 0.8047 |
15+
| Lite model | 0.8805 | 0.8227 |
16+
417
## TrecQA (clean)
518

619
```
7-
python main.py trecqa.mpcnn.lite.model --arch mpcnn_no_per_dim_no_multi_pooling --dataset trecqa --epochs 5 --lr 0.0002 --regularization 0.000126 --wide-conv --attention basic --dropout 0.3 --sparse-features
20+
python main.py trecqa.mpcnn.lite.model --arch mpcnn_no_per_dim_no_multi_pooling --dataset trecqa --epochs 5 --lr 0.00037 --regularization 0.0017304 --dropout 0
821
```
922

10-
| Split | MAP | MRR |
11-
| ----------- |:-------:| ------:|
12-
| Dev | 0.8028 | 0.8750 |
13-
| Test | 0.7903 | 0.8449 |
23+
| Split | MAP | MRR |
24+
| ------------------ |:----------:| --------:|
25+
| Full model (paper) | 0.762 | 0.854 |
26+
| Lite model | 0.795 | 0.889 |
1427

1528
## WikiQA
1629

1730
```
18-
python main.py wikiqa.mpcnn.lite.model --arch mpcnn_no_per_dim_no_multi_pooling --dataset wikiqa --epochs 5 --lr 0.00038 --regularization 0.0003396 --wide-conv --attention basic --dropout 0.5 --sparse-features
31+
python main.py wikiqa.mpcnn.lite.model --arch mpcnn_no_per_dim_no_multi_pooling --dataset wikiqa --epochs 5 --lr 0.00025 --regularization 0.0001088 --wide-conv --attention basic --sparse-features
1932
```
2033

21-
| Split | MAP | MRR |
22-
| ----------- |:-------:| ------:|
23-
| Dev | 0.7552 | 0.7639 |
24-
| Test | 0.7106 | 0.7270 |
34+
| Split | MAP | MRR |
35+
| ------------------ |:----------:| --------:|
36+
| Full model (paper) | 0.693 | 0.709 |
37+
| Lite model | 0.696 | 0.708 |

0 commit comments

Comments
 (0)