VSRAJHAVEL commited on
Commit
5f0bfe6
·
verified ·
1 Parent(s): 2b8523d

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +150 -0
README.md CHANGED
@@ -1,3 +1,153 @@
1
  ---
 
 
2
  license: mit
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
+ language:
3
+ - en
4
  license: mit
5
+ library_name: pytorch
6
+ tags:
7
+ - chemistry
8
+ - graph-neural-networks
9
+ - drug-discovery
10
+ - rdkit
11
+ - pytorch-geometric
12
+ - moleculenet
13
+ - cheminformatics
14
+ - toxicology
15
+ - admet
16
+ - mpnn
17
+ - gin
18
+ - gcn
19
+ - gat
20
+ datasets:
21
+ - esol
22
+ - bbbp
23
+ - tox21
24
+ metrics:
25
+ - rmse
26
+ - roc_auc
27
+ model-index:
28
+ - name: BioGraph-AI-MPNN-ESOL
29
+ results:
30
+ - task:
31
+ type: tabular-regression
32
+ name: Aqueous Solubility Prediction
33
+ dataset:
34
+ name: ESOL Delaney
35
+ type: esol
36
+ metrics:
37
+ - type: rmse
38
+ value: 1.010
39
+ name: Test RMSE
40
+ - name: BioGraph-AI-MPNN-Tox21
41
+ results:
42
+ - task:
43
+ type: tabular-classification
44
+ name: Multi-Task Nuclear Receptor & Stress Pathway Toxicity
45
+ dataset:
46
+ name: Tox21
47
+ type: tox21
48
+ metrics:
49
+ - type: roc_auc
50
+ value: 0.757
51
+ name: Test ROC-AUC
52
+ - name: BioGraph-AI-RF-BBBP
53
+ results:
54
+ - task:
55
+ type: tabular-classification
56
+ name: Blood-Brain Barrier Penetration
57
+ dataset:
58
+ name: BBBP
59
+ type: bbbp
60
+ metrics:
61
+ - type: roc_auc
62
+ value: 0.853
63
+ name: Test ROC-AUC
64
  ---
