298 Publications

pmwd: A Differentiable Cosmological Particle-Mesh N-body Library

Y. Li, L. Lu, C. Modi, Drew Jamieson, Yucheng Zhang, Yu Feng, W. Zhou, Ngai Pok Kwan, François Lanusse, L. Greengard

The formation of the large-scale structure, the evolution and distribution of galaxies, quasars, and dark matter on cosmological scales, requires numerical simulations. Differentiable simulations provide gradients of the cosmological parameters, that can accelerate the extraction of physical information from statistical analyses of observational data. The deep learning revolution has brought not only myriad powerful neural networks, but also breakthroughs including automatic differentiation (AD) tools and computational accelerators like GPUs, facilitating forward modeling of the Universe with differentiable simulations. Because AD needs to save the whole forward evolution history to backpropagate gradients, current differentiable cosmological simulations are limited by memory. Using the adjoint method, with reverse time integration to reconstruct the evolution history, we develop a differentiable cosmological particle-mesh (PM) simulation library pmwd (particle-mesh with derivatives) with a low memory cost. Based on the powerful AD library JAX, pmwd is fully differentiable, and is highly performant on GPUs.

Show Abstract

Learning Feynman Diagrams with Tensor Trains

Yuriel Nunez-Fernandez, Matthieu Jeannin, Philipp T. Dumitrescu, Thomas Kloss, J. Kaye, Olivier Parcollet, Xavier Waintal

We use tensor network techniques to obtain high-order perturbative diagrammatic expansions for the quantum many-body problem at very high precision. The approach is based on a tensor train parsimonious representation of the sum of all Feynman diagrams, obtained in a controlled and accurate way with the tensor cross interpolation algorithm. It yields the full time evolution of physical quantities in the presence of any arbitrary time-dependent interaction. Our benchmarks on the Anderson quantum impurity problem, within the real-time nonequilibrium Schwinger-Keldysh formalism, demonstrate that this technique supersedes diagrammatic quantum Monte Carlo by orders of magnitude in precision and speed, with convergence rates \(1/N2\) or faster, where N is the number of function evaluations. The method also works in parameter regimes characterized by strongly oscillatory integrals in high dimension, which suffer from a catastrophic sign problem in quantum Monte Carlo calculations. Finally, we also present two exploratory studies showing that the technique generalizes to more complex situations: a double quantum dot and a single impurity embedded in a two-dimensional lattice.

Show Abstract

libdlr: Efficient imaginary time calculations using the discrete Lehmann representation

J. Kaye, Kun Chen , Hugo U. R. Strand

We introduce libdlr, a library implementing the recently introduced discrete Lehmann representation (DLR) of imaginary time Green's functions. The DLR basis consists of a collection of exponentials chosen by the interpolative decomposition to ensure stable and efficient recovery of Green's functions from imaginary time or Matsubara frequency samples. The library provides subroutines to build the DLR basis and grids, and to carry out various standard operations. The simplicity of the DLR makes it straightforward to incorporate into existing codes as a replacement for less efficient representations of imaginary time Green's functions, and libdlr is intended to facilitate this process. libdlr is written in Fortran, provides a C header interface, and contains a Python module pydlr. We also introduce a stand-alone Julia implementation, Lehmann.jl

Show Abstract

High-Resolution EEG Source Reconstruction with Boundary Element Fast Multipole Method Using Reciprocity Principle and TES Forward Model Matrix

William A. Wartman, Tommi Raij, M. Rachh, Fa-Hsuan Lin, Konstantin Weise, Thomas Knoesche, Burkhard Maess, Carsten H. Wolters, Aapo R. Nummenmaa, Sergey N. Makaroff, Matti Hämäläinen

