Bragman, FJS; Tanno, R; Ourselin, S; Alexander, DC; Cardoso, MJ; (2019) Learning task-specific and shared representations in medical imaging. In: Medical Image Computing and Computer Assisted Intervention – MICCAI 2019. (pp. pp. 374-383). Springer Nature: Cham, Switzerland.

Preview

Text StochasticFilterGroups_MICCAI_2019.pdf - Accepted Version Download (1MB) | Preview

Abstract

The performance of multi-task learning hinges on the design of feature sharing between tasks; a process which is combinatorial in the network depth and task count. Hand-crafting an architecture based on human intuitions of task relationships is therefore suboptimal. In this paper, we present a probabilistic approach to learning task-specific and shared representations in Convolutional Neural Networks (CNNs) for multi-task learning of semantic tasks. We introduce Stochastic Filter Groups; which is a mechanism that groups convolutional kernels into task-specific and shared groups to learn an optimal kernel allocation. They facilitate learning optimal shared and task specific representations. We employ variational inference to learn the posterior distribution over the possible grouping of kernels and CNN weights. Experiments on MRI-based prostate radiotherapy organ segmentation and CT synthesis demonstrate that the proposed method learns optimal task allocations that are inline with human-optimised networks whilst improving performance over competing baselines.

Download activity - last month

Export as XMLCSVJSON

Download activity - last 12 months

Export as JSONCSVXML

Downloads by country - last 12 months

Export as XMLJSONCSV

Archive Staff Only

View Item