Publications

Ouail Kitouni, Niklas Nolte, Diane Bouchacourt, Adina Williams, Mike Rabbat, Mark Ibrahim (2024). The Factorization Curse: Which Tokens You Predict Underlie the Reversal Curse and More.

ArXiv

Ouail Kitouni, Niklas Nolte, Victor Samuel Perez-Diaz, Sokratis Trifinopoulos, Mike Williams (2024). From Neurons to Neutrons: A Case Study in Interpretability.

ArXiv

Hansley Narasiah, Ouail Kitouni, Andrea Scorsoglio, Bernd Sturdza, Shawn Hatcher, Dolores Garcia, Matt Kusner (2023). ML Optimization for Concentrated Solar Power Plants. NeurIPS2023 ML4PS.

PDF

Ouail Kitouni, Niklas Nolte, James Hensman, Bhaskar Mitra (2023). DiSK: Diffusion Model for Structured Knowledge. ICML2024.

ArXiv

Ouail Kitouni, Niklas Nolte, Sokratis Trifinopoulos, Subhash Kantamneni, Mike Williams (2023). NuCLR: Nuclear Co-Learned Representations. ICML2023 SynS ML.

PDF

Ouail Kitouni, Niklas Nolte, Mike Williams (2022). NEEMo: Geometric Fitting using Neural Estimation of the Energy Mover's Distance. NeurIPS2022 ML4PS.

PDF Code Poster Slides

Ouail Kitouni, Niklas Nolte, Mike Williams (2022). Robust and Provably Monotonic Networks. ICLR2023.

PDF Code Project Poster NeurIPS Abstract

Ziming Liu, Ouail Kitouni, Niklas Nolte, Eric Michaud, Mike Williams, Max Tegmark (2022). Towards Understanding Grokking: An Effective Theory of Representation Learning. NeurIPS2022.

PDF

Ouail Kitouni, Benjamin Nachman, Constantin Weisser, Mike Williams (2020). Controlling Classifier Bias with Moment Decomposition: A Method to Enhance Searches for Resonances. NeurIPS2020 ML4PS.

PDF Code Project Poster Video INSPIRE-HEP NeurIPS Abstract