Powerful and Theoretically Guaranteed Independence Testing on Heterogeneous Federated Clients
YiXin Ren, Hongquan Liu, Juncai Zhang, Yewei Xia, Zichuan Lin, Deheng Ye, Hao Zhang, Jihong Guan, Shuigeng Zhou
Abstract
In this paper, we present a novel federated independence testing (FedIT) method that addresses both theoretical and practical challenges arising from client heterogeneity. We begin by revisiting existing federated independence testing methods and showing why they fail to provide valid guarantees or maintain statistical power under data distributional shift across clients. Building on this analysis, we develop a copula-based marginal alignment technique together with a stacking-based aggregation strategy that amplifies intra-client dependence while mitigating inter-client variation, resulting in a theoretically sound and powerful global test. For practicality, we further accelerate the aggregation step and incorporate a privacy-preserving mechanism. On the theoretical side, we prove both the correctness of our method and the validity of the test. Empirically, we conduct extensive experiments on both synthetic and real-world datasets, which demonstrate the superiority of our solution over existing methods.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext b842f99c-d4ce-4d41-80fe-00b5b8bf0621Builds on12
- Learning Deep Kernels for Non-Parametric Two-Sample TestsFeng Liu, Wenkai Xu, Jie Lu, Guangquan Zhang et al.ICML 2020 · 213 citations
- Self-Supervised Learning with Kernel Dependence MaximizationYazhe Li, Roman Pogodin, Danica J. Sutherland, Arthur GrettonNeurIPS 2021 · 107 citations
- Efficient Aggregated Kernel Tests using Incomplete -statisticsAntonin Schrab, Ilmun Kim, Benjamin Guedj, Arthur GrettonNeurIPS 2022 · 42 citations
- Learning Kernel Tests Without Data SplittingJonas M. Kübler, Wittawat Jitkrittum, Bernhard Schölkopf, Krikamol MuandetNeurIPS 2020 · 27 citations
- Federated Causal Discovery from Heterogeneous DataLoka Li, Ignavier Ng, Gongxu Luo, Biwei Huang et al.ICLR 2024 · 16 citations
Related papers
- Local Learning Matters: Rethinking Data Heterogeneity in Federated LearningMatías Mendieta, Taojiannan Yang, Pu Wang, Minwoo Lee et al.CVPR 2022 · 176 citations
- FedDisco: Federated Learning with Discrepancy-Aware CollaborationRui Ye, Mingkai Xu, Jianyu Wang, Chenxin Xu et al.ICML 2023 · 136 citations
- FedCDWA: Decoupled Federated Prototype Distillation with Hierarchical Wasserstein AggregationZhenshen Liu, Kai Fan, Wenjie Li, Kuan Zhang et al.ICML 2026
- FedCND: Federated Graph-Level Clustering under Inter-Client Cluster Number DiscrepancyJunlong Wu, Renda Han, Wenxuan Tu, Jingxin Liu et al.WWW 2026
- PA3Fed: Period-Aware Adaptive Aggregation for Improved Federated LearningChengxiang Huang, Bingyan LiuAAAI 2025 · 4 citations
