← Search

Xuheng Li

6 accepted papers

2026

Best-of-Majority: Minimax-Optimal Strategy for Pass@k Inference Scaling

ICLR 2026poster

LLM inference often generates a batch of candidates for a prompt and selects one via strategies like majority voting or Best-of- N (BoN). For difficult tasks, this single-shot selection often underperforms. Consequently, evaluations commonly report Pass@$k$: the agent may submit up to $k$ responses,…

Cited by 0SourceScholar
2026

Dimension-Independent Convergence of Underdamped Langevin Monte Carlo in KL Divergence

ICML 2026poster

Underdamped Langevin dynamics (ULD) is a widely-used sampler for Gibbs distributions $\pi\propto e^{-V}$, and is often empirically effective in high dimensions. However, existing non-asymptotic convergence guarantees for discretized ULD typically scale polynomially with the ambient dimension $d$, le…

Cited by 0SourceScholar
2025

Automated Proof Generation for Rust Code via Self-Evolution

ICLR 2025poster

Ensuring correctness is crucial for code generation. Formal verification offers a definitive assurance of correctness, but demands substantial human effort in proof construction and hence raises a pressing need for automation. The primary obsta- cle lies in the severe lack of data—there is much fewe…

Cited by 4SourcePDFScholar
2024

Risk Bounds of Accelerated SGD for Overparameterized Linear Regression

ICLR 2024poster

Accelerated stochastic gradient descent (ASGD) is a workhorse in deep learning and often achieves better generalization performance than SGD. However, existing optimization theory can only explain the faster convergence of ASGD, but cannot explain its better generalization. In this paper, we study t…

Cited by 3SourcePDFScholar