As mentioned in section 2.2, top-K s responses from K l -width beam search benefit from properties of higher probability and diversity.
k here is set as k s , so that a partial response with positive future reward is also associated with above desirable properties.