65
+
66
+ # BioGraph AI: Deep Graph Neural Networks for Molecular Property Prediction & ADMET Profiling
67
+
68
+ [![GitHub Stars](https://img.shields.io/badge/GitHub-BioGraph--AI-10b981?logo=github)](https://github.com/VSRAJHAVEL/BioGraph-AI)
69
+ [![License: MIT](https://img.shields.io/badge/License-MIT-blue.svg)](https://opensource.org/licenses/MIT)
70
+ [![PyTorch](https://img.shields.io/badge/PyTorch-2.0+-ee4c2c?logo=pytorch)](https://pytorch.org/)
71
+ [![PyG](https://img.shields.io/badge/PyG-PyTorch_Geometric-3b82f6)](https://pyg.org/)
72
+ [![RDKit](https://img.shields.io/badge/Cheminformatics-RDKit-green)](https://www.rdkit.org/)
73
+
74
+ **BioGraph AI** is an enterprise-grade Computer-Aided Drug Design (CADD) and Molecular Machine Learning platform. This repository hosts pre-trained model weights for **15 state-of-the-art graph architectures and classical baselines** evaluated under strict Bemis-Murcko scaffold splitting across three canonical MoleculeNet benchmarks (**ESOL, BBBP, Tox21**).
75
+
76
+ > **Handcrafted by [RAJHAVEL V S](https://vsrajhavel.github.io)**
77
+ > **Portfolio:** [vsrajhavel.github.io](https://vsrajhavel.github.io) | **LinkedIn:** [in/rajhavelvs](https://in.linkedin.com/in/rajhavelvs) | **GitHub:** [@VSRAJHAVEL](https://github.com/VSRAJHAVEL)
78
+
79
+ ---
80
+
81
+ ## 🔬 Benchmark Performance (Bemis-Murcko Scaffold Split 80/10/10)
82
+
83
+ All models were trained, validated, and evaluated on local Intel CPU physical threads. Zero scaffold leakage was mathematically confirmed across all training and test partitions.
84
+
85
+ ### 1. Aqueous Solubility (ESOL — Delaney Regression, $\log S$ in $\text{mol/L}$)
86
+ | Model Architecture | Test RMSE ↓ | Test MAE ↓ | Test $R^2$ ↑ | Checkpoint File |
87
+ | :--- | :---: | :---: | :---: | :--- |
88
+ | **MPNN (Edge-Aware)** | **1.010** | **0.777** | **0.784** | `checkpoints/esol_mpnn_best.pt` |
89
+ | **GIN (Weisfeiler-Lehman)** | 1.054 | 0.812 | 0.764 | `checkpoints/esol_gin_best.pt` |
90
+ | **GCN (Kipf-Welling)** | 1.080 | 0.811 | 0.753 | `checkpoints/esol_gcn_best.pt` |
91
+ | **Baseline (RF + Morgan)** | 1.109 | 0.859 | 0.739 | `checkpoints/esol_baseline_best.pkl` |
92
+ | **GAT (Multi-Head)** | 1.208 | 0.938 | 0.691 | `checkpoints/esol_gat_best.pt` |
93
+
94
+ ### 2. Blood-Brain Barrier Penetration (BBBP — Binary Classification)
95
+ | Model Architecture | Test ROC-AUC ↑ | Test PR-AUC ↑ | Checkpoint File |
96
+ | :--- | :---: | :---: | :--- |
97
+ | **Baseline (RF + Morgan)** | **0.853** | **0.978** | `checkpoints/bbbp_baseline_best.pkl` |
98
+ | **GIN** | 0.723 | 0.936 | `checkpoints/bbbp_gin_best.pt` |
99
+ | **GCN** | 0.719 | 0.951 | `checkpoints/bbbp_gcn_best.pt` |
100
+ | **MPNN** | 0.665 | 0.944 | `checkpoints/bbbp_mpnn_best.pt` |
101
+ | **GAT** | 0.624 | 0.928 | `checkpoints/bbbp_gat_best.pt` |
102
+
103
+ ### 3. Toxicity Profiling (Tox21 — 12 Multi-Task Toxicity Pathways)
104
+ *Evaluated across 12 nuclear receptor (NR) and stress response (SR) pathways with missing label masking.*
105
+ | Model Architecture | Test ROC-AUC ↑ | Test PR-AUC ↑ | Checkpoint File |
106
+ | :--- | :---: | :---: | :--- |
107
+ | **MPNN (Edge-Aware)** | **0.757** | 0.304 | `checkpoints/tox21_mpnn_best.pt` |
108
+ | **GCN** | 0.754 | 0.310 | `checkpoints/tox21_gcn_best.pt` |
109
+ | **GIN** | 0.747 | 0.306 | `checkpoints/tox21_gin_best.pt` |
110
+ | **Baseline (RF)** | 0.730 | **0.349** | `checkpoints/tox21_baseline_best.pkl` |
111
+ | **GAT** | 0.711 | 0.264 | `checkpoints/tox21_gat_best.pt` |
112
+
113
+ ---
114
+
115
+ ## 🚀 Quickstart: Loading Checkpoints in Python
116
+
117
+ ### Installation
118
+ ```bash
119
+ pip install torch torch-geometric rdkit huggingface_hub
120
+ ```
121
+
122
+ ### Download Pretrained Weights
123
+ ```python
124
+ from huggingface_hub import hf_hub_download
125
+ import torch
126
+
127
+ # Download best ESOL MPNN checkpoint
128
+ checkpoint_path = hf_hub_download(
129
+ repo_id="VSRAJHAVEL/BioGraph-AI-Models",
130
+ filename="checkpoints/esol_mpnn_best.pt"
131
+ )
132
+
133
+ state_dict = torch.load(checkpoint_path, map_location="cpu")
134
+ print("Successfully loaded BioGraph AI MPNN checkpoint:", len(state_dict), "weight tensors")
135
+ ```
136
+
137
+ ---
138
+
139
+ ## ���� Architectural Overview
140
+
141
+ 1. **Message Passing Neural Networks (MPNN)**: Formulated with edge-network conditioned message functions $M(h_v, h_w, e_{vw})$ incorporating bond stereo, conjugation, and aromaticity. Top performer for steric-dependent thermodynamic properties.
142
+ 2. **Graph Isomorphism Network (GIN)**: Employs sum-aggregation with multi-layer perceptrons approximating the 1-Weisfeiler-Lehman topological graph isomorphism test.
143
+ 3. **Graph Convolutional Networks (GCN)**: First-order localized spectral convolutions with symmetric normalized Laplacian aggregations.
144
+ 4. **Graph Attention Networks (GAT)**: LeakyReLU-parameterized multi-head attention over bonded neighborhoods.
145
+ 5. **Random Forest Baselines**: 500-estimator ensembles trained on 2048-bit ECFP4 (Morgan radius 2) fingerprints.
146
+
147
+ ---
148
+
149
+ ## 🔗 Project Resources
150
+
151
+ - **Full Application Repository**: [https://github.com/VSRAJHAVEL/BioGraph-AI](https://github.com/VSRAJHAVEL/BioGraph-AI)
152
+ - **Interactive Web App**: Includes 3D Mol* Conformer Studio, ADMET Spider Radar, MMP Bioisosteric Optimization, PCA Chemical Space Explorer, and GNNExplainer Attribution.
153
+ - **License**: MIT License (Copyright 2026 RAJHAVEL V S)