Abstract:Tensor-based representations are being increasingly used to represent complex data types such as imaging data, due to their appealing properties such as dimension reduction and the preservation of spatial information. Recently, there is a growing literature on using Bayesian scalar-on-tensor regression techniques that use tensor-based representations for high-dimensional and spatially distributed covariates to predict continuous outcomes. However surprisingly, there is limited development on corresponding Bayesian classification methods relying on tensor-valued covariates. Standard approaches that vectorize the image are not desirable due to the loss of spatial structure, and alternate methods that use extracted features from the image in the predictive model may suffer from information loss. We propose a novel data augmentation-based Bayesian classification approach relying on tensor-valued covariates, with a focus on imaging predictors. We propose two data augmentation schemes, one resulting in a support vector machine (SVM) type of classifier, and another yielding a logistic regression classifier. While both types of classifiers have been proposed independently in literature, our contribution is to extend such existing methodology to accommodate high-dimensional tensor valued predictors that involve low rank decompositions of the coefficient matrix while preserving the spatial information in the image. An efficient Markov chain Monte Carlo (MCMC) algorithm is developed for implementing these methods. Simulation studies show significant improvements in classification accuracy and parameter estimation compared to routinely used classification methods. We further illustrate our method in a neuroimaging application using cortical thickness MRI data from Alzheimer's Disease Neuroimaging Initiative, with results displaying better classification accuracy throughout several classification tasks, including classification on pairs of the three diagnostic groups: normal control, AD patients, and MCI patients; gender classification (males vs females); and cognitive performance based on high and low levels of MMSE scores.

Bayesian tensor regression using the Tucker decomposition for sparse spatial modeling

Bayesian Sparse Tucker Models for Dimension Reduction and Tensor Completion

Bayesian tensor-on-tensor regression with efficient computation

Tucker Tensor Regression and Neuroimaging Analysis

Tensor Regression Using Low-rank and Sparse Tucker Decompositions

Bayesian Nonparametric Models for Multiway Data Analysis

Infinite Tucker Decomposition: Nonparametric Bayesian Models for Multiway Data Analysis

Sparse Symmetric Tensor Regression for Functional Connectivity Analysis

Low-Rank Tucker-2 Model for Multi-Subject fMRI Data Decomposition With Spatial Sparsity Constraint

Bayesian Tensor Modeling for Image-based Classification of Alzheimer's Disease

Sparse regularized low-rank tensor regression with applications in genomic data analysis

A tensor based varying-coefficient model for multi-modal neuroimaging data analysis

Bayesian Tensor Analysis

Tucker tensor factor models: matricization and mode-wise PCA estimation

Functional Bayesian Tucker Decomposition for Continuous-indexed Tensor Data

Tensor decompositions and sparse log-linear models

Four-Dimensional Sparse Bayesian Tensor Decomposition for Gene Expression Data

Tensor Graphical Model: Non-Convex Optimization and Statistical Inference

Scalable Bayesian Tensor Ring Factorization for Multiway Data Analysis

Dintucker: Scaling Up Gaussian Process Models On Large Multidimensional Arrays

High-dimensional Grouped-regression using Bayesian Sparse Projection-posterior