-
Capturing dynamical correlations using implicit neural representations
Authors:
Sathya Chitturi,
Zhurun Ji,
Alexander Petsch,
Cheng Peng,
Zhantao Chen,
Rajan Plumley,
Mike Dunne,
Sougata Mardanya,
Sugata Chowdhury,
Hongwei Chen,
Arun Bansil,
Adrian Feiguin,
Alexander Kolesnikov,
Dharmalingam Prabhakaran,
Stephen Hayden,
Daniel Ratner,
Chun**g Jia,
Youssef Nashed,
Joshua Turner
Abstract:
The observation and description of collective excitations in solids is a fundamental issue when seeking to understand the physics of a many-body system. Analysis of these excitations is usually carried out by measuring the dynamical structure factor, S(Q, $ω$), with inelastic neutron or x-ray scattering techniques and comparing this against a calculated dynamical model. Here, we develop an artific…
▽ More
The observation and description of collective excitations in solids is a fundamental issue when seeking to understand the physics of a many-body system. Analysis of these excitations is usually carried out by measuring the dynamical structure factor, S(Q, $ω$), with inelastic neutron or x-ray scattering techniques and comparing this against a calculated dynamical model. Here, we develop an artificial intelligence framework which combines a neural network trained to mimic simulated data from a model Hamiltonian with automatic differentiation to recover unknown parameters from experimental data. We benchmark this approach on a Linear Spin Wave Theory (LSWT) simulator and advanced inelastic neutron scattering data from the square-lattice spin-1 antiferromagnet La$_2$NiO$_4$. We find that the model predicts the unknown parameters with excellent agreement relative to analytical fitting. In doing so, we illustrate the ability to build and train a differentiable model only once, which then can be applied in real-time to multi-dimensional scattering data, without the need for human-guided peak finding and fitting algorithms. This prototypical approach promises a new technology for this field to automatically detect and refine more advanced models for ordered quantum systems.
△ Less
Submitted 8 April, 2023;
originally announced April 2023.
-
CryoAI: Amortized Inference of Poses for Ab Initio Reconstruction of 3D Molecular Volumes from Real Cryo-EM Images
Authors:
Axel Levy,
Frédéric Poitevin,
Julien Martel,
Youssef Nashed,
Ariana Peck,
Nina Miolane,
Daniel Ratner,
Mike Dunne,
Gordon Wetzstein
Abstract:
Cryo-electron microscopy (cryo-EM) has become a tool of fundamental importance in structural biology, hel** us understand the basic building blocks of life. The algorithmic challenge of cryo-EM is to jointly estimate the unknown 3D poses and the 3D electron scattering potential of a biomolecule from millions of extremely noisy 2D images. Existing reconstruction algorithms, however, cannot easily…
▽ More
Cryo-electron microscopy (cryo-EM) has become a tool of fundamental importance in structural biology, hel** us understand the basic building blocks of life. The algorithmic challenge of cryo-EM is to jointly estimate the unknown 3D poses and the 3D electron scattering potential of a biomolecule from millions of extremely noisy 2D images. Existing reconstruction algorithms, however, cannot easily keep pace with the rapidly growing size of cryo-EM datasets due to their high computational and memory cost. We introduce cryoAI, an ab initio reconstruction algorithm for homogeneous conformations that uses direct gradient-based optimization of particle poses and the electron scattering potential from single-particle cryo-EM data. CryoAI combines a learned encoder that predicts the poses of each particle image with a physics-based decoder to aggregate each particle image into an implicit representation of the scattering potential volume. This volume is stored in the Fourier domain for computational efficiency and leverages a modern coordinate network architecture for memory efficiency. Combined with a symmetrized loss function, this framework achieves results of a quality on par with state-of-the-art cryo-EM solvers for both simulated and experimental data, one order of magnitude faster for large datasets and with significantly lower memory requirements than existing methods.
△ Less
Submitted 30 August, 2022; v1 submitted 15 March, 2022;
originally announced March 2022.
-
Combining imitation and deep reinforcement learning to accomplish human-level performance on a virtual foraging task
Authors:
Vittorio Giammarino,
Matthew F Dunne,
Kylie N Moore,
Michael E Hasselmo,
Chantal E Stern,
Ioannis Ch. Paschalidis
Abstract:
We develop a simple framework to learn bio-inspired foraging policies using human data. We conduct an experiment where humans are virtually immersed in an open field foraging environment and are trained to collect the highest amount of rewards. A Markov Decision Process (MDP) framework is introduced to model the human decision dynamics. Then, Imitation Learning (IL) based on maximum likelihood est…
▽ More
We develop a simple framework to learn bio-inspired foraging policies using human data. We conduct an experiment where humans are virtually immersed in an open field foraging environment and are trained to collect the highest amount of rewards. A Markov Decision Process (MDP) framework is introduced to model the human decision dynamics. Then, Imitation Learning (IL) based on maximum likelihood estimation is used to train Neural Networks (NN) that map human decisions to observed states. The results show that passive imitation substantially underperforms humans. We further refine the human-inspired policies via Reinforcement Learning (RL) using the on-policy Proximal Policy Optimization (PPO) algorithm which shows better stability than other algorithms and can steadily improve the policies pretrained with IL. We show that the combination of IL and RL can match human results and that good performance strongly depends on combining the allocentric information with an egocentric representation of the environment.
△ Less
Submitted 6 December, 2022; v1 submitted 11 March, 2022;
originally announced March 2022.