Skip to main content

Phylodynamic paramater and model inference using pretrained deep neural networks.

Project description

PhyloDeep

PhyloDeep is a python library for parameter estimation and model selection from phylogenetic trees, based on deep learning.

For more information on the method, including the covered parameter subspace, please refer to the preprint here: bioRxiv

Together with the phylodeep package code (in the folder phylodeep), we provide:

  • all data shown in the preprint (in the folder data_publication)
  • used simulators to train deep learners (in the folder simulators)
  • the tree analyzed in the study as a showcase application (in the folder test_tree_HIV_Zurich, for description and original reference see below)

The data are extensive (including 50.000 testing trees), we thus do not recommend copying the whole repository.

The installation time of the package can be up to several minutes, including downloading dependencies. The run time should be a couple of seconds. The package was tested in Linux (Ubuntu 18.08), Windows 10 and MacOS.

Installation

PhyloDeep is available for Python 3.6 on pip.

Windows

For Windows users, we recommend installing phylodeep via Cygwin environment. First instal Python 3.6 and pip3 from the Cygwin packages. Then install phylodeep:

pip3 install phylodeep

All other platforms

You can install phylodeep for Python 3.6 with or without conda, following the procedures described below:

Installing with conda

Once you have conda installed, create an environment for phylodeep with Python 3.6 (here we name it phyloenv):

conda create --name phyloenv python=3.6

Then activate it:

conda activate phyloenv

Then install phylodeep in it:

pip install phylodeep

Installing without conda

Make sure that Pyhon 3.6 and pip3 are installed, then install phylodeep:

pip3 install phylodeep

Usage

If you installed phylodeep with conda, do not forget to activate the corresponding environment (e.g. phyloenv) before using PhyloDeep:

conda activate phyloenv

We recommend to perform a priori model adequacy first to assess whether the input data resembles well the simulations on which the neural networks were trained.

Here, we use an HIV tree reconstructed from 200 sequences, published in Phylodynamics on local sexual contact networks by Rasmussen et al in PloS Computational Biology in 2017, and that you can find at github and in this repository.

Python

from phylodeep import BD, BDEI, BDSS, SUMSTATS, FULL
from phylodeep.checkdeep import checkdeep
from phylodeep.modeldeep import modeldeep
from phylodeep.paramdeep import paramdeep


path_to_tree = './Zurich.trees'

# set presumed sampling probability
sampling_proba = 0.25

# a priori check for models BD, BDEI, BDSS
checkdeep(path_to_tree, model=BD, outputfile_png='BD_a_priori_check.png')
checkdeep(path_to_tree, model=BDEI, outputfile_png='BDEI_a_priori_check.png')
checkdeep(path_to_tree, model=BDSS, outputfile_png='BDSS_a_priori_check.png')


# model selection
model_BDEI_vs_BD_vs_BDSS = modeldeep(path_to_tree, sampling_proba, vector_representation=FULL)

# the selected model is BDSS

# parameter inference
param_BDSS = paramdeep(path_to_tree, sampling_proba, model=BDSS, vector_representation=FULL, 
                                 ci_computation=True)

# for the interpretation of results, please see below

Command line

# we use here a tree of 200 tips

# a priori model adequacy check: highly recommended
checkdeep -t ./Zurich.trees -m BD -o BD_model_adequacy.png
checkdeep -t ./Zurich.trees -m BDEI -o BDEI_model_adequacy.png
checkdeep -t ./Zurich.trees -m BDSS -o BDSS_model_adequacy.png

# model selection
modeldeep -t ./Zurich.trees -p 0.25 -v CNN_FULL_TREE -o model_selection.csv

# parameter inference
paramdeep -t ./Zurich.trees -p 0.25 -m BDSS -v CNN_FULL_TREE -o HIV_Zurich_BDSS_CNN.csv
paramdeep -t ./Zurich.trees -p 0.25 -m BDSS -v FFNN_SUMSTATS -o HIV_Zurich_BDSS_FFNN_CI.csv -c

Example of output and interpretations

Here, we use an HIV tree reconstructed from 200 sequences, published in Phylodynamics on local sexual contact networks by Rasmussen et al in PloS Computational Biology in 2017, and that you can find at github

The a priori model adequacy check results in the following figures:

BD model adequacy test

BD model adequacy

BDEI model adequacy test

BDEI model adequacy

BDSS model adequacy test

BDSS model adequacy

For the three models (BD, BDEI and BDSS), HIV tree datapoint (represented by a red star) is well inside the data cloud of simulations, where warm colors correspond to high density of simulations. The simulations and HIV tree datapoint were in the form of summary statistics prior to applying PCA. All three models thus pass the model adequacy check.

We then apply model selection using the full tree representation and obtain the following result:

Model Probability BDEI Probability BD Probability BDSS
Predicted probability 0.00 0.00 1.00

The BDSS probability is by far the highest: it is the BDSS model that is confidently selected

Finally, under the selected model BDSS, we predict parameter values together with 95% CIs:

R naught Infectious period X transmission Superspreading fraction
predicted value 1.69 9.78 9.34 0.079
CI 2.5% 1.40 8.12 6.65 0.050
CI 97.5% 2.08 12.26 10 0.133

The point estimates for parameters that are no time related (R naught, X transmission and Superspreading fraction) are well inside the parameter ranges of simulations and thus seem valid (R naught between 1 and 5, x transmission between 3 and 10, superspreading fraction between 0.05 and 0.20).

The time related parameters (infectious and eventually incubation period for BDEI model) are in the same units as the branches of input tree, here in years (9.78 years). The covered parameter space for time related parameters is large due to internal rescaling of all input trees. It should apply to any tree.

Preprint

Voznica J, Zhukova A, Boskova V, Saulnier E, Lemoine F, Moslonka-Lefebvre M, Gascuel O (2021) Deep learning from phylogenies to uncover the transmission dynamics of epidemics. bioRxiv

Project details


Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

phylodeep-0.2.57.tar.gz (37.1 kB view hashes)

Uploaded Source

Built Distribution

phylodeep-0.2.57-py3-none-any.whl (4.5 MB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page