From: Incorporating medical code descriptions for diagnosis prediction in healthcare
Dataset | @k | MLP | MLP + | RNN | RNN + | RNNa | RNNa+ | Dipole | Dipole + | RETAIN | RETAIN + | GRAM | GRAM + |
---|---|---|---|---|---|---|---|---|---|---|---|---|---|
MIMIC-III | 5 | 0.6939 | 0.7124 | 0.6616 | 0.7160 | 0.6504 | 0.7083 | 0.6599 | 0.7074 | 0.6835 | 0.7167 ∗ | 0.6885 | 0.7132 |
10 | 0.6441 | 0.6603 | 0.6145 | 0.6565 | 0.6021 | 0.6527 | 0.6116 | 0.6539 | 0.6361 | 0.6623 ∗ | 0.6424 | 0.6596 | |
15 | 0.6812 | 0.6926 ∗ | 0.6546 | 0.6906 | 0.6412 | 0.6856 | 0.6524 | 0.6903 | 0.6777 | 0.6918 | 0.6828 | 0.6918 | |
20 | 0.7420 | 0.7544 ∗ | 0.7199 | 0.7511 | 0.7109 | 0.7455 | 0.7159 | 0.7483 | 0.7403 | 0.7501 | 0.7434 | 0.7513 | |
25 | 0.7939 | 0.8070 ∗ | 0.7755 | 0.8019 | 0.7697 | 0.8009 | 0.7723 | 0.8020 | 0.7912 | 0.8010 | 0.7941 | 0.8028 | |
30 | 0.8357 | 0.8460 | 0.8186 | 0.8456 | 0.8142 | 0.8445 | 0.8169 | 0.8453 | 0.8335 | 0.8445 | 0.8377 | 0.8468 ∗ | |
Heart failure | 5 | 0.4451 | 0.4947 | 0.4890 | 0.5172 | 0.4976 | 0.5103 | 0.4964 | 0.5111 | 0.3751 | 0.5140 | 0.5341 | 0.5365 ∗ |
10 | 0.6122 | 0.6206 | 0.6585 | 0.6879 | 0.6675 | 0.6817 | 0.6689 | 0.6829 | 0.5378 | 0.6828 | 0.7123 | 0.7159 ∗ | |
15 | 0.6996 | 0.7060 | 0.7436 | 0.7683 | 0.7496 | 0.7631 | 0.7514 | 0.7648 | 0.6372 | 0.7613 | 0.7901 | 0.7939 ∗ | |
20 | 0.7606 | 0.7643 | 0.8006 | 0.8213 | 0.8050 | 0.8174 | 0.8070 | 0.8167 | 0.7088 | 0.8143 | 0.8402 | 0.8442 ∗ | |
25 | 0.8100 | 0.8140 | 0.8425 | 0.8593 | 0.8453 | 0.8560 | 0.8476 | 0.8557 | 0.7655 | 0.8533 | 0.8761 | 0.8789 ∗ | |
30 | 0.8477 | 0.8511 | 0.8743 | 0.8879 | 0.8770 | 0.8857 | 0.8785 | 0.8846 | 0.8102 | 0.8826 | 0.9025 | 0.9047 ∗ |