Molecular Sparse Representation by a 3D Ellipsoid Radial Basis Function Neural Network via L1 Regularization

The three-dimensional structures and shapes of biomolecules provide essential information about their interactions and functions. Unfortunately, the computational cost of bio-molecular shape representation is an active challenge which increases rapidly as the number of atoms increase. Recent developments in sparse representation and deep learning have shown significant improvements in terms of time and space. A sparse representation of molecular shape is also useful in various other applications, such as molecular structure alignment, docking, and coarse-grained molecular modeling. We have developed an ellipsoid radial basis function neural network (ERBFNN) and an algorithm for sparsely representing molecular shape. To evaluate a sparse representation model of molecular shape, the Gaussian density map of the molecule is approximated using ERBFNN with a relatively small number of neurons. The deep learning models were trained by optimizing a nonlinear loss function with L1 regularization. Experimental results reveal that our algorithm can represent the original molecular shape with a relatively higher accuracy and fewer scale of ERBFNN. Our network in principle is applicable to the multiresolution sparse representation of molecular shape and coarse-grained molecular modeling. Executable files are available at https://github.com/SGUI-LSEC/SparseGaussianMolecule. The program was implemented in PyTorch and was run on Linux.