Global Optimality in Separable Dictionary Learning with Applications to the Analysis of Diffusion MRI

  • 2019-09-19 15:31:47
  • Evan Schwab, Benjamin D. Haeffele, RenĂ© Vidal, Nicolas Charon
  • 0

Abstract

Sparse dictionary learning is a popular method for representing signals aslinear combinations of a few elements from a dictionary that is learned fromthe data. In the classical setting, signals are represented as vectors and thedictionary learning problem is posed as a matrix factorization problem wherethe data matrix is approximately factorized into a dictionary matrix and asparse matrix of coefficients. However, in many applications in computer visionand medical imaging, signals are better represented as matrices or tensors(e.g. images or videos), where it may be beneficial to exploit themulti-dimensional structure of the data to learn a more compact representation.One such approach is separable dictionary learning, where one learns separatedictionaries for different dimensions of the data. However, typicalformulations involve solving a non-convex optimization problem; thusguaranteeing global optimality remains a challenge. In this work, we propose aframework that builds upon recent developments in matrix factorization toprovide theoretical and numerical guarantees of global optimality for separabledictionary learning. We propose an algorithm to find such a globally optimalsolution, which alternates between following local descent steps and checking acertificate for global optimality. We illustrate our approach on diffusionmagnetic resonance imaging (dMRI) data, a medical imaging modality thatmeasures water diffusion along multiple angular directions in every voxel of anMRI volume. State-of-the-art methods in dMRI either learn dictionaries only forthe angular domain of the signals or in some cases learn spatial and angulardictionaries independently. In this work, we apply the proposed separabledictionary learning framework to learn spatial and angular dMRI dictionariesjointly and provide preliminary validation on denoising phantom and real dMRIbrain data.

 

Quick Read (beta)

loading the full paper ...