Skip to main content
. 2021 Dec 8;5(12):e20767. doi: 10.2196/20767

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.