← Search

Gurpreet Gosal

4 accepted papers

2026

Don't Drop Dropout: Optimizing Layer Sparsity for Efficient LLM Training and Inference

ICML 2026poster

Layer dropout (a.k.a.\ stochastic depth) has been shown to enable faster training, higher accuracy, and robustness to zero-shot layer pruning in both language and vision transformers. However, as models and datasets have scaled, dropout---particularly layer dropout---has largely disappeared from LLM…

Cited by 0SourceScholar
2026

Scaling with Collapse: Efficient and Predictable Training of LLM Families

ICLR 2026poster

Effective LLM training relies on *consistency*, meaning that key quantities—such as final losses and optimal hyperparameters—scale predictably across model sizes. Qiu et al. (2025) recently showed that this consistency extends beyond scalars: whole training loss curves can *collapse* onto a universa…

Cited by 0SourceScholar
2025

Power Lines: Scaling laws for weight decay and batch size in LLM pre-training

NeurIPS 2025poster

Efficient LLM pre-training requires well-tuned hyperparameters (HPs), including learning rate η and weight decay λ. We study scaling laws for HPs: formulas for how to scale HPs as we scale model size N, dataset size D, and batch size B. Recent work suggests the AdamW timescale, τ = B/(ηλD), should…

Cited by 0SourceScholar
2025

Straight to Zero: Why Linearly Decaying the Learning Rate to Zero Works Best for LLMs

ICLR 2025poster

LLMs are commonly trained with a learning rate (LR) warmup, followed by cosine decay to 10% of the maximum (10x decay). In a large-scale empirical study, we show that under an optimal peak LR, a simple linear decay-to-zero (D2Z) schedule consistently outperforms other schedules when training at comp…

Cited by 2SourcePDFScholar