Conference and Journal Papers

2025

  1. TAPNext: Tracking Any Point (TAP) as Next Token Prediction
    , Carl Doersch, Yi Yang, Skanda Koppula, Viorica Patraucean, Xu Owen He, Ignacio Rocco, Mehdi S. M. Sajjadi, , and Ross Goroshin
    International Conference on Computer Vision (ICCV), 2025.
    #Other
    [website], [arXiv], [code], [huggingface], [YouTube]

  2. Manifold Metric: A Loss Landscape Approach for Predicting Model Performance
    , , Ariside Baratin, , and
    Conference on Lifelong Learning Agents (CoLLAs), 2025.
    #DL
    [arXiv]

  3. Combining Domain and Alignment Vectors Provides Better Knowledge-Safety Trade-offs in LLMs
    , , Matthew Riemer, Pin-Yu Chen, Amal Zouaq, Payel Das, and
    Annual Meeting of the Association for Computational Linguistics (ACL), 2025.
    #NLP
    [arXiv]

  4. Small Encoders Can Rival Large Decoders in Detecting Groundedness
    , , , Fernando Rodriguez, Alaa Boukhary, Adam Elwood, and
    Findings of the Association for Computational Linguistics (ACL), 2025.
    #NLP

  5. Do Robot Snakes Dream like Electric Sheep? Investigating the Effects of Architectural Inductive Biases on Hallucination
    , Prasanna Parthasarathi, Mehdi Rezagholizadeh, Boxing Chen, and
    Findings of the Association for Computational Linguistics (ACL), 2025.
    #NLP
    [arXiv]

  6. IDAT: A Multi-Modal Dataset and Toolkit for Building and Evaluating Interactive Task-Solving Agents
    Shrestha Mohanty, Negar Arabzadeh, Andrea Tupini, Yuxuan Sun, Alexey Skrynnik, , Marc-Alexandre Côté, and Julia Kiseleva
    ACM SIGIR Conference on Research and Development in Information Retrieval, 2025.
    #NLP
    [arXiv]

  7. NeoBERT: A Next Generation BERT
    , , Mariam El Mezouar, and
    Transactions on Machine Learning Research (TMLR), 2025.
    #NLP
    [openreview]

  8. A Generalist Hanabi Agent
    , , , Miao Liu, , and
    International Conference on Learning Representations (ICLR), 2025.
    #RL, #Other
    [website], [openreview], [code]

  9. BindGPT: A Scalable Framework for 3D Molecular Design via Language Modeling and Reinforcement Learning
    , Maksim Kuznetsov, Roman Schutski, Rim Shayakhmetov, Daniil Polykovskiy, , and Alex Zhavoronkov
    AAAI Conference on Artificial Intelligence (AAAI), 2025. [Best poster award]
    #DL, #RL
    [arXiv], [website], [YouTube]

  10. ChartGemma: Visual Instruction-tuning for Chart Reasoning in the Wild
    Ahmed Masry*, , Aayush Bajaj, Aaryaman Kartha, Enamul Hoque, and Shafiq Joty
    International Conference on Computational Linguistics (COLING) Industry Track, 2025.
    #NLP
    [acl], [arXiv], [code]

