Table 3.
Performance comparison based on MIMIC-III
| missing rate | |||
|---|---|---|---|
| Model | 0% | 20% | 40% |
| Next-admission diagnosis prediction (Accuracy@20) | |||
| INVASE | 0.571 ± 0.015 | 0.609 ± 0.019 | 0.589 ± 0.017 |
| LSPIN | 0.615± 0.022 | 0.600 ± 0.023 | 0.598 ± 0.025 |
| gI | 0.602 ± 0.014 | 0.600 ± 0.017 | 0.591 ± 0.018 |
| GroupFS | 0.592 ± 0.012 | 0.589 ± 0.012 | 0.586 ± 0.014 |
![]() |
0.610 ± 0.019 | 0.592 ± 0.019 | 0.586 ± 0.018 |
![]() |
0.589 ± 0.011 | 0.576 ± 0.016 | 0.571 ± 0.012 |
![]() |
0.575 ± 0.09 | 0.572 ± 0.011 | 0.561 ± 0.015 |
![]() |
0.646± 0.016 | 0.621 ± 0.015 | 0.614 ± 0.017 |
| Hi-BEHRT | 0.631 ± 0.013 | 0.618± 0.012 | 0.610 ± 0.011 |
| Mortality prediction (AUC) | |||
| INVASE | 0.872 ± 0.021 | 0.839 ± 0.019 | 0.812 ± 0.020 |
| LSPIN | 0.944 ± 0.010 | 0.880 ± 0.015 | 0.877 ± 0.016 |
| gI | 0.917 ± 0.014 | 0.835 ± 0.014 | 0.810 ± 0.017 |
| GroupFS | 0.845 ± 0.011 | 0.837 ± 0.013 | 0.823 ± 0.013 |
![]() |
0.859 ± 0.008 | 0.848 ± 0.010 | 0.836 ± 0.012 |
![]() |
0.896 ± 0.015 | 0.892 ± 0.017 | 0.881 ± 0.014 |
![]() |
0.862 ± 0.017 | 0.855 ± 0.022 | 0.853 ± 0.013 |
![]() |
0.945 ± 0.010 | 0.879 ± 0.012 | 0.878 ± 0.011 |
| Hi-BEHRT | 0.865 ± 0.021 | 0.832 ± 0.025 | 0.824 ± 0.023 |
FlexGPC-Trans-MLP is abbreviated as FlexGPC







