Add BPNet model ENCSR759WXZ (ENCSR054JMQ)

.gitattributes

@@ -33,3 +33,8 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+fold_0/saved_model/variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
+fold_1/saved_model/variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
+fold_2/saved_model/variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
+fold_3/saved_model/variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
+fold_4/saved_model/variables/variables.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,137 @@
+---
+license: cc-by-4.0
+library_name: bpnet
+tags:
+  - bpnet
+  - dna
+  - genomics
+  - transcription-factor-binding
+  - chip-seq
+  - encode
+  - encode-bpnet-atlas
+  - hg38
+  - qc-passed
+  - REST
+---
+
+# ENCODE BPNet Atlas
+
+As part of the ENCODE 4 Project, we trained BPNet models on 2,339 ENCODE
+transcription factor ChIP-seq experiments spanning 788 targets across
+175 biosamples. Here, we provide all models for open-source use.
+
+For more information about the models, see:
+
+- Main ENCODE 4 Paper
+- A unified lexicon of predictive DNA sequence motifs from ENCODE transcription
+  factor binding and chromatin accessibility assays (Deshpande et al., Zenodo 2025)
+- Base-resolution models of transcription-factor binding reveal soft motif syntax
+  (Avsec et al., Nat Genet 2021)
+
+## BPNet model: REST ChIP-seq in K562 (ENCSR054JMQ)
+
+- Model: BPNet
+- Assay: TF ChIP-seq
+- Target: REST
+- Experiment: [ENCSR054JMQ](https://www.encodeproject.org/experiments/ENCSR054JMQ/)
+- Model annotation: [ENCSR759WXZ](https://www.encodeproject.org/annotations/ENCSR759WXZ/)
+- Biosample: K562 (Full name: Homo sapiens K562 genetically modified (insertion) using CRISPR targeting H. sapiens REST)
+- Cell slim(s): hematopoietic cell, leukocyte, cancer cell
+- Organ slim(s): bodily fluid, blood
+- Developmental slim(s): mesoderm
+- System slim(s): immune system
+- Assembly: hg38
+
+## QC
+
+- Status: passed
+- Notes: Found direct motif (counts, profile);
+
+## Directory structure
+
+5-fold cross-validation. Each `fold_*/` contains the trained BPNet model in two formats:
+
+- `fold_0/model.h5` — BPNet model in .h5 (Keras) format
+- `fold_0/saved_model/` — BPNet model in TensorFlow SavedModel format (a directory; load directly)
+- `config.json` — training / architecture parameters
+
+## Instructions
+
+BPNet takes a one-hot DNA sequence plus control (bias) inputs and predicts
+stranded profile logits and total logcounts. The control inputs come from the
+matched WCE/Input DNA control and **can be passed as zeros**.
+
+### 1. Loading the SavedModel and making predictions
+
+```python
+import numpy as np
+import tensorflow as tf
+from scipy.special import logsumexp
+
+model = tf.saved_model.load("fold_0/saved_model")
+# sequence: (N, 2114, 4) one-hot [A,C,G,T]
+# profile_bias_input: (N, 1000, 2) per-base profile bias from WCE/Input control, or zeros
+# counts_bias_input:  (N, 2) log2 total counts from WCE/Input control, or zeros
+predictions = model.signatures["serving_default"](**{
+    "sequence": sequence.astype("float32"),
+    "profile_bias_input_0": profile_bias_input.astype("float32"),
+    "counts_bias_input_0": counts_bias_input.astype("float32")})
+# predictions["profile_predictions"]:   (N, 1000, 2) logits (strands NOT independent)
+# predictions["logcounts_predictions"]: (N, 1) total logcount
+
+output_len = 1000
+def vectorized_prediction_to_profile(predictions):
+    logits_arr = predictions["profile_predictions"]
+    counts_arr = predictions["logcounts_predictions"]
+    pred_profile_logits = np.reshape(logits_arr, [-1, 1, output_len * 2])
+    probVals_array = np.exp(pred_profile_logits - logsumexp(
+        pred_profile_logits, axis=2).reshape([len(logits_arr), 1, 1]))
+    profile_predictions = np.multiply(
+        np.exp(counts_arr).reshape([len(counts_arr), 1, 1]), probVals_array)
+    plus  = np.reshape(profile_predictions, [len(counts_arr), output_len, 2])[:, :, 0]
+    minus = np.reshape(profile_predictions, [len(counts_arr), output_len, 2])[:, :, 1]
+    return plus, minus, counts_arr
+
+plus, minus, logcounts = vectorized_prediction_to_profile(predictions)
+```
+
+### 2. Loading the .h5 (Keras) and making predictions
+
+```python
+import numpy as np
+import tensorflow as tf
+import tensorflow.keras.backend as kb
+from tensorflow.keras.models import load_model
+from tensorflow.keras.utils import CustomObjectScope
+from bpnet.model.custommodel import CustomModel
+
+def get_model(model_path):
+    with CustomObjectScope({"kb": kb, "tf": tf, "CustomModel": CustomModel}):
+        return load_model(model_path)
+
+model = get_model("fold_0/model.h5")
+N = sequence.shape[0]
+predictions = model.predict([
+    sequence,                     # (N, 2114, 4)
+    np.zeros((N, 1000, 2)),       # profile_bias_input (or real WCE/Input control values)
+    np.zeros((N, 2))])            # counts_bias_input  (or real control log2 counts)
+# predictions[0]: (N, 1000, 2) logits;  predictions[1]: (N, 1) logcounts
+# convert with the same vectorized_prediction_to_profile() (predictions[0], predictions[1])
+```
+
+## Docker image to load and use the models
+
+`kundajelab/bpnet-atlas` (placeholder — image forthcoming).
+
+## Code
+
+- Code: https://github.com/kundajelab/bpnet/
+- Toolbox & downstream analysis: https://github.com/kundajelab/bpnet/wiki
+
+## License & citation
+
+External data users may freely download, analyze and publish results based on any
+ENCODE data without restrictions.
+
+Released under the ENCODE data-use policy. Please cite the ENCODE Project
+Consortium and the model software: BPNet (Avsec et al., Nat Genet 2021).

config.json

@@ -0,0 +1,45 @@
+{
+    "input_len": 2114,
+    "output_profile_len": 1000,
+    "motif_module_params": {
+        "filters": [64],
+        "kernel_sizes": [21],
+        "padding": "valid"
+    },
+    "syntax_module_params": {
+        "num_dilation_layers": 8,
+        "filters": 64,
+        "kernel_size": 3,
+        "padding": "valid",
+        "pre_activation_residual_unit": true
+    },
+    "profile_head_params": {
+        "filters": 1,
+        "kernel_size":  75,
+        "padding": "valid"
+    },
+    "counts_head_params": {
+        "filters": 1,
+        "kernel_size":  75,
+        "padding": "valid",
+        "units": [1],
+        "activations":["linear"],
+		"dropouts":[0]
+		
+    },
+    "profile_bias_module_params": {
+        "kernel_sizes": [1]
+    },
+    "counts_bias_module_params": {
+    },
+    "use_attribution_prior": false,
+    "attribution_prior_params": {
+        "frequency_limit": 150,
+        "limit_softness": 0.2,
+        "grad_smooth_sigma": 3,
+        "profile_grad_loss_weight": 200,
+        "counts_grad_loss_weight": 100        
+    },
+    "loss_weights": [1, 115.85285299806577],
+    "counts_loss": "MSE"
+}

fold_0/model.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:30594bc8dc1b702f97c330f5174de439490830ebbf7f1ce90fc0b655e5885260
+size 561784

fold_0/saved_model/saved_model.pb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a5e82e56f32aaf35260d0b2ca63ebacb9a6974b588c0f4cf480caac57c5ce233
+size 750202

fold_0/saved_model/variables/variables.data-00000-of-00001

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c326c18875b71cfc9ced9cc7fac501fe552ca60806ff1cf34db6f36717f003ac
+size 1393322

fold_1/model.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c938d50f8c44e8c5f659e629f97446fdb18ee52b7391b4d2a8f3c8f3dbd5302a
+size 561784

fold_1/saved_model/saved_model.pb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fdcda4fda2b3bcd8935b2662ab2ff5c703f6aa032c259c9453a5cd45cfda2998
+size 750202

fold_1/saved_model/variables/variables.data-00000-of-00001

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:e1b76e5650408ceaac5413e2f769b5a98da6809851b2e8a8f2c9a72b666ae672
+size 1393322

fold_2/model.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bd778b737bdbc0aab5b3ca702e048484999e1208eece8351c804eecba1bd77aa
+size 561784

fold_2/saved_model/saved_model.pb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7acdf11a127a418cdd31ab0101a29d9e79add9dd59539052ba399380bb768cae
+size 750202

fold_2/saved_model/variables/variables.data-00000-of-00001

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:178a111c862de9124aad70d7dea3062f53a930f9bd37e5ae61ca8431546754ea
+size 1393322

fold_3/model.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d4e69296552a04149c819c096ba6c4ddf406b8e8f5e2141c02b22e365cc680bd
+size 561784

fold_3/saved_model/saved_model.pb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:deeeeaa597d62864068ed3e79d2a11f99a21cae81df6706161b609374463631b
+size 750202

fold_3/saved_model/variables/variables.data-00000-of-00001

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f0f3dfd44e16b683b2465a1765f43636cf02a9051a7aaed1ed437411fd7deb8b
+size 1393322

fold_4/model.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6d2f3438f6d35dc2bb481feb4af234b4bc3f8cf5b6b91a138864c6a9a8d5024d
+size 561784

fold_4/saved_model/saved_model.pb

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7324f9e19671a1a0ba8e0bb856cd2d10625b7d5f62901c47f2a6dc677b28063c
+size 750202

fold_4/saved_model/variables/variables.data-00000-of-00001

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1bc4af188a7c7bbc6b6b6e78c08b779e28bd237c5b86dbc1aaf8155b9be431b9
+size 1393322