← Search

Nikita Kotelevskii

6 accepted papers

2026

Neural Optimal Transport Meets Multivariate Conformal Prediction

ICLR 2026poster

We propose a framework for conditional vector quantile regression (CVQR) that combines neural optimal transport with amortized optimization, and apply it to multivariate conformal prediction. Classical quantile regression does not extend naturally to multivariate responses, while existing approaches…

Cited by 0SourceScholar
2026

Position: Agentic AI systems should be making Bayes-consistent decisions

ICML 2026poster

LLMs excel at predictive tasks and complex reasoning tasks, but many high-value deployments rely on decisions under uncertainty, for example, which tool to call, which expert to consult, or how many resources to invest. While the usefulness and feasibility of Bayesian approaches remain unclear for L…

Cited by 0SourceScholar
2025

From Risk to Uncertainty: Generating Predictive Uncertainty Measures via Bayesian Estimation

ICLR 2025poster

There are various measures of predictive uncertainty in the literature, but their relationships to each other remain unclear. This paper uses a decomposition of statistical pointwise risk into components associated with different sources of predictive uncertainty: namely, aleatoric uncertainty (inhe…

Cited by 1SourcePDFScholar
2024

Dirichlet-based Uncertainty Quantification for Personalized Federated Learning with Improved Posterior Networks

IJCAI 2024poster

In modern federated learning, one of the main challenges is to account for inherent heterogeneity and the diverse nature of data distributions for different clients. This problem is often addressed by introducing personalization of the models towards the data distribution of the particular client. H…

Cited by 7SourcePDFScholar
2024

Efficient Conformal Prediction under Data Heterogeneity

AISTATS 2024poster

Conformal prediction (CP) stands out as a robust framework for uncertainty quantification, which is crucial for ensuring the reliability of predictions. However, common CP methods heavily rely on the data exchangeability, a condition often violated in practice. Existing approaches for tackling non-e…

Cited by 4SourcePDFScholar
2021

Monte Carlo Variational Auto-Encoders

ICML 2021spotlight

Variational auto-encoders (VAE) are popular deep latent variable models which are trained by maximizing an Evidence Lower Bound (ELBO). To obtain tighter ELBO and hence better variational approximations, it has been proposed to use importance sampling to get a lower variance estimate of the evidence…