Distributed mediation analysis with communication efficiency
Shaomin Li
Abstract
We study the mediation analysis under the distributed framework, where data are stored and processed across different worker machines due to storage limitations or privacy concerns. Building upon the classic Sobel's test and MaxP test, we introduce the distributed Sobel's test and distributed MaxP test, respectively. These tests are both communication-efficient and easy to implement. Theoretical analysis and numerical experiments show that, compared to the global test obtained by pooling all data together, the proposed tests achieve nearly identical power, independent of the number of machines. Furthermore, based on these two distributed test statistics, many enhanced mediation tests derived from the Sobel's or MaxP tests can be easily adapted to the distributed system. We apply our method to an educational study, testing whether the effect of high school mathematics on college-level Probability and Mathematical Statistics courses is mediated by Calculus. Our method successfully detects the mediation effect, which would not be identifiable using data from only the first or second class, highlighting the advantage of our approach.
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.
Builds on4
- FuseFL: One-Shot Federated Learning through the Lens of Causality with Progressive Model FusionZhenheng Tang, Yonggang Zhang, Peijie Dong, Yiu-ming Cheung et al.NeurIPS 2024 · 28 citations
- Federated Causal Discovery from Heterogeneous DataLoka Li, Ignavier Ng, Gongxu Luo, Biwei Huang et al.ICLR 2024 · 16 citations
- Collaborative Heterogeneous Causal Inference Beyond Meta-analysisTianyu Guo, Sai Praneeth Karimireddy, Michael I. JordanICML 2024 · 4 citations
- AMDP: An Adaptive Detection Procedure for False Discovery Rate Control in High-Dimensional Mediation AnalysisJiarong Ding, Xuehu ZhuNeurIPS 2023 · 2 citations
Related papers
- Simultaneous Inference for Massive Data: Distributed BootstrapYang Yu, Shih-Kang Chao, Guang ChengICML 2020 · 17 citations
- Sharp Bounds for Generalized Causal Sensitivity AnalysisDennis Frauen, Valentyn Melnychuk, Stefan FeuerriegelNeurIPS 2023 · 36 citations
- Secure Multi-party Computation of Differentially Private MedianJonas Böhler, Florian KerschbaumUSENIX Security 2020
- A Reinforcement Learning Framework for Dynamic Mediation AnalysisLin Ge, Jitao Wang, Chengchun Shi, Zhenke Wu et al.ICML 2023 · 6 citations
- Conditional Distributional Treatment Effect with Kernel Conditional Mean Embeddings and U-Statistic RegressionJunhyung Park, Uri Shalit, Bernhard Schölkopf, Krikamol MuandetICML 2021 · 46 citations
