Abstract
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.
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 language | English |
|---|---|
| Journal | Proceedings of Machine Learning Research |
| Volume | 2022 |
| Issue number | 1 |
| Pages (from-to) | 1-11 |
| ISSN | 2640-3498 |
| Publication status | Published - 2022 |
| Event | GeoMedIA Workshop 2022: Geometric Deep Learning in Medical Image Analysis - Amsterdam, Netherlands Duration: 18 Nov 2022 → … |
Conference
| Conference | GeoMedIA Workshop 2022 |
|---|---|
| Country/Territory | Netherlands |
| City | Amsterdam |
| Period | 18/11/2022 → … |