BENK: The Beran Estimator with Neural Kernels for Estimating the Heterogeneous Treatment Effect
Not on the map. Why?
Abstract (arXiv)
A method for estimating the conditional average treatment effect under condition of censored time-to-event data called BENK (the Beran Estimator with Neural Kernels) is proposed. The main idea behind the method is to apply the Beran estimator for estimating the survival functions of controls and treatments. Instead of typical kernel functions in the Beran estimator, it is proposed to implement kernels in the form of neural networks of a specific form called the neural kernels. The conditional average treatment effect is estimated by using the survival functions as outcomes of the control and treatment neural networks which consists of a set of neural kernels with shared parameters. The neural kernels are more flexible and can accurately model a complex location structure of feature vectors. Various numerical simulation experiments illustrate BENK and compare it with the well-known T-learner, S-learner and X-learner for several types of the control and treatment outcome functions based on the Cox models, the random survival forest and the Nadaraya-Watson regression with Gaussian kernels. The code of proposed algorithms implementing BENK is available in https://github.com/Stasychbr/BENK.
Key facts. Cited by 1 paper in the map (data through 14 Oct 2026). Builds on 48 of its 100 references that are in the map. First posted on arXiv 19 Nov 2022.
Builds on
48 of 100 references.
References · 48
- 2015Estimation and Inference of Heterogeneous Treatment Effects using Random Forestscited by 600
- 2017Meta-learners for Estimating Heterogeneous Treatment Effects using Machine Learningcited by 478
- 2016DeepSurv: Personalized Treatment Recommender System Using A Cox Proportional Hazards Deep Neural Networkcited by 358
- 2016Generalized Random Forestscited by 312
- 2015Recursive Partitioning for Heterogeneous Causal Effectscited by 255
- 2021Nonparametric Estimation of Heterogeneous Treatment Effects: From Theory to Learning Algorithmscited by 144
- 2019An Attentive Survey of Attention Modelscited by 124
- 2022Causal Transformer for Estimating Counterfactual Outcomescited by 123
- 2017Machine Learning for Survival Analysis: A Surveycited by 112
- 2021Dive into Deep Learningcited by 111
- 2019Learning Counterfactual Representations for Estimating Individual Dose-Response Curvescited by 90
- 2020Estimating the Effects of Continuous-valued Interventions using Generative Adversarial Networkscited by 78
36 more
- 2016Deep Survival Analysiscited by 61
- 2021VCNet and Functional Targeted Regularization For Learning Causal Effects of Continuous Treatmentscited by 61
- 2017Some methods for heterogeneous treatment effect estimation in high-dimensionscited by 48
- 2021Conditional Distributional Treatment Effect with Kernel Conditional Mean Embeddings and U-Statistic Regressioncited by 40
- 2015Residual Weighted Learning for Estimating Individualized Treatment Rulescited by 38
- 2020A unified survey of treatment effect heterogeneity modeling and uplift modelingcited by 36
- 2022Combining Observational and Randomized Data for Estimating Heterogeneous Treatment Effectscited by 32
- 2020Estimating heterogeneous treatment effects with right-censored data via causal survival forestscited by 29
- 2021On Locality of Local Explanation Modelscited by 25
- 2021SurvITE: Learning Heterogeneous Treatment Effects from Time-to-Event Datacited by 25
- 2022Exploring Transformer Backbones for Heterogeneous Treatment Effect Estimationcited by 25
- 2016Unbiased split variable selection for random survival forests using maximally selected rank statisticscited by 24
- 2017Estimating Individual Treatment Effect in Observational Data Using Random Forest Methodscited by 21
- 2019DNNSurv: Deep Neural Networks for Survival Analysis Using Pseudo Valuescited by 15
- 2021NCoRE: Neural Counterfactual Representation Learning for Combinations of Treatmentscited by 14
- 2018Transfer Learning for Estimating Causal Effects using Neural Networkscited by 13
- 2021Multi-Source Causal Inference Using Control Variatescited by 13
- 2022Counterfactual Phenotyping with Censored Time-to-Eventscited by 13
- 2020Estimating heterogeneous survival treatment effect in observational data using machine learningcited by 12
- 2019Targeted Estimation of Heterogeneous Treatment Effect in Observational Survival Analysiscited by 9
- 2021CETransformer: Casual Effect Estimation via Transformer Based Representation Learningcited by 8
- 2018Learning Data-adaptive Nonparametric Kernelscited by 7
- 2022BITES: Balanced Individual Treatment Effect for Survival datacited by 7
- 2022Efficient Heterogeneous Treatment Effect Estimation With Multiple Experiments and Multiple Outcomescited by 6
- 2019Learning from Irregularly Sampled Data for Endomicroscopy Super-resolution: A Comparative Study of Sparse and Dense Approachescited by 5
- 2022AGBoost: Attention-based Modification of Gradient Boosting Machinecited by 5
- 2018A Fundamental Measure of Treatment Effect Heterogeneitycited by 4
- 2022Heterogeneous Treatment Effect with Trained Kernels of the Nadaraya-Watson Regressioncited by 4
- 2018Heterogeneous Treatment Effect Estimation through Deep Learningcited by 3
- 2019On the estimation of average treatment effects with right-censored time to event outcome and competing riskscited by 3
- 2020Enabling Counterfactual Survival Analysis with Balanced Representationscited by 3
- 2021Dimension-Free Average Treatment Effect Inference with Deep Neural Networkscited by 3
- 2022Individual Treatment Effect Estimation Through Controlled Neural Network Training in Two Stagescited by 3
- 2022Causal Inference from Small High-dimensional Datasetscited by 3
- 2022Learning to Infer Counterfactuals: Meta-Learning for Estimating Multiple Imbalanced Treatment Effectscited by 2
- 2021Ranking of average treatment effects with generalized random forests for time-to-event outcomescited by 1
Built on by
Citing papers per year: 2026: 1.
1 paper.