English

Key-Embedded Privacy for Decentralized AI in Biomedical Omics

Machine Learning 2026-03-31 v1 Distributed, Parallel, and Cluster Computing

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.

Keywords

Cite

@article{arxiv.2603.28334,
  title  = {Key-Embedded Privacy for Decentralized AI in Biomedical Omics},
  author = {Rongyu Zhang and Hongyu Dong and Gaole Dai and Ziqi Qiao and Shenli Zheng and Yuan Zhang and Aosong Cheng and Xiaowei Chi and Jincai Luo and Pin Li and Li Du and Dan Wang and Yuan Du and Xudong Xing and Jianxu Chen and Shanghang Zhang},
  journal= {arXiv preprint arXiv:2603.28334},
  year   = {2026}
}
R2 v1 2026-07-01T11:43:58.396Z