LSTM-GNN: A Multi-Channel Model for Molecular Properties Prediction

Ziyang Wen,Han Zhou,Xiumin Shi,Jiaqi Peng,Lu Wang
DOI: https://doi.org/10.1145/3674658.3674677
2024-01-01
Abstract:Predicting molecular properties is crucial in drug design and development. In recent years, machine learning and deep learning models have been widely used for this purpose. However, the insights gained from these methods are often not exhaustive, leading to results that fall short of expectations. In this paper, we propose a multi-channel molecular property prediction model, LSTM-GNN, which orchestrates a symphony of a bidirectional long short-term memory network and a graph attention network-based graph convolutional network to comprehensively unravel the intricacies of molecular structures. By fusing these disparate information channels, our model provides a holistic understanding of molecular properties, enabling more accurate prediction. The effectiveness of our model was evaluated using the HIV and MUV datasets, two publicly available datasets from MoleculeNet. We used the ROC-AUC metric to evaluate the performance of the model. The results show an improvement of 1.8% for HIV and 2.2% for MUV compared to the baseline models. Ablation experiments were performed on the model, demonstrating the complementarity of features in the two channels.
What problem does this paper attempt to address?