DANIEL: A Distributed and Scalable Approach for Global Representation Learning with EHR Applications
Fuente:
arXiv
Saved in:
| Main Authors: | , , , , , , |
|---|---|
| Format: | Preprint |
| Published: |
2025
|
| Subjects: | |
| Online Access: | |
| Tags: |
Add Tag
No Tags, Be the first to tag this record!
|
| _version_ | 1866911248693592064 |
|---|---|
| author | Wang, Zebin Gan, Ziming Tang, Weijing Xia, Zongqi Cai, Tianrun Cai, Tianxi Lu, Junwei |
| author_facet | Wang, Zebin Gan, Ziming Tang, Weijing Xia, Zongqi Cai, Tianrun Cai, Tianxi Lu, Junwei |
| contents | Classical probabilistic graphical models face fundamental challenges in modern data environments, which are characterized by high dimensionality, source heterogeneity, and stringent data-sharing constraints. In this work, we revisit the Ising model, a well-established member of the Markov Random Field (MRF) family, and develop a distributed framework that enables scalable and privacy-preserving representation learning from large-scale binary data with inherent low-rank structure. Our approach optimizes a non-convex surrogate loss function via bi-factored gradient descent, offering substantial computational and communication advantages over conventional convex approaches. We evaluate our algorithm on multi-institutional electronic health record (EHR) datasets from 58,248 patients across the University of Pittsburgh Medical Center (UPMC) and Mass General Brigham (MGB), demonstrating superior performance in global representation learning and downstream clinical tasks, including relationship detection, patient phenotyping, and patient clustering. These results highlight a broader potential for statistical inference in federated, high-dimensional settings while addressing the practical challenges of data complexity and multi-institutional integration. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2511_02754 |
| institution | arXiv |
| publishDate | 2025 |
| record_format | arxiv |
| spellingShingle | DANIEL: A Distributed and Scalable Approach for Global Representation Learning with EHR Applications Wang, Zebin Gan, Ziming Tang, Weijing Xia, Zongqi Cai, Tianrun Cai, Tianxi Lu, Junwei Methodology Machine Learning Classical probabilistic graphical models face fundamental challenges in modern data environments, which are characterized by high dimensionality, source heterogeneity, and stringent data-sharing constraints. In this work, we revisit the Ising model, a well-established member of the Markov Random Field (MRF) family, and develop a distributed framework that enables scalable and privacy-preserving representation learning from large-scale binary data with inherent low-rank structure. Our approach optimizes a non-convex surrogate loss function via bi-factored gradient descent, offering substantial computational and communication advantages over conventional convex approaches. We evaluate our algorithm on multi-institutional electronic health record (EHR) datasets from 58,248 patients across the University of Pittsburgh Medical Center (UPMC) and Mass General Brigham (MGB), demonstrating superior performance in global representation learning and downstream clinical tasks, including relationship detection, patient phenotyping, and patient clustering. These results highlight a broader potential for statistical inference in federated, high-dimensional settings while addressing the practical challenges of data complexity and multi-institutional integration. |
| title | DANIEL: A Distributed and Scalable Approach for Global Representation Learning with EHR Applications |
| topic | Methodology Machine Learning |
| url | https://arxiv.org/abs/2511.02754 |