A hybrid convolutional and recurrent neural network for predicting the function of DNA sequences
DanQ is a hybrid convolutional and recurrent neural network model for predicting the function of DNA de novo from sequence.
Quang, D. and Xie, X. ``DanQ: a hybrid convolutional and recurrent neural network for predicting the function of DNA sequences'', NAR, 2015.
DanQ uses a lot of bleeding edge software packages, and very often these software packages are not backwards compatible when they are updated. Therefore, I have included the most recent version numbers of the software packages for the configuration that worked for me. For the record, I am using Ubuntu Linux 14.04 LTS with an NVIDIA Titan Z GPU.
$ git clone git://github.com/Theano/Theano.git $ cd Theano $ python setup.py develop
keras (0.2.0). Deep learning package that uses Theano backend. I'm in the process of upgrading to version 0.3.0 with the Tensorflow backend.
seya (???). I had to modify the source code of this package a little bit. You can try getting the latest version from Github, but for your convenience I've uploaded my copy of the package. You can install it as follows:
$ tar zxvf DanQ_seya.tar.gz $ cd DanQ_seya $ python setup.py install
I will likely improve DanQ soon and drop the dependency on seya.
CUDA (6.5). Theano can use either CPU or GPU, but using a GPU is almost entirely necessary for a network and dataset this large.
cuDNN (2). Significantly speeds up convolution operations.
You need to first download the training, validation, and testing sets from DeepSEA. You can download the datasets from here. After you have extracted the contents of the tar.gz file, move the 3 .mat files into the data/ folder.
If you have everything installed, you can train a model as follows:
$ python DanQ_train.py
On my system, each epoch took about 6 hours. Whenever the validation loss is reaches a new minimum at the end of a training epoch, the best weights are stored in DanQ_bestmodel.hdf5. I've already uploaded the fully trained model in the hyperlink. You can see motif results, including visualizations and TOMTOM comparisons to known motifs, in the motifs/ folder. Likewise, you can also train a much larger model where about half of the motifs are initialized with JASPAR motifs:
$ python DanQ-JASPAR_train.py
Weights are saved to the fight DanQ-JASPAR_bestmodel.hdf5 whenever the validation loss is lowered. Motif results for this model are also stored in the motifs/ folder.
For your convenience, I've posted the current ROC AUC and PR AUC statistics comparing DanQ and DanQ-JASPAR with DeepSEA.
If you do not want to train a model from scratch and just want to do predictions, I've included test scripts for both models and the file example.h5 in the data folder. This is the same hdf5 file that is generated using the example from the DeepSEA package. The test scripts here have the same input and output formats as the prediction script from DeepSEA, so you can replace the prediction step of the DeepSEA pipeline (i.e. the 2_DeepSEA.lua script) with the test scripts here:
$ python DanQ_test.py data/example.h5 data/example_DanQ_pred.h5