Key-Embedded Privacy for Decentralized AI in Biomedical Omics

Authors:Rongyu Zhang, Hongyu Dong, Gaole Dai, Ziqi Qiao, Shenli Zheng, Yuan Zhang, Aosong Cheng, Xiaowei Chi, Jincai Luo, Pin Li, Li Du, Dan Wang, Yuan Du, Xudong Xing, Jianxu Chen, Shanghang Zhang

View PDF HTML (experimental)

Abstract:The rapid adoption of data-driven methods in biomedicine has intensified concerns over privacy, governance, and regulation, limiting raw data sharing and hindering the assembly of representative cohorts for clinically relevant AI. This landscape necessitates practical, efficient privacy solutions, as cryptographic defenses often impose heavy overhead and differential privacy can degrade performance, leading to sub-optimal outcomes in real-world settings. Here, we present a lightweight federated learning method, INFL, based on Implicit Neural Representations that addresses these challenges. Our approach integrates plug-and-play, coordinate-conditioned modules into client models, embeds a secret key directly into the architecture, and supports seamless aggregation across heterogeneous sites. Across diverse biomedical omics tasks, including cohort-scale classification in bulk proteomics, regression for perturbation prediction in single-cell transcriptomics, and clustering in spatial transcriptomics and multi-omics with both public and private data, we demonstrate that INFL achieves strong, controllable privacy while maintaining utility, preserving the performance necessary for downstream scientific and clinical applications.

Subjects:

Machine Learning (cs.LG); Distributed, Parallel, and Cluster Computing (cs.DC)

Cite as: arXiv:2603.28334 [cs.LG]

(or arXiv:2603.28334v1 [cs.LG] for this version)

https://doi.org/10.48550/arXiv.2603.28334

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Rongyu Zhang [view email] [v1] Mon, 30 Mar 2026 12:04:50 UTC (38,126 KB)