Supplement to " Learning Individualized Treatment Rules with Many Treatments: A Supervised Clustering Approach Using Adaptive Fusion "

Haixu Ma Department of Statistics and Operations Research University of North Carolina at Chapel Hill Chapel Hill, NC 27516 haixuma@live.unc.edu A.1 Estimation of the main effect We briefly discuss how to obtain the estimation of the main effect function M For nonparametric regression, we follow [ 3 ] to divide the training data into M folds based on the assigned treatment. Then p E r Y |Z,A " a s is obtained from the regression forest [ 4 ] on Y Z with the dataset tp y We refer to [ 3 ] for more discussions about the case of misspecifying the main effect, and the corresponding robust and efficient method to solve the misspecification problem. A.2 Implementation details for the adaptive proximal gradient algorithm Recall that U " diag pX The main steps of the proposed algorithm for SCAF are summarized as below. In particular, the experiments were run on a Linux-based computing server.