Skip to main content

DeepRM: Deep Learning for RNA Modification Detection using Nanopore Direct RNA Sequencing

Project description

DeepRM

Deep learning for RNA Modification

GitHub Repo CC BY-NC-SA 4.0 GitHub Repo stars GitHub last commit GitHub code size in bytes GitHub contributors GitHub language count

deeprm.png

Table of Contents

✨ Introduction

DeepRM is a deep learning-based framework for RNA modification detection using Nanopore direct RNA sequencing. This repository contains the source code for training and running DeepRM.

🎯 Key Features

  • High accuracy: Achieves state-of-the-art accuracy in RNA modification detection and stoichiometry measurement.
  • Single-molecule resolution: Provides single-molecule level predictions for RNA modifications.
  • End-to-end pipeline: Easy-to-use pipeline from raw reads to site-level predictions.
  • Customizable: Supports training of custom models.

📦 Installation

Prerequisites

  • Linux x86_64
  • Python 3.9+
  • Pytorch 2.0+

Optional

  • Torchmetrics 0.9.0+ (only for training)

    • python -m pip install torchmetrics
      
  • Dorado 0.7.3+ (optional, for basecalling)

  • SAMtools 1.16.1+ (optional, for BAM file processing)

  • Python package requirements are listed in requirements.txt and will be installed automatically when you install DeepRM.

Installation options

  • Estimated time: ~10 minutes
  1. Install via PIP (recommended)
python -m pip install deeprm
  1. Install from source (GitHub)
git clone https://github.com/vadanamu/deeprm
cd deeprm
python -m pip install -U pip
python -m pip install -e .
  • If installation fails on old OS (e.g., CentOS 7) due to NumPy, you can try installing older versions of NumPy first:
  •  python -m pip install "numpy<2.3.0,>2.0.0"
     python -m pip install -e .
    

Verify Installation

deeprm --version
deeprm check
  • If everything is installed correctly, you should see the version of DeepRM and a message indicating that the installation is successful.
  • If you encounter CUDA or torch-related errors, make sure you have installed the correct version of PyTorch with CUDA support.

Build from Source

  • DeepRM can use a C++-based preprocessing tool for acceleration, which is both provided as a precompiled binary and source code.
  • Depending on your system configuration, you may need to build the C++ preprocessing tool from source, located in the cpp directory of the DeepRM repository.
  • Please refer to the cpp/README.md page for detailed build instructions.

🚀 Quickstart

  • For demonstration purposes, you can use examples POD5 and BAM files provided in the examples directory of the repository.
  • You can also use your own POD5 and BAM files.

RNA Modification Detection

  • Estimated time: ~1 hours

1️⃣ Prepare data

deeprm call prep -p inference_example.pod5 -b inference_example.bam -o <prep_dir>
  • (Alternative) To supply your own POD5 file:
    dorado basecaller --reference <ref_fasta> --min-qscore 0 --emit-moves rna004_130bps_sup@v5.0.0 <pod5_dir> | \
    tee <bam_path> | deeprm call prep -p <pod5_dir> -b - -o <prep_dir>
    
    • If Dorado fails due to "illegal memory access", try adding --chunksize <chunk_size> option (e.g., chunk_size=12000).

2️⃣ Run inference

deeprm call run -b inference_example.bam -i <prep_dir> -o <pred_dir> -s 1000
  • Adjust the -s (batch size) parameter according to your GPU memory capacity (default: 10000).
  • Expected output file:
    • Site-level detection result file (.bed)
    • Molecule-level detection result file (.npz)

Model Training

  • Estimated time: ~1 hours

1️⃣ Prepare unmodified & modified training data

deeprm train prep -p training_a_example.pod5 -b training_a_example.bam -o <prep_dir>/a
deeprm train prep -p training_m6a_example.pod5 -b training_m6a_example.bam -o <prep_dir>/m6a

2️⃣ Compile training data

deeprm train compile -n <prep_dir>/a/data -p <prep_dir>/m6a/data -o <prep_dir>/compiled

3️⃣ Run training

deeprm train run -d <prep_dir>/compiled -o <output_dir> --batch 64
  • Adjust the --batch parameter according to your GPU memory capacity (default: 1024).
  • Expected output file:
    • Trained DeepRM model file (.pt)

💻 Usage

Inference usage

deeprm_inference_pipeline.png

Prepare Data

Accelerated preparation (recommended, default)
  • This method uses precompiled C++ binary for accelerating the preprocessing step.
