Multiway sparse distance weighted discrimination

TL;DR

This paper introduces a flexible multiway classification framework that handles any number of dimensions and sparsity levels, improving accuracy for structured high-dimensional data like MRS and gene expression.

Contribution

It extends multiway DWD to arbitrary dimensions and sparsity, providing a robust, interpretable classification method for complex structured data.

Findings

Improved classification accuracy with multiway data

Robustness to sparsity levels in data

Successful application to MRS and gene expression datasets

Abstract

Modern data often take the form of a multiway array. However, most classification methods are designed for vectors, i.e., 1-way arrays. Distance weighted discrimination (DWD) is a popular high-dimensional classification method that has been extended to the multiway context, with dramatic improvements in performance when data have multiway structure. However, the previous implementation of multiway DWD was restricted to classification of matrices, and did not account for sparsity. In this paper, we develop a general framework for multiway classification which is applicable to any number of dimensions and any degree of sparsity. We conducted extensive simulation studies, showing that our model is robust to the degree of sparsity and improves classification accuracy when the data have multiway structure. For our motivating application, magnetic resonance spectroscopy (MRS) was used to measure the abundance of several metabolites across multiple neurological regions and across multiple time points in a mouse model of Friedreich's ataxia, yielding a four-way data array. Our method reveals a robust and interpretable multi-region metabolomic signal that discriminates the groups of interest. We also successfully apply our method to gene expression time course data for multiple sclerosis treatment. An R implementation is available in the package MultiwayClassification at http://github.com/lockEF/MultiwayClassification .