Skip to main content

This is a preprint.

It has not yet been peer reviewed by a journal.

The National Library of Medicine is running a pilot to include preprints that result from research funded by NIH in PMC and PubMed.

bioRxiv logoLink to bioRxiv
[Preprint]. 2024 Apr 2:2023.06.09.544263. Originally published 2023 Jun 11. [Version 2] doi: 10.1101/2023.06.09.544263

Gauge equivariant convolutional neural networks for diffusion mri

Uzair Hussain, Ali R Khan
PMCID: PMC10274917  PMID: 37333315

Abstract

Diffusion MRI (dMRI) is an imaging technique widely used in neuroimaging research, where the signal carries directional information of underlying neuronal fibres based on the diffusivity of water molecules. One of the shortcomings of dMRI is that numerous images, sampled at gradient directions on a sphere, must be acquired to achieve a reliable angular resolution for model-fitting, which translates to longer scan times, higher costs, and barriers to clinical adoption. In this work we introduce gauge equivariant convolutional neural network (gCNN) layers for dMRI that overcome the challenges associated with the signal being acquired on a sphere with antipodal points identified. This is done by noting that the domain is equivalent to the real projective plane, ℝ P 2 , which is a non-euclidean and a non-orientable manifold. This is in stark contrast to a rectangular grid which typical convolutional neural networks (CNNs) are designed for. We apply our method to upsample angular resolution for predicting diffusion tensor imaging (DTI) parameters from just six diffusion gradient directions. The symmetries introduced allow gCNNs the ability to train with fewer subjects as compared to a baseline model that involves only 3D convolutions.

Full Text Availability

The license terms selected by the author(s) for this preprint version do not permit archiving in PMC. The full text is available from the preprint server.


Articles from bioRxiv are provided here courtesy of Cold Spring Harbor Laboratory Preprints

RESOURCES