3 papers
stat.ME2026
Generalized Hierarchical Conformal Prediction
Soham Mallick, Eric Tchetgen Tchetgen, Edgar Dobriban +1
Many prediction problems arise with data collected in groups. In this setting, hierarchical conformal prediction (HCP) (Lee et al., 2026) provides distribution-free prediction sets…
cs.AI2026
Statistical Early Stopping for Reasoning Models
Yangxinyu Xie, Tao Wang, Soham Mallick +5
While LLMs have seen substantial improvement in reasoning capabilities, they also sometimes overthink, generating unnecessary reasoning steps, particularly under uncertainty, given…
cs.AI2025
Foundations of Top- Decoding For Language Models
Georgy Noarov, Soham Mallick, Tao Wang +5
Top- decoding is a widely used method for sampling from LLMs: at each token, only the largest next-token-probabilities are kept, and the next token is sampled after re-norma…