Neural Supervised Clustering for Learning Individualized Treatment Rules with Many Treatments

Daixi Huang, Yixin Yuan, Yifan Dai +1

Learning an individualized treatment rule (ITR) with many treatments requires estimating treatment-specific effects from limited observations per arm. Treatments with similar condi…