arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 1904.04460cs.LGcs.CL
收藏

基于注意力的多示例神经网络在不完全低质量数据医疗诊断中的应用

Attention-based Multi-instance Neural Network for Medical Diagnosis from Incomplete and Low Quality Data

The University of Sydney(悉尼大学);Renmin University of China(中国人民大学);Beijing Medicinovo Technology Co.,Ltd.(北京美迪诺沃科技有限公司)

Zeyuan Wang, Josiah Poon, Shiding Sun, Simon Poon

摘要:

从临床记录中提取模式的一种方法是将每个患者记录视为具有各种数量的症状形式的实例的袋子。医学诊断就是先发现信息量,然后将其映射到一种或多种疾病。在许多情况下,患者被表示为某些特征空间中的向量,并且分类器被应用于生成诊断结果。然而,在许多现实世界的情况下,数据往往是低质量的,由于各种原因,如数据的一致性,完整性,完整性,准确性等,在本文中,我们提出了一种新的方法,基于注意力的多实例神经网络(AMI-Net),使单疾病分类仅基于现有的和有效的信息在现实世界的门诊记录。在患者的上下文中,它将一个包的实例作为输入,并以端到端的方式直接输出包标签。首先采用嵌入层,将实例映射到代表个体病情的嵌入空间。通过多头注意力Transformer、实例级多实例池和袋级多实例池来获取实例之间的相关性及其对最终分类的重要性。该方法在两个非标准化和高度不平衡的数据集上进行了测试,一个在传统中医药(TCM)领域,另一个在西医(WM)领域。我们的初步结果表明,所提出的方法优于所有基线结果的显着保证金。

英文摘要:

One way to extract patterns from clinical records is to consider each patient record as a bag with various number of instances in the form of symptoms. Medical diagnosis is to discover informative ones first and then map them to one or more diseases. In many cases, patients are represented as vectors in some feature space and a classifier is applied after to generate diagnosis results. However, in many real-world cases, data is often of low-quality due to a variety of reasons, such as data consistency, integrity, completeness, accuracy, etc. In this paper, we propose a novel approach, attention based multi-instance neural network (AMI-Net), to make the single disease classification only based on the existing and valid information in the real-world outpatient records. In the context of a patient, it takes a bag of instances as input and output the bag label directly in end-to-end way. Embedding layer is adopted at the beginning, mapping instances into an embedding space which represents the individual patient condition. The correlations among instances and their importance for the final classification are captured by multi-head attention transformer, instance-level multi-instance pooling and bag-level multi-instance pooling. The proposed approach was test on two non-standardized and highly imbalanced datasets, one in the Traditional Chinese Medicine (TCM) domain and the other in the Western Medicine (WM) domain. Our preliminary results show that the proposed approach outperforms all baselines results by a significant margin.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