Hierarchical Label Distribution Learning for Disease Prediction

Yi Ren,Jing Xia,Ziyi Yu,Zhenchuan Zhang,Tianshu Zhou,Yu Tian,Jingsong Li
DOI: https://doi.org/10.3233/SHTI231066
2024-01-25
Abstract:The prediction of disease can facilitate early intervention, comprehensive diagnosis and treatment, thereby benefiting healthcare and reducing medical costs. While single class and multi-class learning methods have been applied for disease prediction, they are inadequate in distinguishing between primary and secondary diagnoses, which is crucial for treatments. In this paper, label distribution is suggested to describe the diagnosis, which assigns the description degree to quantify the diagnosis. Additionally, a novel hierarchical label distribution learning (HLDL) model is proposed to make fine-grained predictions based on the hierarchical classification of diseases, taking into account the relationship among diseases. The experimental results on real-world datasets demonstrate that the HLDL model outperforms the baselines with statistical significance.
What problem does this paper attempt to address?