CATE with continuous treatment and categorical outcome

[hadi-gharibi]

I was trying to use DML for continuous treatment (price) and binary outcome(churn). Based on the docs, its not possible to use any of these techniques to this case. Is there any way to adjust any of these algorithms for this setup? If there is a paper that I could develop and add to this library I'm down to it as well.

[PhilippBach]

Dear @hadi-gharibi ,

thanks for opening this issue. Here you can find two references:

• PLR with binary outcome (logistic regression): https://academic.oup.com/ectj/article/24/3/559/6296639 . I think there's also R code available ; for heterogeneous you'd have to specify interaction terms in the regression model, but I'm not really sure if you can really interpret them as CATEs
• IRM with continuous treatment : https://arxiv.org/abs/2004.03036 ; I think using that paper, which is pretty advanced, it should also be possible to combine the output with the current CATE implementation as of https://academic.oup.com/ectj/article/24/2/264/5899048

I'm not really sure if I understand your question right:

Do you rather want to estimate dose-response relationships for continuous treatments and binary outcomes or do you really want to estimate CATEs after such an estimation. I think the interpretation can be a bit tricky in the latter case, maybe @SvenKlaassen has some thoughts on this

[hadi-gharibi]

Really good point. Maybe calculating CATE is not necessary after all estimating dose-response relationships would be sufficient enough.
Let me explain my problem in more detail.I was calculating the treatment as a percentage increase from the current price (imagine some personalized subscription price). For example, 0.2 means 20% increase. In this case, explaining CATE should be fine, but after thinking about your comment, I think it might not be important after all. Estimating dose-response relationships should be good enough.

Since I’m new to the topic, could you please tell me if DML is even the correct tool for me given this problem? Am I on the right track?

[SvenKlaassen]

Yes, I think DML would be suitable to estimate dose-response relationships.
But currently, we have no model which specifically enforces the combination of continous treatment and binary outcomes.
I think reference Double/debiased machine learning for logistic partially linear model posted by @PhilippBach seems really helpful for this setting.

[hadi-gharibi]

Thanks for the response @PhilippBach and @SvenKlaassen
Since I need this, I have to make it work, one way or another and the setup of this problem is really common in the industry I belive it would be a nice feature. If you think it would fit in DML library setup, I can give it a shot.
I can expand the DoubleMLPLR to support continuous treatment and binary outcomes, but I think It needs a different _check_learner and lots of if and else here and there. I don't think it would be a good approach.

Another way is to have a separate class to support that, like DoubleMLLPLR for the logistic partially linear model. it would work just fine.

Yet another idea could be to keep the DoubleMLPLR as a strategy class that picks the correct approach( linear or logistic class) based on the dataset.

[SvenKlaassen]

That would be really great.
I would also agree that a separate class DoubleMLLPLR would be the better choice.
Especially as the class would be more complicated due to the nested type of cross-fitting structure.