Table 2.
Recurrent neural network condensation.
| Model | Parameters, n | File size (kb) | Inference (seconds per sample) | Training time (seconds; 20 epochs) | Test AUROCa (last epoch) |
| Baseline LSTMb |
8081 |
129 |
523 |
4890 |
0.836 |
| Pruned LSTM | 3273 | 73 | 318 | 4990 | 0.853 |
| Hidden-layer LSTM | 6993 | 111 | 254 | 3000 | 0.860 |
aAUROC: area under the receiver operating curve.
bLSTM: long short-term memory.