TLRNet: Estimating Individual Treatment Effect based on Local Information and Single Learner Structure

Abstract:Causal inference has become a central issue across various fields, including computer science, statistics, economics, education, healthcare, and medicine. The broad applicability of this discipline has garnered increased research funding and attention. In recent years, the estimation of causal effects from observational data has gained traction due to the vast amounts of collected data and the lower costs compared to randomized controlled trials. Advances in causal effect estimation methods have enhanced service personalization tools. For instance, these tools can help identify the most effective type of treatment (considering both cost and success rate) for each patient among different medical service options. This paper proposes an innovative method for estimating the heterogeneity of treatment effects. The structure of the proposed model is based on a deep neural network and a pseudo-single learner. The proposed method has been compared with other state-of-the-art methods on the IHDP benchmark. Acceptable results have been obtained by using one estimator to estimate the potential outcomes of two treatment groups. Accordingly, this paves the way for further development and improvement of the proposed method.

7 pages, 1 figure, 2 tables. This is the author-accepted manuscript of the paper presented at the 2nd International Conference on Artificial Intelligence and Software Engineering (AI-SOFT 2024), Shiraz University, Iran

Machine Learning (stat.ML); Machine Learning (cs.LG)

arXiv:2607.22762 [stat.ML]

(or arXiv:2607.22762v1 [stat.ML] for this version)

https://doi.org/10.48550/arXiv.2607.22762 arXiv-issued DOI via DataCite (pending registration)