KL Divergence Comparison for MRI (averaged over 1 runs):
+---------------------------+---------------------+-----------------------+
| Method                    | Average KL (Prob)   | Average KL (Argmax)   |
+===========================+=====================+=======================+
| Original                  | 4.59e-02 ± 0.00e+00 | 1.77e-01 ± 0.00e+00   |
+---------------------------+---------------------+-----------------------+
| MCal_CE (Cross-Entropy)   | 6.73e-06 ± 0.00e+00 | 4.29e-03 ± 0.00e+00   |
+---------------------------+---------------------+-----------------------+
| PatchCutout-trained Model | 4.31e-03 ± 0.00e+00 | 2.29e-02 ± 0.00e+00   |
+---------------------------+---------------------+-----------------------+