Treatment Effect Estimation with Differentiated Networked Effect on Graph Data

2026-05-22

AI Summary

This study addresses the bias in individual treatment effect (ITE) estimation from graph-structured observational data, which often arises from neglecting heterogeneous neighbor influences that induce differential network effects (DNE). To tackle this issue, the work proposes a novel graph neural network–based framework that explicitly models DNE for the first time. The approach integrates two attentive modules and a message amplifier to dynamically assess each neighbor’s contribution to interference and adaptively modulate interference intensity based on its magnitude. By moving beyond the conventional assumption of homogeneous interference effects, the method achieves significantly superior performance over existing approaches across three real-world graph datasets, demonstrating that accurately modeling DNE is crucial for enhancing both ITE estimation accuracy and downstream decision quality.

Abstract

Estimating individual treatment effect (ITE) from observational graph data is crucial for decision-making in the fields such as commerce and medicine. This task is challenging due to interference, where individual outcomes can be influenced by the treatments and covariates of their neighbors. Existing methods attempt to model such interference for accurate ITE estimation. However, a critical issue is often overlooked: differentiated networked effect (DNE), an effect caused by local networks consisting of neighbors with varying importance and scales. Capturing DNE is vital; otherwise, we will end up with imprecise ITE estimation due to an erroneous characterization of interference, which can result in misguided decisions. To address this challenge, we propose a novel interference modeling mechanism that incorporates two partial attention mechanisms and a message amplifier. The partial attention mechanisms automatically estimate the importance of different neighbors in contributing to interference, while the message amplifier adjusts the results of the interference modeling mechanism based on the scale of neighbors, all of which enables the model to capture DNE. Experiments on three real-world graphs demonstrate that our methods outperform existing approaches for ITE estimation from graph data, which corroborates the importance of explicitly capturing DNE.

Problem

• individual treatment effect
• interference
• differentiated networked effect
• graph data
• observational study

Innovation

• Differentiated Networked Effect
• Individual Treatment Effect
• Partial Attention Mechanism
• Message Amplifier
• Interference Modeling