Volume Preserving Image Segmentation with Entropic Regularization Optimal Transport and Its Applications in Deep Learning

2019-09-22 · Haifeng Li, Jun Liu, Li Cui, Hai-yang Huang, Xue-Cheng Tai

Image segmentation with a volume constraint is an important prior for many real applications. In this work, we present a novel volume preserving image segmentation algorithm, which is based on the framework of entropic regularized optimal transport theory. The classical Total Variation (TV) regularizer and volume preserving are integrated into a regularized optimal transport model, and the volume and classification constraints can be regarded as two measures preserving constraints in the optimal transport problem. By studying the dual problem, we develop a simple and efficient dual algorithm for our model. Moreover, to be different from many variational based image segmentation algorithms, the proposed algorithm can be directly unrolled to a new Volume Preserving and TV regularized softmax (VPTV-softmax) layer for semantic segmentation in the popular Deep Convolution Neural Network (DCNN). The experiment results show that our proposed model is very competitive and can improve the performance of many semantic segmentation nets such as the popular U-net.

📄 PDF Abstract BibTeX arXiv:1909.09931

Code (0)

등록된 구현이 없습니다.

Tasks

Image Segmentation Segmentation Semantic Segmentation

Methods 이 논문이 사용한 방법론

Softmax The Softmax output function transforms a previous layer's output into a vector of probabilities. It is commonly used for multiclass classification. Given an input vector $x$…

Convolution A convolution is a type of matrix operation, consisting of a kernel, a small matrix of weights, that slides over input data performing element-wise multiplication with the…

Training generative models from privatized data

2023-06-15 · Daria Reshetova, Wei-Ning Chen, Ayfer Özgür

Local differential privacy is a powerful method for privacy-preserving data collection. In this paper, we develop a framework for training Generative Adversarial Networks (GANs) on differentially privatized data. We show…

Privacy Preserving

Dual-Scale Volume Priors with Wasserstein-Based Consistency for Semi-Supervised Medical Image Segmentation

2025-09-04 · Junying Meng, Gangxuan Zhou, Jun Liu, Weihong Guo arxiv

Despite signi cant progress in semi-supervised medical image segmentation, most existing segmentation networks overlook e ective methodological guidance for feature extraction and important prior information from dataset…

Semi-supervised Medical Image Segmentation

HeartVolMesh: Cardiac Volumetric Mesh Reconstruction via Covariance-Guided Graph Deformation

2026-07-05 · Fengming Lin, Arezoo Zakeri, Haoran Dou, Zherui Zhou 외 arxiv

Accurate patient-specific tetrahedral cardiac meshes are essential for in-silico trials, yet common segmentation-then-modelling pipelines can blur thin-wall anatomy and offer limited cross-case correspondence. We propose…

Topology-Preserving 3D Image Segmentation Based On Hyperelastic Regularization

2021-03-31 · Daoping Zhang, Lok Ming Lui

Image segmentation is to extract meaningful objects from a given image. For degraded images due to occlusions, obscurities or noises, the accuracy of the segmentation result can be severely affected. To alleviate this pr…