David Alvarez-Melis

(he/him/his)

Assistant Professor, Harvard University (SEAS)

150 Western Av. Room 2-332, Allston MA 02134

[three initials]@seas.harvard.edu

About

I'm an Assistant Professor of Computer Science at Harvard SEAS where I lead the Data-Centric Machine Learning (DCML) group. I'm also an Associate Faculty at the Kempner Institute, and have affiliations with the Harvard Data Science Initiative. I am also a researcher at Microsoft Research New England.

In my group, we study how data—not just models—shapes the behavior and reliability of AI systems. Our research develops foundational principles and methods for characterizing, transforming, and optimizing datasets to make learning more efficient, interpretable, and adaptive. By treating datasets as structured, dynamic entities rather than static inputs, we aim to build a rigorous science of data for machine learning: a framework that formally connects data properties to model outcomes. In other words, we seek to reivindicate the role of data in AI and ML, recognizing the reality that in this day and age data is often the most scarce, valuable, and manipulable resource in learning systems, yet we still lack a systematic understanding of how to harness it effectively, robustly, and predictably.

Our work integrates tools from optimal transport, information theory, and generative modeling, and applies them across domains including scientific data, language, and vision. For example, we have developed methods based on optimal transport to align datasets from different domains, compare complex data distributions, generate synthetic data that preserves key characteristics, and transform datasets to improve model robustness. We have also explored techniques for interpreting black-box models, enhancing their trustworthiness through robust explanations, and incorporating human-centered design principles.

If you share these goals, please reach out. We are always looking for collaborators and new group members!

Prospective lab members: If you are interested in joining my group at Harvard, please read this.

Bio

I obtained a PhD in computer science from MIT, where I worked at CSAIL on various topics in machine learning and natural language processing. I also hold BSc (Licenciatura) and MS degrees in mathematics from ITAM and Courant Institute (NYU), respectively. During the latter, I worked on semidefinite programming for domain adaptation for my thesis. Between Master's and PhD, I spent a year at IBM's T.J. Watson Research Center, working in the Speech Recognition and NLP teams.

News

Publications

Most recent publications on Google Scholar.

Bridging Compute- and Data-Optimal Pretraining

Tian Qin, Kimia Hamidieh, David Alvarez-Melis

NeurIPS'26: Neural Information Processing Systems. 2026. To appear. Oral.

Let's (not) just put things in Context: Test-time Training for Long-context LLMs

Rachit Bansal, Aston Zhang, Rishabh Tiwari, Lovish Madaan, Venkata Sai Surya Subramanyam Duvvuri, Devvrit Khatri, David Brandfonbrener, David Alvarez-Melis, Prajjwal Bhargava, Mihir Kale, Samy Jelassi

ICLR'26: International Conference on Learning Representations. 2026.

RL Excursions during Pre-training: How early is too early for on-policy learning?

Rachit Bansal, Clara Mohri, Tian Qin, David Alvarez-Melis, Sham Kakade

NeurIPS'26: Neural Information Processing Systems. 2026. To appear.

Guided Speculative Inference for Efficient Test-Time Alignment of LLMs

Jonathan Geuter, Youssef Mroueh, David Alvarez-Melis

ICLR'26: International Conference on Learning Representations. 2026.

Domain-Aware Scaling Laws Uncover Data Synergy

Kimia Hamidieh, Lester Mackey, David Alvarez-Melis

COLM'26: Conference on Language Modeling. 2026.

Boomerang Distillation Enables Zero-Shot Model Size Interpolation

Sara Kangaslahti, Nihal V. Nayak, Jonathan Geuter, Marco Fumero, Francesco Locatello, David Alvarez-Melis

ICLR'26: International Conference on Learning Representations. 2026.

Understanding Layer Patching in Model Size Interpolation

Sara Kangaslahti, Jonathan Geuter, Nihal V. Nayak, Marco Fumero, Francesco Locatello, David Alvarez-Melis

NeurIPS'26: Neural Information Processing Systems. 2026. To appear.

Stop Training for the Worst: Progressive Unmasking Accelerates Masked Diffusion Training

Jaeyeon Kim, Jonathan Geuter, David Alvarez-Melis, Sham M. Kakade, Sitan Chen

ICML'26: International Conference on Machine Learning. 2026.

Three-operator Splitting with Stale Gradients for Faster Non-linear Optimal Transport

Jacob Lindbäck, David Alvarez-Melis, Mikael Johansson

AISTATS'26: International Conference on Artificial Intelligence and Statistics. 2026.

