Ujjwal Jain

Trainable Calibration Measures for Neural Networks from Kernel Mean Embeddings

Modern neural networks have recently been found to be poorly calibrated, primarily in the direction of over-confidence. Methods like entropy penalty and temperature smoothing improve calibration by clamping confidence, but in doing so compromise the many legitimately confident predictions. We propos…