← Search

Udaya Ghai

7 accepted papers

2025

How Does Critical Batch Size Scale in Pre-training?

ICLR 2025poster

Training large-scale models under given resources requires careful design of parallelism strategies. In particular, the efficiency notion of critical batch size (CBS), concerning the compromise between time and compute, marks the threshold beyond which greater data parallelism leads to diminishing r…

2025

Mind the Gap: Examining the Self-Improvement Capabilities of Large Language Models

ICLR 2025oral

Self-improvement is a mechanism in Large Language Model (LLM) pre-training, post-training and test-time inference. We explore a framework where the model verifies its own outputs, filters or reweights data based on this verification, and distills the filtered data. Despite several empirical success…

Cited by 4SourcePDFScholar
2023

Online Nonstochastic Model-Free Reinforcement Learning

NeurIPS 2023poster

We investigate robust model-free reinforcement learning algorithms designed for environments that may be dynamic or even adversarial. Traditional state-based policies often struggle to accommodate the challenges imposed by the presence of unmodeled disturbances in such settings. Moreover, optimizing…

Cited by 11SourcePDFScholar