Add BPNet model ENCSR601GCZ (ENCSR871NPV)

.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
+  - JUNB
+---
+
+# 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: JUNB ChIP-seq in A549 (ENCSR871NPV)
+
+- Model: BPNet
+- Assay: TF ChIP-seq
+- Target: JUNB
+- Experiment: [ENCSR871NPV](https://www.encodeproject.org/experiments/ENCSR871NPV/)
+- Model annotation: [ENCSR601GCZ](https://www.encodeproject.org/annotations/ENCSR601GCZ/)
+- Biosample: A549 (Full name: Homo sapiens A549 treated with 100 nM dexamethasone (agonist) for 3 hours)
+- Cell slim(s): cancer cell
+- Organ slim(s): lung
+- Developmental slim(s): endoderm
+- System slim(s): respiratory 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, 62.14762576398684],
+    "counts_loss": "MSE"
+}

fold_0/model.h5

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

fold_0/saved_model/saved_model.pb

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

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

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

fold_1/model.h5

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

fold_1/saved_model/saved_model.pb

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

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

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

fold_2/model.h5

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

fold_2/saved_model/saved_model.pb

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

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

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

fold_3/model.h5

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

fold_3/saved_model/saved_model.pb

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

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

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

fold_4/model.h5

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

fold_4/saved_model/saved_model.pb

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

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

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