Modify the probability calculation to accelerate the algorithm