Tensor Networks for Medical Image Classification

04/21/2020
by   Raghavendra Selvan, et al.
0

With the increasing adoption of machine learning tools like neural networks across several domains, interesting connections and comparisons to concepts from other domains are coming to light. In this work, we focus on the class of Tensor Networks, which has been a work horse for physicists in the last two decades to analyse quantum many-body systems. Building on the recent interest in tensor networks for machine learning, we extend the Matrix Product State tensor networks (which can be interpreted as linear classifiers operating in exponentially high dimensional spaces) to be useful in medical image analysis tasks. We focus on classification problems as a first step where we motivate the use of tensor networks and propose adaptions for 2D images using classical image domain concepts such as local orderlessness of images. With the proposed locally orderless tensor network model (LoTeNet), we show that tensor networks are capable of attaining performance that is comparable to state-of-the-art deep learning methods. We evaluate the model on two publicly available medical imaging datasets and show performance improvements with fewer model hyperparameters and lesser computational resources compared to relevant baseline methods.

READ FULL TEXT
research
09/25/2020

Locally orderless tensor networks for classifying two- and three-dimensional medical images

Tensor networks are factorisations of high rank tensors into networks of...
research
09/15/2021

Patch-based medical image segmentation using Quantum Tensor Networks

Tensor networks are efficient factorisations of high dimensional tensors...
research
02/13/2021

Segmenting two-dimensional structures with strided tensor networks

Tensor networks provide an efficient approximation of operations involvi...
research
11/16/2022

Neurodevelopmental Phenotype Prediction: A State-of-the-Art Deep Learning Model

A major challenge in medical image analysis is the automated detection o...
research
09/16/2022

Deep tensor networks with matrix product operators

We introduce deep tensor networks, which are exponentially wide neural n...
research
07/24/2023

Compact Capable: Harnessing Graph Neural Networks and Edge Convolution for Medical Image Classification

Graph-based neural network models are gaining traction in the field of r...
research
05/15/2019

Number-State Preserving Tensor Networks as Classifiers for Supervised Learning

We propose a restricted class of tensor network state, built from number...

Please sign up or login with your details

Forgot password? Click here to reset