dorado basecaller --reference <ref_fasta> --min-qscore 0 --emit-moves rna004_130bps_sup@v5.0.0 <pod5_dir> | \
tee <bam_path> | deeprm call prep -p <pod5_dir> -b - -o <prep_dir>
  • If Dorado fails due to "illegal memory access", try adding --chunksize <chunk_size> option (e.g., chunk_size=12000).
  • If the precompiled binary does not work on your system, please refer to the cpp/README.md page for detailed build instructions.
  • Adjust the -g (--filter-flag) parameter according to your needs. If using a genomic reference, you may want to use -g 260.
Sequential preparation
  • This method is slower than the accelerated preparation method, but is supported for cases such as:

    • The POD5 files are already basecalled to BAM files with move tags.
    • You want to run basecalling and preprocessing in separate machines.
  • Basecall the POD5 files to BAM files with move tags (skip if already done):

    • If Dorado fails due to "illegal memory access", try adding --chunksize <chunk_size> option (e.g., chunk_size=12000).
dorado basecaller --reference <reference_path> --min-qscore 0 --emit-moves rna004_130bps_sup@v5.0.0 <pod5_dir> > <raw_bam_path>"
  • Filter, sort, and index the BAM files:
    • Adjust the -F parameter according to your needs. If using a genomic reference, you may want to use -F 260.
samtools view -@ <threads> -bh -F 276 -o <bam_path> <raw_bam_path>
samtools sort -@ <threads> -o <bam_path> <bam_path>
samtools index -@ <threads> <bam_path>
  • To preprocess the inference data (transcriptome), run the following command:
deeprm call prep -p <input_POD5_dir> -b <bam_path> -o <prep_dir>
  • This will create the npz files for inference.

Run Inference

  • The trained DeepRM model file is attached in the repository: weight/deeprm_weights.pt.
  • For inference, run the following command:
    • Adjust the -s (batch size) parameter according to your GPU memory capacity (default: 10000).
deeprm call run --model <model_file> --data <data_dir> --output <prediction_dir> --gpu-pool <gpu_pool>
  • This will create a directory with the site-level and molecule-level result files.
  • Optionally, if you used a transcriptomic reference for alignment, you can convert the result to genomic coordinates by supplying a RefFlat/GenePred/RefGene file (--annot <annotation_file>).

BED file format

Molecule-level NPZ file format

  • The output NPZ file contains the following arrays:
    1. read_id
    2. label_id
    3. pred: modification score (between 0 and 1)
  • Read ID specification:
    • The UUID4 format read ID (128 bits) is converted to two 64-bit integers for NumPy compatibility.
    • You can convert the two 64-bit integers back to UUID4 using the following Python code:
      import numpy as np
      import uuid
      def int_to_uuid(high, low):
          return uuid.UUID(bytes=b"".join([high.tobytes(),low.tobytes()]))
      
  • Label ID specification:
    • Label ID contains the reference, position, and strand information.
    • You can decode the label ID using the following Python code:
    import numpy as np
    def decode_label_id(label_id, label_div = 10**9):
        strand = np.sign(label_id)
        label_id_abs = np.abs(label_id) - 1
        ref_id = label_id_abs // label_div
        pos = label_id_abs % label_div
        return ref_id, pos, strand
    
    • Reference ID is extracted from the input BAM file header.

Training usage

deeprm_train_pipeline.png

Prepare Data

  • You can skip this step if your POD5 files are already basecalled to BAM files with move tags.
dorado basecaller --min-qscore 0 --emit-moves rna004_130bps_sup@v5.0.0 <pod5_dir> > <bam_path>
samtools index -@ <threads> <bam_path>
  • To preprocess the training data (synthetic oligonucleotide), run the following command:
deeprm train prep --input <input_POD5_dir> --output <output_file>
  • This will create:
    • Training dataset: /block
  • To compile the training dataset, run the following command:
deeprm train compile --input <input_POD5_dir> --output <output_file>
  • This will create:
    • Training dataset: /block

Run Training

  • To train the model, run the following command:
deeprm train run --model deeprm_model --data <data_dir> --output <output_dir> --gpu-pool <gpu_pool>
  • Adjust the --batch parameter according to your GPU memory capacity (default: 1024).
  • This will create a directory with the trained model file.