2024

  1. WorkArena++: Towards Compositional Planning and Reasoning-based Common Knowledge Work Tasks
    Leo Boisvert*, , Maxime Gasse, Massimo Caccia, Thibault Le Sellier De Chezelles, Quentin Cappart, Nicolas Chapados, Alexandre Lacoste, and Alexandre Drouin
    Conference on Neural Information Processing Systems (NeurIPS) Datasets and Benchmarks Track, 2024.
    #NLP
    [openreview], [arXiv], [code]

  2. Balancing Context Length and Mixing Times for Reinforcement Learning at Scale
    Matthew Riemer, Khimya Khetarpal, , and
    Conference on Neural Information Processing Systems (NeurIPS), 2024.
    #RL
    [openreview]

  3. Exploring Quantization for Efficient Pre-Training of Transformer Language Models
    , , , and
    Findings of the Association for Computational Linguistics (EMNLP), 2024.
    #NLP, #DL
    [acl], [arXiv]

  4. Context-Aware Assistant Selection for Improved Inference Acceleration with Large Language Models
    , Prasanna Parthasarathi, Mehdi Rezagholizadeh, and
    Conference on Empirical Methods in Natural Language Processing (EMNLP), 2024.
    #NLP
    [acl], [arXiv]

  5. Do Large Language Models Know How Much They Know?
    , , Prasanna Parthasarathi, Shagun Sodhani, and
    Conference on Empirical Methods in Natural Language Processing (EMNLP), 2024.
    #NLP
    [acl]

  6. Sharpness-Aware Minimization Scaled by Outlier Normalization for Robust DNNs on In-Memory Computing Accelerators
    Sébastien Henwood, , Yvon Savaria, , and François Leduc-Primeau
    Asilomar Conference on Signals, Systems, and Computers, 2024.
    [Conference on Lifelong Learning Agents (CoLLAs) Workshop Track, 2022]
    [Edge Intelligence Workshop (EIW), 2022]
    #DL
    [arXiv]

  7. Toward Debugging Deep Reinforcement Learning Programs with RLExplorer
    Rached Bouchoucha, Ahmed Haj Yahmed, , , Amin Nikanjam, , and Foutse Khomh
    International Conference on Software Maintenance and Evolution (ICSME), 2024.
    #RL
    [arXiv]

  8. Should We Attend More or Less? Modulating Attention for Fairness
    , , Samira Shabanian, and
    Conference on Language Modeling (COLM), 2024.
    #NLP
    [openreview], [arXiv]

  9. Are self-explanations from Large Language Models faithful?
    , , and Siva Reddy
    Findings of the Association for Computational Linguistics (ACL), 2024.
    #NLP
    [acl], [arXiv], [code], [YouTube]

  10. A deep-dive into the tradeoffs of preference alignment with PEFT
    , , Matthew Riemer, Pin-Yu Chen, Amal Zouaq, Payel Das, and
    Annual Meeting of the Association for Computational Linguistics (ACL), 2024.
    #NLP
    [acl], [arXiv]

  11. Why Don’t Prompt-Based Fairness Metrics Correlate?
    , , Ioana Baldini, and
    Annual Meeting of the Association for Computational Linguistics (ACL), 2024.
    #NLP
    [acl], [arXiv], [YouTube]

  12. Sub-goal Distillation: A Method to Improve Small Language Agents
    , Elias Stengel-Eskin, , and Marc-Alexandre Cote
    Conference on Lifelong Learning Agents (CoLLAs), 2024. [Oral presentation.]
    #RL, #NLP
    [arXiv]

  13. Partial Models for Building Adaptive Model-Based Reinforcement Learning Agents
    Safa Alver, , and Doina Precup
    Conference on Lifelong Learning Agents (CoLLAs), 2024.
    #RL
    [arXiv]

  14. Lookbehind-SAM: k steps back, 1 step forward
    , , Aristide Baratin, and
    International Conference on Machine Learning (ICML), 2024.
    #DL
    [pmlr], [arXiv], [code], [YouTube]

  15. Faithfulness Measurable Masked Language Models
    , Siva Reddy, and
    International Conference on Machine Learning (ICML), 2024. [Spotlight award - top 3.5%]
    #NLP
    [pmlr], [arXiv], [code], [YouTube], [blogpost]

  16. Contrast-agnostic Spinal Cord Segmentation: A Comparative Study of ConvNets and Vision Transformers
    , Sandrine Bedard, Jan Valosek, , and Julien Cohen-Adad
    Medical Imaging with Deep Learning (MIDL), 2024.
    #DL, #Other
    [openreview]

  17. Promoting Exploration in Memory-Augmented Adam using Critical Momenta
    , , Aristide Baratin, Reza Babanezhad Harikandeh, , Simon Lacoste-Julien, Razvan Pascanu, and
    Transactions on Machine Learning Research (TMLR), 2024.
    #DL
    [openreview], [arXiv]

  18. A Responsible Framework for Applying Artificial Intelligence on Medical Images and Signals at the Point-of-care: the PACS-AI Platform
    Pascal Theriault-Lauzier, Denis Cobin, Olivier Tastet, Elodie Labrecque Langlais, Bahareh Taji, Guson Kang, Aun-Yeong Chong, Derek So, An Tang, Judy Wawira Gichoya, , Pierre-Luc Déziel, Julie G Hussin, Samuel Kadoury, and Robert Avram
    Canadian Journal of Cardiology, 2024.
    #DL, #Other
    [paper]

  19. MVP: Minimal Viable Phrase for Long Text Understanding
    , Amal Zouaq, and
    Joint International Conference on Computational Linguistics, Language, Resources and Evaluation (LREC-COLING), 2024.
    #NLP
    [acl]

  20. Mastering Memory Tasks with World Models
    , , , and
    International Conference on Learning Representations (ICLR), 2024. [Oral presentation.]
    #RL, #DL
    [openreview], [arXiv]

  21. Intelligent Switching for Reset-Free RL
    , , Glen Berseth, and
    International Conference on Learning Representations (ICLR), 2024.
    #RL
    [openreview], [arXiv]

  22. On the Costs and Benefits of Adopting Lifelong Learning for Software Analytics - Empirical Study on Brown Build and Risk Prediction
    Doriane Olewicki, Sarra Habchi, Mathieu Nayrolles, , , and Bram Adams
    International Conference on Software Engineering (ICSE) - Software Engineering in Practice Track, 2024. [ICSE24 SEIP Distinguished Paper Award.]
    #DL
    [arXiv]

  23. Fast and Accurate Output Error Estimation for Memristor-Based Deep Neural Networks
    Jonathan Kern, Sébastien Henwood, , Elsa Dupraz, Abdeldjalil Aïssa-El-Bey, Yvon Savaria, and François Leduc-Primeau
    IEEE Transactions on Signal Processing, 2024.
    #DL
    [paper]

  24. Fairness-Aware Structured Pruning in Transformers
    , , Samira Shabanian, Ioana Baldini, and
    AAAI Conference on Artificial Intelligence (AAAI), 2024.
    #NLP
    [aaai], [arXiv], [YouTube]

  25. Learning Conditional Policies for Crystal Design Using Offline Reinforcement Learning
    , Santiago Miret, , Mariano Phielipp, , and
    Digital Discovery Journal, 2024.
    #RL
    [paper]

