ContextCare: Incorporating Contextual Information Networks to Representation Learning on Medical Forum Data

ContextCare: Incorporating Contextual Information Networks to Representation Learning on Medical Forum Data

Stan Zhao, Meng Jiang, Quan Yuan, Bing Qin, Ting Liu, ChengXiang Zhai

Proceedings of the Twenty-Sixth International Joint Conference on Artificial Intelligence
Main track. Pages 3497-3503. https://doi.org/10.24963/ijcai.2017/489

Online users have generated a large amount of health-related data on medical forums and search engines. However, exploiting these rich data for orienting patient online and assisting medical checkup offline is nontrivial due to the sparseness of existing symptom-disease links, which caused by the natural and chatty expressions of symptoms. In this paper, we propose a novel and general representation learning method ContextCare for human generated health-related data, which learns the latent relationship between symptoms and diseases from the symptom-disease diagnosis network for disease prediction, disease category prediction and disease clustering. To alleviate the network sparseness, ContextCare adopts regularizations from rich contextual information networks including a symptom co-occurrence network and a disease evolution network. Therefore, our representations of symptoms and diseases incorporate knowledge from these three networks. Extensive experiments on medical forum data demonstrate that ContextCare outperforms the state-of-the-art methods in disease category prediction, disease prediction and disease clustering.
Keywords:
Machine Learning: Data Mining
Multidisciplinary Topics and Applications: AI and Social Sciences