Dedieu, Antoine

14 publications

TMLR 2025 Diffusion Model Predictive Control Guangyao Zhou, Sivaramakrishnan Swaminathan, Rajkumar Vasudeva Raju, J Swaroop Guntupalli, Wolfgang Lehrach, Joseph Ortiz, Antoine Dedieu, Miguel Lazaro-Gredilla, Kevin Patrick Murphy

ICML 2025 Improving Transformer World Models for Data-Efficient RL Antoine Dedieu, Joseph Ortiz, Xinghua Lou, Carter Wendelken, J Swaroop Guntupalli, Wolfgang Lehrach, Miguel Lazaro-Gredilla, Kevin Patrick Murphy

ICLRW 2025 Improving Transformer World Models for Data-Efficient RL Antoine Dedieu, Joseph Ortiz, Xinghua Lou, Carter Wendelken, Wolfgang Lehrach, J Swaroop Guntupalli, Miguel Lazaro-Gredilla, Kevin Patrick Murphy

NeurIPS 2024 DMC-VB: A Benchmark for Representation Learning for Control with Visual Distractors Joseph Ortiz, Antoine Dedieu, Wolfgang Lehrach, J. Swaroop Guntupalli, Carter Wendelken, Ahmad Humayun, Guangyao Zhou, Sivaramakrishnan Swaminathan, Miguel Lázaro-Gredilla, Kevin Murphy

ICML 2024 Learning Cognitive Maps from Transformer Representations for Efficient Planning in Partially Observed Environments Antoine Dedieu, Wolfgang Lehrach, Guangyao Zhou, Dileep George, Miguel Lazaro-Gredilla

MLOSS 2024 PGMax: Factor Graphs for Discrete Probabilistic Graphical Models and Loopy Belief Propagation in JAX Guangyao Zhou, Antoine Dedieu, Nishanth Kumar, Wolfgang Lehrach, Shrinu Kushagra, Dileep George, Miguel Lázaro-Gredilla

ICML 2023 Learning Noisy or Bayesian Networks with Max-Product Belief Propagation Antoine Dedieu, Guangyao Zhou, Dileep George, Miguel Lazaro-Gredilla

NeurIPS 2023 Schema-Learning and Rebinding as Mechanisms of In-Context Learning and Emergence Sivaramakrishnan Swaminathan, Antoine Dedieu, Rajkumar Vasudeva Raju, Murray Shanahan, Miguel Lazaro-Gredilla, Dileep George

JMLR 2022 Solving L1-Regularized SVMs and Related Linear Programs: Revisiting the Effectiveness of Column and Constraint Generation Antoine Dedieu, Rahul Mazumder, Haoyue Wang

JMLR 2021 Learning Sparse Classifiers: Continuous and Mixed Integer Optimization Perspectives Antoine Dedieu, Hussein Hazimeh, Rahul Mazumder

NeurIPS 2021 Perturb-and-Max-Product: Sampling and Learning in Discrete Energy-Based Models Miguel Lazaro-Gredilla, Antoine Dedieu, Dileep George

AAAI 2021 Query Training: Learning a Worse Model to Infer Better Marginals in Undirected Graphical Models with Hidden Variables Miguel Lázaro-Gredilla, Wolfgang Lehrach, Nishad Gothoskar, Guangyao Zhou, Antoine Dedieu, Dileep George

AAAI 2021 Sample-Efficient L0-L2 Constrained Structure Learning of Sparse Ising Models Antoine Dedieu, Miguel Lázaro-Gredilla, Dileep George

AISTATS 2019 Error Bounds for Sparse Classifiers in High-Dimensions Antoine Dedieu