If you look at the examples in Google’s Nature paper, their examples boosts the 4th choice, “durian”, to the same likelihood as the 2nd the 3rd choices, “lychee” and “papaya”, even though the rankings from the LLM probability distribution (PLM) were:
mango 0.50
lychee 0.30
papaya .015
durian 0.05