Skip to content
Log in
Open access

DynU-Net: Dynamic Uncertainty-Aware Multi-task U-Net for Joint Lesion Segmentation and Classification in Medical Imaging

Ngoc Ly Tran Thi Thu Thuy Nguyen Ba-Hung Ngo Tae Jong Choi
Jul 2026 · Journal of Computational Design and Engineering · 0 citations

TL;DR

A Dynamic Uncertainty-aware Network (DynU-Net) is proposed, a multi-task framework that adaptively balances segmentation and classification through learnable per-task uncertainty parameters that consistently outperforms both single-task and existing multi-task baselines.

Abstract

Segmentation and classification of lesion is essential for computer-aided diagnosis (CAD) in grayscale medical imaging, enabling precise localization and reliable categorization of various pathological conditions. However, conventional multi-task learning (MTL) frameworks often suffer from suboptimal task interactions and require manual loss weight tuning, limiting their effectiveness on challenging modalities such as ultrasound, MRI, and CT. To address these limitations, we propose a Dynamic Uncertainty-aware Network (DynU-Net), a multi-task framework that adaptively balances segmentation and classification through learnable per-task uncertainty parameters. This mechanism eliminates the need for manual hyperparameter tuning while mitigating gradient imbalance during optimization. Extensive experiments on three public datasets, including Brain Stroke CT, Brain Tumor MRI, and BUSI (ultrasound), demonstrate that DynU-Net consistently outperforms both single-task and existing multi-task baselines. In particular, it achieves a Dice score of 88.90% ± 0.33% and a macro-F1 score of 96.73% ± 0.74% on the Brain Stroke CT dataset while maintaining favorable computational efficiency compared to competing architectures. These results highlight the effectiveness of dynamic uncertainty-aware optimization in enabling robust and balanced multi-task learning for joint lesion segmentation and classification on public medical imaging benchmarks.

DOI Semantic Scholar
Read PDF

Similar papers

Conference Jul 2026

Automated 3D Brain Tumor Segmentation using Attention U-Net with Multi-Task Regularization on Multiparametric MRI

In order to make accurate distinctions within gliomas - specifically necrosis, edema, and enhancement - accurate segmentation of these regions from multiple MRI types is required. This task has proven difficult because of how heavily class imbalance affects voxels in this imaging modality, combined with its high degree of morphologic variability. In response to this challenge, a new compact 3D attention-based U-Net is developed that uses multi-task regularization, which is used for the volumetric segmentation of gliomas using data from the BraTS 2023 competition. By applying attention gates at each skip connection in the proposed model, non-tumor-related activations are suppressed before the output of the decoder is concatenated together. Additionally, the addition of a lightweight binary auxiliary classification layer after the first bottleneck in the model can act as an inductive bias to prevent the encoder from collapsing into shortcut texture features based solely on local information when faced with severe class imbalance. The model was tested on a held-out set of 251 patients, resulting in Dice similarity coefficients of 0.927 (whole tumor), 0.877 (tumor core), and 0.831 (enhancing tumor). Furthermore, the median HD95 found for whole tumor boundary localization was 1.57 mm. The model has approximately 5.64 million trainable parameters and performs inference in approximately 0.18 s per patient on a single GPU. This represents a 70% decrease in the number of trainable parameters compared to standard 3D U-Net architectures performing similarly well on segmentation tasks, thus making it suitable for use in resource-constrained deployment settings.

Naina Kokate, Sarvesh Garude, Yashraj Hande et al. · 0 citations
Jul 2026

Semi-supervised Medical Image Segmentation via Perturbation-Aware Mutual Learning and Edge-Aware Uncertainty Loss for Accurate Anatomical Delineation.

This paper proposes a novel framework that effectively leverages unlabeled data to improve segmentation performance in cardiac structures and applies a novel consistency constraint by a dual fine-grained boundary loss that provide global characteristics-based guidance from the transition of the boundary region and an edge-aware uncertainty loss.

Waqas Anwaar, Van Manh, Wufeng Xue et al. · 0 citations
Open access Aug 2026

REC-UNet: a 2D U-Net model with residual cross-dimensional attention for liver tumor segmentation

A novel residual “Enhancement-Calibration” U-Net architecture, termed REC-UNet, which achieves high overall segmentation accuracy across diverse lesion sizes and contrast conditions without relying on explicit size-stratified optimization.

Zhiyuan Wang, Lijun Liang, Wei Wu et al. · 0 citations