Discriminating sample groups with multi-way data

TL;DR

This paper introduces a novel framework for high-dimensional multi-way data classification, extending linear classifiers like SVM and DWD to handle multi-dimensional measurements with low-rank coefficient structures, demonstrated through biomedical applications.

Contribution

The paper develops multi-way versions of SVM and DWD that incorporate low-rank structures, enabling effective classification of multi-dimensional biomedical data.

Findings

Improved classification performance over naive methods.

Enhanced interpretability of model coefficients.

Successful application to clinical datasets.

Abstract

High-dimensional linear classifiers, such as the support vector machine (SVM) and distance weighted discrimination (DWD), are commonly used in biomedical research to distinguish groups of subjects based on a large number of features. However, their use is limited to applications where a single vector of features is measured for each subject. In practice data are often multi-way, or measured over multiple dimensions. For example, metabolite abundance may be measured over multiple regions or tissues, or gene expression may be measured over multiple time points, for the same subjects. We propose a framework for linear classification of high-dimensional multi-way data, in which coefficients can be factorized into weights that are specific to each dimension. More generally, the coefficients for each measurement in a multi-way dataset are assumed to have low-rank structure. This framework extends existing classification techniques, and we have implemented multi-way versions of SVM and DWD. We describe informative simulation results, and apply multi-way DWD to data for two very different clinical research studies. The first study uses metabolite magnetic resonance spectroscopy data over multiple brain regions to compare patients with and without spinocerebellar ataxia, the second uses publicly available gene expression time-course data to compare treatment responses for patients with multiple sclerosis. Our method improves performance and simplifies interpretation over naive applications of full rank linear classification to multi-way data. An R package is available at https://github.com/lockEF/MultiwayClassification .