2023

  1. Self-Influence Guided Data Reweighting for Language Model Pre-training
    , Tolga Bolukbasi, Sriram Ganapathy, Shikhar Vashishth, , and Partha Talukdar
    Conference on Empirical Methods in Natural Language Processing (EMNLP), 2023.
    #NLP
    [acl], [openreview], [arXiv]

  2. EpiK-Eval: Evaluation for Language Models as Epistemic Models
    , , Prasanna Parthasarathi, Shagun Sodhani, and
    Conference on Empirical Methods in Natural Language Processing (EMNLP), 2023.
    #NLP
    [acl], [openreview], [arXiv], [code]

  3. Measuring the Knowledge Acquisition-Utilization Gap in Pretrained Language Models
    Amirhossein Kazemnejad, Mehdi Rezagholizadeh, Prasanna Parthasarathi, and
    Findings of the Association for Computational Linguistics (EMNLP), 2023.
    #NLP
    [acl], [openreview], [arXiv]

  4. Training DNNs Resilient to Adversarial and Random Bit-Flips by Learning Quantization Ranges
    , , Jean Pierre David, and François Leduc-Primeau
    Transactions on Machine Learning Research (TMLR), 2023.
    #DL
    [openreview], [code]

  5. Replay Buffer with Local Forgetting for Adapting to Local Environment Changes in Deep Model-Based Reinforcement Learning
    , , Ida Momennejad, Harm van Seijen, and
    Conference on Lifelong Learning Agents (CoLLAs), 2023.
    [Deep Reinforcement Learning Workshop, NeurIPS, 2022]
    #RL
    [pmlr], [arXiv]

  6. Towards Few-shot Coordination: Revisiting Ad-hoc Teamplay Challenge In the Game of Hanabi
    , , , Miao Liu, and
    Conference on Lifelong Learning Agents (CoLLAs), 2023.
    #RL
    [pmlr], [arXiv]

  7. Dealing With Non-stationarity in Decentralized Cooperative Multi-Agent Deep Reinforcement Learning via Multi-Timescale Learning
    , , Amit Sinha, Mohammad Amini, , Aditya Mahajan, and
    Conference on Lifelong Learning Agents (CoLLAs), 2023.
    #RL
    [pmlr], [arXiv]

  8. Conditionally Optimistic Exploration for Cooperative Deep Multi-Agent Reinforcement Learning
    , Yangchen Pan, Chenjun Xiao, , and
    Conference on Uncertainty in Artificial Intelligence (UAI), 2023.
    #RL
    [pmlr], [arXiv]

  9. An Empirical Investigation of the Role of Pre-training in Lifelong Learning
    Sanket Vaibhav Mehta, , , and Emma Strubell
    Journal of Machine Learning Research, 2023.
    #DL
    [jmlr], [arXiv]

  10. Multi-Agent Reinforcement Learning for Fast-Timescale Demand Response of Residential Loads
    Vincent Mai, Philippe Maisonneuve, Tianyu Zhang, , Liam Paull, and Antoine Lesage-Landry
    International Conference on Autonomous Agents and Multiagent Systems (AAMAS), 2023.
    #RL
    [arXiv]

  11. Deep Learning on a Healthy Data Diet: Finding Important Examples for Fairness
    , Prasanna Parthasarathi, , Hamid Palangi, Samira Shabanian, and
    AAAI Conference on Artificial Intelligence (AAAI), 2023.
    #NLP
    [aaai], [arXiv], [YouTube]

  12. DEUP: Direct Epistemic Uncertainty Prediction
    Moksh Jain, Salem Lahlou, , Victor Butoi, Paul Bertin, , Maksym Korablyov, and Yoshua Bengio
    Transactions on Machine Learning Research (TMLR), 2023.
    #DL
    [openreview], [arXiv], [code]

  13. Label fusion and training methods for reliable representation of inter-rater uncertainty
    Andreanne Lemay, Charley Gros, , and Julien Cohen-Adad
    The Journal of Machine Learning for Biomedical Imaging (MELBA), 2023.
    #DL
    [paper]

