About Me

I am a postdoc with Yoshua Bengio studying the causal discovery of cell dynamics at Mila in Montreal. This work is in collaboration with Fabian Theis through the newly formed Helmholtz International Lab, a German-Canadian collaboration.

I completed my PhD in the computer science department at Yale University in 2021 where I was advised by Smita Krishnaswamy. My dissertation can be found here. My research interests are in generative modeling, deep learning, and optimal transport. I’m working on applying ideas from generative modeling, causal discovery, optimal transport, and graph signal processing to understand how cells develop and respond to changing conditions. I’m also interested in generative models for protein design and cofounded a startup to work on these problems.

I grew up in Seattle, Washington, USA, and graduated from Tufts University in 2017 with a BS and MS in computer science. Outside of work, I love sailing and running. I am the 2019 junior North American champion in the 505 class, and I recently ran my first 50 mile race the Vermont 50!

Interests
  • Generative Modeling
  • Optimal Transport
  • Single Cell Dynamics
  • Graph Signal Processing
  • Protein Design
  • Causal Discovery
Education

  • PhD in Computer Science, 2021

    Yale University

  • MPhil in Computer Science, 2020

    Yale University

  • MS in Computer Science, 2017

    Tufts University

  • BS in Computer Science, 2017

    Tufts University

Preprints

Tara Akhound-Sadegh, Jarrid Rector-Brooks, Avishek Joey Bose, Sarthak Mittal, Pablo Lemos, Cheng-Hao Liu, Marcin Sendera, Siamak Ravanbakhsh, Gauthier Gidel, Yoshua Bengio, Nikolay Malkin, Alex Tong (2024). Iterated Denoising Energy Matching for Sampling from Boltzmann Densities. arXiv.

PDF Cite Code

Kirill Neklyudov, Rob Brekelmans, Alex Tong, Lazar Atanackovic, Qiang Liu, Alireza Makhzani (2023). A Computational Framework for Solving Wasserstein Lagrangian Flows. arXiv.

PDF Cite Code

Trang Nguyen, Alex Tong, Kanika Madan, Yoshua Bengio, Dianbo Liu (2023). Causal Inference in Gene Regulatory Networks with GFlowNet: Towards Scalability in Large Systems. arXiv.

PDF Cite

Alex Tong, Manik Kuchroo, Shabarni Gupta, Aarthi Venkat, Beatriz P. San Juan, Laura Rangel, Brandon Zhu, John G. Lock, Christine L. Chaffer, Smita Krishnaswamy (2023). Learning Transcriptional and Regulatory Dynamics Driving Cancer Cell Plasticity Using Neural ODE-Based Optimal Transport. BioRxiv.

PDF Cite DOI

Publications

Avishek Joey Bose, Tara Akhound-Sadegh, Kilian Fatras, Guillaume Huguet, Jarrid Rector-Brooks, Cheng-Hao Liu, Andrei Cristian Nica, Maksym Korablyov, Michael Bronstein, Alex Tong (2024). SE(3)-Stochastic Flow Matching for Protein Backbone Generation. To appear in ICLR (Spotlight).

PDF Cite Code MoML Video Logg Video VantAI Video

Alex Tong, Nikolay Malkin, Kilian Fatras, Lazar Atanackovic, Yanlei Zhang, Guillaume Huguet, Guy Wolf, Yoshua Bengio (2024). Simulation-Free Schrodinger Bridges via Score and Flow Matching. To appear in AISTATS.
Also presented at Frontiers4LCD Workshop @ ICML 2023.

PDF Cite Code

Alex Tong, Frederik Wenkel, Dhananjay Bhaskar, Kincaid Macdonald, Jackson Grady, Michael Perlmutter, Smita Krishnaswamy, Guy Wolf (2024). Learnable Filters for Geometric Scattering Modules. Accepted in IEEE Trans. Sig. Proc..

PDF Cite arXiv

