|
1 | 1 | MP-CNN Lite Results |
2 | 2 | =================== |
3 | 3 |
|
| 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 | + |
4 | 17 | ## TrecQA (clean) |
5 | 18 |
|
6 | 19 | ``` |
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 |
8 | 21 | ``` |
9 | 22 |
|
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 | |
14 | 27 |
|
15 | 28 | ## WikiQA |
16 | 29 |
|
17 | 30 | ``` |
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 |
19 | 32 | ``` |
20 | 33 |
|
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