arXiv Machine Learning June 15, 2026 By Lev V. Utkin, Andrei V. Konstantinov, Stanislav K. Kogan, Natalya M. Verbova, Maksim I. Goriunov

Attention-Based Estimation of the Individual Treatment Benefit Probability under Dose Variation

Read the original on arXiv Machine Learning →

arXiv:2606. 13821v1 Announce Type: new Abstract: Estimating the probability that a treatment outperforms a control for an individual patient, called the Individual Probability of Treatment Benefit (IPTB), offers a clinically intuitive alternative to population-average metrics.

Summary generated by The Flow from the publisher's feed. The full article lives at arXiv Machine Learning.

arXiv Machine Learning

Aug 7

Surv-IPTB: An Attention-Based Model for Estimating Individual Probability of Treatment Benefit with Survival Data

arXiv:2608. 06288v1 Announce Type: new Abstract: This work presents a novel attention-based framework for estimating the Individual Probability of Treatment Benefit (IPTB) in survival analysis contexts.

By Lev V. Utkin, Stanislav K. Kogan, Andrei V. Konstantinov

arXiv Machine Learning

Jun 3

Tailoring Strictly Proper Scoring Rules for Downstream Tasks: An Application to Causal Inference

arXiv:2606. 03332v1 Announce Type: new Abstract: Probabilistic models are typically trained using task-agnostic objectives like log-loss, which can lead to significant errors in downstream estimation.

By Roman Plaud, Alexandre Perez-Lebel, Antoine Saillenfest, Thomas Bonald, Marine Le Morvan, Ga\"el Varoquaux, Matthieu Labeau

benchmarks safety

arXiv Machine Learning

Jul 30

Uncertainty-Guided LLM Semantic Augmentation for Heterogeneous Treatment Effect Estimation

arXiv:2607. 26599v1 Announce Type: new Abstract: Estimating heterogeneous treatment effects is central to targeted interventions, such as personalized promotions and precision medicine.

By Jialu Xu, Mengkun Liang, Guannan Liu, Xiaojie Mao, Junjie Wu

llms benchmarks

arXiv Machine Learning

Jul 28

MOCA: A Transformer-based Modular Causal Inference Framework with One-way Cross-attention and Cutting Feedback

arXiv:2604. 23107v2 Announce Type: replace-cross Abstract: Causal effect estimation from observational data requires careful adjustment for confounding.

By Lei Wang, Debashis Ghosh

llms agents benchmarks

arXiv AI

Jul 29

Annotation-Assisted Learning of Treatment Policies From Multimodal Electronic Health Records

arXiv:2507. 20993v4 Announce Type: replace-cross Abstract: We study how to learn treatment policies from multimodal electronic health records (EHRs) that consist of tabular data and clinical text.

By Henri Arno, Thomas Demeester

multimodal

arXiv Machine Learning

Jul 10

Improving RCT-Based Treatment Effect Estimation Under Covariate Mismatch via Calibrated Alignment

arXiv:2603. 19186v3 Announce Type: replace Abstract: Randomized controlled trials (RCTs) are the gold standard for estimating treatment effects, yet they are often underpowered for detecting effect heterogeneity.

By Amir Asiaee, Samhita Pal

rag safety

Related stories

Surv-IPTB: An Attention-Based Model for Estimating Individual Probability of Treatment Benefit with Survival Data

Tailoring Strictly Proper Scoring Rules for Downstream Tasks: An Application to Causal Inference

Uncertainty-Guided LLM Semantic Augmentation for Heterogeneous Treatment Effect Estimation

MOCA: A Transformer-based Modular Causal Inference Framework with One-way Cross-attention and Cutting Feedback

Annotation-Assisted Learning of Treatment Policies From Multimodal Electronic Health Records

Improving RCT-Based Treatment Effect Estimation Under Covariate Mismatch via Calibrated Alignment