← Search

Mengxin Yu

3 accepted papers

2026

Statistical Early Stopping for Reasoning Models

ICML 2026poster

While LLMs have seen substantial improvement in reasoning capabilities, they also sometimes overthink, generating unnecessary reasoning steps, particularly under uncertainty, given ill-posed or ambiguous queries. We introduce statistically principled early stopping methods that monitor uncertainty s…

Cited by 0SourceScholar
2025

Foundations of Top-$k$ Decoding for Language Models

NeurIPS 2025poster

Top-$k$ decoding is a widely used method for sampling from LLMs: at each token, only the largest $k$ next-token-probabilities are kept, and the next token is sampled after re-normalizing them to sum to unity. Top-$k$ and other sampling methods are motivated by the intuition that true next-token dist…

Cited by 0SourceScholar
2024

Uncertainty in Language Models: Assessment through Rank-Calibration

EMNLP 2024main

Language Models (LMs) have shown promising performance in natural language generation. However, as LMs often generate incorrect or hallucinated responses, it is crucial to correctly quantify their uncertainty in responding to given inputs. In addition to verbalized confidence elicited via prompting,…