María Ramos Zapatero, Alex Tong, Jahangir Sufi, Petra Vlckova, Ferran Cardoso Rodriguez, Callum Nattress, Xiao Qin, Daniel Hochhauser, Smita Krishnaswamy, Christopher J. Tape (2023). Trellis tree-based analysis reveals stromal regulation of patient-derived organoid drug responses. Cell.

PDF Cite DOI code reproducible code Tutorial Cancer Research Spotlight Yale News News-Medical

Guillaume Huguet, Alex Tong, Edward De Brouwer, Yanlei Zhang, Guy Wolf, Ian Adelstein, Smita Krishnaswamy (2023). A Heat Diffusion Perspective on Geodesic Preserving Dimensionality Reduction. In NeurIPS.

PDF Cite Code

Lazar Atanackovic, Alex Tong, Jason Hartford, Leo J. Lee, Bo Wang, Yoshua Bengio (2023). DynGFN: Bayesian Dynamic Causal Discovery Using Generative Flow Networks. In NeurIPS.
Also presented at Frontiers4LCD Workshop @ NeurIPS 2022.

PDF Cite Code

Michael Perlmutter, Alex Tong, Feng Gao, Guy Wolf, Matthew Hirn (2023). Understanding Graph Neural Networks with Generalized Geometric Scattering Transforms. In SIMODS.

PDF Cite Code

Guillaume Huguet, Alex Tong, María Ramos Zapatero, Christpher J. Tape, Guy Wolf, Smita Krishnaswamy (2023). Geodesic Sinkhorn for Fast and Accurate Optimal Transport on Manifolds. In IEEE MLSP.

PDF Cite

Oluwadamilola Fasina, Guillaume Huguet, Alex Tong, Yanlei Zhang, Guy Wolf, Maximilian Nickel, Ian Adelstein, Smita Krishnaswamy (2023). Neural FIM for Learning Fisher Information Metrics from Point Cloud Data. In ICML.

PDF Cite

Guillaume Huguet, Alex Tong, Bastian Rieck, Jessie Huang, Manik Kuchroo, Matthew Hirn, Guy Wolf, Smita Krishnaswamy (2023). Time-Inhomogeneous Diffusion Geometry and Topology. In SIMODS.

PDF Cite SIMODS

Manik Kuchroo, Marcello DiStasio, Eric Song, Eda Calapkulu, Le Zhang, Maryam Ige, Amar H. Sheth, Abdelilah Majdoubi, Madhvi Menon, Alex Tong, Abhinav Godavarthi, Yu Xing, Scott Gigante, Holly Steach, Jessie Huang, Guillaume Huguet, Janhavi Narain, Kisung You, George Mourgkos, Rahul M. Dhodapkar, Matthew J. Hirn, Bastian Rieck, Guy Wolf, Smita Krishnaswamy, Brian P. Hafler (2023). Single-Cell Analysis Reveals Inflammatory Interactions Driving Macular Degeneration. Nature Communications.

PDF Cite DOI

Alex Tong, Nikolay Malkin, Guillaume Huguet, Yanlei Zhang, Jarrid Rector-Brooks, Kilian Fatras, Guy Wolf, Yoshua Bengio (2023). Improving and Generalizing Flow-Based Generative Models with Minibatch Optimal Transport. Accepted in TMLR
Also presented at Frontiers4LCD Workshop @ ICML 2023.

PDF Cite Code Video

Sam Leone, Alex Tong, Guillaume Huguet, Guy Wolf, Smita Krishnaswamy (2023). Graph Fourier MMD for signals on data graphs. In SAMPTA.

PDF Cite

Guillaume Huguet, D. S. Magruder, Alex Tong, Oluwadamilola Fasina, Manik Kuchroo, Guy Wolf, Smita Krishnaswamy (2022). Manifold Interpolating Optimal-Transport Flows for Trajectory Inference. In NeurIPS.

PDF Cite Code NeurIPS Arxiv

