Please join usonlinefor a talk by,Assistant Professorof Statistics at PurdueUniversity. She willԳ“Federated Learning for Feature Mismatch” on Wednesday, September 23, at 1:30pm Central Time. For access to this seminar, contact Cierra Streeter.
Dr. Xue develops and analyzes advanced statistical methodology at the intersection of data integration, mobile health, missing data, high-dimensional inference, and statistical genetics. She received her PhD in Statistics from the University of Illinois Urbana–Champaign in 2019, followed by postdoctoral work at the University of Pennsylvania.
Abstract
We consider a framework that involves multiple sites with privacy constraints and heterogeneity, where different sites collect data from independent subjects, and site-specific, mismatched feature sets (e.g., hospitals with differing equipment and workflows). Most existing methods assume either aligned covariates or overlappingcohorts, andthus are inapplicable when both samples aredisjointand the covariates are mismatched. We propose Fed-PATE, a three-stage framework: local estimation at each site using its own observed features; a cross-site projection aggregation that recovers a consensus full-feature coefficients among similar sites; and a penalized transfer estimator that feeds the aggregated information back to each site. To accommodate heterogeneity and prevent negative transfer, we introduce a weighting scheme for model selection that adapts to inter-site similarity. Our proposed approach shows a lower prediction risk in theoretical analyses. Simulations show significant reductions inpredictionMSE and robustness to inter-site heterogeneity. In a real-world application to Serum Creatinine (SCr) prediction using EHR variables, Fed-PATE improves performance over existing approaches forthe majority ofsites.