Recovery using Equation 9 is more efficient than the method in Section 5.3, as we recover \(K\) logits \(z_{i_{1}},z_{i_{2}},\cdots,z_{i_{K}}\) rather than just \(K-1\) logits. However, if \(B\) is large, numerical stability may be an issue. (And, if \(B\) is small, the logit bias may be insufficient to force the API to output the desired tokens by placing them in the top \(K\).) Specifically, as \(B\to\infty\), we have \((1-e^{-B})\sum_{j}^{K}\exp(a_{i_{j}}(z,b))\to 1\) and so the logarithm in Equation 9 tends to \(\log(1-1)=-\infty\); this means we may have catastrophic cancellation.