FeARH: Federated machine learning with anonymous random hybridization on electronic medical records.

Journal: Journal of biomedical informatics
Published Date:

Abstract

Electrical medical records are restricted and difficult to centralize for machine learning model training due to privacy and regulatory issues. One solution is to train models in a distributed manner that involves many parties in the process. However, sometimes certain parties are not trustable, and in this project, we aim to propose an alternative method to traditional federated learning with central analyzer in order to conduct training in a situation without a trustable central analyzer. The proposed algorithm is called "federated machine learning with anonymous random hybridization (abbreviated as 'FeARH')", using mainly hybridization algorithm to degenerate the integration of connections between medical record data and models' parameters by adding randomization into the parameter sets shared to other parties. Based on our experiment, our new algorithm has similar AUCROC and AUCPR results compared with machine learning in a centralized manner and original federated machine learning.

Authors

  • Jianfei Cui
    Viterbi School of Engineering, University of Southern California, Los Angeles, CA 90007, United States.
  • He Zhu
    State Key Laboratory of Bioelectronics, School of Biological Science and Medical Engineering, Southeast University, Nanjing 210096, China.
  • Hao Deng
    Faculty of Information Technology, Macau University of Science and Technology, Avenida Wai Long, Taipa, Macau, China.
  • Ziwei Chen
    Beijing Jiaotong University, Beijing, China. Electronic address: zwchen@bjtu.edu.cn.
  • Dianbo Liu
    Computer Science and Artificial Intelligence Laboratory, MIT, Cambridge.