CEHMR: Curriculum learning enhanced hierarchical multi-label classification for medication recommendation

doi:10.1016/j.artmed.2023.102613

	CEHMR: Curriculum learning enhanced hierarchical multi-label classification for medication recommendation
	Sun, Mengxuan 1,2; Niu, Jinghao1 ; Yang, Xuebing1 ; Gu, Yifan 1,2; Zhang, Wensheng1,2,3
发表期刊	ARTIFICIAL INTELLIGENCE IN MEDICINE
ISSN	0933-3657
	2023-09-01
卷号	143 页码:13
通讯作者	Yang, Xuebing(yangxuebing2013@ia.ac.cn) ; Zhang, Wensheng(zhangwenshengia@hotmail.com)
摘要	The medication recommendation (MR) or medication combination prediction task aims to predict effective prescriptions given accurate patient representations derived from electronic health records (EHRs), which contributes to improving the quality of clinical decision-making, especially for patients with multi-morbidity. Although in recent years deep learning technology has achieved great success in MR, the performance of current multi-label based MR solutions is unsatisfactory. They mainly focus on improving the patient representation module and modeling the medication label dependencies such as drug-drug interaction (DDI) correlation and co-occurrence relationship. However, the hierarchical dependency among medication labels and diversity of difficulty among MR training examples lack sufficient consideration. In this paper, we propose a framework of Curriculum learning Enhanced Hierarchical multi-label classification for MR (CEHMR). Motivated by the category hierarchy of medications which organizes standard medication codes in a hierarchical structure, we utilize it to provide more trustworthy prior knowledge for modeling label dependency. Specifically, we design a hierarchical multi-label classifier with a learnable gate fusion layer, to simultaneously capture the level-independent (local) and level-dependent (global) hierarchical information in the medication hierarchy. In addition, to overcome the diversity of training example difficulties, and progressively achieve a smoother training process, we introduce a bootstrap-based curriculum learning strategy. Hence, the example difficulty can be measured based on the predictive performance of the MR model, and then all training examples would be retrained from easy to hard under the guidance of a predefined training scheduler. Experiments on the real-world medical MIMIC-III database demonstrate that the proposed framework can achieve state-of-theart performance compared with seven representative baselines, and extensive ablation studies validate the effectiveness of each component of CEHMR.
关键词	Medication recommendation EHR Hierarchical multi-label classification Curriculum learning
DOI	10.1016/j.artmed.2023.102613
收录类别	SCI
语种	英语
资助项目	National Key R&D Program of China[2021ZD0111005] ; National Key R&D Program of China[61976212] ; National Key R&D Program of China[62006139] ; National Key R&D Program of China[62203437] ; National Key R&D Program of China[61976213]
项目资助者	National Key R&D Program of China
WOS研究方向	Computer Science ; Engineering ; Medical Informatics
WOS类目	Computer Science, Artificial Intelligence ; Engineering, Biomedical ; Medical Informatics
WOS记录号	WOS:001061845300001
出版者	ELSEVIER
引用统计	被引频次：2[WOS] [WOS记录] [WOS相关记录]
文献类型	期刊论文
条目标识符	http://ir.ia.ac.cn/handle/173211/53171
专题	多模态人工智能系统全国重点实验室
通讯作者	Yang, Xuebing; Zhang, Wensheng
作者单位	1.Chinese Acad Sci, Inst Automat, State Key Lab Multimodal Artificial Intelligence S, Beijing, Peoples R China 2.Univ Chinese Acad Sci, Beijing, Peoples R China 3.Guangzhou Univ, Guangzhou, Peoples R China
第一作者单位	中国科学院自动化研究所
通讯作者单位	中国科学院自动化研究所
推荐引用方式 GB/T 7714	Sun, Mengxuan,Niu, Jinghao,Yang, Xuebing,et al. CEHMR: Curriculum learning enhanced hierarchical multi-label classification for medication recommendation[J]. ARTIFICIAL INTELLIGENCE IN MEDICINE,2023,143:13.
APA	Sun, Mengxuan,Niu, Jinghao,Yang, Xuebing,Gu, Yifan,&Zhang, Wensheng.(2023).CEHMR: Curriculum learning enhanced hierarchical multi-label classification for medication recommendation.ARTIFICIAL INTELLIGENCE IN MEDICINE,143,13.
MLA	Sun, Mengxuan,et al."CEHMR: Curriculum learning enhanced hierarchical multi-label classification for medication recommendation".ARTIFICIAL INTELLIGENCE IN MEDICINE 143(2023):13.