KL Divergence Comparison for MedMCQA (averaged over 2 runs):
+----------------+---------------------+-----------------------+
| Method         | Average KL (Prob)   | Average KL (Argmax)   |
+================+=====================+=======================+
| Token Dropping | 1.17e-01 ± 0.00e+00 | 7.50e-01 ± 0.00e+00   |
+----------------+---------------------+-----------------------+