# Post-DAE: Anatomically Plausible Segmentation via Post-Processing with Denoising Autoencoders

arXiv: https://arxiv.org/abs/2006.13791 · HTML: https://mlsanity.com/paper/2006.13791/

## Key facts

- Cited by 39 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 24 Jun 2020.

## Built on by (most cited first, 30 of 39)

- [Deep Learning for Chest X-ray Analysis: A Survey](https://mlsanity.com/paper/2103.08700/) (2021)
- [CycleMix: A Holistic Strategy for Medical Image Segmentation from Scribble Supervision](https://mlsanity.com/paper/2203.01475/) (2022)
- [Robust machine learning segmentation for large-scale analysis of heterogeneous clinical brain MRI datasets](https://mlsanity.com/paper/2209.02032/) (2022)
- [CheXmask: a large-scale dataset of anatomical segmentation masks for multi-center chest x-ray images](https://mlsanity.com/paper/2307.03293/) (2023)
- [ShapePU: A New PU Learning Framework Regularized by Global Consistency for Scribble Supervised Cardiac Segmentation](https://mlsanity.com/paper/2206.02118/) (2022)
- [ScribFormer: Transformer Makes CNN Work Better for Scribble-based Medical Image Segmentation](https://mlsanity.com/paper/2402.02029/) (2024)
- [ScribbleVC: Scribble-supervised Medical Image Segmentation with Vision-Class Embedding](https://mlsanity.com/paper/2307.16226/) (2023)
- [Medical Image Segmentation with Limited Supervision: A Review of Deep Network Models](https://mlsanity.com/paper/2103.00429/) (2021)
- [A survey on shape-constraint deep learning for medical image segmentation](https://mlsanity.com/paper/2101.07721/) (2021)
- [Anatomically-aware Uncertainty for Semi-supervised Image Segmentation](https://mlsanity.com/paper/2310.16099/) (2023)
- [Joint Motion Correction and Super Resolution for Cardiac Segmentation via Latent Optimisation](https://mlsanity.com/paper/2107.03887/) (2021)
- [Multi-view Hybrid Graph Convolutional Network for Volume-to-mesh Reconstruction in Cardiovascular MRI](https://mlsanity.com/paper/2311.13706/) (2023)
- [CF-Seg: Counterfactuals meet Segmentation](https://mlsanity.com/paper/2506.16213/) (2025)
- [Re-using Adversarial Mask Discriminators for Test-time Training under Distribution Shifts](https://mlsanity.com/paper/2108.11926/) (2021)
- [Learning With Context Feedback Loop for Robust Medical Image Segmentation](https://mlsanity.com/paper/2103.02844/) (2021)
- [Robust Classification from Noisy Labels: Integrating Additional Knowledge for Chest Radiography Abnormality Assessment](https://mlsanity.com/paper/2104.05261/) (2021)
- [Temporally Constrained Neural Networks (TCNN): A framework for semi-supervised video semantic segmentation](https://mlsanity.com/paper/2112.13815/) (2021)
- [Robust Segmentation of Brain MRI in the Wild with Hierarchical CNNs and no Retraining](https://mlsanity.com/paper/2203.01969/) (2022)
- [Leveraging Labeling Representations in Uncertainty-based Semi-supervised Segmentation](https://mlsanity.com/paper/2203.05682/) (2022)
- [Improved post-hoc probability calibration for out-of-domain MRI segmentation](https://mlsanity.com/paper/2208.02870/) (2022)
- [Progressive Test Time Energy Adaptation for Medical Image Segmentation](https://mlsanity.com/paper/2503.16616/) (2025)
- [Stop Throwing Away Discriminators! Re-using Adversaries for Test-Time Training](https://mlsanity.com/paper/2108.12280/) (2021)
- [Shape-intensity knowledge distillation for robust medical image segmentation](https://mlsanity.com/paper/2409.17503/) (2024)
- [GREN: Graph-Regularized Embedding Network for Weakly-Supervised Disease Localization in X-ray Images](https://mlsanity.com/paper/2107.06442/) (2021)
- [Supervision by Denoising for Medical Image Segmentation](https://mlsanity.com/paper/2202.02952/) (2022)
- [Label Refinement Network from Synthetic Error Augmentation for Medical Image Segmentation](https://mlsanity.com/paper/2209.06353/) (2022)
- [Universal Topology Refinement for Medical Image Segmentation with Polynomial Feature Synthesis](https://mlsanity.com/paper/2409.09796/) (2024)
- [Localizing the conceptual difference of two scenes using deep learning for house keeping usages](https://mlsanity.com/paper/2208.04884/) (2022)
- [ScribbleVS: Scribble-Supervised Medical Image Segmentation via Dynamic Competitive Pseudo Label Selection](https://mlsanity.com/paper/2411.10237/) (2024)
- [Leveraging Global Binary Masks for Structure Segmentation in Medical Images](https://mlsanity.com/paper/2205.09107/) (2022)

## Abstract (arXiv)

We introduce Post-DAE, a post-processing method based on denoising autoencoders (DAE) to improve the anatomical plausibility of arbitrary biomedical image segmentation algorithms. Some of the most popular segmentation methods (e.g. based on convolutional neural networks or random forest classifiers) incorporate additional post-processing steps to ensure that the resulting masks fulfill expected connectivity constraints. These methods operate under the hypothesis that contiguous pixels with similar aspect should belong to the same class. Even if valid in general, this assumption does not consider more complex priors like topological restrictions or convexity, which cannot be easily incorporated into these methods. Post-DAE leverages the latest developments in manifold learning via denoising autoencoders. First, we learn a compact and non-linear embedding that represents the space of anatomically plausible segmentations. Then, given a segmentation mask obtained with an arbitrary method, we reconstruct its anatomically plausible version by projecting it onto the learnt manifold. The proposed method is trained using unpaired segmentation mask, what makes it independent of intensity information and image modality. We performed experiments in binary and multi-label segmentation of chest X-ray and cardiac magnetic resonance images. We show how erroneous and noisy segmentation masks can be improved using Post-DAE. With almost no additional computation cost, our method brings erroneous segmentations back to a feasible space.
