Time After Time: Deep-Q Effect Estimation for Interventions on When and What to do

Publication date

2025-01-22

Authors

Wald, Yoav
Goldstein, Mark
Efroni, Yonathan
van Amsterdam, Wouter A.C.ORCID 0000-0002-3181-0810
Ranganath, Rajesh

Editors

Advisors

Supervisors

DOI

Document Type

Part of book

Collections

Open Access logo

License

taverne

Abstract

Problems in fields such as healthcare, robotics, and finance requires reasoning about the value both of what decision or action to take and when to take it. The prevailing hope is that artificial intelligence will support such decisions by estimating the causal effect of policies such as how to treat patients or how to allocate resources over time. However, existing methods for estimating the effect of a policy struggle with irregular time. They either discretize time, or disregard the effect of timing policies. We present a new deep-Q algorithm that estimates the effect of both when and what to do called Earliest Disagreement Q-Evaluation (EDQ). EDQ makes use of recursion for the Q-function that is compatible with flexible sequence models, such as transformers. EDQ provides accurate estimates under standard assumptions. We validate the approach through experiments on survival time and tumor growth tasks.

Keywords

Taverne, Language and Linguistics, Computer Science Applications, Education, Linguistics and Language

Citation

Wald, Y, Goldstein, M, Efroni, Y, van Amsterdam, W A C & Ranganath, R 2025, Time After Time : Deep-Q Effect Estimation for Interventions on When and What to do. in 13th International Conference on Learning Representations, ICLR 2025. 13th International Conference on Learning Representations, ICLR 2025, International Conference on Learning Representations, ICLR, pp. 10422-10435, 13th International Conference on Learning Representations, ICLR 2025, Singapore, Singapore, 24/04/25. < https://openreview.net/forum?id=5yDS32hKJc >, conference