3D CNN as an Approach to Predict the Cerebral Performance of Comatose Patients

Rafael Teodoro Ors Quixal1, Elisa Ramirez2, Samuel Ruiperez-Campillo3, Francisco Castells Ramon4, Jose Millet4
1Polytechnic University of Valencia, 2Institute ITACA, Universitat Politecnica de Valencia, 3Stanford University, 4BioITACA-UPV


Abstract

Comatose patients after cardiac arrest require urgent and accurate classification to guide their management and predict their outcomes. In this study, we propose a new parameter for comatose patient classification based on microstate maps of frequency distribution. The objective of this study is to evaluate the performance of the presented parameter for comatose patient classification compared to existing parameters, and to explore the potential of incorporating additional parameters and using more advanced machine learning models to improve classification accuracy. We performed FFT on standardized channels of electroencephalogram recordings from comatose patients after cardiac arrest, and used microstate clustering to obtain maps of frequency distribution. We then sorted the microstate maps to minimize cross-entropy and averaged the cluster centres, which were further clustered into 5 microstates by cerebral performance category (CPC). The resulting microstate centres were used for classification using averaged cross-correlation. We compared the performance of our proposed parameter to those obtained from amplitude-microstates and frequency-microstates. We plan to train a 3D-CNN and SVM models to incorporate temporal-spatial information and extract additional parameters, including evoked potentials and transition probabilities of the dynamic matrix extracted from the microstates segmentation, which were observed to have significative differences between outcomes classification. Our suggested parameter performed better than those obtained from amplitude-microstates and frequency-microstates, increasing the score of a dummy model by 0.8 on local cross-validation. The presented parameter achieved a score of 0.382 in our local run and 0.24 on the platform. Fig-ure 1 shows the good specificity of the proposed parameter across CPC classification in a confusion matrix. The suggested parameter based on microstate maps of frequency distribution shows promise for improving comatose patient classification after cardiac arrest. Incorporating additional parameters and using more advanced machine learning models as described above, we expect a considerable increase in the accuracy of the classification