Current browse context:
cs.LG
Change to browse by:
References & Citations
Computer Science > Machine Learning
Title: Self-Supervised Learning for 3D Medical Image Analysis using 3D SimCLR and Monte Carlo Dropout
(Submitted on 29 Sep 2021 (v1), last revised 1 Oct 2021 (this version, v2))
Abstract: Self-supervised learning methods can be used to learn meaningful representations from unlabeled data that can be transferred to supervised downstream tasks to reduce the need for labeled data. In this paper, we propose a 3D self-supervised method that is based on the contrastive (SimCLR) method. Additionally, we show that employing Bayesian neural networks (with Monte-Carlo Dropout) during the inference phase can further enhance the results on the downstream tasks. We showcase our models on two medical imaging segmentation tasks: i) Brain Tumor Segmentation from 3D MRI, ii) Pancreas Tumor Segmentation from 3D CT. Our experimental results demonstrate the benefits of our proposed methods in both downstream data-efficiency and performance.
Submission history
From: Yamen Ali [view email][v1] Wed, 29 Sep 2021 09:15:41 GMT (1438kb,D)
[v2] Fri, 1 Oct 2021 08:33:36 GMT (1439kb,D)
Link back to: arXiv, form interface, contact.