Alex Tong, Guillaume Huguet, Dennis Shung, Amine Natik, Manik Kuchroo, Guillaume Lajoie, Guy Wolf, Smita Krishnaswamy (2022). Embedding Signals on Knowledge Graphs with Unbalanced Diffusion Earth Mover's Distance. In ICASSP.

PDF Cite Code Arxiv

Ana Luisa Perdigoto, Songyan Deng, Katherine C. Du, Manik Kuchroo, Daniel B. Burkhardt, Alex Tong, Gary Israel, Marie E. Robert, Stuart P. Weisberg, Nancy Kirkiles-Smith, Angeliki M. Stamatouli, Harriet M. Kluger, Zoe Quandt, Arabella Young, Mei-Ling Yang, Mark J. Mamula, Jordan S. Pober, Mark S. Anderson, Smita Krishnaswamy, Kevan C. Herold (2022). Immune Cells and Their Inflammatory Mediators Modify Beta Cells and Cause Checkpoint Inhibitor-Induced Diabetes. JCI Insight.

PDF Cite DOI

Brian Palmer Hafler, Manik Kuchroo, Marcello DiStasio, Eric Song, Le Zhang, Maryam Ige, Amar Sheth, Madhvi Menon, Alex Tong, Yu Xing, Scott Gigante, Jessie Huang, George Mourgkos, Smita Krishnaswamy, Rahul Dhodapkar, Guy Wolf (2022). Topological Analysis of Single-Cell Hierarchy Reveals Inflammatory Glial Landscape of Macular Degeneration. Investigative Ophthalmology & Visual Science.

Cite

Manik Kuchroo, Jessie Huang, Patrick Wong, Jean-Christophe Grenier, Dennis Shung, Alex Tong, Carolina Lucas, Jon Klein, Daniel Burkhardt, Scott Gigante, Abhinav Godavarthi, Benjamin Israelow, Tianyang Mao, Ji Eun Oh, Julio Silva, Takehiro Takahashi, Camila D. Odio, Arnau Casanovas-Massana, John Fournier, Yale IMPACT Team, Shelli Farhadian, Charles S. Dela Cruz, Albert I. Ko, F. Perry Wilson, Julie Hussin, Guy Wolf, Akiko Iwasaki, Smita Krishnaswamy (2022). Multiscale PHATE identifies multimodal signatures of COVID-19. Nature Biotechnology.

PDF Cite DOI

Michal Gerasimiuk, Dennis L. Shung, Alexander Tong, Adrian J. Stanley, Machael Shultz, Jeffrey Ngu, Loren Laine, Guy Wolf, Smita Krishnaswamy (2021). MURAL: An Unsupervised Random Forest-Based Embedding for Electronic Health Record Data. In IEEE Big Data.

Cite Code Arxiv Tweetorial

Malte D Luecken, Daniel B Burkhardt, Robrecht Cannoodt, Christopher Lance, Aditi Agrawal, Hananeh Aliee, Ann T Chen, Louise Deconinck, Angela M Detweiler, Alejandro Granados, Shelly Huynh, Laura Isacco, Yang Joon Kim, Sunil Kuppasani, Heiko Lickert, Aaron McGeever, Honey Mekonen, Joaquin Caceres, Maurizio Morri, Michaela Mueller, Norma F Neff, Sheryl Paul, Kaylie Schneider, Scott Steelman, Michael Sterr, Dan J Treacy, Alex Tong, Alexandra-Chloé Villani, Guilin Wang, Jia Yan, Ce Zhang, Angela O Pisco, Smita Krishnaswamy, Fabian J Theis, Jonathan M Bloom (2021). A sandbox for prediction and integration of DNA, RNA, and protein data in single cells. In NeurIPS Datasets and Benchmarks.

PDF Cite Code Competition Dataset

Alex Tong, Guy Wolf, Smita Krishnaswamy (2021). Fixing Bias in Reconstruction-based Anomaly Detection with Lipschitz Discriminators. In Journal of Signal Processing Systems.

