E2CB2former: Effecitve and Explainable Transformer for CB2 Receptor Ligand Activity Prediction

TL;DR

CB2former is an innovative, interpretable transformer-based model that significantly improves the prediction of CB2 receptor ligand activity and identifies key molecular features for drug discovery.

Contribution

This work introduces CB2former, a novel GCN-Transformer framework that enhances predictive accuracy and interpretability in CB2 receptor ligand activity prediction.

Findings

Achieved R squared of 0.685 and AUC of 0.940, outperforming baseline models.

Provided molecular insights through attention weight analysis.

Demonstrated potential for accelerating drug discovery processes.

Abstract

Accurate prediction of CB2 receptor ligand activity is pivotal for advancing drug discovery targeting this receptor, which is implicated in inflammation, pain management, and neurodegenerative conditions. Although conventional machine learning and deep learning techniques have shown promise, their limited interpretability remains a significant barrier to rational drug design. In this work, we introduce CB2former, a framework that combines a Graph Convolutional Network with a Transformer architecture to predict CB2 receptor ligand activity. By leveraging the Transformer's self attention mechanism alongside the GCN's structural learning capability, CB2former not only enhances predictive performance but also offers insights into the molecular features underlying receptor activity. We benchmark CB2former against diverse baseline models including Random Forest, Support Vector Machine, K Nearest Neighbors, Gradient Boosting, Extreme Gradient Boosting, Multilayer Perceptron, Convolutional Neural Network, and Recurrent Neural Network and demonstrate its superior performance with an R squared of 0.685, an RMSE of 0.675, and an AUC of 0.940. Moreover, attention weight analysis reveals key molecular substructures influencing CB2 receptor activity, underscoring the model's potential as an interpretable AI tool for drug discovery. This ability to pinpoint critical molecular motifs can streamline virtual screening, guide lead optimization, and expedite therapeutic development. Overall, our results showcase the transformative potential of advanced AI approaches exemplified by CB2former in delivering both accurate predictions and actionable molecular insights, thus fostering interdisciplinary collaboration and innovation in drug discovery.