A benchmarking framework for improving machine learning with fNIRS neuroimaging data

Benerradi, Johann (2024) A benchmarking framework for improving machine learning with fNIRS neuroimaging data. PhD thesis, University of Nottingham.

[thumbnail of PhD_Thesis_Johann.pdf]
Preview
PDF (Thesis - as examined) - Requires a PDF viewer such as GSview, Xpdf or Adobe Acrobat Reader
Available under Licence Creative Commons Attribution.
Download (6MB) | Preview

Abstract

Functional near-infrared spectroscopy (fNIRS) is a non-invasive neuroimaging technology that presents attractive advantages such as a good compromise between temporal resolution, spatial resolution, and portability. Thanks to its properties and relative robustness to motion, it is often used in naturalistic settings and one of its key domains of application is the study of working memory and mental workload. However fNIRS is less popular and tested than other technologies such as electroencephalography (EEG) or functional magnetic resonance imaging (fMRI) which can nowadays be used in a clinical context. As such, there is no real consensus yet in the research community as to how fNIRS data should be used and analysed. While efforts to establish best practices with fNIRS have been published, there are still no community standards for using machine learning with fNIRS data. Moreover, the lack of open source benchmarks and standard expectations for reporting means that published works often claim high generalisation capabilities, but with poor methodology practices or reporting with missing details. These issues make it hard to evaluate the performance of models when it comes to choosing them for brain-computer interface (BCI) applications.

In this thesis, we present the creation of an open-source benchmarking framework called BenchNIRS to establish a best practice machine learning methodology to develop, evaluate, and compare models for classification from fNIRS data, using open-access datasets from the literature. This framework makes the implementation of a robust machine learning methodology for fNIRS much simpler and less time-consuming.

We demonstrate the utility of the framework by presenting a benchmarking of 6 baseline machine learning models (linear discriminant analysis (LDA), support vector machine (SVM), k-nearest neighbours (kNN), artificial neural network (ANN), convolutional neural network (CNN) and long short-term memory (LSTM)) on 5 open-access datasets and investigate the influence of different factors on the classification performance. Those benchmarks set a solid basis for future comparisons of machine learning approaches for fNIRS classification and reveal that most models have lower performances than expected when evaluating them in an unbiased way.

We then go on to use the framework in a specific challenging use case, the classification of mental workload, and demonstrate how it is used to develop machine learning models tailored to a specific task or dataset. We show that models using as inputs longer durations of fNIRS recordings did not necessarily predict n-back levels better and that the classification from fNIRS data on this task is relatively challenging, despite the tailoring of deep learning models to specific device configurations being promising.

Finally, we go beyond supervised learning and extend the framework to study how unlabelled segments of the data can be used to improve classification. This leads us to perform transfer learning with a self-supervised representation learning pretext task and study to what extent it can be useful to fNIRS data classification. We explore the use of opposite hemoglobin type reconstruction as a pretext task and extend the framework to support the exploration of more pretext tasks for transfer learning in the future.

Item Type: Thesis (University of Nottingham only) (PhD)
Supervisors: Wilson, Max L.
Valstar, Michel F.
Clos, Jeremie
Keywords: fNIRS, machine learning, mental workload
Subjects: Q Science > QA Mathematics > QA 75 Electronic computers. Computer science
R Medicine > R Medicine (General) > R855 Medical technology. Biomedical engineering. Electronics
Faculties/Schools: UK Campuses > Faculty of Science > School of Computer Science
Item ID: 78005
Depositing User: Benerradi, Johann
Date Deposited: 23 Jul 2024 04:40
Last Modified: 23 Jul 2024 04:40
URI: https://eprints.nottingham.ac.uk/id/eprint/78005

Actions (Archive Staff Only)

Edit View Edit View