-
Scribble-based Domain Adaptation via Co-segmentation
Authors:
Reuben Dorent,
Samuel Joutard,
Jonathan Shapey,
Sotirios Bisdas,
Neil Kitchen,
Robert Bradford,
Shakeel Saeed,
Marc Modat,
Sebastien Ourselin,
Tom Vercauteren
Abstract:
Although deep convolutional networks have reached state-of-the-art performance in many medical image segmentation tasks, they have typically demonstrated poor generalisation capability. To be able to generalise from one domain (e.g. one imaging modality) to another, domain adaptation has to be performed. While supervised methods may lead to good performance, they require to fully annotate addition…
▽ More
Although deep convolutional networks have reached state-of-the-art performance in many medical image segmentation tasks, they have typically demonstrated poor generalisation capability. To be able to generalise from one domain (e.g. one imaging modality) to another, domain adaptation has to be performed. While supervised methods may lead to good performance, they require to fully annotate additional data which may not be an option in practice. In contrast, unsupervised methods don't need additional annotations but are usually unstable and hard to train. In this work, we propose a novel weakly-supervised method. Instead of requiring detailed but time-consuming annotations, scribbles on the target domain are used to perform domain adaptation. This paper introduces a new formulation of domain adaptation based on structured learning and co-segmentation. Our method is easy to train, thanks to the introduction of a regularised loss. The framework is validated on Vestibular Schwannoma segmentation (T1 to T2 scans). Our proposed method outperforms unsupervised approaches and achieves comparable performance to a fully-supervised approach.
△ Less
Submitted 6 August, 2020; v1 submitted 7 July, 2020;
originally announced July 2020.
-
Generalised Wasserstein Dice Score for Imbalanced Multi-class Segmentation using Holistic Convolutional Networks
Authors:
Lucas Fidon,
Wenqi Li,
Luis C. Garcia-Peraza-Herrera,
**endra Ekanayake,
Neil Kitchen,
Sebastien Ourselin,
Tom Vercauteren
Abstract:
The Dice score is widely used for binary segmentation due to its robustness to class imbalance. Soft generalisations of the Dice score allow it to be used as a loss function for training convolutional neural networks (CNN). Although CNNs trained using mean-class Dice score achieve state-of-the-art results on multi-class segmentation, this loss function does neither take advantage of inter-class re…
▽ More
The Dice score is widely used for binary segmentation due to its robustness to class imbalance. Soft generalisations of the Dice score allow it to be used as a loss function for training convolutional neural networks (CNN). Although CNNs trained using mean-class Dice score achieve state-of-the-art results on multi-class segmentation, this loss function does neither take advantage of inter-class relationships nor multi-scale information. We argue that an improved loss function should balance misclassifications to favour predictions that are semantically meaningful. This paper investigates these issues in the context of multi-class brain tumour segmentation. Our contribution is threefold. 1) We propose a semantically-informed generalisation of the Dice score for multi-class segmentation based on the Wasserstein distance on the probabilistic label space. 2) We propose a holistic CNN that embeds spatial information at multiple scales with deep supervision. 3) We show that the joint use of holistic CNNs and generalised Wasserstein Dice scores achieves segmentations that are more semantically meaningful for brain tumour segmentation.
△ Less
Submitted 29 August, 2017; v1 submitted 3 July, 2017;
originally announced July 2017.
-
Fair Pipelines
Authors:
Amanda Bower,
Sarah N. Kitchen,
Laura Niss,
Martin J. Strauss,
Alexander Vargas,
Suresh Venkatasubramanian
Abstract:
This work facilitates ensuring fairness of machine learning in the real world by decoupling fairness considerations in compound decisions. In particular, this work studies how fairness propagates through a compound decision-making processes, which we call a pipeline. Prior work in algorithmic fairness only focuses on fairness with respect to one decision. However, many decision-making processes re…
▽ More
This work facilitates ensuring fairness of machine learning in the real world by decoupling fairness considerations in compound decisions. In particular, this work studies how fairness propagates through a compound decision-making processes, which we call a pipeline. Prior work in algorithmic fairness only focuses on fairness with respect to one decision. However, many decision-making processes require more than one decision. For instance, hiring is at least a two stage model: deciding who to interview from the applicant pool and then deciding who to hire from the interview pool. Perhaps surprisingly, we show that the composition of fair components may not guarantee a fair pipeline under a $(1+\varepsilon)$-equal opportunity definition of fair. However, we identify circumstances that do provide that guarantee. We also propose numerous directions for future work on more general compound machine learning decisions.
△ Less
Submitted 2 July, 2017;
originally announced July 2017.
-
Scalable multimodal convolutional networks for brain tumour segmentation
Authors:
Lucas Fidon,
Wenqi Li,
Luis C. Garcia-Peraza-Herrera,
**endra Ekanayake,
Neil Kitchen,
Sebastien Ourselin,
Tom Vercauteren
Abstract:
Brain tumour segmentation plays a key role in computer-assisted surgery. Deep neural networks have increased the accuracy of automatic segmentation significantly, however these models tend to generalise poorly to different imaging modalities than those for which they have been designed, thereby limiting their applications. For example, a network architecture initially designed for brain parcellati…
▽ More
Brain tumour segmentation plays a key role in computer-assisted surgery. Deep neural networks have increased the accuracy of automatic segmentation significantly, however these models tend to generalise poorly to different imaging modalities than those for which they have been designed, thereby limiting their applications. For example, a network architecture initially designed for brain parcellation of monomodal T1 MRI can not be easily translated into an efficient tumour segmentation network that jointly utilises T1, T1c, Flair and T2 MRI. To tackle this, we propose a novel scalable multimodal deep learning architecture using new nested structures that explicitly leverage deep features within or across modalities. This aims at making the early layers of the architecture structured and sparse so that the final architecture becomes scalable to the number of modalities. We evaluate the scalable architecture for brain tumour segmentation and give evidence of its regularisation effect compared to the conventional concatenation approach.
△ Less
Submitted 25 June, 2017;
originally announced June 2017.
-
Cohomology of Standard Modules on Partial Flag Varieties
Authors:
S. N. Kitchen
Abstract:
Cohomological induction gives an algebraic method for constructing representations of a real reductive Lie group $G$ from irreducible representations of reductive subgroups. Beilinson-Bernstein localization alternatively gives a geometric method for constructing Harish-Chandra modules for $G$ from certain representations of a Cartan subgroup. The duality theorem of Hecht, Mili\vcić, Schmid and Wol…
▽ More
Cohomological induction gives an algebraic method for constructing representations of a real reductive Lie group $G$ from irreducible representations of reductive subgroups. Beilinson-Bernstein localization alternatively gives a geometric method for constructing Harish-Chandra modules for $G$ from certain representations of a Cartan subgroup. The duality theorem of Hecht, Mili\vcić, Schmid and Wolf establishes a relationship between modules cohomologically induced from minimal parabolics and the cohomology of the $\ms{D}$-modules on the complex flag variety for $G$ determined by the Beilinson-Bernstein construction. The main results of this paper give a generalization of the duality theorem to partial flag varieties, which recovers cohomologically induced modules arising from nonminimal parabolics.
△ Less
Submitted 15 January, 2011;
originally announced January 2011.