A Hybrid Multi-Factor Network with Dynamic Sequence Modeling for Early Warning of Intraoperative Hypotension
This repository provides the implementation of HMF, a Hybrid Multi-Factor framework for dynamic intraoperative hypotension (IOH) prediction.
The HMF framework is designed to predict IOH using Mean Arterial Pressure (MAP) as a key indicator. Traditional approaches often use static models that fail to capture the dynamic nature of physiological signals. In contrast, our framework employs advanced techniques to address challenges such as distributional drift and temporal dependencies in physiological data.
- Transformer Encoder: Captures the temporal evolution of MAP through patch-based input representation.
- Symmetric Normalization and De-Normalization: Mitigates distributional drift in data to ensure robust performance across varying conditions.
- Sequence Decomposition: Disaggregates input signals into trend and seasonal components for improved sequence modeling.
- State-of-the-Art Performance: Extensively validated on two real-world datasets, outperforming competitive baselines.
- Python 3.7 or later
- PyTorch >= 1.10
- Other dependencies listed in
requirements.txt
-
Clone the repository:
git clone https://github.com/yourusername/HMF-IOH.git cd HMF-IOH -
Install dependencies:
pip install -r requirements.txt
- Place your physiological signal datasets in the
data/directory. - Ensure the data format follows the structure outlined in
docs/data_format.md.
To train the model, run:
python train.py --config configs/default.yamlEvaluate the trained model using:
python evaluate.py --model checkpoints/best_model.pth --data data/test_data.csvModify configs/default.yaml to customize training parameters, such as learning rate, batch size, and model architecture.
The HMF framework demonstrates state-of-the-art performance on two real-world datasets. See the paper for detailed experimental results.
We welcome contributions! Please read our CONTRIBUTING.md file before submitting pull requests.
This project is licensed under the MIT License. See the LICENSE file for details.