首页|Cafe: Improved Federated Data Imputation by Leveraging Missing Data Heterogeneity
Cafe: Improved Federated Data Imputation by Leveraging Missing Data Heterogeneity
扫码查看
点击上方二维码区域，可以放大扫码查看
原文链接
NETL
NSTL
IEEE
Federated learning (FL), a decentralized machine learning approach, offers great performance while alleviating autonomy and confidentiality concerns. Despite FL’s popularity, how to deal with missing values in a federated manner is not well understood. In this work, we initiate a study of federated imputation of missing values, particularly in complex scenarios, where missing data heterogeneity exists and the state-of-the-art (SOTA) approaches for federated imputation suffer from significant loss in imputation quality. We propose Cafe, a personalized FL approach for missing data imputation. Cafe is inspired from the observation that heterogeneity can induce differences in observable and missing data distribution across clients, and that these differences can be leveraged to improve the imputation quality. Cafe computes personalized weights that are automatically calibrated for the level of heterogeneity, which can remain unknown, to develop personalized imputation models for each client. An extensive empirical evaluation over a variety of settings demonstrates that Cafe matches the performance of SOTA baselines in homogeneous settings while significantly outperforming the baselines in heterogeneous settings.
ImputationData modelsDistributed databasesHospitalsGlucosePredictive modelsComputational modelingBiological system modelingMathematical modelsProtocols
Sitao Min、Hafiz Asif、Xinyue Wang、Jaideep Vaidya
展开 >
Rutgers University, Newark, NJ, USA
Rutgers University, Newark, NJ, USA|Hofstra University, Hempstead, NY, USA
Center for Applied Statistics and School of Statistics, Renmin University of China, Beijing, China
NETL
NSTL
IEEE
