DFML: Dynamic Federated Meta-Learning for Rare Disease Prediction

被引:6
作者
Chen, Bingyang [1 ]
Chen, Tao [1 ]
Zeng, Xingjie [1 ]
Zhang, Weishan [1 ]
Lu, Qinghua [2 ]
Hou, Zhaoxiang [1 ]
Zhou, Jiehan [3 ]
Helal, Sumi [4 ]
机构
[1] China Univ Petr, Sch Comp Sci & Technol, Qingdao 266580, Peoples R China
[2] CSIRO, Sydney, NSW 2601, Australia
[3] Oulu Univ, Oulu 90570, Finland
[4] Univ Florida, Gainesville, FL 32611 USA
基金
中国国家自然科学基金;
关键词
Diseases; Predictive models; Data models; Servers; Hospitals; Task analysis; Federated learning; Inaccuracy-focused meta-learning; dynamic weight-based fusion strategy; federated meta-learning; rare disease prediction; digital health; MODEL; SYSTEM;
D O I
10.1109/TCBB.2023.3239848
中图分类号
Q5 [生物化学];
学科分类号
071010 ; 081704 ;
摘要
Millions of patients suffer from rare diseases around the world. However, the samples of rare diseases are much smaller than those of common diseases. Hospitals are usually reluctant to share patient information for data fusion due to the sensitivity of medical data. These challenges make it difficult for traditional AI models to extract rare disease features for disease prediction. In this paper, we propose a Dynamic Federated Meta-Learning (DFML) approach to improve rare disease prediction. We design an Inaccuracy-Focused Meta-Learning (IFML) approach that dynamically adjusts the attention to different tasks according to the accuracy of base learners. Additionally, a dynamic weight-based fusion strategy is proposed to further improve federated learning, which dynamically selects clients based on the accuracy of each local model. Experiments on two public datasets show that our approach outperforms the original federated meta-learning algorithm in accuracy and speed with as few as five shots. The average prediction accuracy of the proposed model is improved by 13.28% compared with each hospital's local model.
引用
收藏
页码:880 / 889
页数:10
相关论文
共 37 条
  • [1] A smart healthcare monitoring system for heart disease prediction based on ensemble deep learning and feature fusion
    Ali, Farman
    El-Sappagh, Shaker
    Islam, S. M. Riazul
    Kwak, Daehan
    Ali, Amjad
    Imran, Muhammad
    Kwak, Kyung-Sup
    [J]. INFORMATION FUSION, 2020, 63 : 208 - 222
  • [2] Boughorbel S., 2019, arXiv
  • [3] Federated learning of predictive models from federated Electronic Health Records
    Brisimi, Theodora S.
    Chen, Ruidi
    Mela, Theofanie
    Olshevsky, Alex
    Paschalidis, Ioannis Ch.
    Shi, Wei
    [J]. INTERNATIONAL JOURNAL OF MEDICAL INFORMATICS, 2018, 112 : 59 - 67
  • [4] Chen F, 2019, Arxiv, DOI arXiv:1802.07876
  • [5] Time series forecasting of COVID-19 transmission in Canada using LSTM networks
    Chimmula, Vinay Kumar Reddy
    Zhang, Lei
    [J]. CHAOS SOLITONS & FRACTALS, 2020, 135
  • [6] Using recurrent neural network models for early detection of heart failure onset
    Choi, Edward
    Schuetz, Andy
    Stewart, Walter F.
    Sun, Jimeng
    [J]. JOURNAL OF THE AMERICAN MEDICAL INFORMATICS ASSOCIATION, 2017, 24 (02) : 361 - 370
  • [7] Fallah A., 2020, P ADV NEUR INF PROC, P35657
  • [8] Fallah A, 2020, Arxiv, DOI arXiv:2002.07948
  • [9] Finn C, 2017, PR MACH LEARN RES, V70
  • [10] RDAD: A Machine Learning System to Support Phenotype-Based Rare Disease Diagnosis
    Jia, Jinmeng
    Wang, Ruiyuan
    An, Zhongxin
    Guo, Yongli
    Ni, Xi
    Shi, Tieliu
    [J]. FRONTIERS IN GENETICS, 2018, 9