Semi-Supervised Factored Logistic Regression for High-Dimensional Neuroimaging Data

Existing neuroimaging methods typically perform either discovery of unknown neural structure or testing of neural structure associated with mental tasks. We therefore propose to blend representation modelling and task classification into a unified statistical learning problem. A multinomial logistic regression is introduced that is constrained by factored coefficients and coupled with an autoencoder. We show that this approach yields more accurate and interpretable neural models of psychological tasks in a reference dataset, as well as better generalization to other datasets.