speed ups for training data removal