COSMIT + DOC modified_huber probabilities in SGD