PPaperPicks

Yoshua Bengio

University of Montréal, Department of Computer Science and Operations Research, QC, Canada

51 papers at tracked venues · 45 at CORE A* · active 2024–2026

Venues

Frequent coauthors

Nikolay Malkin
×15
Minsu Kim
×10
Moksh Jain
×7
Emmanuel Bengio
×6
Dinghuai Zhang
×6
Tianyu Zhang
×5
Sungjin Ahn
×4
Jarrid Rector-Brooks
×4
Jinkyoo Park
×4
Alexander Tong
×4
Marcin Sendera
×4
Pablo Lemos
×4

Papers

  1. Extendable Planning via Multiscale Diffusion
    AAAI 2026 ·
    Chang Chen
    ,
    Hany Hamed
    ,
    Doojin Baek
    ,
    Taegu Kang
    ,
    Samyeul Noh
    , Yoshua Bengio,
    Sungjin Ahn
  2. AI for Global Climate Cooperation: Modeling Global Climate Negotiations, Agreements, and Long-Term Cooperation in RICE-N
    ICML 2025 ·
    Tianyu Zhang
    ,
    Andrew Robert Williams
    ,
    Phillip Wozny
    ,
    Kai-Hendrik Cohrs
    ,
    Koen Ponse
    ,
    Marco Jiralerspong
    ,
    Soham R. Phade
    ,
    Sunil Srinivasa
    ,
    Lu Li
    ,
    Yang Zhang
    ,
    Prateek Gupta
    ,
    Erman Acar
    ,
    Irina Rish
    , Yoshua Bengio,
    Stephan Zheng
  3. Action abstractions for amortized sampling
    ICLR 2025 ·
    Oussama Boussif
    ,
    Léna Néhale Ezzine
    ,
    Joseph D. Viviano
    ,
    Michal Koziarski
    ,
    Moksh Jain
    ,
    Nikolay Malkin
    ,
    Emmanuel Bengio
    ,
    Rim Assouel
    , Yoshua Bengio
  4. Adaptive Inference-Time Scaling via Cyclic Diffusion Search
    NeurIPS 2025 ·
    Gyubin Lee
    ,
    Bao Truong
    ,
    Jaesik Yoon
    ,
    Dongwoo Lee
    ,
    Minsu Kim
    , Yoshua Bengio,
    Sungjin Ahn
  5. Adaptive teachers for amortized samplers
    ICLR 2025 ·
    Minsu Kim
    ,
    Sanghyeok Choi
    ,
    Taeyoung Yun
    ,
    Emmanuel Bengio
    ,
    Leo Feng
    ,
    Jarrid Rector-Brooks
    ,
    Sungsoo Ahn
    ,
    Jinkyoo Park
    ,
    Nikolay Malkin
    , Yoshua Bengio
  6. AlignVLM: Bridging Vision and Language Latent Spaces for Multimodal Document Understanding
    NeurIPS 2025 ·
    Ahmed Masry
    ,
    Juan A. Rodríguez
    ,
    Tianyu Zhang
    ,
    Suyuchen Wang
    ,
    Chao Wang
    ,
    Aarash Feizi
    ,
    Akshay Kalkunte Suresh
    ,
    Abhay Puri
    ,
    Xiangru Jian
    ,
    Pierre-André Noël
    ,
    Sathwik Tejaswi Madhusudhan
    ,
    Marco Pedersoli
    ,
    Bang Liu
    ,
    Nicolas Chapados
    , Yoshua Bengio,
    Enamul Hoque
    ,
    Chris Pal
    ,
    Issam Hadj Laradji
    ,
    David Vázquez
    ,
    Perouz Taslakian
    ,
    Spandana Gella
    ,
    Sai Rajeswar Mudumba
  7. Ant Colony Sampling with GFlowNets for Combinatorial Optimization
    AISTATS 2025 ·
    Minsu Kim
    ,
    Sanghyeok Choi
    ,
    Hyeonah Kim
    ,
    Jiwoo Son
    ,
    Jinkyoo Park
    , Yoshua Bengio
  8. AssembleFlow: Rigid Flow Matching with Inertial Frames for Molecular Assembly
    ICLR 2025 ·
    Hongyu Guo
    , Yoshua Bengio,
    Shengchao Liu
  9. Bringing SAM to new heights: leveraging elevation data for tree crown segmentation from drone imagery
    NeurIPS 2025 ·
    Mélisande Teng
    ,
    Arthur Ouaknine
    ,
    Etienne Laliberté
    , Yoshua Bengio,
    David Rolnick
    ,
    Hugo Larochelle
  10. Can a Bayesian Oracle Prevent Harm from an Agent?
    UAI 2025 · Yoshua Bengio,
    Michael K. Cohen
    ,
    Nikolay Malkin
    ,
    Matt MacDermott
    ,
    Damiano Fornasiere
    ,
    Pietro Greiner
    ,
    Younesse Kaddar
  11. EarthView: A Large Scale Remote Sensing Dataset for Self-Supervision
    WACV 2025 ·
    Diego A. Velázquez
    ,
    Pau Rodríguez López
    ,
    Sergio Alonso
    ,
    Josep M. Gonfaus
    ,
    Jordi Gonzàlez
    ,
    Gerardo Richarte
    ,
    Javier Marín
    , Yoshua Bengio,
    Alexandre Lacoste
  12. Efficient Diversity-Preserving Diffusion Alignment via Gradient-Informed GFlowNets
    ICLR 2025 ·
    Zhen Liu
    ,
    Tim Z. Xiao
    ,
    Weiyang Liu
    , Yoshua Bengio,
    Dinghuai Zhang
  13. Fast Monte Carlo Tree Diffusion: 100× Speedup via Parallel and Sparse Planning
    NeurIPS 2025 ·
    Jaesik Yoon
    ,
    Hyeonseo Cho
    , Yoshua Bengio,
    Sungjin Ahn
  14. Geometric Signatures of Compositionality Across a Language Model's Lifetime
    ACL 2025 ·
    Jin Hwa Lee
    ,
    Thomas Jiralerspong
    ,
    Lei Yu
    , Yoshua Bengio,
    Emily Cheng
  15. HarmAug: Effective Data Augmentation for Knowledge Distillation of Safety Guard Models
    ICLR 2025 ·
    Seanie Lee
    ,
    Haebin Seong
    ,
    Dong Bok Lee
    ,
    Minki Kang
    ,
    Xiaoyin Chen
    ,
    Dominik Wagner
    , Yoshua Bengio,
    Juho Lee
    ,
    Sung Ju Hwang
  16. Learning Diverse Attacks on Large Language Models for Robust Red-Teaming and Safety Tuning
    ICLR 2025 ·
    Seanie Lee
    ,
    Minsu Kim
    ,
    Lynn Cherif
    ,
    David Dobre
    ,
    Juho Lee
    ,
    Sung Ju Hwang
    ,
    Kenji Kawaguchi
    ,
    Gauthier Gidel
    , Yoshua Bengio,
    Nikolay Malkin
    ,
    Moksh Jain
  17. MAP: Low-compute Model Merging with Amortized Pareto Fronts via Quadratic Approximation
    ICLR 2025 ·
    Lu Li
    ,
    Tianyu Zhang
    ,
    Zhiqi Bu
    ,
    Suyuchen Wang
    ,
    Huan He
    ,
    Jie Fu
    ,
    Yonghui Wu
    ,
    Jiang Bian
    ,
    Yong Chen
    , Yoshua Bengio
  18. Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold
    ICLR 2025 ·
    Lazar Atanackovic
    ,
    Xi Zhang
    ,
    Brandon Amos
    ,
    Mathieu Blanchette
    ,
    Leo J. Lee
    , Yoshua Bengio,
    Alexander Tong
    ,
    Kirill Neklyudov
  19. Monte Carlo Tree Diffusion for System 2 Planning
    ICML 2025 ·
    Jaesik Yoon
    ,
    Hyeonseo Cho
    ,
    Doojin Baek
    , Yoshua Bengio,
    Sungjin Ahn
  20. On the Transfer of Object-Centric Representation Learning
    ICLR 2025 ·
    Aniket Rajiv Didolkar
    ,
    Andrii Zadaianchuk
    ,
    Anirudh Goyal
    ,
    Michael Curtis Mozer
    , Yoshua Bengio,
    Georg Martius
    ,
    Maximilian Seitzer
  21. Outsourced Diffusion Sampling: Efficient Posterior Inference in Latent Spaces of Generative Models
    ICML 2025 ·
    Siddarth Venkatraman
    ,
    Mohsin Hasan
    ,
    Minsu Kim
    ,
    Luca Scimeca
    ,
    Marcin Sendera
    , Yoshua Bengio,
    Glen Berseth
    ,
    Nikolay Malkin
  22. RL, but don't do anything I wouldn't do
    UAI 2025 ·
    Michael K. Cohen
    ,
    Marcus Hutter
    , Yoshua Bengio,
    Stuart Russell
  23. Rejecting Hallucinated State Targets during Planning
    ICML 2025 ·
    Harry Zhao
    ,
    Tristan Sylvain
    ,
    Romain Laroche
    ,
    Doina Precup
    , Yoshua Bengio
  24. Structure Language Models for Protein Conformation Generation
    ICLR 2025 ·
    Jiarui Lu
    ,
    Xiaoyin Chen
    ,
    Stephen Zhewen Lu
    ,
    Chence Shi
    ,
    Hongyu Guo
    , Yoshua Bengio,
    Jian Tang
  25. Towards Improving Exploration through Sibling Augmented GFlowNets
    ICLR 2025 ·
    Kanika Madan
    ,
    Alex Lamb
    ,
    Emmanuel Bengio
    ,
    Glen Berseth
    , Yoshua Bengio
  26. Towards a Formal Theory of Representational Compositionality
    ICML 2025 ·
    Eric Elmoznino
    ,
    Thomas Jiralerspong
    , Yoshua Bengio,
    Guillaume Lajoie
  27. Trajectory Balance with Asynchrony: Decoupling Exploration and Learning for Fast, Scalable LLM Post-Training
    NeurIPS 2025 ·
    Brian R. Bartoldson
    ,
    Siddarth Venkatraman
    ,
    James Diffenderfer
    ,
    Moksh Jain
    ,
    Tal Ben-Nun
    ,
    Seanie Lee
    ,
    Minsu Kim
    ,
    Johan S. Obando-Ceron
    , Yoshua Bengio,
    Bhavya Kailkhura
  28. VCR: A Task for Pixel-Level Complex Reasoning in Vision Language Models via Restoring Occluded Text
    ICLR 2025 ·
    Tianyu Zhang
    ,
    Suyuchen Wang
    ,
    Lu Li
    ,
    Ge Zhang
    ,
    Perouz Taslakian
    ,
    Sai Rajeswar
    ,
    Jie Fu
    ,
    Bang Liu
    , Yoshua Bengio
  29. Amortizing intractable inference in diffusion models for vision, language, and control
    NeurIPS 2024 ·
    Siddarth Venkatraman
    ,
    Moksh Jain
    ,
    Luca Scimeca
    ,
    Minsu Kim
    ,
    Marcin Sendera
    ,
    Mohsin Hasan
    ,
    Luke Rowe
    ,
    Sarthak Mittal
    ,
    Pablo Lemos
    ,
    Emmanuel Bengio
    ,
    Alexandre Adam
    ,
    Jarrid Rector-Brooks
    , Yoshua Bengio,
    Glen Berseth
    ,
    Nikolay Malkin
  30. Amortizing intractable inference in large language models
    ICLR 2024 ·
    Edward J. Hu
    ,
    Moksh Jain
    ,
    Eric Elmoznino
    ,
    Younesse Kaddar
    ,
    Guillaume Lajoie
    , Yoshua Bengio,
    Nikolay Malkin
  31. Consciousness-Inspired Spatio-Temporal Abstractions for Better Generalization in Reinforcement Learning
    ICLR 2024 ·
    Harry Zhao
    ,
    Safa Alver
    ,
    Harm van Seijen
    ,
    Romain Laroche
    ,
    Doina Precup
    , Yoshua Bengio
  32. Cycle Consistency Driven Object Discovery
    ICLR 2024 ·
    Aniket Rajiv Didolkar
    ,
    Anirudh Goyal
    , Yoshua Bengio
  33. Delta-AI: Local objectives for amortized inference in sparse graphical models
    ICLR 2024 ·
    Jean-Pierre R. Falet
    ,
    Hae Beom Lee
    ,
    Nikolay Malkin
    ,
    Chen Sun
    ,
    Dragos Secrieru
    ,
    Dinghuai Zhang
    ,
    Guillaume Lajoie
    , Yoshua Bengio
  34. Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory optimization
    ICLR 2024 ·
    Dinghuai Zhang
    ,
    Ricky T. Q. Chen
    ,
    Cheng-Hao Liu
    ,
    Aaron C. Courville
    , Yoshua Bengio
  35. Discrete Probabilistic Inference as Control in Multi-path Environments
    UAI 2024 ·
    Tristan Deleu
    ,
    Padideh Nouri
    ,
    Nikolay Malkin
    ,
    Doina Precup
    , Yoshua Bengio
  36. Expected flow networks in stochastic environments and two-player zero-sum games
    ICLR 2024 ·
    Marco Jiralerspong
    ,
    Bilun Sun
    ,
    Danilo Vucetic
    ,
    Tianyu Zhang
    , Yoshua Bengio,
    Gauthier Gidel
    ,
    Nikolay Malkin
  37. Improved off-policy training of diffusion samplers
    NeurIPS 2024 ·
    Marcin Sendera
    ,
    Minsu Kim
    ,
    Sarthak Mittal
    ,
    Pablo Lemos
    ,
    Luca Scimeca
    ,
    Jarrid Rector-Brooks
    ,
    Alexandre Adam
    , Yoshua Bengio,
    Nikolay Malkin
  38. Improving Gradient-Guided Nested Sampling for Posterior Inference
    ICML 2024 ·
    Pablo Lemos
    ,
    Nikolay Malkin
    ,
    Will Handley
    , Yoshua Bengio,
    Yashar Hezaveh
    ,
    Laurence Perreault Levasseur
  39. Iterated Denoising Energy Matching for Sampling from Boltzmann Densities
    ICML 2024 ·
    Tara Akhound-Sadegh
    ,
    Jarrid Rector-Brooks
    ,
    Avishek Joey Bose
    ,
    Sarthak Mittal
    ,
    Pablo Lemos
    ,
    Cheng-Hao Liu
    ,
    Marcin Sendera
    ,
    Siamak Ravanbakhsh
    ,
    Gauthier Gidel
    , Yoshua Bengio,
    Nikolay Malkin
    ,
    Alexander Tong
  40. Learning to Scale Logits for Temperature-Conditional GFlowNets
    ICML 2024 ·
    Minsu Kim
    ,
    Joohwan Ko
    ,
    Taeyoung Yun
    ,
    Dinghuai Zhang
    ,
    Ling Pan
    ,
    Woochang Kim
    ,
    Jinkyoo Park
    ,
    Emmanuel Bengio
    , Yoshua Bengio
  41. Local Search GFlowNets
    ICLR 2024 ·
    Minsu Kim
    ,
    Taeyoung Yun
    ,
    Emmanuel Bengio
    ,
    Dinghuai Zhang
    , Yoshua Bengio,
    Sungsoo Ahn
    ,
    Jinkyoo Park
  42. Memory Efficient Neural Processes via Constant Memory Attention Block
    ICML 2024 ·
    Leo Feng
    ,
    Frederick Tung
    ,
    Hossein Hajimirsadeghi
    , Yoshua Bengio,
    Mohamed Osama Ahmed
  43. Metacognitive Capabilities of LLMs: An Exploration in Mathematical Problem Solving
    NeurIPS 2024 ·
    Aniket Didolkar
    ,
    Anirudh Goyal
    ,
    Nan Rosemary Ke
    ,
    Siyuan Guo
    ,
    Michal Valko
    ,
    Timothy P. Lillicrap
    ,
    Danilo Jimenez Rezende
    , Yoshua Bengio,
    Michael C. Mozer
    ,
    Sanjeev Arora
  44. Object centric architectures enable efficient causal representation learning
    ICLR 2024 ·
    Amin Mansouri
    ,
    Jason S. Hartford
    ,
    Yan Zhang
    , Yoshua Bengio
  45. PhyloGFN: Phylogenetic inference with generative flow networks
    ICLR 2024 ·
    Ming-Yang Zhou
    ,
    Zichao Yan
    ,
    Elliot Layne
    ,
    Nikolay Malkin
    ,
    Dinghuai Zhang
    ,
    Moksh Jain
    ,
    Mathieu Blanchette
    , Yoshua Bengio
  46. Pre-Training and Fine-Tuning Generative Flow Networks
    ICLR 2024 ·
    Ling Pan
    ,
    Moksh Jain
    ,
    Kanika Madan
    , Yoshua Bengio
  47. RGFN: Synthesizable Molecular Generation Using GFlowNets
    NeurIPS 2024 ·
    Michal Koziarski
    ,
    Andrei Rekesh
    ,
    Dmytro Shevchuk
    ,
    Almer van der Sloot
    ,
    Piotr Gainski
    , Yoshua Bengio,
    Cheng-Hao Liu
    ,
    Mike Tyers
    ,
    Robert A. Batey
  48. Regeneration Learning: A Learning Paradigm for Data Generation
    AAAI 2024 ·
    Xu Tan
    ,
    Tao Qin
    ,
    Jiang Bian
    ,
    Tie-Yan Liu
    , Yoshua Bengio
  49. Simulation-Free Schrödinger Bridges via Score and Flow Matching
    AISTATS 2024 ·
    Alexander Tong
    ,
    Nikolay Malkin
    ,
    Kilian Fatras
    ,
    Lazar Atanackovic
    ,
    Yanlei Zhang
    ,
    Guillaume Huguet
    ,
    Guy Wolf
    , Yoshua Bengio
  50. Trajectory Flow Matching with Applications to Clinical Time Series Modelling
    NeurIPS 2024 ·
    Xi Zhang
    ,
    Yuan Pu
    ,
    Yuki Kawamura
    ,
    Andrew Loza
    , Yoshua Bengio,
    Dennis L. Shung
    ,
    Alexander Tong
  51. Tree Cross Attention
    ICLR 2024 ·
    Leo Feng
    ,
    Frederick Tung
    ,
    Hossein Hajimirsadeghi
    , Yoshua Bengio,
    Mohamed Osama Ahmed