🔧 Troubleshooting

  • If installation fails on old OS (e.g., CentOS 7) due to a NumPy-related error, you can try installing older versions of NumPy first:
    python -m pip install "numpy<2.3.0,>2.0.0"
    python -m pip install -e .
    
  • If you encounter CUDA or torch-related errors, make sure you have installed the correct version of PyTorch with correct CUDA version support.
  • If Dorado fails due to "illegal memory access", try adding --chunksize <chunk_size> option (e.g., chunk_size=12000).
  • If DeepRM call fails due to memory error, try reducing the batch size (-s option, default: 10000).
  • If DeepRM train fails due to memory error, try reducing the batch size (--batch option, default: 1024).
  • If DeeepRM call preprocess fails due to libssl.so.1.1 not found error in newer versons of Ubuntu, try installing libssl1.1 package:
    wget <libssl_file>
    sudo dpkg <libssl_file>
    
  • If DeepRM call preprocess fails due to memory error, try reducing the number of threads (-t option), the preprocessing batch size (-n option), or the output chunk size (-k option).
  • If DeepRM train does not output training-related metrics, try installing torchmetrics package:
    python -m pip install torchmetrics
    

📐 Architecture

deeprm_architecture.png

📝 Citation

If you use DeepRM in your research, please cite the following paper:

:class: nohighlight
@article{
  title={Comprehensive single-molecule resolution discovery of m6A RNA modification sites in the human transcriptome},
  author={Gihyeon Kang, Hyeonseo Hwang, Hyeonseong Jeon, Heejin Choi, Hee Ryung Chang, Nagyeong Yeo, Junehee Park, Narae Son, Eunkyeong Jeon, Jungmin Lim, Jaeung Yun, Wook Choi, Jae-Yoon Jo, Jong-Seo Kim, Sangho Park, Yoon Ki Kim, Daehyun Baek},
  journal={In review},
  year={In review},
  publisher={In review}
}

📝 License

Creative Commons License
DeepRM is licensed under a Creative Commons Attribution-NonCommercial-ShareAlike 4.0 International License by Seoul National University R&DB Foundation and Genome4me Inc.

See the LICENSE file for details.

🏛️ Contributors

This repository is developed and maintained by the following organization:

  • Laboratory of Computational Biology, School of Biological Sciences, Seoul National University
    • Principal Investigator: Prof. Daehyun Baek
  • Genome4me, Inc., Seoul, Republic of Korea

🏛️ Acknowledgements

This study was supported by the National Research Foundation of Korea (NRF) funded by the Ministry of Science and ICT, Republic of Korea (MSIT) (RS-2019-NR037866, RS-2020-NR049252, RS-2020-NR049538, and RS-2022-NR067483), by a grant of Korean ARPA-H Project through the Korea Health Industry Development Institute (KHIDI), funded by the Ministry of Health & Welfare, Republic of Korea (RS-2025-25422732), by Artificial Intelligence Industrial Convergence Cluster Development Project funded by MSIT and Gwangju Metropolitan City, by National IT Industry Promotion Agency (NIPA) funded by MSIT, and by Korea Research Environment Open Network (KREONET) managed and operated by Korea Institute of Science and Technology Information (KISTI).

Project details


Download files

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

Source Distribution

deeprm-1.0.6.tar.gz (33.1 MB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

deeprm-1.0.6-py3-none-any.whl (31.7 MB view details)

Uploaded Python 3

File details

Details for the file deeprm-1.0.6.tar.gz.

File metadata

  • Download URL: deeprm-1.0.6.tar.gz
  • Upload date:
  • Size: 33.1 MB
  • Tags: Source
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for deeprm-1.0.6.tar.gz
Algorithm Hash digest
SHA256 0ca1e8774469b58c46bf43de80c12162d9e11616226c2ab948e93621b0c607a7
MD5 55632bc7333b8f6acc5370f4a8448fd2
BLAKE2b-256 ab419b05a404fe73cf07c2b07bc1adbe95ebe81937a676c392d07256624bb357

See more details on using hashes here.

Provenance

The following attestation bundles were made for deeprm-1.0.6.tar.gz:

Publisher: publish.yml on vadanamu/DeepRM

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file deeprm-1.0.6-py3-none-any.whl.

File metadata

  • Download URL: deeprm-1.0.6-py3-none-any.whl
  • Upload date:
  • Size: 31.7 MB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? Yes
  • Uploaded via: twine/6.1.0 CPython/3.13.7

File hashes

Hashes for deeprm-1.0.6-py3-none-any.whl
Algorithm Hash digest
SHA256 dd0691bb06e881250dd4544dc530981020b5b82d24220d1a6b901909c5563e70
MD5 9f9e42f9b354d3e7c8330e01e57b4480
BLAKE2b-256 d667eac23133fa1894b73264db9b97505c3fad7a1b22f3f219dbd47d70683aeb

See more details on using hashes here.

Provenance

The following attestation bundles were made for deeprm-1.0.6-py3-none-any.whl:

Publisher: publish.yml on vadanamu/DeepRM

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

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