VITA: 'Carefully Chosen and Weighted Less' Is Better in Medication Recommendation (2312.12100v1)
Abstract: We address the medication recommendation problem, which aims to recommend effective medications for a patient's current visit by utilizing information (e.g., diagnoses and procedures) given at the patient's current and past visits. While there exist a number of recommender systems designed for this problem, we point out that they are challenged in accurately capturing the relation (spec., the degree of relevance) between the current and each of the past visits for the patient when obtaining her current health status, which is the basis for recommending medications. To address this limitation, we propose a novel medication recommendation framework, named VITA, based on the following two novel ideas: (1) relevant-Visit selectIon; (2) Target-aware Attention. Through extensive experiments using real-world datasets, we demonstrate the superiority of VITA (spec., up to 5.56% higher accuracy, in terms of Jaccard, than the best competitor) and the effectiveness of its two core ideas. The code is available at https://github.com/jhheo0123/VITA.
- Neural Machine Translation by Jointly Learning to Align and Translate. In Proceedings of the International Conference on Learning Representations (ICLR).
- Personalizing Medication Recommendation with a Graph-Based Approach. ACM Transactions on Information Systems, 40: 1–23.
- SMR: Medical Knowledge Graph Embedding for Safe Medicine Recommendation. Big Data Research, 23: 100174.
- Distilling the knowledge in a neural network. arXiv preprint arXiv:1503.02531.
- Categorical Reparameterization with Gumbel-Softmax. In Proceedings of the International Conference on Learning Representations (ICLR).
- MIMIC-IV, a freely accessible electronic health record dataset. Scientific data, 10: 1.
- MIMIC-III, a freely accessible critical care database. Scientific data, 3: 1–9.
- Acoustic Scene Classification for Mismatched Recording Devices Using Heated-Up Softmax and Spectrum Correction. In Proceedings of the IEEE International Conference on Acoustics, Speech and Signal Processing (ICASSP), 126–130.
- Pre-training of Graph Augmented Transformers for Medication Recommendation. In Proceedings of the International Joint Conference on Artificial Intelligence (IJCAI), 5953–5959.
- GAMENet: Graph Augmented MEmory Networks for Recommending Medication Combination. In Proceedings of the AAAI Conference on Artificial Intelligence (AAAI), volume 33, 1126–1133.
- Debiased, Longitudinal and Coordinated Drug Recommendation through Multi-Visit Clinic Records. In Proceedings of the International Conference on Neural Information Processing Systems (NeurIPS), volume 35, 27837–27849.
- 4sdrug: Symptom-based set-to-set small and safe drug recommendation. In Proceedings of the ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (KDD), 3970–3980.
- Attention is All you Need. In Proceedings of the International Conference on Neural Information Processing Systems (NeurIPS), volume 30, 5998–6008.
- Personalized Prescription for Comorbidity. In Proceedings of the International Conference on Database Systems for Advanced Applications (DASFAA), 3–19.
- Self-Supervised Adversarial Distribution Regularization for Medication Recommendation. In Proceedings of the International Joint Conference on Artificial Intelligence (IJCAI), 3134–3140.
- Multi-hop Reading on Memory Neural Network with Selective Coverage for Medication Recommendation. In Proceedings of the International ACM Conference on Information and Knowledge Management (CIKM), 2020–2029.
- Leveraging multiple types of domain knowledge for safe and effective drug recommendation. In Proceedings of the International ACM Conference on Information and Knowledge Management (CIKM), 2169–2178.
- Conditional Generation Net for Medication Recommendation. In Proceedings of the international ACM conference on World Wide Web (WWW), 935–945.
- Change Matters: Medication Change Prediction with Recurrent Residual Networks. In Proceedings of the International Joint Conference on Artificial Intelligence (IJCAI), 3728–3734.
- SafeDrug: Dual Molecular Graph Encoders for Recommending Effective and Safe Drug Combinations. In Proceedings of the International Joint Conference on Artificial Intelligence (IJCAI), 3735–3741.
- LEAP: Learning to Prescribe Effective and Safe Treatment Combinations for Multimorbidity. In Proceedings of the ACM SIGKDD International Conference on Knowledge Discovery and Data Mining (KDD), 1315–1324.