Local–Global Memory Neural Network for Medication Prediction

Jun Song,Yueyang Wang,Siliang Tang,Yin Zhang,Zhigang Chen,Zhongfei Zhang,Tong Zhang,Fei Wu
DOI: https://doi.org/10.1109/tnnls.2020.2989364
IF: 14.255
2021-04-01
IEEE Transactions on Neural Networks and Learning Systems
Abstract:Electronic medical records (EMRs) play an important role in medical data mining and sequential data learning. In this article, we propose to use a sequential neural network with dynamic content-based memories to predict future medications, given EMRs. The local–global memory neural network contains two layers of memories: the local memory and the global memory. Particularly, our method learns the hidden knowledge within EMRs by locally remembering individual patterns of a patient (via local memory) and globally remembering group evidence of disease (via global memory). In addition, we show how our model can be modified to classify the hidden states of EMRs from different patients at each time step into different phases that indicate the progressions of medications in terms of a specific disease, in an unsupervised manner. Experimental results on real EMRs data sets show that, by learning EMRs with external local and global memories, with regard to a given disease, our model improves the prediction performance compared with several alternative methods.
computer science, artificial intelligence, theory & methods,engineering, electrical & electronic, hardware & architecture
What problem does this paper attempt to address?