Group Convolutional Neural Networks for DWI Segmentation

Research output: Contribution to journalConference articleResearch

Documents

  • Fulltext

    Final published version, 2.45 MB, PDF document

We present a Group Convolutional Network for Segmentation of Diffusion Weighted Imaging data (DWI). The network incorporates group actions that are natural for this type of data, in the form of
equivariant convolutions, i.e., roto-translation equivariant convolutions. The equivariance property provides an important inductive bias and may alleviate the need for data augmentation strategies. Instead of performing group equivariant convolutions via spectral (Fourier-based) approaches, as is common for
equivariance, we implement direct and light-weight regular group convolutions. We study the effect of equivariance and weight sharing over
on performances of the networks on DWI scans from the Human Connectome project. We show how that full
equivariance improves segmentations, while limiting the number of learnable parameters.
Original languageEnglish
JournalProceedings of Machine Learning Research
Volume2022
Issue number1
Pages (from-to)1-11
ISSN2640-3498
Publication statusPublished - 2022
EventGeoMedIA Workshop 2022: Geometric Deep Learning in Medical Image Analysis - Amsterdam, Netherlands
Duration: 18 Nov 2022 → …

Conference

ConferenceGeoMedIA Workshop 2022
CountryNetherlands
CityAmsterdam
Period18/11/2022 → …

ID: 339166879