Publications

(*) denotes equal contribution

Accepted

2026

  1. Effectiveness of Curriculum Learning Depends on Reward Sparsity and Competing Optima
    John J. Vastola, Ann Huang, Satpreet H. Singh, Samuel J. Gershman, and Kanaka Rajan
    In Advances in Neural Information Processing Systems 2026
  2. Temporal Smoothness Constraints on Efficient Neurobiological Codes Imply Temporal Specialization
    John J. Vastola, and Samuel J. Gershman
    In Advances in Neural Information Processing Systems 2026
  3. Quantifying the failure of symmetry learning in linear diffusion models due to initialization and finite-sample effects
    John J. Vastola, and Kanaka Rajan
    In NeurIPS 2026 Workshop on Symmetry and Geometry in Neural Representations 2026
  4. Solvable models of learning to pursue a moving target
    John J. Vastola, and Kanaka Rajan
    Reinforcement Learning Journal 2026

Publications

2026

  1. Implications of hierarchical Markov models of behavior: on irreversibility, predictability, and dimensionality
    John J. Vastola, and Kanaka Rajan
    In 9th Annual Conference on Cognitive Computational Neuroscience 2026

2025

  1. Gradient Descent as Loss Landscape Navigation: a Normative Framework for Deriving Learning Rules
    John J. Vastola, Samuel J Gershman, and Kanaka Rajan
    In Advances in Neural Information Processing Systems 2025
  2. A Variational Manifold Embedding Framework for Nonlinear Dimensionality Reduction
    John J. Vastola, Samuel J. Gershman, and Kanaka Rajan
    In NeurIPS 2025 Workshop on Symmetry and Geometry in Neural Representations 2025
  3. Deep RL Needs Deep Behavior Analysis: Exploring Implicit Planning by Model-Free Agents in Open-Ended Environments
    Riley Simmons-Edler, Ryan P. Badman, Felix Berg, Raymond Chua, John J. Vastola, Joshua Lunger, William Qian, and Kanaka Rajan
    In Advances in Neural Information Processing Systems 2025
  4. Monod: model-based discovery and integration through fitting stochastic transcriptional dynamics to single-cell sequencing data
    Gennady Gorin, Tara Chari, Maria Carilli, John J. Vastola, and Lior Pachter
    Nature Methods Nov 2025
  5. Generalization through variance: how noise shapes inductive biases in diffusion models
    John J. Vastola
    In The Thirteenth International Conference on Learning Representations Nov 2025
  6. Dynamical symmetries in the fluctuation-driven regime: an application of Noether’s theorem to noisy dynamical systems
    John J. Vastola
    In Proceedings of the 3rd NeurIPS Workshop on Symmetry and Geometry in Neural Representations Nov 2025

2024

  1. The Unreasonable Effectiveness of Gaussian Score Approximation for Diffusion Models and its Applications
    Binxu Wang, and John J. Vastola
    Transactions on Machine Learning Research Nov 2024
  2. Optimal packing of attractor states in neural representations
    John J. Vastola
    In Proceedings of the 2nd NeurIPS Workshop on Symmetry and Geometry in Neural Representations Nov 2024

2023

  1. Studying stochastic systems biology of the cell with single-cell genomics data
    Gennady Gorin, John J. Vastola, and Lior Pachter
    Cell Systems Oct 2023
  2. Causal inference during closed-loop navigation: parsing of self- and object-motion
    Jean-Paul Noel, Johannes Bill, Haoran Ding, John J. Vastola, Gregory C. DeAngelis, Dora E. Angelaki, and Jan Drugowitsch
    Philosophical Transactions of the Royal Society B: Biological Sciences Oct 2023

2022

  1. Interpretable and tractable models of transcriptional noise for the rational design of single-molecule quantification experiments
    Gennady Gorin*, John J. Vastola*, Meichen Fang, and Lior Pachter
    Nature Communications Dec 2022
  2. Is the information geometry of probabilistic population codes learnable?
    John J. Vastola, Zach Cohen, and Jan Drugowitsch
    In NeurIPS 2022 Workshop on Symmetry and Geometry in Neural Representations Dec 2022

2021

  1. Solving the chemical master equation for monomolecular reaction systems and beyond: a Doi-Peliti path integral view
    John J. Vastola
    Journal of Mathematical Biology Oct 2021

2020

  1. Chemical Langevin equation: A path-integral view of Gillespie’s derivation
    John J. Vastola, and William R. Holmes
    Phys. Rev. E Mar 2020

Submitted

2026

  1. A theory of noisy linear recurrent neural network learning dynamics
    John J. Vastola, and Kanaka Rajan
    2026
  2. On the stability of learning
    John J. Vastola
    2026
  3. Field-theoretic diffusion models and sampling as spatial pattern formation
    John J. Vastola, and Binxu Wang
    2026
  4. Spatial Reward Sparsity Drives Slow Policy Learning While Temporal Reward Sparsity Delays Refinement
    John J. Vastola, and Kanaka Rajan
    2026
  5. Curriculum learning in brains and machines
    John J. Vastola, Andrew Saxe, Christopher Summerfield, and Kanaka Rajan
    Submitted to Nature Reviews Neuroscience 2026

2025

  1. Storing long-lived memories via molecular error correction: a minimal mathematical model of Crick’s memory switch
    John J. Vastola, Tejas Ramdas, and Samuel J. Gershman
    bioRxiv 2025