sharktide/ohca-predictor-v1
CVDGaurd: A Cross-Attention 17 Million Parameter Deep Neural Network for the Early Prediciton of Out-Of-Hospital Cardiac Arrest
This repository contains a deep multimodal cardiovascular risk monitoring architecture designed to predict Out-of-Hospital Cardiac Arrest (OHCA) events using simultaneous time-series physiological waveforms and static clinical tabular baselines.
The model is built using TensorFlow/Keras and is wrapped into the global Hugging Face ecosystem via the unified AutoModel entry point using custom remote code execution.
Model Details
- Architecture Name: TFCVDPredictorForOHCADetection
- Domain: Cardiovascular Disease (CVD) & Emergency Medicine
- Task: Binary classification / Risk estimation of impending Out-of-Hospital Cardiac Arrest (OHCA)
- Framework: TensorFlow 2.16+ / Keras 3 - CVDPredict Library
- Primary Inputs: Continuous 1D/2D sensor waveforms (ECG, PPG, Accelerometer, Gyroscope, Vitals) + Patient tabular metadata
Performance Metrics (Validation Cohort)
Evaluated independently on a standardized clinical baseline cohort (196 validation samples, post-rate = 28.1%):
Note: The cross-attention transformer components are highly sensitive to physiological anomalies and morphological variations in raw continuous wave signals.
Data Schema & Input Specifications
To initialize prediction profiles properly or prevent pipeline shape truncation errors, inputs should be structured around a ~185-second monitoring window (windowdurationhours ≈ 0.05138) conforming to the following target sampling specs:
1. Continuous Time-Series Signals (Waveforms)
2. Clinical Context Vectors (Tabular Baseline Matrices)
- demographics: NumPy array of shape (24,) — Normalized age, sex, and baseline physiological markers.
- medications: NumPy array of shape (14,) — Encoded binary medication history indicators.
- comorbidities: NumPy array of shape (14,) — Encoded binary chronic disease matrix.
- lab_values: NumPy array of shape (8,) — Standardized baseline blood chemistry values.
3. Vital Parameter Scalars
- heartrate (float), rhythm (int), activitystate (int)
- spo2mean (float), sbpmean (float), dbpmean (float), respirationmean (float)
Quickstart / Deployment Usage
Prerequisites
Before running inference, you must have your matching domain execution library (CVDPredict/ohca_predictor), transformers, and tensorflow installed on your host system:
pip install git+https://github.com/sharktide/CVD-Predict.git@v1.0.0
pip install "transformers>=5"
# Install tensorflow for your system by following the instructions at https://tensorflow.org/installInference Code Example
You can load the model directly from the internet via the standard Hugging Face pipeline in just a few lines of code:
import numpy as np
import warnings
warnings.filterwarnings("ignore")
from transformers import AutoModel
from ohca_predictor.utils.io_utils import WindowSample
# 1. Download and map the custom model wrapper from the cloud Hub repository
model = AutoModel.from_pretrained("sharktide/ohca-predictor-v1", trust_remote_code=True)2. Package raw incoming data streams into a structured WindowSample instance
patient_record = WindowSample(
ecg=np.random.randn(24050).astype(np.float32),
accelerometer=np.zeros((9620, 3), dtype=np.float32),
gyroscope=np.zeros((9620, 3), dtype=np.float32),
ppg=np.random.randn(9250).astype(np.float32),
respiration=np.zeros(4625, dtype=np.float32),
spo2=np.zeros(1850, dtype=np.float32),
temperature=np.zeros(185, dtype=np.float32),
demographics=np.zeros(24, dtype=np.float32),
medications=np.zeros(14, dtype=np.float32),
comorbidities=np.zeros(14, dtype=np.float32),
lab_values=np.zeros(8, dtype=np.float32),
heart_rate=72.0, rhythm=0, activity_state=0,
spo2_mean=97.5, sbp_mean=120.0, dbp_mean=80.0,
patient_id="live-monitor-case-001", signal_quality={"ecg": 1.0},
window_start_hours=0.0, window_duration_hours=0.05138,
ohca_label=0.0, event_indicator=0.0, time_to_event=0.0
)
# 3. Dispatches forward pass execution natively
prediction = model(patient_record)
print(f"Prediction Success!")
print(f"Calculated Patient OHCA Risk: {prediction['ohca_risk'].numpy().item():.4f}")Security Notice
This model relies on Custom Remote Code Execution (trust_remote_code=True) to execute the pipeline routing files (modelingohca.py and configurationohca.py) directly from the Hugging Face hub. Always ensure you are requesting a pinned repository commit hash if deploying this wrapper framework inside production or clinical environments.