Background Accurate high-resolution EEG source reconstruction (localization) is important for several tasks, including rigorous and rapid mental health screening.Objective The present study has developed, validated, and applied a new source localization algorithm utilizing a charge-based boundary element fast multipole method (BEM-FMM) coupled with the Helmholtz reciprocity principle and the transcranial electrical stimulation (TES) forward solution.Methods The unknown cortical dipole density is reconstructed over the entire cortical surface by expanding into global basis functions in the form of cortical fields of active TES electrode pairs. These pairs are constructed from the reading electrodes. An analog of the minimum norm estimation (MNE) equation is obtained after substituting this expansion into the reciprocity principle written in terms of measured electrode voltages. Delaunay (geometrically balanced) triangulation of the electrode cap is introduced first. Basis functions for all electrode pairs connected by the edges of a triangular mesh are precomputed and stored in memory. A smaller set of independent basis functions is then selected and employed at every time instant. This set is based on the highest voltage differences measured.Results The method is validated against the classic, yet challenging problem of median nerve stimulation and the tangential cortical sources located at the posterior wall of the central sulcus for an N20/P20 peak (2 scanned subjects). The method is further applied to perform source reconstruction of synthesized tangential cortical sources located at the posterior wall of the central sulcus (12 different subjects). In the second case, an average source reconstruction error of 7 mm is reported for the best possible noiseless scenario.Conclusions Once static preprocessing with TES electrodes has been done (the basis functions have been computed), our method requires fractions of a second to complete the accurate high-resolution source localization.Competing Interest StatementThe authors have declared no competing interest.

Show Abstract
bioRxiv
November 1, 2022

Catching actin proteins in action

P. Cossio, Glen M. Hocky

Two groups have visualized actin — the protein polymer that gives cells their shape — at high resolution. The structures provide in-depth views of the polymer as it adopts fleeting states and undergoes conformational changes.

Show Abstract

Protocol for iterative optimization of modified peptides bound to protein targets

Rodrigo Ochoa, P. Cossio, Thomas Fox

Peptides are commonly used as therapeutic agents. However, they suffer from easy degradation and instability. Replacing natural by non-natural amino acids can avoid these problems, and potentially improve the affinity towards the target protein. Here, we present a computational pipeline to optimize peptides based on adding non-natural amino acids while improving their binding affinity. The workflow is an iterative computational evolution algorithm, inspired by the PARCE protocol, that performs single-point mutations on the peptide sequence using modules from the Rosetta framework. The modifications can be guided based on the structural properties or previous knowledge of the biological system. At each mutation step, the affinity to the protein is estimated by sampling the complex conformations and applying a consensus metric using various open protein-ligand scoring functions. The mutations are accepted based on the score differences, allowing for an iterative optimization of the initial peptide. The sampling/scoring scheme was benchmarked with a set of protein-peptide complexes where experimental affinity values have been reported. In addition, a basic application using a known protein-peptide complex is also provided. The structure- and dynamic-based approach allows users to optimize bound peptides, with the option to personalize the code for further applications. The protocol, called mPARCE, is available at: https://github.com/rochoa85/mPARCE/.

Show Abstract

Forced and spontaneous symmetry breaking in cell polarization

Pearson Miller , D. Fortunato , Cyrill Muratov, L. Greengard, S. Shvartsman

How does breaking the symmetry of an equation alter the symmetry of its solutions? Here, we systematically examine how reducing underlying symmetries from spherical to axisymmetric influences the dynamics of an archetypal model of cell polarization, a key process of biological spatial self-organization. Cell polarization is characterized by nonlinear and non-local dynamics, but we overcome the theory challenges these traits pose by introducing a broadly applicable numerical scheme allowing us to efficiently study continuum models in a wide range of geometries. Guided by numerical results, we discover a dynamical hierarchy of timescales that allows us to reduce relaxation to a purely geometric problem of area-preserving geodesic curvature flow. Through application of variational results, we analytically construct steady states on a number of biologically relevant shapes. In doing so, we reveal non-trivial solutions for symmetry breaking.

Show Abstract

