Adapting protein language models and contrastive learning for highly-accurate drug-target interaction prediction. http://conplex.csail.mit.edu
  • Python 97.6%
  • Makefile 2.4%
Find a file
Repository files (latest commit first)
Filename Latest commit message Latest commit date
Samuel Sledzieski 61ab7ad8c0
Update build.yml
Until we figure out how to run tests without cuda on github actions, don't run tests for build
2023-06-12 09:43:35 -07:00
.github Update build.yml 2023-06-12 09:43:35 -07:00
assets/images update readme and tests 2023-04-27 15:44:33 -04:00
config update config and docs 2023-04-26 17:44:17 -04:00
conplex_dti Update versioning and cli parser info 2023-06-12 11:47:39 -04:00
docs add utils to api documentation 2023-04-27 15:45:49 -04:00
tests Update versioning and cli parser info 2023-06-12 11:47:39 -04:00
.editorconfig Initial commit 2023-03-11 12:16:44 -05:00
.gitignore adding cli training and documentation skeleton 2023-04-26 15:53:38 -04:00
.pre-commit-config.yaml Initial commit 2023-03-11 12:16:44 -05:00
.readthedocs.yaml apparently invalid python version? 2023-04-26 15:54:58 -04:00
CODE_OF_CONDUCT.md Initial commit 2023-03-11 12:16:44 -05:00
CONTRIBUTING.md Initial commit 2023-03-11 12:16:44 -05:00
cookiecutter-config-file.yml Initial commit 2023-03-11 12:16:44 -05:00
LICENSE Initial commit 2023-03-11 12:16:44 -05:00
Makefile Update Makefile 2023-06-12 09:34:52 -07:00
poetry.lock Update versioning and cli parser info 2023-06-12 11:47:39 -04:00
pyproject.toml Update versioning and cli parser info 2023-06-12 11:47:39 -04:00
README.md Merge branch 'main' into dev 2023-06-12 08:51:16 -07:00
requirements.txt Update versioning and cli parser info 2023-06-12 11:47:39 -04:00
SECURITY.md Initial commit 2023-03-11 12:16:44 -05:00
setup.cfg Initial commit 2023-03-11 12:16:44 -05:00

ConPLex

ConPLex Schematic

ConPLex Releases PyPI Documentation Status License Code style: black

🚧🚧 Please note that ConPLex v0.1.0 is currently a pre-release and is actively being developed. For the code used to generate our PNAS results, see the manuscript code 🚧🚧

Abstract

Sequence-based prediction of drug-target interactions has the potential to accelerate drug discovery by complementing experimental screens. Such computational prediction needs to be generalizable and scalable while remaining sensitive to subtle variations in the inputs. However, current computational techniques fail to simultaneously meet these goals, often sacrificing performance on one to achieve the others. We develop a deep learning model, ConPLex, successfully leveraging the advances in pre-trained protein language models ("PLex") and employing a novel protein-anchored contrastive co-embedding ("Con") to outperform state-of-the-art approaches. ConPLex achieves high accuracy, broad adaptivity to unseen data, and specificity against decoy compounds. It makes predictions of binding based on the distance between learned representations, enabling predictions at the scale of massive compound libraries and the human proteome. Experimental testing of 19 kinase-drug interaction predictions validated 12 interactions, including four with sub-nanomolar affinity, plus a novel strongly-binding EPHB1 inhibitor (K_D = 1.3nM). Furthermore, ConPLex embeddings are interpretable, which enables us to visualize the drug-target embedding space and use embeddings to characterize the function of human cell-surface proteins. We anticipate ConPLex will facilitate novel drug discovery by making highly sensitive in-silico drug screening feasible at genome scale.

Installation

Install from PyPI

pip install conplex-dti
conplex-dti --help

Compile from Source

git clone https://github.com/samsledje/ConPLex.git
cd ConPLex
conda create -n conplex-dti python=3.9
conda activate conplex-dti
make poetry-download
export PATH=[poetry install location]:PATH
export PYTHON_KEYRING_BACKEND=keyring.backends.null.Keyring
make install
conplex-dti --help

Usage

Download benchmark data sets

...

Run benchmark training

conplex-dti train --run-id TestRun --config config/default_config.yaml

Make predictions with a trained model

...

Visualize co-embedding space

...

Reference

If you use ConPLex, please cite “Contrastive learning in protein language space predicts interactions between drugs and protein targets” by Rohit Singh*, Samuel Sledzieski*, Bryan Bryson, Lenore Cowen and Bonnie Berger, currently in press at PNAS.

TBD .bibtex citation

Manuscript Code

Code used to generate results in the manuscript can be found in the development repository