← Search

Krunoslav Lehman Pavasovic

4 accepted papers

2026

Overshoot and Shrinkage in Classifier-Free Guidance: From Theory to Practice

ICLR 2026poster

Classifier-Free Guidance (CFG) is widely used in diffusion and flow-based generative models for high-quality conditional generation, yet its theoretical properties remain incompletely understood. By connecting CFG to the high-dimensional framework of diffusion regimes, we show that in sufficiently h…

Cited by 0SourceScholar
2025

A Differentiable Rank-Based Objective for Better Feature Learning

ICLR 2025poster

In this paper, we leverage existing statistical methods to better understand feature learning from data. We tackle this by modifying the model-free variable selection method, Feature Ordering by Conditional Independence (FOCI), which is introduced in Azadkia & Chatterjee (2021). While FOCI is based…

Cited by 0SourcePDFScholar
2023

Approximate Heavy Tails in Offline (Multi-Pass) Stochastic Gradient Descent

NeurIPS 2023spotlight

A recent line of empirical studies has demonstrated that SGD might exhibit a heavy-tailed behavior in practical settings, and the heaviness of the tails might correlate with the overall performance. In this paper, we investigate the emergence of such heavy tails. Previous works on this problem only…

2023

MARS: Meta-learning as Score Matching in the Function Space

ICLR 2023top-25%

Meta-learning aims to extract useful inductive biases from a set of related datasets. In Bayesian meta-learning, this is typically achieved by constructing a prior distribution over neural network parameters. However, specifying families of computationally viable prior distributions over the high-di…