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%): 

Metric	Value
AUROC	0.932 [0.895 - 0.972]
AUPRC	0.892 [0.828 - 0.949]
Optimal Risk Threshold	0.302
Sensitivity @ Optimal	0.855
Specificity @ Optimal	0.844
Brier Score	0.0904

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 (window_duration_hours ≈ 0.05138) conforming to the following target sampling specs: 

1. Continuous Time-Series Signals (Waveforms)

Parameter Name	Dimension	Frequency	Description / Layout
ecgShape	(24050,)	130 HzRaw	continuous 1D ECG trace array
ppgShape	(9250,)	50 Hz	Continuous photoplethysmogram wave
accelerometer	(9620, 3)	52 Hz	3-axis continuous accelerometer matrix
gyroscope	(9620, 3)	52 Hz	3-axis continuous gyroscope telemetry
respiration	(4625,)	25 Hz	Continuous chest expansion respiration belt wave
spo2Shape (1850,)	10 Hz	Continuous blood oxygen saturation array stream
temperature	(185,)	1 Hz	Continuous core body temperature array log

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

• heart_rate (float), rhythm (int), activity_state (int)
• spo2_mean (float), sbp_mean (float), dbp_mean (float), respiration_mean (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/install

Inference 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 (modeling_ohca.py and configuration_ohca.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.