Skip to main content
. 2024 Mar 18;14:6425. doi: 10.1038/s41598-024-56983-6

Table 12.

Performance comparison of models on original and augmented data (CNN, A-CNN, LSTM, A-LSTM, GRU, A-GRU).

Model Data set Parameters Complexity Assessment Metrics
Optimizer LR SC TC Acc (%) Sp (%) Sn (%) Pre (%) F1-S (%) AUC (%)
CNN Original SGD 0.0001 28.03M 5.05h 92.33 89.02 90.08 96.34 98.30 96.05
ADAM 0.0001 28.01M 5.03h 92.84 93.30 95.34 97.00 97.89 97.93
CNN Augmented 27.06M 4.07h 93.87 98.03 95.90 98.64 97.00 98.34
- 27.07M 4.09h 93.98 99.10 97.34 98.89 98.33 98.90
A-CNN Augmented 28.06M 5.02h 94.00 98.03 95.90 98.64 97.00 98.64
28.08M 4.04h 95.78 98.77 98.94 99.09 99.34 99.00
LSTM Original 39.02M 4.01h 96.30 90.50 92.28 98.98 98.00 98.84
39.05M 9.08h 97.54 93.00 95.49 99.00 98.08 99.00
LSTM Augmented - 38.03M 5.08h 97.44 94.60 96.88 97.58 98.90 90.04
39.02M 5.03h 97.54 95.77 97.01 98.66 98.98 99.70
A-LSTM Augmented 37.01M 5.03h 98.00 97.82 97.18 87.00 99.30 99.04
- 41.01M 5.04h 98.64 97.47 98.91 98.96 98.98 98.90
GRU Original 110.01M 5.04h 97.90 98.09 99.00 98.34 99.00 98.97
110.05M 5.04h 98.02 99.00 99.17 99.03 99.69 98.99
GRU Augmented 113.05M 6.05h 97.92 96.39 94.09 94.87 98.58 98.06
- 115.08M 6.04h 98.40 90.98 97.56 98.78 99.01 98.79
A-GRU Original 121.07M 6.06h 98.44 99.59 97.60 99.03 98.30 97.95
- 122.06M 7.09h 98.79 99.80 98.87 99.66 99.65 98.99
A-GRU Augmented 122.03M 7.03h 98.97 99.35 98.89 99.99 99.96 99.66
122.07M 7.08h 99.32 99.78 99.12 100 99.01 99.89

The number of trainable parameters in a model represents its space complexity. M stands for million. The more trainable parameters there are, the higher the space complexity (SC). The models’ training time (measured in hours) is known as the time complexity (TC).

Significant values are in [bold].