Implementation of SurfCon model for Paper "SurfCon: Synonym Discovery on Privacy-Aware Clinical Data", which studies Synonym Discovery on Privacy-Aware Clinical Data.
This repository is the implementation for the SurfCon model that utilizes the surface form and global context information to mine synonyms for medical terms extracted from clinical texts, such as Electronic Medical Records (EMRs).
The surface form information provides connections between medical terms in the surface form level. Not surprisingly, the surface form information is critical in the medical domain. For example, term hypocupremia is the synonym of copper deficiency in which hypo means deficiency, cupre means copper and mia is connected with blood. Inspired by such intuition, we design a bi-level surface form encoder to capture the information in both character and word levels of the medical terms.
Moreover, the global context information provides semantic information between medical terms. To deal with OOV problem (e.g., query terms do not show up in existing data), we leverage an inductive context prediction module to predict the close neighbors for terms and aggregate the predicted neighbors to calculate a context score between terms.
Please refer to our paper for more detailed information.
The dataset used in current experiments contains medical term-term co-occurrence graphs extracted from EMRs. The dataset can be downloaded from the original paper, Building the graph of medicine from millions of clinical narratives
More importantly, you can apply our model to your own data. Our model and problem setting can be extended to any other text corpus with the privacy concerns as long as a co-occurrence graph is provided.
-
For testing our pretrained SurfCon model to discover synonyms, please download the pretrained model and parameters and ensure the correct file paths.
-
The pretrained embeddings can be downloaded here: GloVe, charNgram, node features.
-
If you want to test quickly with existing medicla terms, please refer to here and put them in the corresponding path of data folder. If you have your own terms, please reivse the argument
--cand_terms_path
intest_surfcon.sh
with a text file (one term per line). -
Testing our pretrained SurfCon model to discover synonyms:
> bash test_surfcon.sh
Steps for trianing the model from scratch:
Step 1: Training the inductive context prediction model:
> bash train_ctx_pred.sh
Step 2: Training the final ranking model:
> bash train_surfcon.sh
If you have any questions, please feel free to contact us!
@inproceedings{wang2019surfcon,
title={SurfCon: Synonym Discovery on Privacy-Aware Clinical Data},
author={Wang, Zhen and Yue, Xiang and Moosavinasab, Soheil and Huang, Yungui and Lin, Simon and Sun, Huan},
booktitle={Proceedings of the 25th ACM SIGKDD International Conference on Knowledge Discovery \& Data Mining},
year={2019},
organization={ACM}
}