Discretely-constrained deep network for weakly supervised segmentation
An efficient strategy for weakly-supervised segmentation is to impose constraints or regularization priors on target regions. Recent efforts have focused on incorporating such constraints in the training of convolutional neural networks (CNN), however this has so far been done within a continuous optimization framework. Yet, various segmentation constraints and regularization can be modeled and optimized more efficiently in a discrete formulation. This paper proposes a method, based on the alternating direction method of multipliers (ADMM) algorithm, to train a CNN with discrete constraints and regularization priors. This method is applied to the segmentation of medical images with weak annotations, where both size constraints and boundary length regularization are enforced. Experiments on a benchmark cardiac segmentation dataset show our method to yield a performance near to full supervision.
Code (0)
등록된 구현이 없습니다.
Tasks
Cardiac SegmentationSegmentationWeakly supervised segmentationSimilar Papers 제목 키워드 기반
ACCL: Adversarial constrained-CNN loss for weakly supervised medical image segmentation
We propose adversarial constrained-CNN loss, a new paradigm of constrained-CNN loss methods, for weakly supervised medical image segmentation. In the new paradigm, prior knowledge is encoded and depicted by reference mas…
Image SegmentationMedical Image SegmentationPseudo LabelSegmentation+2Constrained Convolutional Neural Networks for Weakly Supervised Segmentation
We present an approach to learn a dense pixel-wise labeling from image-level tags. Each image-level tag imposes constraints on the output labeling of a Convolutional Neural Network (CNN) classifier. We propose Constraine…
Image SegmentationSemantic SegmentationTAGWeakly supervised segmentationDiscretization and Machine Learning Approximation of BSDEs with a Constraint on the Gains-Process
We study the approximation of backward stochastic differential equations (BSDEs for short) with a constraint on the gains process. We first discretize the constraint by applying a so-called facelift operator at times of …
BIG-bench Machine LearningConstrained-CNN losses for weakly supervised segmentation
Weakly-supervised learning based on, e.g., partially labelled images or image-tags, is currently attracting significant attention in CNN segmentation as it can mitigate the need for full and laborious pixel/voxel annotat…
Medical Image SegmentationSegmentationSemantic SegmentationWeakly-supervised Learning+2A Unified Variational Framework for Deep Weakly Supervised Image Segmentation
We propose a unified variational framework for image segmentation under sparse pixel-level supervision. Our method is based on a simplex-constrained Potts model with a smooth perimeter regularizer, yielding a convex, smo…
Image Segmentation