PDF Cite Code Slides Video

Alex Tong, Frederik Wenkel, Kincaid Macdonald, Smita Krishnaswamy, Guy Wolf (2021). Data-Driven Learning of Geometric Scattering Networks. In IEEE MLSP.
Also presented at ML4M Workshop @ NeurIPS 2020.

PDF Cite Code Slides Video ArXiv Tweetorial

Manik Kuchroo, Abhinav Godavarthi, Alex Tong, Smita Krishnaswamy, Guy Wolf (2021). Multimodal data visualization and denoising with integrated diffusion. In IEEE MLSP.

PDF Cite Arxiv

Alex Tong, Guillaume Huguet, Amine Natik, Kincaid Macdonald, Manik Kuchroo, Ronald Coifman, Guy Wolf, Smita Krishnaswamy (2021). Diffusion Earth Mover's Distance and Distribution Embeddings. In ICML.
Also presented at LMRL Workshop @ NeurIPS 2020.

PDF Cite Code Poster Slides Video Arxiv ICML Page Tweetorial Workshop Poster

Remi Flamary, Nicolas Courty, Alexandre Gramfort, Mokhtar Z Alaya, Aurelie Boisbunon, Stanislas Chambon, Laetitia Chapel, Adrien Corenflos, Kilian Fatras, Nemo Fournier, Leo Gautheron, Nathalie T H Gayraud, Hicham Janati, Alain Rakotomamonjy, Ievgen Redko, Antoine Rolet, Antony Schutz, Vivien Seguy, Danica J Sutherland, Romain Tavenard, Alex Tong, Titouan Vayer (2021). POT: Python Optimal Transport. In JMLR.

PDF Cite Code

Daniel B. Burkhardt, Jay S. Stanley, Alex Tong, Ana Luisa Perdigoto, Scott A. Gigante, Kevan C. Herold, Guy Wolf, Antonio J. Giraldez, David van Dijk, Smita Krishnaswamy (2021). Quantifying the effect of experimental perturbations in single-cell RNA-sequencing data using graph signal processing. In Nature Biotechnology.

Cite Code DOI BioRxiv

Egbert Castro, Andrew Benz, Alex Tong, Guy Wolf, Smita Krishnaswamy (2020). Uncovering the Folding Landscape of RNA Secondary Structure with Deep Graph Embeddings. In IEEE Big Data.
Also at GRLB Workshop @ ICML 2020.

PDF Cite Code

Alex Tong, Guy Wolf, Smita Krishnaswamy (2020). Fixing Bias in Reconstruction-based Anomaly Detection with Lipschitz Discriminators. In IEEE MLSP.
Best Student Paper Award.

PDF Cite Code Slides Video Yale Daily Article Best Paper Award

Alex Tong, Jessie Huang, Guy Wolf, David van Dijk, Smita Krishnaswamy (2020). TrajectoryNet: A Dynamic Optimal Transport Network for Modeling Cellular Dynamics. In ICML.
Also at LMRL Workshop @ NeurIPS 2019.

PDF Cite Code Project Slides Video Arxiv ICML Page Introductory Talk

Alex Tong, David van Dijk, Jay S. Stanley III, Matthew Amodio, Kristina Yim, Rebecca Muhle, James Noonan, Guy Wolf, Smita Krishnaswamy (2020). Interpretable Neuron Structuring with Graph Spectral Regularization. In IDA
Also presented at RLGM Workshop @ ICLR 2019.

PDF Cite Code Slides Video DOI

David van Dijk, Daniel B. Burkhardt, Matthew Amodio, Alex Tong, Guy Wolf, Smita Krishnaswamy (2019). Finding Archetypal Spaces Using Neural Networks. In IEEE Big Data.

PDF Cite Code DOI Arxiv

James Aspnes, Bernhard Haeupler, Alex Tong, Philipp Woelfel (2018). Allocate-On-Use Space Complexity of Shared-Memory Algorithms. In DISC.

PDF Cite DOI