2022

  1. Evaluating the Faithfulness of Importance Measures in NLP by Recursively Masking Allegedly Important Tokens and Retraining
    , Nicholas Meade, Vaibhav Adlakha, and Siva Reddy
    Findings of the Association for Computational Linguistics (EMNLP), 2022.
    [BlackboxNLP Workshop, 2022]
    #NLP
    [acl], [arXiv], [code]

  2. Detecting Languages Unintelligible to Multilingual Models through Local Structure Probes
    , Prasanna Parthasarathi, Amal Zouaq, and
    Findings of the Association for Computational Linguistics (EMNLP), 2022.
    #NLP
    [acl]

  3. Local Structure Matters Most in Most Languages
    , Prasanna Parthasarathi, Amal Zouaq, and
    Conference of the Asia-Pacific Chapter of the Association for Computational Linguistics and the International Joint Conference on Natural Language Processing (AACL-IJCNLP), 2022.
    #NLP
    [acl]

  4. TAG: Task-based Accumulated Gradients for Lifelong Learning
    , Balaraman Ravindran, and
    Conference on Lifelong Learning Agents (CoLLAs), 2022.
    [Workshop on Theory and Foundation of Continual Learning, ICML, 2021]
    #DL
    [pmlr], [arXiv], [code]

  5. Improving Meta-Learning Generalization with Activation-Based Early-Stopping
    , Christopher Pal, , and
    Conference on Lifelong Learning Agents (CoLLAs), 2022.
    #DL
    [pmlr], [arXiv], [code], [YouTube]

  6. Combining Reinforcement Learning and Constraint Programming for Sequence-Generation Tasks with Hard Constraints
    , , and Gilles Pesant
    International Conference on Principles and Practice of Constraint Programming (CP), 2022.
    #RL
    [paper]

  7. Biological Sequence Design with GFlowNets
    Moksh Jain, Emmanuel Bengio, Alex-Hernandez Garcia, , Bonaventure F. P. Dossou, Chanakya Ekbote, Jie Fu, Tianyu Zhang, Micheal Kilgour, Dinghuai Zhang, Lena Simine, Payel Das, and Yoshua Bengio
    International Conference on Machine Learning (ICML), 2022.
    #DL
    [pmlr], [arXiv], [code]

  8. Towards Evaluating Adaptivity of Model-Based Reinforcement Learning Methods
    , , , Ida Momennejad, , and Harm van Seijen
    International Conference on Machine Learning (ICML), 2022.
    #RL
    [pmlr], [arXiv], [code]

  9. Post-hoc Interpretability for Neural NLP: A Survey
    , Siva Reddy, and
    ACM Computing Surveys, 2022.
    #NLP
    [acm], [arXiv]

  10. Local Structure Matters Most: Perturbation Study in NLU
    , Prasanna Parthasarathi, Amal Zouaq, and
    Findings of the Association for Computational Linguistics (ACL), 2022.
    #NLP
    [acl], [arXiv]

  11. Memory Augmented Optimizers for Deep Learning
    , Prasanna Parthasarathi, Mido Assran, and
    International Conference on Learning Representations (ICLR), 2022.
    #DL
    [openreview], [arXiv], [code]

  12. PatchUp: A Feature-Space Block-Level Regularization Technique for Convolutional Neural Networks
    , Mohammad Amini, , Vikas Verma, and
    AAAI Conference on Artificial Intelligence (AAAI), 2022.
    #DL
    [aaai], [arXiv], [code]