A PDE-free, neural network-based eddy viscosity model coupled with RANS equations

Ruiying Xu, Xu-Hui Zhou, J. Han, Richard P Dwight, Heng Xiao

In fluid dynamics, constitutive models are often used to describe the unresolved turbulence and to close the Reynolds averaged Navier–Stokes (RANS) equations. Traditional PDE-based constitutive models are usually too rigid to calibrate with a large set of high-fidelity data. Moreover, commonly used turbulence models are based on the weak equilibrium assumption, which cannot adequately capture the nonlocal physics of turbulence. In this work, we propose using a vector-cloud neural network (VCNN) to learn the nonlocal constitutive model, which maps a regional mean flow field to the local turbulence quantities without solving the transport PDEs. The network is strictly invariant to coordinate translation, rotation, and uniform motion, as well as ordering of the input points. The VCNN-based nonlocal constitutive model is trained and evaluated on flows over a family of parameterized periodic hills. Numerical results demonstrate its predictive capability on target turbulence quantities of turbulent kinetic energy k and dissipation ɛ. More importantly, we investigate the robustness and stability of the method by coupling the trained model back to RANS solver. The solver shows good convergence with the simulated velocity field comparable to that based on k–ɛ model when starting from a reasonable initial condition. This study, as a proof of concept, highlights the feasibility of using a nonlocal, frame-independent, neural network-based constitutive model to close the RANS equations, paving the way for the further emulation of the Reynolds stress transport models.

Show Abstract

Toward a Geometrical Understanding of Self-supervised Contrastive Learning

Romain Cosentino, A. Sengupta, Salman Avestimehr, Mahdi Soltanolkotabi, Antonio Ortega, Ted Willke, M. Tepper

Self-supervised learning (SSL) is currently one of the premier techniques to create data representations that are actionable for transfer learning in the absence of human annotations. Despite their success, the underlying geometry of these representations remains elusive, which obfuscates the quest for more robust, trustworthy, and interpretable models. In particular, mainstream SSL techniques rely on a specific deep neural network architecture with two cascaded neural networks: the encoder and the projector. When used for transfer learning, the projector is discarded since empirical results show that its representation generalizes more poorly than the encoder's. In this paper, we investigate this curious phenomenon and analyze how the strength of the data augmentation policies affects the data embedding. We discover a non-trivial relation between the encoder, the projector, and the data augmentation strength: with increasingly larger augmentation policies, the projector, rather than the encoder, is more strongly driven to become invariant to the augmentations. It does so by eliminating crucial information about the data by learning to project it into a low-dimensional space, a noisy estimate of the data manifold tangent plane in the encoder representation. This analysis is substantiated through a geometrical perspective with theoretical and empirical results.

Show Abstract

Generation and motion of interfaces in a mass-conserving reaction-diffusion system

Pearson W. Miller, D. Fortunato , Matteo Novaga, Stanislav Y. Shvartsman, Cyrill B. Muratov

Reaction-diffusion models with nonlocal constraints naturally arise as limiting cases of coupled bulk-surface models of intracellular signalling. In this paper, a minimal, mass-conserving model of cell-polarization on a curved membrane is analyzed in the limit of slow surface diffusion. Using the tools of formal asymptotics and calculus of variations, we study the characteristic wave-pinning behavior of this system on three dynamical timescales. On the short timescale, generation of an interface separating high- and low-concentration domains is established under suitable conditions. Intermediate timescale dynamics is shown to lead to a uniform growth or shrinking of these domains to sizes which are fixed by global parameters. Finally, the long time dynamics reduces to area-preserving geodesic curvature flow that may lead to multi-interface steady state solutions. These results provide a foundation for studying cell polarization and related phenomena in biologically relevant geometries.

Show Abstract
  • Previous Page
  • Viewing
  • Next Page
Advancing Research in Basic Science and MathematicsSubscribe to Flatiron Institute announcements and other foundation updates