.png){kind=icon}
返回
A typical sample-driven learning framework for automatic disease diagnosis
.png){kind=icon}
PRE
AI
DOI：10.1016/j.asoc.2024.111745.png)
.png){kind=icon}
原文求助.png){kind=icon}
分享.png){kind=icon}
收藏
摘要
En 中文
Disease diagnosis mainly depends on the doctor's medical knowledge and clinical experience, which can be treated as a medical text classification task. We observe that existing data-driven methods always suffer from the distribution bias since a small amount of common diseases appear high-frequently, while most diseases are infrequent in real-world, which leads to an unbalanced data distribution in the disease diagnosis task. To address this problem, we propose a new learning framework, T ypical sample-Driven D riven G raph N eural N etwork (TD-GNN) for disease knowledge representation and classification. In our framework, different from previous methods, each disease (label) is concretized and learned from several corresponding well-representative samples rather than full imbalance data. In addition, the contrastive learning strategy is utilized to enhance the distinguishable features learning between different diseases. In this study, we construct a real-world dataset covering 350 common diseases to evaluate the proposed learning method. The experimental results demonstrate that the proposed TD-GNN significantly outperforms the state-of-the-art baselines, especially for the majority of diseases in which only small samples can be collected from the real world. Additionally, our method can provide a sample-based interpretation for disease prediction learning.
Keyword:
Unbalanced classification
Contrastive learning
Disease diagnosis
Sample-based learning
期刊
IF:
6.6
论文数:
1.4W
被引数:
4.8W
.png){kind=icon}