2021

  1. Benchmarking Bias Mitigation Algorithms in Representation Learning through Fairness Metrics
    , Deepak Sharma, Soroush Mehri, Adriana Romero, Samira Shabanian, and Sina Honari
    Conference on Neural Information Processing Systems (NeurIPS) Datasets and Benchmarks Track, 2021.
    #NLP
    [neurips], [openreview], [code]

  2. A Survey of Data Augmentation Approaches for NLP
    Steven Y. Feng, Varun Gangal, Jason Wei, , Soroush Vosoughi, Teruko Mitamura, and Eduard Hovy
    Findings of the Association for Computational Linguistics (ACL-IJCNLP), 2021.
    #NLP
    [acl], [arXiv]

  3. MLMLM: Link Prediction with Mean Likelihood Masked Language Model
    , Philippe Trempe, Amal Zouaq, and
    Findings of the Association for Computational Linguistics (ACL-IJCNLP), 2021.
    #NLP
    [acl], [arXiv]

  4. Continuous Coordination As a Realistic Scenario for Lifelong Learning
    , , Aaron Courville, and
    International Conference on Machine Learning (ICML), 2021.
    #RL
    [pmlr], [arXiv], [code]

  5. A Brief Study on the Effects of Training Generative Dialogue Models with a Semantic loss
    Prasanna Parthasarathi, , Joelle Pineau, and
    Proceedings of the 22nd Annual SIGdial Meeting on Discourse and Dialogue, 2021.
    #NLP
    [acl]

  6. Do Encoder Representations of Generative Dialogue Models Encode Sufficient Information about the Task ?
    Prasanna Parthasarathi, , and Joelle Pineau
    Proceedings of the 22nd Annual SIGdial Meeting on Discourse and Dialogue, 2021.
    #NLP
    [acl]

  7. IIRC: Incremental Implicitly-Refined Classification
    , , Shagun Sodhani, and
    IEEE CVF Conference on Computer Vision and Pattern Recognition (CVPR), 2021.
    #DL
    [paper], [arXiv], [code], [website], [PyPI], [docs]

  8. Towered Actor Critic for Handling Multiple Action Types in Reinforcement Learning For Drug Discovery
    Sai Krishna Gottipati, Yashaswi Pathak, Boris Sattarov, Sahir, Rohan Nuttall, Mohammad Amini, Matthew E. Taylor, and
    AAAI Conference on Artificial Intelligence (AAAI), 2021.
    #RL
    [aaai]

2020

  1. The LoCA Regret: A Consistent Metric to Evaluate Model-Based Behavior in Reinforcement Learning
    Harm van Seijen, , , and
    Conference on Neural Information Processing Systems (NeurIPS), 2020.
    #RL
    [neurips], [arXiv], [code]

  2. Learning To Navigate The Synthetically Accessible Chemical Space Using Reinforcement Learning
    Sai Krishna Gottipati*, Boris Sattarov*, Sufeng Niu, Yashaswi Pathak, Haoran Wei, Shengchao Liu, Karam MJ Thomas, Simon Blackburn, Connor W Coley, Jian Tang, , and Yoshua Bengio
    International Conference on Machine Learning (ICML), 2020.
    #RL
    [pmlr], [arXiv]