Journey to the Centre of Cluster: Harnessing Interior Nodes for A/B Testing under Network Interference
Qianyi Chen, Anpeng Wu, Bo Li, Lu Deng, Yong Wang
Abstract
A/B testing on platforms often faces challenges from network interference, where a unit's outcome depends not only on its own treatment but also on the treatments of its network neighbors. To address this, cluster-level randomization has become standard, enabling the use of network-aware estimators. These estimators typically trim the data to retain only a subset of informative units, achieving low bias under suitable conditions but often suffering from high variance. In this paper, we first demonstrate that the interior nodes—units whose neighbors all lie within the same cluster—constitute the vast majority of the post-trimming subpopulation. In light of this, we propose directly averaging over the interior nodes to construct the mean-in-interior (MII) estimator, which circumvents the delicate reweighting required by existing network-aware estimators and substantially reduces variance in classical settings. However, we show that interior nodes are often not representative of the full population, particularly in terms of network-dependent covariates, leading to notable bias. We then augment the MII estimator with a counterfactual predictor trained on the entire network, allowing us to adjust for covariate distribution shifts between the interior nodes and full population. By rearranging the expression, we reveal that our augmented MII estimator embodies an analytical form of the point estimator within prediction-powered inference framework . This insight motivates a semi-supervised lens, wherein interior nodes are treated as labeled data subject to selection bias. Extensive and challenging simulation studies demonstrate the outstanding performance of our augmented MII estimator across various settings.
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 on6
- Staggered Rollout Designs Enable Causal Inference Under Interference Without Network KnowledgeMayleen Cortez, Matthew Eichhorn, Christina Lee YuNeurIPS 2022 · 29 citations
- Cluster Randomized Designs for One-Sided Bipartite ExperimentsJennifer Brennan, Vahab Mirrokni, Jean Pouget-AbadieNeurIPS 2022 · 17 citations
- Higher-Order Causal Message Passing for Experimentation with Complex InterferenceMohsen Bayati, Yuwei Luo, William Overman, Mohamad Sadegh Shirani Faradonbeh et al.NeurIPS 2024 · 9 citations
- Independent-Set Design of Experiments for Estimating Treatment and Spillover Effects under Network InterferenceChencheng Cai, Xu Zhang, Edoardo M. AiroldiICLR 2024 · 8 citations
- Optimized Covariance Design for AB Test on Social Network under InterferenceQianyi Chen, Bo Li, Lu Deng, Yong WangNeurIPS 2023 · 6 citations
Related papers
- A/B Testing in Network Data with Covariate-Adaptive RandomizationJialu Wang, Ping Li, Feifang HuICML 2023 · 6 citations
- Experimentation on Endogenous GraphsWenshuo Wang, Edvard Bakhitov, Dominic CoeyAAAI 2026
- A Tree-based Model Averaging Approach for Personalized Treatment Effect Estimation from Heterogeneous Data SourcesXiaoqing Tan, Chung-Chou H. Chang, Ling Zhou, Lu TangICML 2022 · 23 citations
- Estimating Heterogeneous Treatment Effects: Mutual Information Bounds and Learning AlgorithmsXingzhuo Guo, Yuchen Zhang, Jianmin Wang, Mingsheng LongICML 2023 · 11 citations
- Causal Network Motifs: Identifying Heterogeneous Spillover Effects in A/B TestsYuan Yuan, Kristen M. Altenburger, Farshad KootiWWW 2021 · 37 citations
