Sex classification from functional brain connectivity: Generalization to multiple datasets

Lisa Wiersch,Patrick Friedrich,Sami Hamdan,Vera Komeyer,Felix Hoffstaedter,Kaustubh R. Patil,Simon B. Eickhoff,Susanne Weis
DOI: https://doi.org/10.1002/hbm.26683
IF: 4.8
2024-04-24
Human Brain Mapping
Abstract:Machine learning is often used to study neuroimaging data. However, ML‐studies often use single datasets. Here, we investigated the generalization performance in sex classification of parcel‐wise classifiers (pwC) using RS‐fMRI data. Our results indicated that compound samples show higher accuracies and better spatial consistency than single sample pwCs. Machine learning (ML) approaches are increasingly being applied to neuroimaging data. Studies in neuroscience typically have to rely on a limited set of training data which may impair the generalizability of ML models. However, it is still unclear which kind of training sample is best suited to optimize generalization performance. In the present study, we systematically investigated the generalization performance of sex classification models trained on the parcelwise connectivity profile of either single samples or compound samples of two different sizes. Generalization performance was quantified in terms of mean across‐sample classification accuracy and spatial consistency of accurately classifying parcels. Our results indicate that the generalization performance of parcelwise classifiers (pwCs) trained on single dataset samples is dependent on the specific test samples. Certain datasets seem to "match" in the sense that classifiers trained on a sample from one dataset achieved a high accuracy when tested on the respected other one and vice versa. The pwCs trained on the compound samples demonstrated overall highest generalization performance for all test samples, including one derived from a dataset not included in building the training samples. Thus, our results indicate that both a large sample size and a heterogeneous data composition of a training sample have a central role in achieving generalizable results.
radiology, nuclear medicine & medical imaging,neurosciences,neuroimaging
What problem does this paper attempt to address?