Nikolay Malkin

Action abstractions for amortized sampling ICLR 2025 Adaptive teachers for amortized samplers ICLR 2025 Can a Bayesian Oracle Prevent Harm from an Agent? UAI 2025 Fast Flow-based Visuomotor Policies via Conditional Optimal Transport Couplings CoRL 2025 Learning Diverse Attacks on Large Language Models for Robust Red-Teaming and Safety Tuning ICLR 2025 Mixtures of In-Context Learners ACL 2025 Outsourced Diffusion Sampling: Efficient Posterior Inference in Latent Spaces of Generative Models ICML 2025 PQMass: Probabilistic Assessment of the Quality of Generative Models using Probability Mass Estimation ICLR 2025 Amortizing intractable inference in diffusion models for vision, language, and control NeurIPS 2024 Amortizing intractable inference in large language models ICLR 2024 Delta-AI: Local objectives for amortized inference in sparse graphical models ICLR 2024 Discrete Probabilistic Inference as Control in Multi-path Environments UAI 2024 Expected flow networks in stochastic environments and two-player zero-sum games ICLR 2024 Improved off-policy training of diffusion samplers NeurIPS 2024 Improving Gradient-Guided Nested Sampling for Posterior Inference ICML 2024 Iterated Denoising Energy Matching for Sampling from Boltzmann Densities ICML 2024 PhyloGFN: Phylogenetic inference with generative flow networks ICLR 2024 Simulation-Free Schrödinger Bridges via Score and Flow Matching AISTATS 2024 A theory of continuous generative flow networks ICML 2023 Better Training of GFlowNets with Local Credit and Incomplete Trajectories ICML 2023 GFlowNet-EM for Learning Compositional Latent Variable Models ICML 2023 GFlowNets and variational inference ICLR 2023 GFlowOut: Dropout with Generative Flow Networks ICML 2023 Joint Bayesian Inference of Graphical Structure and Parameters with a Single Generative Flow Network NeurIPS 2023 Learning GFlowNets From Partial Episodes For Improved Convergence And Stability ICML 2023 Let the Flows Tell: Solving Graph Combinatorial Problems with GFlowNets NeurIPS 2023 ThinkSum: Probabilistic reasoning over sets using large language models ACL 2023 Coherence boosting: When your pretrained language model is not paying enough attention ACL 2022 Diffusion Models as Plug-and-Play Priors NeurIPS 2022 Generative Flow Networks for Discrete Probabilistic Modeling ICML 2022 Resolving label uncertainty with implicit posterior models UAI 2022 Trajectory balance: Improved credit assignment in GFlowNets NeurIPS 2022 GPT Perdetry Test: Generating new meanings for new words NAACL 2021 Studying word order through iterative shuffling EMNLP 2021 Mining self-similarity: Label super-resolution with epitomic representations ECCV 2020