← Search

Jatin Prakash

3 accepted papers

2026

KL-Regularized Reinforcement Learning is Designed to Mode Collapse

ICLR 2026poster

Classical intuitions cast minimizing reverse KL as "mode seeking" and forward KL as "mass covering". In KL-regularized reinforcement learning, however, the regularizer determines _both_ the target distribution's shape _and_ the divergence being implicitly minimized, making its role more nuanced than…

Cited by 0SourceScholar
2024

Enhancing Tail Performance in Extreme Classifiers by Label Variance Reduction

ICLR 2024poster

Extreme Classification (XC) architectures, which utilize a massive One-vs-All (OvA) classifier layer at the output, have demonstrated remarkable performance on problems with large label sets. Nonetheless, these architectures falter on tail labels with few representative samples. This phenomenon has…

Cited by 7SourcePDFScholar
2022

A Stitch in Time Saves Nine: A Train-Time Regularizing Loss for Improved Neural Network Calibration

CVPR 2022oral

Deep Neural Networks (DNNs) are known to make overconfident mistakes, which makes their use problematic in safety-critical applications. State-of-the-art (SOTA) calibration techniques improve on the confidence of predicted labels alone, and leave the confidence of non-max classes (e.g. top-2, top-5)…

Cited by 60PDFcodeScholar