Bundle Geodesic Convolutional Neural Network for DWI Segmentation from Single Scan Learning

Renfei Liu*, Francois Lauze, Kenny Erleben, Sune Darkner

*Corresponding author for this work

Research output: Chapter in Book/Report/Conference proceedingArticle in proceedingsResearchpeer-review

Abstract

We present a tissue classifier for Magnetic Resonance Diffusion Weighted Imaging (DWI) data trained from a single subject with a single b-value. The classifier is based on a Riemannian Deep Learning framework for extracting features with rotational invariance, where we extend a G-CNN learning architecture generically on a Riemannian manifold. We validate our framework using single-shell DWI data with a very limited amount of training data - only 1 scan. The proposed framework mainly consists of three layers: a lifting layer that locally represents and convolves data on tangent spaces to produce a family of functions defined on the rotation groups of the tangent spaces, i.e., a section of a bundle of rotational functions on the manifold; a group convolution layer that convolves this section with rotation kernels to produce a new section; and a projection layer using maximisation to collapse this local data to form new manifold based functions. We present an instantiation on the 2-dimensional sphere where the DWI orientation data is in general represented, and we use it for voxel classification. We show that this allows us to learn a classifier for cerebrospinal fluid (CSF) - subcortical - grey matter - white matter classification from only one scan.

Original languageEnglish
Title of host publicationComputational Diffusion MRI - 12th International Workshop, CDMRI 2021, Held in Conjunction with MICCAI 2021, Proceedings
EditorsSuheyla Cetin-Karayumak, Daan Christiaens, Matteo Figini, Pamela Guevara, Noemi Gyori, Vishwesh Nath, Tomasz Pieciak
PublisherSpringer
Publication date2021
Pages121-132
ISBN (Print)9783030876142
DOIs
Publication statusPublished - 2021
Event12th International Workshop on Computational Diffusion MRI, CDMRI 2021, held in conjunction with 24th International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2021 - Virtual, Online
Duration: 1 Oct 20211 Oct 2021

Conference

Conference12th International Workshop on Computational Diffusion MRI, CDMRI 2021, held in conjunction with 24th International Conference on Medical Image Computing and Computer Assisted Intervention, MICCAI 2021
CityVirtual, Online
Period01/10/202101/10/2021
SeriesLecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
Volume13006 LNCS
ISSN0302-9743

Bibliographical note

Publisher Copyright:
© 2021, Springer Nature Switzerland AG.

Keywords

  • Classification
  • DWI
  • Geodesic CNN
  • Single scan learning

Cite this