Optimizing Treatment Allocation in the Presence of Interference

Daan Caljon, Jente Van Belle, Jeroen Berrevoets, Wouter Verbeke
9/30/2024
cs.SIcs.LGstat.ML

No official code found. Best implementation: daan-caljon/OTAPI (1 star, MIT, last commit 2025-05-05).

Abstract

In Influence Maximization (IM), the objective is to -- given a budget -- select the optimal set of entities in a network to target with a treatment so as to maximize the total effect. For instance, in marketing, the objective is to target the set of customers that maximizes the total response rate, resulting from both direct treatment effects on targeted customers and indirect, spillover, effects that follow from targeting these customers. Recently, new methods to estimate treatment effects in the presence of network interference have been proposed. However, the issue of how to leverage these models to make better treatment allocation decisions has been largely overlooked. Traditionally, in Uplift Modeling (UM), entities are ranked according to estimated treatment effect, and the top entities are allocated treatment. Since, in a network context, entities influence each other, the UM ranking approach will be suboptimal. The problem of finding the optimal treatment allocation in a network setting is \textcolor{red}{NP-hard,} and generally has to be solved heuristically. To fill the gap between IM and UM, we propose OTAPI: Optimizing Treatment Allocation in the Presence of Interference to find solutions to the IM problem using treatment effect estimates. OTAPI consists of two steps. First, a causal estimator is trained to predict treatment effects in a network setting. Second, this estimator is leveraged to identify an optimal treatment allocation by integrating it into classic IM algorithms. We demonstrate that this novel method outperforms classic IM and UM approaches on both synthetic and semi-synthetic datasets.

View on arXivView PDF

Code Implementations(1)

daan-caljon/OTAPICommunity100%
MIT
12PythonMay 7, 20241 years agoMIT

Similar Papers

Cite this paper

@article{caljon2024optimizing,
  title  = {Optimizing Treatment Allocation in the Presence of Interference},
  author = {Daan Caljon and Jente Van Belle and Jeroen Berrevoets and Wouter Verbeke},
  year   = {2024},
  eprint = {2410.00075},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2410.00075v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2410.00075.svg)](https://codeofpaper.com/paper/2410.00075)

Discussion