Estimating Treatment Effects in Networks using Domain Adversarial Training

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Caljon, Daan, Van Belle, Jente, Verbeke, Wouter
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866908610099937280
author Caljon, Daan
Van Belle, Jente
Verbeke, Wouter
author_facet Caljon, Daan
Van Belle, Jente
Verbeke, Wouter
contents Estimating heterogeneous treatment effects in network settings is complicated by interference, meaning that the outcome of an instance can be influenced by the treatment status of others. Existing causal machine learning approaches usually assume a known exposure mapping that summarizes how the outcome of a given instance is influenced by others' treatment, a simplification that is often unrealistic. Furthermore, the interaction between homophily -- the tendency of similar instances to connect -- and the treatment assignment mechanism can induce a network-level covariate shift that may lead to inaccurate treatment effect estimates, a phenomenon that has not yet been explicitly studied. To address these challenges, we propose HINet, a novel method that integrates graph neural networks with domain adversarial training. This combination allows estimating treatment effects under unknown exposure mappings while mitigating the impact of (network-level) covariate shift. An extensive empirical evaluation on synthetic and semi-synthetic network datasets demonstrates the effectiveness of our approach.
format Preprint
id arxiv_https___arxiv_org_abs_2510_21457
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Estimating Treatment Effects in Networks using Domain Adversarial Training
Caljon, Daan
Van Belle, Jente
Verbeke, Wouter
Machine Learning
Estimating heterogeneous treatment effects in network settings is complicated by interference, meaning that the outcome of an instance can be influenced by the treatment status of others. Existing causal machine learning approaches usually assume a known exposure mapping that summarizes how the outcome of a given instance is influenced by others' treatment, a simplification that is often unrealistic. Furthermore, the interaction between homophily -- the tendency of similar instances to connect -- and the treatment assignment mechanism can induce a network-level covariate shift that may lead to inaccurate treatment effect estimates, a phenomenon that has not yet been explicitly studied. To address these challenges, we propose HINet, a novel method that integrates graph neural networks with domain adversarial training. This combination allows estimating treatment effects under unknown exposure mappings while mitigating the impact of (network-level) covariate shift. An extensive empirical evaluation on synthetic and semi-synthetic network datasets demonstrates the effectiveness of our approach.
title Estimating Treatment Effects in Networks using Domain Adversarial Training
topic Machine Learning
url https://arxiv.org/abs/2510.21457