Integrate LabelTree with SGD