A Critical Look at Targeted Instruction Selection: Disentangling What Matters (and What Doesn't)

Nihal V. Nayak, Paula Rodriguez-Diaz, Neha Hulkund, Sara Beery, David Alvarez-Melis

ICML'26: International Conference on Machine Learning. 2026.

Curriculum Learning as Transport: Understanding Curricula with Wasserstein Geodesics

Changho Shin, David Alvarez-Melis

COLM'26: Conference on Language Modeling. 2026.

Integrating chemical structures as treatments improves representations of microscopy images for morphological profiling

Yemin Yu, Emre Hayir, Neil Tenenholtz, Lester Mackey, Ying Wei, David Alvarez-Melis, Ava P. Amini, Alex X. Lu

PLOS Computational Biology. 2026.

Random Scaling of Emergent Capabilities

Rosie Zhao, Tian Qin, David Alvarez-Melis, Sham M. Kakade, Naomi Saphra

ICML'26: International Conference on Machine Learning. 2026.

Thinking at the Right Size: Amortized Distillation Across Post-Trained LLMs

Yan Zhou, Sara Kangaslahti, Jonathan Geuter, Nihal V. Nayak, Marco Fumero, Francesco Locatello, David Alvarez-Melis

EMNLP Findings'26: Findings of the Association for Computational Linguistics: EMNLP. 2026.

Why Larger Models Learn More: Effects of Capacity, Interference, and Rare-Task Retention

Jing Huang, Daniel Wurgaft, Rachit Bansal, Laura Ruis, Naomi Saphra, David Alvarez-Melis, Andrew Kyle Lampinen, Christopher Potts, Ekdeep Singh Lubana

ICML 2026 Workshop on High Dimensional Learning Dynamics.

Can Interpretation Predict Behavior on Unseen Data?

Victoria R. Li, Jenny Kaufmann, Tian Qin, Martin Wattenberg, David Alvarez-Melis, Naomi Saphra

Proceedings of the 9th BlackboxNLP Workshop: Analyzing and Interpreting Neural Networks for NLP, 2026.

Adam or Gauss-Newton? A Comparative Study In Terms of Basis Alignment and SGD Noise

Bingbin Liu, Rachit Bansal, Depen Morwani, David Alvarez-Melis, Sham M. Kakade

ICML 2026 Workshop on High Dimensional Learning Dynamics.

Adam or Gauss-Newton? A Comparative Study In Terms of Basis Alignment and SGD Noise

Bingbin Liu, Rachit Bansal, Depen Morwani, Nikhil Vyas, David Alvarez-Melis, Sham M. Kakade

NeurIPS 2026 Workshop on Optimization for Machine Learning. To appear.

Lossy Superposition: Predicting Compositional Errors Without Seeing Inputs

Jennifer Meng Lu, Ruochen Zhang, Isabelle Lee, David Alvarez-Melis, Ellie Pavlick, Naomi Saphra

ICML 2026 Workshop on Mechanistic Interpretability.

Feeding BabyLMs Macaroni: Code-Switching Curricula Cause Cross-Lingual Convergence

Dries Rooryck, Alexander Dazhen Cai, Yonatan Belinkov, David Alvarez-Melis, Kianté Brantley

Proceedings of the Second BabyLM Workshop, 2026.

Beyond Explainable AI (XAI): An Overdue Paradigm Shift and Post-XAI Research Directions

Saleh Afroogh, Syed Ishtiaque Ahmed, Petra Ahrweiler, David Alvarez-Melis, Mansur Maturidi Arief, Emilia Barakova, Falco J. Bargagli-Stoffi, Erdem Biyik, Hanjie Chen, Xiang 'Anthony' Chen, Robert Alan Clements, Keeley Crockett, Amit Dhurandhar, Fethiye Irmak Dogan, Mollie Dollinger, Motahhare Eslami, Aldo A Faisal, Arya Farahi, Melanie F. Pradier, Saadia Gabriel, Diego Garcia-Olano, Marzyeh Ghassemi, Shaona Ghosh, Hatice Gunes, Ehsan Hajiramezanali, Stefan Haufe, Biwei Huang, Angel Hwang, Md Tauhidul Islam, Junfeng Jiao, Amir-Hossein Karimi, Saber Kazeminasab, Anastasia Kuzminykh, William La Cava, Brian Y. Lim, Xiaofeng Liu, Mohammad R. K. Mofrad, Alicia Parrish, Maria Perez-Ortiz, Shriti Raj, Swabha Swayamdipta, Salmonn Talebi, Kush R. Varshney, Mihaela Vorvoreanu, Lily Weng, Alice Xiang, Yiming Xu, Ding Zhao, Jieyu Zhao

arXiv preprint. 2026.

Releasing the Inner Jev: A Simple Recipe for Open Decision Models

Jonathan Geuter, Andrew Cheng, David Alvarez-Melis

arXiv preprint. 2026.

Why Larger Models Learn More: Effects of Capacity, Interference, and Rare-Task Retention

Jing Huang, Daniel Wurgaft, Rachit Bansal, Laura Ruis, Naomi Saphra, David Alvarez-Melis, Andrew Kyle Lampinen, Christopher Potts, Ekdeep Singh Lubana

arXiv preprint. 2026.

Adversarial Concept Search: Predicting Compositional Errors From Feature Geometry

Jennifer Meng Lu, Ruochen Zhang, Isabelle Lee, David Alvarez-Melis, Ellie Pavlick, Naomi Saphra

arXiv preprint. 2026.

Universal Bounds for Self-Normalized Kullback-Leibler Estimation

Buu Phan, Lester Mackey, David Alvarez-Melis

arXiv preprint. 2026.

Low-Frequency Shortcuts in Texture-Driven Visual Learning

Utku Şirin, Cathy Hou, Despina-Ekaterini Argiropoulos, David Alvarez-Melis, Stratos Idreos

arXiv preprint. 2026.

OT on the Map: Quantifying Domain Shifts in Geographic Space

Haoran Zhang, Livia Betti, Konstantin Klemmer, Esther Rolf, David Alvarez-Melis

arXiv preprint. 2026.

Investigating the interaction of linguistic and mathematical reasoning in language models using multilingual number puzzles

Antara Raaghavi Bhattacharya, Isabel Papadimitriou, Kathryn Davidson, David Alvarez-Melis

EMNLP'25: Empirical Methods in Natural Language Processing. 2025.

scGeneScope: A Treatment-Matched Single Cell Imaging and Transcriptomics Dataset and Benchmark for Treatment Response Modeling

Joel Dapello, Marcel Nassar, Ridvan Eksi, Ban Wang, Jules Gagnon-Marchand, Kenneth Gao, Akram Baharlouei, Kyra Thrush, Nina Riehs, Amy Peterson, Aniket Tolpadi, Abhejit Rajagopal, Henry Miller, Ashley Mae Conard, David Alvarez-Melis, Rory Stark, Simone Bianco, Morgan Levine, Ava P. Amini, Alex X. Lu, Nicolò Fusi, Ravi Pandya, Valentina Pedoia, Hana El-Samad

NeurIPS D&B'25: Neural Information Processing Systems, Datasets and Benchmarks Track. 2025.

DDEQs: Distributional Deep Equilibrium Models through Wasserstein Gradient Flows

Jonathan Geuter, Clément Bonet, Anna Korba, David Alvarez-Melis

AISTATS'25: International Conference on Artificial Intelligence and Statistics. 2025.

Mixture of Parrots: Experts improve memorization more than reasoning

Samy Jelassi, Clara Mohri, David Brandfonbrener, Alex Gu, Nikhil Vyas, Nikhil Anand, David Alvarez-Melis, Yuanzhi Li, Sham M. Kakade, Eran Malach

ICLR'25: International Conference on Learning Representations. 2025.

Continuous Language Model Interpolation yields Dynamic and Controllable Text Generation

Sara Kangaslahti, David Alvarez-Melis

TMLR: Transactions on Machine Learning Research. 2025.

JONES-19: A Cultural Image Dataset Based on The Grammar of Ornament

Linh Pham, Stuart M. Shieber, David Alvarez-Melis, Alexandros Haridis

KUI'25: International Conference on Culture and Computer Science. 2025.

To Backtrack or Not to Backtrack: When Sequential Search Limits Model Reasoning

Tian Qin, David Alvarez-Melis, Samy Jelassi, Eran Malach

COLM'25: Conference on Language Modeling. 2025.

Data Drives Unstable Hierarchical Generalization in LMs

Tian Qin, Naomi Saphra, David Alvarez-Melis

EMNLP'25: Empirical Methods in Natural Language Processing. 2025.

What is the Right Notion of Distance between Predict-then-Optimize Tasks?

Paula Rodriguez-Diaz, Lingkai Kong, Kai Wang, David Alvarez-Melis, Milind Tambe

UAI'25: Uncertainty in Artificial Intelligence. 2025.

Reliable and Responsible Foundation Models: A Comprehensive Survey

Xinyu Yang, Junlin Han, Rishi Bommasani, Jinqi Luo, Wenjie Qu, Wangchunshu Zhou, Adel Bibi, Xiyao Wang, Jaehong Yoon, Elias Stengel-Eskin, Shengbang Tong, Lingfeng Shen, Rafael Rafailov, Runjia Li, Zhaoyang Wang, Yiyang Zhou, Chenhang Cui, Yu Wang, Wenhao Zheng, Huichi Zhou, Jindong Gu, Zhaorun Chen, Peng Xia, Tony Lee, Thomas P. Zollo, Vikash Sehwag, Jixuan Leng, Jiuhai Chen, Yuxin Wen, Huan Zhang, Zhun Deng, Linjun Zhang, Pavel Izmailov, Pang Wei Koh, Yulia Tsvetkov, Andrew Gordon Wilson, Jiaheng Zhang, James Zou, Cihang Xie, Hao Wang, Philip Torr, Julian McAuley, David Alvarez-Melis, Florian Tramèr, Kaidi Xu, Suman Jana, Chris Callison-Burch, René Vidal, Filippos Kokkinos, Mohit Bansal, Beidi Chen, Huaxiu Yao

TMLR: Transactions on Machine Learning Research. 2025.

IGDA: Interactive Graph Discovery through Large Language Model Agents

Alexander Havrilla, David Alvarez-Melis, Nicolò Fusi

ICLR 2025 Workshop on Reasoning and Planning for Large Language Models.

Bispectral OT: Dataset Comparison using Symmetry-Aware Optimal Transport

Annabel Ma, Kaiying Hou, David Alvarez-Melis, Melanie Weber

NeurIPS 2025 Workshop on Symmetry and Geometry in Neural Representations.

Augmenting X-ray Astronomical Representations with Scientific Knowledge through Contrastive Learning

Juan Rafael Martínez-Galarza, Nicolò Oreste Pinciroli Vago, Shivam Raval, Carolina Cuesta-Lazaro, Melanie Weber, David Alvarez-Melis, Alberto Accomazzi, Cecilia Garraffo, Joshua Knutson, Ryan Thill, Christopher B. Green, Imantha Ahangama

ICLR 2025 Second Workshop on Representational Alignment.

Adam or Gauss-Newton? A Comparative Study In Terms of Basis Alignment and SGD Noise

Bingbin Liu, Rachit Bansal, Depen Morwani, Nikhil Vyas, David Alvarez-Melis, Sham M. Kakade

arXiv preprint. 2025.

Decomposing Elements of Problem Solving: What “Math” Does RL Teach?

Tian Qin, Core Francisco Park, Mujin Kwun, Aaron Walsman, Eran Malach, Nikhil Anand, Hidenori Tanaka, David Alvarez-Melis

arXiv preprint. 2025.

KerJEPA: Kernel Discrepancies for Euclidean Self-Supervised Learning

Eric Zimmermann, Harley Wiltzer, Justin Szeto, David Alvarez-Melis, Lester Mackey

arXiv preprint. 2025.

A Label is Worth A Thousand Images in Dataset Distillation

Tian Qin, Zhiwei Deng, David Alvarez-Melis

NeurIPS'24: Neural Information Processing Systems. 2024.

Tag-LLM: Repurposing General-Purpose LLMs for Specialized Domains

Junhong Shen, Neil Tenenholtz, James Brian Hall, David Alvarez-Melis, Nicolo Fusi

ICML'24: International Conference on Machine Learning. 2024.

Adapting Language Models via Token Translation

Zhili Feng, Tanya Marwah, Lester Mackey, David Alvarez-Melis, Nicolò Fusi

NeurIPS 2024 Workshop on Machine Learning and Compression.

CharED: Character-wise Ensemble Decoding for Large Language Models

Kevin Gu, Eva Tuecke, Dmitriy Katz, Raya Horesh, David Alvarez-Melis, Mikhail Yurochkin

ICML 2024 Workshop on Foundation Models in the Wild.

Mixture of Parrots: Mixtures of experts improve memorization more than reasoning

Samy Jelassi, Clara Mohri, David Brandfonbrener, Alex Gu, Nikhil Vyas, Nikhil Anand, David Alvarez-Melis, Yuanzhi Li, Sham M. Kakade, Eran Malach

NeurIPS 2024 Workshop on Mathematics of Modern Machine Learning. Oral.

Distributional Dataset Distillation with Subtask Decomposition

Tian Qin, Zhiwei Deng, David Alvarez-Melis

ICLR 2024 Workshop on Navigating and Addressing Data Problems for Foundation Models.

Understanding the Role of Functional Diversity in Weight-Ensembling with Ingredient Selection and Multidimensional Scaling

Alex Rojas, David Alvarez-Melis

ICML 2024 Workshop on Foundation Models in the Wild.

Strongly Isomorphic Neural Optimal Transport Across Incomparable Spaces

Athina Sotiropoulou, David Alvarez-Melis

Proceedings of the Geometry-grounded Representation Learning and Generative Modeling Workshop (GRaM), 2024.

InfoOT: Information Maximizing Optimal Transport

Ching-Yao Chuang, Stefanie Jegelka, David Alvarez-Melis

ICML'23: International Conference on Machine Learning. 2023.

Domain adaptation using optimal transport for invariant learning using histopathology datasets

Kianoush Falahkheirkhah, Alex Xijie Lu, David Alvarez-Melis, Grace Huynh

MIDL'23: Medical Imaging with Deep Learning. 2023.

Generating Synthetic Datasets by Interpolating along Generalized Geodesics

Jiaojiao Fan, David Alvarez-Melis

UAI'23: Uncertainty in Artificial Intelligence. 2023.

Enhancing Language Models for Technical Domains with Dynamic Token Injection

Giorgio Giannone, Neil Tenenholtz, James Hall, Nicolò Fusi, David Alvarez-Melis

NeurIPS 2023 Workshop on Generative AI and Biology.

Undo Maps: A Tool for Adapting Policies to Perceptual Distortions

Abhi Gupta, Ted Moskovitz, David Alvarez-Melis, Aldo Pacchiano

ICML 2023 Workshop on New Frontiers in Learning, Control, and Dynamical Systems.

Are GANs overkill for NLP?

David Alvarez-Melis, Vikas Garg, Adam Kalai

NeurIPS'22: Neural Information Processing Systems. 2022.

Optimizing Functionals on the Space of Probabilities with Input Convex Neural Networks

David Alvarez-Melis, Yair Schiff, Youssef Mroueh

TMLR: Transactions on Machine Learning Research. 2022.

Earlier version at OTML: NeurIPS'21 Workshop on Optimal Transport in Machine Learning. Spotlight.

Hierarchical Optimal Transport for Comparing Histopathology Datasets

Anna Yeaton, Rahul G. Krishnan, Rebecca Mieloszyk, David Alvarez-Melis, Grace Huynh

MIDL'22: Medical Imaging with Deep Learning. 2022.

Interpretable Distribution Shift Detection using Optimal Transport

Neha Hulkund, Nicolò Fusi, Jennifer Wortman Vaughan, David Alvarez-Melis

ICML 2022 Workshop on DataPerf: Benchmarking Data for Data-Centric AI.

Neural Unbalanced Optimal Transport via Cycle-Consistent Semi-Couplings

Frederike Lübeck, Charlotte Bunne, Gabriele Gut, Jacobo Sarabia del Castillo, Lucas Pelkmans, David Alvarez-Melis

NeurIPS 2022 Workshop on Learning Meaningful Representations of Life.

Budget-Constrained Bounds for Mini-Batch Estimation of Optimal Transport

David Alvarez-Melis, Nicolò Fusi, Lester Mackey, Tal Wagner

arXiv preprint. 2022.

Transfer RL via the Undo Maps Formalism

Abhi Gupta, Ted Moskovitz, David Alvarez-Melis, Aldo Pacchiano

arXiv preprint. 2022.

Dataset Dynamics via Gradient Flows in Probability Space

David Alvarez-Melis, Nicolò Fusi

ICML'21: International Conference on Machine Learning. 2021.

From Human Explanation to Model Interpretability: A Framework Based on Weight of Evidence

David Alvarez-Melis, Harmanpreet Kaur, Hal Daumé III, Hanna Wallach, Jennifer Wortman Vaughan

HCOMP'21: AAAI Conference on Human Computation and Crowdsourcing. 2021.

Geometric Dataset Distances via Optimal Transport

David Alvarez-Melis, Nicolò Fusi

NeurIPS'20: Neural Information Processing Systems. 2020.

Earlier version at AutoML @ ICML 2020.

Unsupervised Hierarchy Matching with Optimal Transport over Hyperbolic Spaces

David Alvarez-Melis, Youssef Mroueh, Tommi Jaakkola

AISTATS'20: International Conference on Artificial Intelligence and Statistics. 2020.

Earlier version at OTML: NeurIPS'18 Workshop on Optimal Transport for Machine Learning. Spotlight.

Towards Optimal Transport with Global Invariances

David Alvarez-Melis, Stefanie Jegelka, Tommi S. Jaakkola

AISTATS'19: International Conference on Artificial Intelligence and Statistics. 2019.

Learning Generative Models across Incomparable Spaces

Charlotte Bunne, David Alvarez-Melis, Andreas Krause, Stefanie Jegelka

ICML'19: International Conference on Machine Learning. 2019.

Earlier version at R2L: NeurIPS'18 Workshop on Relational Representation Learning. Best Paper Award.

Functional Transparency for Structured Data: a Game-Theoretic Approach

Guang-He Lee, Wengong Jin, David Alvarez-Melis, Tommi Jaakkola

ICML'19: International Conference on Machine Learning. 2019.

Towards Robust, Locally Linear Deep Networks

Guang-He Lee, David Alvarez-Melis, Tommi S. Jaakkola

ICLR'19: International Conference on Learning Representations. 2019.

Weight of Evidence as a Basis for Human-Oriented Explanations

David Alvarez-Melis, Hal Daumé III, Jennifer Wortman Vaughan, Hanna Wallach

NeurIPS 2019 Workshop on Human-Centric Machine Learning.

Probabilistic Bias Mitigation in Word Embeddings

Hailey Joren, David Alvarez-Melis

NeurIPS 2019 Workshop on Human-Centric Machine Learning.

Optimal Transport in Structured Domains: Algorithms and Applications

David Alvarez-Melis (advisor: Tommi S. Jaakkola)

PhD Thesis, MIT. 2019.

Gromov-Wasserstein Alignment of Word Embedding Spaces

David Alvarez-Melis, Tommi Jaakkola

EMNLP'18: Empirical Methods in Natural Language Processing. 2018. Oral.

Towards Robust Interpretability with Self-Explaining Neural Networks

David Alvarez-Melis, Tommi S. Jaakkola

NeurIPS'18: Neural Information Processing Systems. 2018.

Structured Optimal Transport

David Alvarez-Melis, Tommi Jaakkola, Stefanie Jegelka

AISTATS'18: International Conference on Artificial Intelligence and Statistics. 2018. Oral.

Earlier version at NIPS Workshop on Optimal Transport for Machine Learning, 2017, as Extended Oral.

On the Robustness of Interpretability Methods

David Alvarez-Melis, Tommi S. Jaakkola

ICML 2018 Workshop on Human Interpretability in Machine Learning.

Learning Generative Models across Incomparable Spaces

Charlotte Bunne, David Alvarez-Melis, Andreas Krause, Stefanie Jegelka

NeurIPS 2018 Workshop on Relational Representation Learning. Best Paper Award.

Game-Theoretic Interpretability for Temporal Modeling

Guang-He Lee, David Alvarez-Melis, Tommi S. Jaakkola

ICML 2018 Workshop on Fairness, Accountability, and Transparency in Machine Learning.

Distributional Adversarial Networks

Chengtao Li, David Alvarez-Melis, Keyulu Xu, Stefanie Jegelka, Suvrit Sra

ICLR 2018 Workshop Track.

A causal framework for explaining the predictions of black-box sequence-to-sequence models

David Alvarez-Melis, Tommi S. Jaakkola

EMNLP'17: Empirical Methods in Natural Language Processing. 2017.

Tree-structured decoding with doubly-recurrent neural networks

David Alvarez-Melis, Tommi S. Jaakkola

ICLR'17: International Conference on Learning Representations. 2017.

The Emotional GAN: Priming Adversarial Generation of Art with Emotion

David Alvarez-Melis, Judith Amores

NIPS 2017 Workshop on Machine Learning for Creativity and Design.

Topic Modeling in Twitter: Aggregating Tweets by Conversations

David Alvarez-Melis, Martin Saveski

ICWSM'16: International AAAI Conference on Web and Social Media. 2016. (Short paper.)

Word Embeddings as Metric Recovery in Semantic Spaces

Tatsunori B. Hashimoto, David Alvarez-Melis, Tommi S. Jaakkola

TACL: Transactions of the Association for Computational Linguistics. 2016. (Presented at ACL'16.)

Word, graph and manifold embedding from Markov processes

Tatsunori B. Hashimoto, David Alvarez-Melis, Tommi S. Jaakkola

NIPS 2015 Workshop on Nonparametric Methods for Large Scale Representation Learning. Oral.

A translation of “The characteristic function of a random phenomenon” by Bruno de Finetti

David Alvarez-Melis, Tamara Broderick

arXiv preprint. 2015.

The Matrix Multiplicative Weights Algorithm for Domain Adaptation

David Alvarez-Melis (advisor: Mehryar Mohri)

MS Thesis, Courant Institute, NYU. 2013.

Lax-Milgram's Theorem: Generalizations and Applications

David Alvarez-Melis (advisor: Carlos Bosch Giral)

BSc Thesis, ITAM. 2011.

Bridging Compute- and Data-Optimal Pretraining

Tian Qin, Kimia Hamidieh, David Alvarez-Melis

NeurIPS'26: Neural Information Processing Systems. 2026. To appear. Oral.

Domain-Aware Scaling Laws Uncover Data Synergy

Kimia Hamidieh, Lester Mackey, David Alvarez-Melis

COLM'26: Conference on Language Modeling. 2026.

Boomerang Distillation Enables Zero-Shot Model Size Interpolation

Sara Kangaslahti, Nihal V. Nayak, Jonathan Geuter, Marco Fumero, Francesco Locatello, David Alvarez-Melis

ICLR'26: International Conference on Learning Representations. 2026.

A Critical Look at Targeted Instruction Selection: Disentangling What Matters (and What Doesn't)

Nihal V. Nayak, Paula Rodriguez-Diaz, Neha Hulkund, Sara Beery, David Alvarez-Melis

ICML'26: International Conference on Machine Learning. 2026.

Curriculum Learning as Transport: Understanding Curricula with Wasserstein Geodesics

Changho Shin, David Alvarez-Melis

COLM'26: Conference on Language Modeling. 2026.

A Label is Worth A Thousand Images in Dataset Distillation

Tian Qin, Zhiwei Deng, David Alvarez-Melis

NeurIPS'24: Neural Information Processing Systems. 2024.

Dataset Dynamics via Gradient Flows in Probability Space

David Alvarez-Melis, Nicolò Fusi

ICML'21: International Conference on Machine Learning. 2021.

Geometric Dataset Distances via Optimal Transport

David Alvarez-Melis, Nicolò Fusi

NeurIPS'20: Neural Information Processing Systems. 2020.

Earlier version at AutoML @ ICML 2020.

Gromov-Wasserstein Alignment of Word Embedding Spaces

David Alvarez-Melis, Tommi Jaakkola

EMNLP'18: Empirical Methods in Natural Language Processing. 2018. Oral.

Towards Robust Interpretability with Self-Explaining Neural Networks

David Alvarez-Melis, Tommi S. Jaakkola

NeurIPS'18: Neural Information Processing Systems. 2018.

Beyond Explainable AI (XAI): An Overdue Paradigm Shift and Post-XAI Research Directions

Saleh Afroogh, Syed Ishtiaque Ahmed, Petra Ahrweiler, David Alvarez-Melis, Mansur Maturidi Arief, Emilia Barakova, Falco J. Bargagli-Stoffi, Erdem Biyik, Hanjie Chen, Xiang 'Anthony' Chen, Robert Alan Clements, Keeley Crockett, Amit Dhurandhar, Fethiye Irmak Dogan, Mollie Dollinger, Motahhare Eslami, Aldo A Faisal, Arya Farahi, Melanie F. Pradier, Saadia Gabriel, Diego Garcia-Olano, Marzyeh Ghassemi, Shaona Ghosh, Hatice Gunes, Ehsan Hajiramezanali, Stefan Haufe, Biwei Huang, Angel Hwang, Md Tauhidul Islam, Junfeng Jiao, Amir-Hossein Karimi, Saber Kazeminasab, Anastasia Kuzminykh, William La Cava, Brian Y. Lim, Xiaofeng Liu, Mohammad R. K. Mofrad, Alicia Parrish, Maria Perez-Ortiz, Shriti Raj, Swabha Swayamdipta, Salmonn Talebi, Kush R. Varshney, Mihaela Vorvoreanu, Lily Weng, Alice Xiang, Yiming Xu, Ding Zhao, Jieyu Zhao

arXiv preprint. 2026.

Releasing the Inner Jev: A Simple Recipe for Open Decision Models

Jonathan Geuter, Andrew Cheng, David Alvarez-Melis

arXiv preprint. 2026.

Why Larger Models Learn More: Effects of Capacity, Interference, and Rare-Task Retention

Jing Huang, Daniel Wurgaft, Rachit Bansal, Laura Ruis, Naomi Saphra, David Alvarez-Melis, Andrew Kyle Lampinen, Christopher Potts, Ekdeep Singh Lubana

arXiv preprint. 2026.

Adversarial Concept Search: Predicting Compositional Errors From Feature Geometry

Jennifer Meng Lu, Ruochen Zhang, Isabelle Lee, David Alvarez-Melis, Ellie Pavlick, Naomi Saphra

arXiv preprint. 2026.

Universal Bounds for Self-Normalized Kullback-Leibler Estimation

Buu Phan, Lester Mackey, David Alvarez-Melis

arXiv preprint. 2026.

Low-Frequency Shortcuts in Texture-Driven Visual Learning

Utku Şirin, Cathy Hou, Despina-Ekaterini Argiropoulos, David Alvarez-Melis, Stratos Idreos

arXiv preprint. 2026.

OT on the Map: Quantifying Domain Shifts in Geographic Space

Haoran Zhang, Livia Betti, Konstantin Klemmer, Esther Rolf, David Alvarez-Melis

arXiv preprint. 2026.

Adam or Gauss-Newton? A Comparative Study In Terms of Basis Alignment and SGD Noise

Bingbin Liu, Rachit Bansal, Depen Morwani, Nikhil Vyas, David Alvarez-Melis, Sham M. Kakade

arXiv preprint. 2025.

Decomposing Elements of Problem Solving: What “Math” Does RL Teach?

Tian Qin, Core Francisco Park, Mujin Kwun, Aaron Walsman, Eran Malach, Nikhil Anand, Hidenori Tanaka, David Alvarez-Melis

arXiv preprint. 2025.

KerJEPA: Kernel Discrepancies for Euclidean Self-Supervised Learning

Eric Zimmermann, Harley Wiltzer, Justin Szeto, David Alvarez-Melis, Lester Mackey

arXiv preprint. 2025.

Budget-Constrained Bounds for Mini-Batch Estimation of Optimal Transport

David Alvarez-Melis, Nicolò Fusi, Lester Mackey, Tal Wagner

arXiv preprint. 2022.

Transfer RL via the Undo Maps Formalism

Abhi Gupta, Ted Moskovitz, David Alvarez-Melis, Aldo Pacchiano

arXiv preprint. 2022.

A translation of “The characteristic function of a random phenomenon” by Bruno de Finetti

David Alvarez-Melis, Tamara Broderick

arXiv preprint. 2015.

Bridging Compute- and Data-Optimal Pretraining

Tian Qin, Kimia Hamidieh, David Alvarez-Melis

NeurIPS'26: Neural Information Processing Systems. 2026. To appear. Oral.

Let's (not) just put things in Context: Test-time Training for Long-context LLMs

Rachit Bansal, Aston Zhang, Rishabh Tiwari, Lovish Madaan, Venkata Sai Surya Subramanyam Duvvuri, Devvrit Khatri, David Brandfonbrener, David Alvarez-Melis, Prajjwal Bhargava, Mihir Kale, Samy Jelassi

ICLR'26: International Conference on Learning Representations. 2026.

RL Excursions during Pre-training: How early is too early for on-policy learning?

Rachit Bansal, Clara Mohri, Tian Qin, David Alvarez-Melis, Sham Kakade

NeurIPS'26: Neural Information Processing Systems. 2026. To appear.

Guided Speculative Inference for Efficient Test-Time Alignment of LLMs

Jonathan Geuter, Youssef Mroueh, David Alvarez-Melis

ICLR'26: International Conference on Learning Representations. 2026.

Domain-Aware Scaling Laws Uncover Data Synergy

Kimia Hamidieh, Lester Mackey, David Alvarez-Melis

COLM'26: Conference on Language Modeling. 2026.

Boomerang Distillation Enables Zero-Shot Model Size Interpolation

Sara Kangaslahti, Nihal V. Nayak, Jonathan Geuter, Marco Fumero, Francesco Locatello, David Alvarez-Melis

ICLR'26: International Conference on Learning Representations. 2026.

Understanding Layer Patching in Model Size Interpolation

Sara Kangaslahti, Jonathan Geuter, Nihal V. Nayak, Marco Fumero, Francesco Locatello, David Alvarez-Melis

NeurIPS'26: Neural Information Processing Systems. 2026. To appear.

Stop Training for the Worst: Progressive Unmasking Accelerates Masked Diffusion Training

Jaeyeon Kim, Jonathan Geuter, David Alvarez-Melis, Sham M. Kakade, Sitan Chen

ICML'26: International Conference on Machine Learning. 2026.

Three-operator Splitting with Stale Gradients for Faster Non-linear Optimal Transport

Jacob Lindbäck, David Alvarez-Melis, Mikael Johansson

AISTATS'26: International Conference on Artificial Intelligence and Statistics. 2026.

A Critical Look at Targeted Instruction Selection: Disentangling What Matters (and What Doesn't)

Nihal V. Nayak, Paula Rodriguez-Diaz, Neha Hulkund, Sara Beery, David Alvarez-Melis

ICML'26: International Conference on Machine Learning. 2026.

Curriculum Learning as Transport: Understanding Curricula with Wasserstein Geodesics

Changho Shin, David Alvarez-Melis

COLM'26: Conference on Language Modeling. 2026.

Integrating chemical structures as treatments improves representations of microscopy images for morphological profiling

Yemin Yu, Emre Hayir, Neil Tenenholtz, Lester Mackey, Ying Wei, David Alvarez-Melis, Ava P. Amini, Alex X. Lu

PLOS Computational Biology. 2026.

Random Scaling of Emergent Capabilities

Rosie Zhao, Tian Qin, David Alvarez-Melis, Sham M. Kakade, Naomi Saphra

ICML'26: International Conference on Machine Learning. 2026.

Thinking at the Right Size: Amortized Distillation Across Post-Trained LLMs

Yan Zhou, Sara Kangaslahti, Jonathan Geuter, Nihal V. Nayak, Marco Fumero, Francesco Locatello, David Alvarez-Melis

EMNLP Findings'26: Findings of the Association for Computational Linguistics: EMNLP. 2026.

Why Larger Models Learn More: Effects of Capacity, Interference, and Rare-Task Retention

Jing Huang, Daniel Wurgaft, Rachit Bansal, Laura Ruis, Naomi Saphra, David Alvarez-Melis, Andrew Kyle Lampinen, Christopher Potts, Ekdeep Singh Lubana

ICML 2026 Workshop on High Dimensional Learning Dynamics.

Adam or Gauss-Newton? A Comparative Study In Terms of Basis Alignment and SGD Noise

Bingbin Liu, Rachit Bansal, Depen Morwani, David Alvarez-Melis, Sham M. Kakade

ICML 2026 Workshop on High Dimensional Learning Dynamics.

Adam or Gauss-Newton? A Comparative Study In Terms of Basis Alignment and SGD Noise

Bingbin Liu, Rachit Bansal, Depen Morwani, Nikhil Vyas, David Alvarez-Melis, Sham M. Kakade

NeurIPS 2026 Workshop on Optimization for Machine Learning. To appear.

Releasing the Inner Jev: A Simple Recipe for Open Decision Models

Jonathan Geuter, Andrew Cheng, David Alvarez-Melis

arXiv preprint. 2026.

Why Larger Models Learn More: Effects of Capacity, Interference, and Rare-Task Retention

Jing Huang, Daniel Wurgaft, Rachit Bansal, Laura Ruis, Naomi Saphra, David Alvarez-Melis, Andrew Kyle Lampinen, Christopher Potts, Ekdeep Singh Lubana

arXiv preprint. 2026.

Universal Bounds for Self-Normalized Kullback-Leibler Estimation

Buu Phan, Lester Mackey, David Alvarez-Melis

arXiv preprint. 2026.

Low-Frequency Shortcuts in Texture-Driven Visual Learning

Utku Şirin, Cathy Hou, Despina-Ekaterini Argiropoulos, David Alvarez-Melis, Stratos Idreos

arXiv preprint. 2026.

OT on the Map: Quantifying Domain Shifts in Geographic Space

Haoran Zhang, Livia Betti, Konstantin Klemmer, Esther Rolf, David Alvarez-Melis

arXiv preprint. 2026.

Investigating the interaction of linguistic and mathematical reasoning in language models using multilingual number puzzles

Antara Raaghavi Bhattacharya, Isabel Papadimitriou, Kathryn Davidson, David Alvarez-Melis

EMNLP'25: Empirical Methods in Natural Language Processing. 2025.

scGeneScope: A Treatment-Matched Single Cell Imaging and Transcriptomics Dataset and Benchmark for Treatment Response Modeling

Joel Dapello, Marcel Nassar, Ridvan Eksi, Ban Wang, Jules Gagnon-Marchand, Kenneth Gao, Akram Baharlouei, Kyra Thrush, Nina Riehs, Amy Peterson, Aniket Tolpadi, Abhejit Rajagopal, Henry Miller, Ashley Mae Conard, David Alvarez-Melis, Rory Stark, Simone Bianco, Morgan Levine, Ava P. Amini, Alex X. Lu, Nicolò Fusi, Ravi Pandya, Valentina Pedoia, Hana El-Samad

NeurIPS D&B'25: Neural Information Processing Systems, Datasets and Benchmarks Track. 2025.

DDEQs: Distributional Deep Equilibrium Models through Wasserstein Gradient Flows

Jonathan Geuter, Clément Bonet, Anna Korba, David Alvarez-Melis

AISTATS'25: International Conference on Artificial Intelligence and Statistics. 2025.

Mixture of Parrots: Experts improve memorization more than reasoning

Samy Jelassi, Clara Mohri, David Brandfonbrener, Alex Gu, Nikhil Vyas, Nikhil Anand, David Alvarez-Melis, Yuanzhi Li, Sham M. Kakade, Eran Malach

ICLR'25: International Conference on Learning Representations. 2025.

Continuous Language Model Interpolation yields Dynamic and Controllable Text Generation

Sara Kangaslahti, David Alvarez-Melis

TMLR: Transactions on Machine Learning Research. 2025.

JONES-19: A Cultural Image Dataset Based on The Grammar of Ornament

Linh Pham, Stuart M. Shieber, David Alvarez-Melis, Alexandros Haridis

KUI'25: International Conference on Culture and Computer Science. 2025.

To Backtrack or Not to Backtrack: When Sequential Search Limits Model Reasoning

Tian Qin, David Alvarez-Melis, Samy Jelassi, Eran Malach

COLM'25: Conference on Language Modeling. 2025.

Data Drives Unstable Hierarchical Generalization in LMs

Tian Qin, Naomi Saphra, David Alvarez-Melis

EMNLP'25: Empirical Methods in Natural Language Processing. 2025.

What is the Right Notion of Distance between Predict-then-Optimize Tasks?

Paula Rodriguez-Diaz, Lingkai Kong, Kai Wang, David Alvarez-Melis, Milind Tambe

UAI'25: Uncertainty in Artificial Intelligence. 2025.

Reliable and Responsible Foundation Models: A Comprehensive Survey

Xinyu Yang, Junlin Han, Rishi Bommasani, Jinqi Luo, Wenjie Qu, Wangchunshu Zhou, Adel Bibi, Xiyao Wang, Jaehong Yoon, Elias Stengel-Eskin, Shengbang Tong, Lingfeng Shen, Rafael Rafailov, Runjia Li, Zhaoyang Wang, Yiyang Zhou, Chenhang Cui, Yu Wang, Wenhao Zheng, Huichi Zhou, Jindong Gu, Zhaorun Chen, Peng Xia, Tony Lee, Thomas P. Zollo, Vikash Sehwag, Jixuan Leng, Jiuhai Chen, Yuxin Wen, Huan Zhang, Zhun Deng, Linjun Zhang, Pavel Izmailov, Pang Wei Koh, Yulia Tsvetkov, Andrew Gordon Wilson, Jiaheng Zhang, James Zou, Cihang Xie, Hao Wang, Philip Torr, Julian McAuley, David Alvarez-Melis, Florian Tramèr, Kaidi Xu, Suman Jana, Chris Callison-Burch, René Vidal, Filippos Kokkinos, Mohit Bansal, Beidi Chen, Huaxiu Yao

TMLR: Transactions on Machine Learning Research. 2025.

IGDA: Interactive Graph Discovery through Large Language Model Agents

Alexander Havrilla, David Alvarez-Melis, Nicolò Fusi

ICLR 2025 Workshop on Reasoning and Planning for Large Language Models.

Augmenting X-ray Astronomical Representations with Scientific Knowledge through Contrastive Learning

Juan Rafael Martínez-Galarza, Nicolò Oreste Pinciroli Vago, Shivam Raval, Carolina Cuesta-Lazaro, Melanie Weber, David Alvarez-Melis, Alberto Accomazzi, Cecilia Garraffo, Joshua Knutson, Ryan Thill, Christopher B. Green, Imantha Ahangama

ICLR 2025 Second Workshop on Representational Alignment.

Adam or Gauss-Newton? A Comparative Study In Terms of Basis Alignment and SGD Noise

Bingbin Liu, Rachit Bansal, Depen Morwani, Nikhil Vyas, David Alvarez-Melis, Sham M. Kakade

arXiv preprint. 2025.

Decomposing Elements of Problem Solving: What “Math” Does RL Teach?

Tian Qin, Core Francisco Park, Mujin Kwun, Aaron Walsman, Eran Malach, Nikhil Anand, Hidenori Tanaka, David Alvarez-Melis

arXiv preprint. 2025.

KerJEPA: Kernel Discrepancies for Euclidean Self-Supervised Learning

Eric Zimmermann, Harley Wiltzer, Justin Szeto, David Alvarez-Melis, Lester Mackey

arXiv preprint. 2025.

A Label is Worth A Thousand Images in Dataset Distillation

Tian Qin, Zhiwei Deng, David Alvarez-Melis

NeurIPS'24: Neural Information Processing Systems. 2024.

Tag-LLM: Repurposing General-Purpose LLMs for Specialized Domains

Junhong Shen, Neil Tenenholtz, James Brian Hall, David Alvarez-Melis, Nicolo Fusi

ICML'24: International Conference on Machine Learning. 2024.

Adapting Language Models via Token Translation

Zhili Feng, Tanya Marwah, Lester Mackey, David Alvarez-Melis, Nicolò Fusi

NeurIPS 2024 Workshop on Machine Learning and Compression.

CharED: Character-wise Ensemble Decoding for Large Language Models

Kevin Gu, Eva Tuecke, Dmitriy Katz, Raya Horesh, David Alvarez-Melis, Mikhail Yurochkin

ICML 2024 Workshop on Foundation Models in the Wild.

Mixture of Parrots: Mixtures of experts improve memorization more than reasoning

Samy Jelassi, Clara Mohri, David Brandfonbrener, Alex Gu, Nikhil Vyas, Nikhil Anand, David Alvarez-Melis, Yuanzhi Li, Sham M. Kakade, Eran Malach

NeurIPS 2024 Workshop on Mathematics of Modern Machine Learning. Oral.

Distributional Dataset Distillation with Subtask Decomposition

Tian Qin, Zhiwei Deng, David Alvarez-Melis

ICLR 2024 Workshop on Navigating and Addressing Data Problems for Foundation Models.

Understanding the Role of Functional Diversity in Weight-Ensembling with Ingredient Selection and Multidimensional Scaling

Alex Rojas, David Alvarez-Melis

ICML 2024 Workshop on Foundation Models in the Wild.

Strongly Isomorphic Neural Optimal Transport Across Incomparable Spaces

Athina Sotiropoulou, David Alvarez-Melis

Proceedings of the Geometry-grounded Representation Learning and Generative Modeling Workshop (GRaM), 2024.

InfoOT: Information Maximizing Optimal Transport

Ching-Yao Chuang, Stefanie Jegelka, David Alvarez-Melis

ICML'23: International Conference on Machine Learning. 2023.

Domain adaptation using optimal transport for invariant learning using histopathology datasets

Kianoush Falahkheirkhah, Alex Xijie Lu, David Alvarez-Melis, Grace Huynh

MIDL'23: Medical Imaging with Deep Learning. 2023.

Generating Synthetic Datasets by Interpolating along Generalized Geodesics

Jiaojiao Fan, David Alvarez-Melis

UAI'23: Uncertainty in Artificial Intelligence. 2023.

Enhancing Language Models for Technical Domains with Dynamic Token Injection

Giorgio Giannone, Neil Tenenholtz, James Hall, Nicolò Fusi, David Alvarez-Melis

NeurIPS 2023 Workshop on Generative AI and Biology.

Undo Maps: A Tool for Adapting Policies to Perceptual Distortions

Abhi Gupta, Ted Moskovitz, David Alvarez-Melis, Aldo Pacchiano

ICML 2023 Workshop on New Frontiers in Learning, Control, and Dynamical Systems.

Are GANs overkill for NLP?

David Alvarez-Melis, Vikas Garg, Adam Kalai

NeurIPS'22: Neural Information Processing Systems. 2022.

Optimizing Functionals on the Space of Probabilities with Input Convex Neural Networks

David Alvarez-Melis, Yair Schiff, Youssef Mroueh

TMLR: Transactions on Machine Learning Research. 2022.

Earlier version at OTML: NeurIPS'21 Workshop on Optimal Transport in Machine Learning. Spotlight.

Hierarchical Optimal Transport for Comparing Histopathology Datasets

Anna Yeaton, Rahul G. Krishnan, Rebecca Mieloszyk, David Alvarez-Melis, Grace Huynh

MIDL'22: Medical Imaging with Deep Learning. 2022.

Interpretable Distribution Shift Detection using Optimal Transport

Neha Hulkund, Nicolò Fusi, Jennifer Wortman Vaughan, David Alvarez-Melis

ICML 2022 Workshop on DataPerf: Benchmarking Data for Data-Centric AI.

Budget-Constrained Bounds for Mini-Batch Estimation of Optimal Transport

David Alvarez-Melis, Nicolò Fusi, Lester Mackey, Tal Wagner

arXiv preprint. 2022.

Transfer RL via the Undo Maps Formalism

Abhi Gupta, Ted Moskovitz, David Alvarez-Melis, Aldo Pacchiano

arXiv preprint. 2022.

Dataset Dynamics via Gradient Flows in Probability Space

David Alvarez-Melis, Nicolò Fusi

ICML'21: International Conference on Machine Learning. 2021.

Geometric Dataset Distances via Optimal Transport

David Alvarez-Melis, Nicolò Fusi

NeurIPS'20: Neural Information Processing Systems. 2020.

Earlier version at AutoML @ ICML 2020.

Unsupervised Hierarchy Matching with Optimal Transport over Hyperbolic Spaces

David Alvarez-Melis, Youssef Mroueh, Tommi Jaakkola

AISTATS'20: International Conference on Artificial Intelligence and Statistics. 2020.

Earlier version at OTML: NeurIPS'18 Workshop on Optimal Transport for Machine Learning. Spotlight.

Towards Optimal Transport with Global Invariances

David Alvarez-Melis, Stefanie Jegelka, Tommi S. Jaakkola

AISTATS'19: International Conference on Artificial Intelligence and Statistics. 2019.

Learning Generative Models across Incomparable Spaces

Charlotte Bunne, David Alvarez-Melis, Andreas Krause, Stefanie Jegelka

ICML'19: International Conference on Machine Learning. 2019.

Earlier version at R2L: NeurIPS'18 Workshop on Relational Representation Learning. Best Paper Award.

Functional Transparency for Structured Data: a Game-Theoretic Approach

Guang-He Lee, Wengong Jin, David Alvarez-Melis, Tommi Jaakkola

ICML'19: International Conference on Machine Learning. 2019.

Towards Robust, Locally Linear Deep Networks

Guang-He Lee, David Alvarez-Melis, Tommi S. Jaakkola

ICLR'19: International Conference on Learning Representations. 2019.

Probabilistic Bias Mitigation in Word Embeddings

Hailey Joren, David Alvarez-Melis

NeurIPS 2019 Workshop on Human-Centric Machine Learning.

Optimal Transport in Structured Domains: Algorithms and Applications

David Alvarez-Melis (advisor: Tommi S. Jaakkola)

PhD Thesis, MIT. 2019.

Gromov-Wasserstein Alignment of Word Embedding Spaces

David Alvarez-Melis, Tommi Jaakkola

EMNLP'18: Empirical Methods in Natural Language Processing. 2018. Oral.

Towards Robust Interpretability with Self-Explaining Neural Networks

David Alvarez-Melis, Tommi S. Jaakkola

NeurIPS'18: Neural Information Processing Systems. 2018.

Structured Optimal Transport

David Alvarez-Melis, Tommi Jaakkola, Stefanie Jegelka

AISTATS'18: International Conference on Artificial Intelligence and Statistics. 2018. Oral.

Earlier version at NIPS Workshop on Optimal Transport for Machine Learning, 2017, as Extended Oral.

On the Robustness of Interpretability Methods

David Alvarez-Melis, Tommi S. Jaakkola

ICML 2018 Workshop on Human Interpretability in Machine Learning.

Learning Generative Models across Incomparable Spaces

Charlotte Bunne, David Alvarez-Melis, Andreas Krause, Stefanie Jegelka

NeurIPS 2018 Workshop on Relational Representation Learning. Best Paper Award.

Game-Theoretic Interpretability for Temporal Modeling

Guang-He Lee, David Alvarez-Melis, Tommi S. Jaakkola

ICML 2018 Workshop on Fairness, Accountability, and Transparency in Machine Learning.

Distributional Adversarial Networks

Chengtao Li, David Alvarez-Melis, Keyulu Xu, Stefanie Jegelka, Suvrit Sra

ICLR 2018 Workshop Track.

A causal framework for explaining the predictions of black-box sequence-to-sequence models

David Alvarez-Melis, Tommi S. Jaakkola

EMNLP'17: Empirical Methods in Natural Language Processing. 2017.

Tree-structured decoding with doubly-recurrent neural networks

David Alvarez-Melis, Tommi S. Jaakkola

ICLR'17: International Conference on Learning Representations. 2017.

The Emotional GAN: Priming Adversarial Generation of Art with Emotion

David Alvarez-Melis, Judith Amores

NIPS 2017 Workshop on Machine Learning for Creativity and Design.

Topic Modeling in Twitter: Aggregating Tweets by Conversations

David Alvarez-Melis, Martin Saveski

ICWSM'16: International AAAI Conference on Web and Social Media. 2016. (Short paper.)

Word, graph and manifold embedding from Markov processes

Tatsunori B. Hashimoto, David Alvarez-Melis, Tommi S. Jaakkola

NIPS 2015 Workshop on Nonparametric Methods for Large Scale Representation Learning. Oral.

A translation of “The characteristic function of a random phenomenon” by Bruno de Finetti

David Alvarez-Melis, Tamara Broderick

arXiv preprint. 2015.

The Matrix Multiplicative Weights Algorithm for Domain Adaptation

David Alvarez-Melis (advisor: Mehryar Mohri)

MS Thesis, Courant Institute, NYU. 2013.

Three-operator Splitting with Stale Gradients for Faster Non-linear Optimal Transport

Jacob Lindbäck, David Alvarez-Melis, Mikael Johansson

AISTATS'26: International Conference on Artificial Intelligence and Statistics. 2026.

Curriculum Learning as Transport: Understanding Curricula with Wasserstein Geodesics

Changho Shin, David Alvarez-Melis

COLM'26: Conference on Language Modeling. 2026.

OT on the Map: Quantifying Domain Shifts in Geographic Space

Haoran Zhang, Livia Betti, Konstantin Klemmer, Esther Rolf, David Alvarez-Melis

arXiv preprint. 2026.

What is the Right Notion of Distance between Predict-then-Optimize Tasks?

Paula Rodriguez-Diaz, Lingkai Kong, Kai Wang, David Alvarez-Melis, Milind Tambe

UAI'25: Uncertainty in Artificial Intelligence. 2025.

Bispectral OT: Dataset Comparison using Symmetry-Aware Optimal Transport

Annabel Ma, Kaiying Hou, David Alvarez-Melis, Melanie Weber

NeurIPS 2025 Workshop on Symmetry and Geometry in Neural Representations.

Strongly Isomorphic Neural Optimal Transport Across Incomparable Spaces

Athina Sotiropoulou, David Alvarez-Melis

Proceedings of the Geometry-grounded Representation Learning and Generative Modeling Workshop (GRaM), 2024.

InfoOT: Information Maximizing Optimal Transport

Ching-Yao Chuang, Stefanie Jegelka, David Alvarez-Melis

ICML'23: International Conference on Machine Learning. 2023.

Domain adaptation using optimal transport for invariant learning using histopathology datasets

Kianoush Falahkheirkhah, Alex Xijie Lu, David Alvarez-Melis, Grace Huynh

MIDL'23: Medical Imaging with Deep Learning. 2023.

Generating Synthetic Datasets by Interpolating along Generalized Geodesics

Jiaojiao Fan, David Alvarez-Melis

UAI'23: Uncertainty in Artificial Intelligence. 2023.

Optimizing Functionals on the Space of Probabilities with Input Convex Neural Networks

David Alvarez-Melis, Yair Schiff, Youssef Mroueh

TMLR: Transactions on Machine Learning Research. 2022.

Earlier version at OTML: NeurIPS'21 Workshop on Optimal Transport in Machine Learning. Spotlight.

Hierarchical Optimal Transport for Comparing Histopathology Datasets

Anna Yeaton, Rahul G. Krishnan, Rebecca Mieloszyk, David Alvarez-Melis, Grace Huynh

MIDL'22: Medical Imaging with Deep Learning. 2022.

Interpretable Distribution Shift Detection using Optimal Transport

Neha Hulkund, Nicolò Fusi, Jennifer Wortman Vaughan, David Alvarez-Melis

ICML 2022 Workshop on DataPerf: Benchmarking Data for Data-Centric AI.

Neural Unbalanced Optimal Transport via Cycle-Consistent Semi-Couplings

Frederike Lübeck, Charlotte Bunne, Gabriele Gut, Jacobo Sarabia del Castillo, Lucas Pelkmans, David Alvarez-Melis

NeurIPS 2022 Workshop on Learning Meaningful Representations of Life.

Budget-Constrained Bounds for Mini-Batch Estimation of Optimal Transport

David Alvarez-Melis, Nicolò Fusi, Lester Mackey, Tal Wagner

arXiv preprint. 2022.

Dataset Dynamics via Gradient Flows in Probability Space

David Alvarez-Melis, Nicolò Fusi

ICML'21: International Conference on Machine Learning. 2021.

Geometric Dataset Distances via Optimal Transport

David Alvarez-Melis, Nicolò Fusi

NeurIPS'20: Neural Information Processing Systems. 2020.

Earlier version at AutoML @ ICML 2020.

Unsupervised Hierarchy Matching with Optimal Transport over Hyperbolic Spaces

David Alvarez-Melis, Youssef Mroueh, Tommi Jaakkola

AISTATS'20: International Conference on Artificial Intelligence and Statistics. 2020.

Earlier version at OTML: NeurIPS'18 Workshop on Optimal Transport for Machine Learning. Spotlight.

Towards Optimal Transport with Global Invariances

David Alvarez-Melis, Stefanie Jegelka, Tommi S. Jaakkola

AISTATS'19: International Conference on Artificial Intelligence and Statistics. 2019.

Learning Generative Models across Incomparable Spaces

Charlotte Bunne, David Alvarez-Melis, Andreas Krause, Stefanie Jegelka

ICML'19: International Conference on Machine Learning. 2019.

Earlier version at R2L: NeurIPS'18 Workshop on Relational Representation Learning. Best Paper Award.

Optimal Transport in Structured Domains: Algorithms and Applications

David Alvarez-Melis (advisor: Tommi S. Jaakkola)

PhD Thesis, MIT. 2019.

Gromov-Wasserstein Alignment of Word Embedding Spaces

David Alvarez-Melis, Tommi Jaakkola

EMNLP'18: Empirical Methods in Natural Language Processing. 2018. Oral.

Structured Optimal Transport

David Alvarez-Melis, Tommi Jaakkola, Stefanie Jegelka

AISTATS'18: International Conference on Artificial Intelligence and Statistics. 2018. Oral.

Earlier version at NIPS Workshop on Optimal Transport for Machine Learning, 2017, as Extended Oral.

Learning Generative Models across Incomparable Spaces

Charlotte Bunne, David Alvarez-Melis, Andreas Krause, Stefanie Jegelka

NeurIPS 2018 Workshop on Relational Representation Learning. Best Paper Award.

Can Interpretation Predict Behavior on Unseen Data?

Victoria R. Li, Jenny Kaufmann, Tian Qin, Martin Wattenberg, David Alvarez-Melis, Naomi Saphra

Proceedings of the 9th BlackboxNLP Workshop: Analyzing and Interpreting Neural Networks for NLP, 2026.

Lossy Superposition: Predicting Compositional Errors Without Seeing Inputs

Jennifer Meng Lu, Ruochen Zhang, Isabelle Lee, David Alvarez-Melis, Ellie Pavlick, Naomi Saphra

ICML 2026 Workshop on Mechanistic Interpretability.

Beyond Explainable AI (XAI): An Overdue Paradigm Shift and Post-XAI Research Directions

Saleh Afroogh, Syed Ishtiaque Ahmed, Petra Ahrweiler, David Alvarez-Melis, Mansur Maturidi Arief, Emilia Barakova, Falco J. Bargagli-Stoffi, Erdem Biyik, Hanjie Chen, Xiang 'Anthony' Chen, Robert Alan Clements, Keeley Crockett, Amit Dhurandhar, Fethiye Irmak Dogan, Mollie Dollinger, Motahhare Eslami, Aldo A Faisal, Arya Farahi, Melanie F. Pradier, Saadia Gabriel, Diego Garcia-Olano, Marzyeh Ghassemi, Shaona Ghosh, Hatice Gunes, Ehsan Hajiramezanali, Stefan Haufe, Biwei Huang, Angel Hwang, Md Tauhidul Islam, Junfeng Jiao, Amir-Hossein Karimi, Saber Kazeminasab, Anastasia Kuzminykh, William La Cava, Brian Y. Lim, Xiaofeng Liu, Mohammad R. K. Mofrad, Alicia Parrish, Maria Perez-Ortiz, Shriti Raj, Swabha Swayamdipta, Salmonn Talebi, Kush R. Varshney, Mihaela Vorvoreanu, Lily Weng, Alice Xiang, Yiming Xu, Ding Zhao, Jieyu Zhao

arXiv preprint. 2026.

Adversarial Concept Search: Predicting Compositional Errors From Feature Geometry

Jennifer Meng Lu, Ruochen Zhang, Isabelle Lee, David Alvarez-Melis, Ellie Pavlick, Naomi Saphra

arXiv preprint. 2026.

Interpretable Distribution Shift Detection using Optimal Transport

Neha Hulkund, Nicolò Fusi, Jennifer Wortman Vaughan, David Alvarez-Melis

ICML 2022 Workshop on DataPerf: Benchmarking Data for Data-Centric AI.

From Human Explanation to Model Interpretability: A Framework Based on Weight of Evidence

David Alvarez-Melis, Harmanpreet Kaur, Hal Daumé III, Hanna Wallach, Jennifer Wortman Vaughan

HCOMP'21: AAAI Conference on Human Computation and Crowdsourcing. 2021.

Functional Transparency for Structured Data: a Game-Theoretic Approach

Guang-He Lee, Wengong Jin, David Alvarez-Melis, Tommi Jaakkola

ICML'19: International Conference on Machine Learning. 2019.

Weight of Evidence as a Basis for Human-Oriented Explanations

David Alvarez-Melis, Hal Daumé III, Jennifer Wortman Vaughan, Hanna Wallach

NeurIPS 2019 Workshop on Human-Centric Machine Learning.

Towards Robust Interpretability with Self-Explaining Neural Networks

David Alvarez-Melis, Tommi S. Jaakkola

NeurIPS'18: Neural Information Processing Systems. 2018.

On the Robustness of Interpretability Methods

David Alvarez-Melis, Tommi S. Jaakkola

ICML 2018 Workshop on Human Interpretability in Machine Learning.

Game-Theoretic Interpretability for Temporal Modeling

Guang-He Lee, David Alvarez-Melis, Tommi S. Jaakkola

ICML 2018 Workshop on Fairness, Accountability, and Transparency in Machine Learning.

A causal framework for explaining the predictions of black-box sequence-to-sequence models

David Alvarez-Melis, Tommi S. Jaakkola

EMNLP'17: Empirical Methods in Natural Language Processing. 2017.

Optimal Transport in Structured Domains: Algorithms and Applications

David Alvarez-Melis (advisor: Tommi S. Jaakkola)

PhD Thesis, MIT. 2019.

The Matrix Multiplicative Weights Algorithm for Domain Adaptation

David Alvarez-Melis (advisor: Mehryar Mohri)

MS Thesis, Courant Institute, NYU. 2013.

Lax-Milgram's Theorem: Generalizations and Applications

David Alvarez-Melis (advisor: Carlos Bosch Giral)

BSc Thesis, ITAM. 2011.

Research Group and Advisees

Current Group Members
Former Students, Interns, and Advisees

Teaching

Past, current, and upcoming courses:

Teaching Philosophy
Explaining is understanding.

The following extract is from David Goldstein's book on Feynman:

"Feynman was a truly great teacher. He prided himself on being able to devise ways to explain even the most profound ideas to beginning students. Once, I said to him, "Dick, explain to me, so that I can understand it, why spin one-half particles obey Fermi-Dirac statistics." Sizing up his audience perfectly, Feynman said, "I'll prepare a freshman lecture on it." But he came back a few days later to say, "I couldn't do it. I couldn't reduce it to the freshman level. That means we don't really understand it."

Vitæ

Full CV in PDF (or a shorter Resumé).

  • Harvard University 2023 --
    Assistant Professor

  • Microsoft Research, New England 2021 -- now
    Senior Researcher

  • Microsoft Research, New England 2019 -- 2021
    Postdoctoral Researcher

  • MIT CSAIL 2014 - 2019
    Ph.D. in Computer Science
    Minor: Mathematical Optimization
    Thesis Advisor: Tommi Jaakkola
  • Microsoft Research, NYC Summer 2018
    Research Intern
    Mentors: H. Wallach, J.W. Vaughan, H. Daume III
  • Microsoft Research, Redmond Summer 2016
    Research Intern
    Mentors: S. Yih, M.W. Chang, K. Toutanova, C. Meek
  • IBM Research 2013 - 2014
    Supplemental Researcher
    Speech Recognition Group
  • Courant Institute, NYU 2011 - 2013
    MS in Mathematics
    Thesis Advisor: Mehryar Mohri
  • ITAM 2006 - 2011
    BSc in Applied Mathematics
    Thesis Advisor: Carlos Bosch

Prospective Students/Postdocs/Interns

I am always looking for motivated students and postdocs to join my group. Unfortunately, I am not able to respond to all emails. So, depending on your situation, please follow one of the follwing routes:

  • You are a prospective PhD student: please apply directly to our PhD program and list me as a faculty of interest. Applying to CS or Applied Math are both fine. Admissions are centralized and made by committee.
  • You are a student already at Harvard: send me an email with the subject line "[Prospect] Interested in joining your group (Harvard Student)" and a brief description of your interests and background.
  • You are a student interested in an internship or visiting position : send me an email with the subject line "[Prospect] Interested in joining your group (External Student)" and a brief description of your interests and background. Update: There are no openings for remote students for 2025-2026 in my group!
  • You are a prospective Postdoc: send me an email with the subject line "[Prospect] Interested in joining your group (Postdoc)" with a CV, link to your publications, and a brief description of your interests and background.
In any of the situations above, please include the word "Monge" anywhere in your the main body of your email, to show me you've read this page and are not just sending a generic email to all faculty (I learnt this trick from Finale).

If your email is not formatted as above, my filters won't catch it so I will almost certainly not see it.

Misc

Outside of research, I enjoy running, brewing beer and playing guitar. I also like quotes. Here's a few more:

"We cannot solve our problems with the same thinking we used when we created them." - A. Einstein
"The real danger is not that computers will begin to think like men, but that men will begin to think like computers" - Syndey J. Harris

Meta

This website was built with jekyll based on a template by the one and only Martin Saveski.