# Estimating Treatment Effects Under Heterogeneous Interference

arXiv: https://arxiv.org/abs/2309.13884 · HTML: https://mlsanity.com/paper/2309.13884/

## Key facts

- Cited by 7 papers in the map (data through 14 Oct 2026).
- Builds on 23 of its 46 references that are in the map.
- First posted on arXiv 25 Sep 2023.

## Builds on (most cited first, 23 of 23)

- [Attention Is All You Need](https://mlsanity.com/paper/1706.03762/) (2017)
- [Semi-Supervised Classification with Graph Convolutional Networks](https://mlsanity.com/paper/1609.02907/) (2016)
- [Graph Attention Networks](https://mlsanity.com/paper/1710.10903/) (2017)
- [SimCSE: Simple Contrastive Learning of Sentence Embeddings](https://mlsanity.com/paper/2104.08821/) (2021)
- [Heterogeneous Graph Attention Network](https://mlsanity.com/paper/1903.07293/) (2019)
- [Ups and Downs: Modeling the Visual Evolution of Fashion Trends with One-Class Collaborative Filtering](https://mlsanity.com/paper/1602.01585/) (2016)
- [Learning Representations for Counterfactual Inference](https://mlsanity.com/paper/1605.03661/) (2016)
- [MAGNN: Metapath Aggregated Graph Neural Network for Heterogeneous Graph Embedding](https://mlsanity.com/paper/2002.01680/) (2020)
- [A Survey on Causal Inference](https://mlsanity.com/paper/2002.02770/) (2020)
- [A Survey on Heterogeneous Graph Embedding: Methods, Techniques, Applications and Sources](https://mlsanity.com/paper/2011.14867/) (2020)
- [Learning Individual Causal Effects from Networked Observational Data](https://mlsanity.com/paper/1906.03485/) (2019)
- [Identification and estimation of treatment and interference effects in observational studies on networks](https://mlsanity.com/paper/1609.06245/) (2016)
- [Graph Infomax Adversarial Learning for Treatment Effect Estimation with Networked Observational Data](https://mlsanity.com/paper/2106.02881/) (2021)
- [Causal Inference under Networked Interference and Intervention Policy Enhancement](https://mlsanity.com/paper/2002.08506/) (2020)
- [An Attention-based Collaboration Framework for Multi-View Network Representation Learning](https://mlsanity.com/paper/1709.06636/) (2017)
- [Learning Causal Effects on Hypergraphs](https://mlsanity.com/paper/2207.04049/) (2022)
- [Auto-G-Computation of Causal Effects on a Network](https://mlsanity.com/paper/1709.01577/) (2017)
- [Linked Causal Variational Autoencoder for Inferring Paired Spillover Effects](https://mlsanity.com/paper/1808.03333/) (2018)
- [Policy Targeting under Network Interference](https://mlsanity.com/paper/1906.10258/) (2019)
- [Learning Individual Treatment Effects under Heterogeneous Interference in Networks](https://mlsanity.com/paper/2210.14080/) (2022)
- [Causal Inference in the Presence of Interference in Sponsored Search Advertising](https://mlsanity.com/paper/2010.07458/) (2020)
- [Semiparametric Estimation of Treatment Effects in Observational Studies with Heterogeneous Partial Interference](https://mlsanity.com/paper/2107.12420/) (2021)
- [Deep Treatment-Adaptive Network for Causal Inference](https://mlsanity.com/paper/2112.13502/) (2021)

## Built on by (most cited first, 7 of 7)

- [Uplift Modeling Under Limited Supervision](https://mlsanity.com/paper/2403.19289/) (2024)
- [Enhancing Treatment Effect Estimation via Active Learning: A Counterfactual Covering Perspective](https://mlsanity.com/paper/2505.05242/) (2025)
- [Learning Exposure Mapping Functions for Inferring Heterogeneous Peer Effects](https://mlsanity.com/paper/2503.01722/) (2025)
- [Estimating Heterogeneous Causal Effect on Networks via Orthogonal Learning](https://mlsanity.com/paper/2509.18484/) (2025)
- [Neural Networks with Causal Graph Constraints: A New Approach for Treatment Effects Estimation](https://mlsanity.com/paper/2404.12238/) (2024)
- [Treatment Effect Estimation with Differentiated Networked Effect on Graph Data](https://mlsanity.com/paper/2605.24358/) (2026)
- [Peer Effects in Signed Networks: Separating Influence Through Positive and Negative Ties](https://mlsanity.com/paper/2610.02872/) (2026)

## Abstract (arXiv)

Treatment effect estimation can assist in effective decision-making in e-commerce, medicine, and education. One popular application of this estimation lies in the prediction of the impact of a treatment (e.g., a promotion) on an outcome (e.g., sales) of a particular unit (e.g., an item), known as the individual treatment effect (ITE). In many online applications, the outcome of a unit can be affected by the treatments of other units, as units are often associated, which is referred to as interference. For example, on an online shopping website, sales of an item will be influenced by an advertisement of its co-purchased item. Prior studies have attempted to model interference to estimate the ITE accurately, but they often assume a homogeneous interference, i.e., relationships between units only have a single view. However, in real-world applications, interference may be heterogeneous, with multi-view relationships. For instance, the sale of an item is usually affected by the treatment of its co-purchased and co-viewed items. We hypothesize that ITE estimation will be inaccurate if this heterogeneous interference is not properly modeled. Therefore, we propose a novel approach to model heterogeneous interference by developing a new architecture to aggregate information from diverse neighbors. Our proposed method contains graph neural networks that aggregate same-view information, a mechanism that aggregates information from different views, and attention mechanisms. In our experiments on multiple datasets with heterogeneous interference, the proposed method significantly outperforms existing methods for ITE estimation, confirming the importance of modeling heterogeneous interference.
