Skip to content
Sepsis Prediction on MIMIC
Branch: master
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Type Name Latest commit message Commit time
Failed to load latest commit information.
configs added the paper hyperparameter configs for retraining on the given sp… Sep 3, 2019
src renamed code to src Sep 3, 2019
Makefile renamed code to src Sep 3, 2019 fixed bad formatting style in README Sep 3, 2019
requirements.txt started public repo Jul 8, 2019

Sepsis prediction on MIMIC dataset

This repository comprises code concerning sepsis prediction (code for labelling, extraction and proposed method) of the Paper:

Moor, M., Horn, M., Rieck, B., Roqueiro, D., & Borgwardt, K. (2019). Early Recognition of Sepsis with Gaussian Process Temporal Convolutional Networks and Dynamic Time Warping. arXiv preprint arXiv:1902.01659.

Quick start on simulated data

To get a quick start of our proposed method MGP-TCN (Multi-task Gaussian Process Adapted Temporal Convolutional Networks) with simulated data, simply run:

$ make mock_data_results

from the main directory of this repository. It runs the MGP-TCN method on simulated health records data without the need to get access to, download, install, and query the MIMIC database.

You should receive the following output (after some initial updates):

>>>>>> 656128
>>>>>> Starting epoch 0
>>>>>> Batch 0/45, took: 3.349, loss: 3575.63257
>>>>>> Batch 1/45, took: 1.005, loss: 2463.76196
>>>>>> Batch 2/45, took: 1.076, loss: 2280.01270

To actually run our method on the MIMIC sepsis cohort (or generally use this cohort due to its high-resolution sepsis-3 label), follow the full instructions below (which as a warning will take several hours and considerably longer than just running the test script above).

Throughout this repo, you can find instructive textfiles named 'info.txt'. In there you find information on how to run the scripts in the corresponding directory (e.g. query etc.).

The codebase is structured in three sequential steps:

  1. Querying the MIMIC relational database to compute the sepsis label and extract input data
  2. Preprocessing Steps
  3. Experiments

Query and real-world experiments

This repository provides a postgresql-pipeline to extract vital time series of sepsis cases and controls from the MIMIC database following the recent SEPSIS-3 definition.

  1. Requirements for MIMIC: a) Requesting Access to MIMIC (publicly available, however with permission procedure) b) Downloading and installing the MIMIC database according to documentation:

  2. Run the pipeline: a) Once the requirements are fulfilled, open the Makefile and - specify your username and database name

    To run the query, type:

     $ make query

    b) For running the entire pipeline (including query and a mgp-tcn experiment), simply run:

     $ make

    c) Given the query was executed before and you wish to execute an mgp-tcn experiment (as configured in configs), use:

     $ make run_experiments    

    d) For running an experiment with the exact hyperparameters we found in our hyperparameter search, use one of the configs in paper_configs like this:

     $ python3 -m src.mgp_tcn.mgp_tcn_fit with configs/paper_configs/config_split0.json


    • This step might take several hours as the rather involved query is highly relational and therefore easier to perform in postgresql rather than processing all data in a dynamic environment.
      • A faster, more dynamic implementation outside of postgres would be favorable, however most of our queries are extensions of the mimic-code repository (in postgres)

Library Versions (which were used for development)

If you cannot by default run the above code, make sure your libraries fit our used versions:

for the experiments a GPU-based tensorflow environment is assumed. For packages see requirements.txt and run:

$ pip install -r requirements.txt

for the queries, make sure you follow the mimic tutorial instructions for setting it up. We used

  • PostgreSQL 9.3.22 on x86_64-unknown-linux-gnu, compiled by gcc (Ubuntu 4.8.4-2ubuntu1~14.04.4) 4.8.4, 64-bit


Depending on which part of the codebase you use, please also cite our main sources:

You can’t perform that action at this time.