Skip to content

DENG-MIT/CRNN

Repository files navigation

CRNN (Chemical Reaction Neural Network)

CRNN is an interpretable neural network architecture for autonomously inference chemical reaction pathways in various chemical systems. It is designed based on the following two fundamental physics laws: the Law of Mass Action and Arrhenius Law. It is also possible to incorporate other physics laws to adapt CRNN to a specific domain.

You can find the common questions regarding CRNN in the FAQs.

Structure of this repo

This repo provides the case studies presented in the original CRNN paper as well as ongoing preliminary results on other systems. Currently, we are actively working on the following systems:

Inside each folder, such as case 1/2/3, you will find at least two Julia codes. One for training and the other one for weight pruning. Those two files are identical, except that the weight pruning includes a function to prune the CRNN weights to further encourage sparsity.

Get Started

Have a look at the code for case 2. The script consists of the following parts:

  • Hyper-parameter settings
  • Generate synthesized data
  • Define the neural ODE problem
  • Train CRNN using ADAM

We strongly recommend using Julia 1.6 with the CRNN code included in this repository. Newer versions may lead to indexing issues and convergence to incorrect mechanisms.

Cite

Ji, Weiqi, and Deng, Sili. "Autonomous Discovery of Unknown Reaction Pathways from Data by Chemical Reaction Neural Network." The Journal of Physical Chemistry A, (2021), 125, 4, 1082–1092, acs/arXiv