| 2025 | ICML | Improving Transformer World Models for Data-Efficient RL. | Antoine Dedieu, Joseph Ortiz, Xinghua Lou, Carter Wendelken, J. Swaroop Guntupalli, Wolfgang Lehrach, Miguel Lzaro-Gredilla, Kevin Patrick Murphy |
| 2024 | ICML | Learning Cognitive Maps from Transformer Representations for Efficient Planning in Partially Observed Environments. | Antoine Dedieu, Wolfgang Lehrach, Guangyao Zhou, Dileep George, Miguel Lzaro-Gredilla |
| 2023 | ICML | Learning Noisy OR Bayesian Networks with Max-Product Belief Propagation. | Antoine Dedieu, Guangyao Zhou, Dileep George, Miguel Lzaro-Gredilla |
| 2021 | AAAI | Sample-Efficient L0-L2 Constrained Structure Learning of Sparse Ising Models. | Antoine Dedieu, Miguel Lzaro-Gredilla, Dileep George |
| 2021 | AAAI | Query Training: Learning a Worse Model to Infer Better Marginals in Undirected Graphical Models with Hidden Variables. | Miguel Lzaro-Gredilla, Wolfgang Lehrach, Nishad Gothoskar, Guangyao Zhou, Antoine Dedieu, Dileep George |
| 2019 | AISTATS | Error bounds for sparse classifiers in high-dimensions. | Antoine Dedieu |
| 2018 | CIKM | Hierarchical Modeling and Shrinkage for User Session LengthPrediction in Media Streaming. | Antoine Dedieu, Rahul Mazumder, Zhen Zhu, Hossein Vahabi |