/Cuff_less_BP_Prediction

Prediction of Blood Pressure from ECG and PPG signals using regression methods.

Primary LanguagePythonMIT LicenseMIT

Cuff less Blood Pressure Prediction

This repository hosts the code for Prediction of Blood Pressure from ECG and PPG signals using two methods.

  1. Feature Extraction and Regression using Machine Learning Methods. Paper

  2. Deep learning based regression.

Getting Started:

  • Clone this repo:
git clone https://github.com/jeya-maria-jose/Cuff_less_BP_Prediction
cd Cuff_less_BP_Prediction

Dataset:

Dataset : Link

This database consist of a cell array of matrices, each cell is one record part.

In each matrix each row corresponds to one signal channel:

1: PPG signal, FS=125Hz; photoplethysmograph from fingertip

2: ABP signal, FS=125Hz; invasive arterial blood pressure (mmHg)

3: ECG signal, FS=125Hz; electrocardiogram from channel II

Processed version of the data from UCI repository used for our experiments: Link

  1. Cleaned folder contains BP records after thresholding them according to Kauchee et al. 2017

  2. GT contains the ground truth SBP, DBP, MAP and class number (depending on the threshold). Ignore the class column as it has not been used for any experiments reported in the papers.

  3. data folder is the whole data as taken from UCI repository.

Feature Extraction and Machine Learning based method:

Prerequisites:

  • MATLAB
  • Python 3
  • Scikit-learn

Feature Extraction

The features taken are explained here

seven_features.m - Code to extract the features : (WN,PIR,PTT,HR,IH,IL,Meu)

ppg_features.m - Code to extract the PPG features

PTT_final.m - Code to extract the PTT

The extracted features are saved in a CSV file from MATLAB.

The CSV file : Link The columns denote the features and BP GT in the same order as extracted.

Machine Learning models

cd models_ML
python rf.py

Using the DL Code:

Prerequisites:

  • Linux
  • Python 3
  • Pytorch

Training

cd models_DL/cnn_lstm_concat
python cnn_multitask.py

Testing

cd models_DL/cnn_lstm_concat
python cnn_test.py

Disclaimer

The code is not completely clean as the data directories are initialized manually. Please make sure the directories are changed according to the remote server where the code is run.

Citation

If you use this , please cite our paper Investigation on the effect of Womersley number, ECG and PPG features for cuff less blood pressure estimation using machine learning:

ML Experiments and Womersley number Paper -

@article{thambiraj2020investigation,
  title={Investigation on the effect of Womersley number, ECG and PPG features for cuff less blood pressure estimation using machine learning},
  author={Thambiraj, Geerthy and Gandhi, Uma and Mangalanathan, Umapathy and Jose, V Jeya Maria and Anand, M},
  journal={Biomedical Signal Processing and Control},
  volume={60},
  pages={101942},
  year={2020},
  publisher={Elsevier}
}
}

Results for DL Experiments - Coming Soon

This work was done while at National Institute of Technology, Tiruchirapalli; India