Predicting Coma Recovery After Cardiac Arrest With Residual Neural Networks

Aims: Interpretation of continuous EEG is a demanding task that requires the expertise of trained neurologists. However, these experts are not always available in many medical centers. As part of the 2023 George B. Moody PhysioNet Challenge, we developed a deep learning based method for analyzing EEG data of comatose patients and predicting prognosis following cardiac arrest. Methods: Our approach is a two-step pipeline that consists of a prediction model and a decision-making strategy. The prediction model is a residual neural network (ResNet-18) that extracts features and makes a prediction based on a short 5-minute EEG recording. In the second step, a majority vote over multiple predictions made for several EEG recordings of a patient determines the final prognosis. Results: Based on 10-fold cross-validation on the training set, we achieved a true positive rate (TPR) of 0.41 for predicting poor outcome while keeping the false positive rate below 0.05 at 72 hours after recovery of spontaneous circulation. On the official challenge leaderboard, our team ZIB_Visual scored 0.426 TPR. Conclusion: Our approach, while simple to implement and execute, faced overfitting challenges during the official competition phase. In this paper, we discuss our implementation and potential improvements to address these issues.