Jul 2026· Journal of Computational Design and Engineering· Vol 13, pp. 109-132· 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.
Medical image segmentation plays a critical role in computer-aided diagnosis by enabling precise delineation of abnormalities in medical images. Accurate segmentation of these images is essential for early diagnosis and treatment planning. However, conventional models such as U-Net face challenges including boundary uncertainty, variations in image quality and limited generalization across datasets.
This study presents MRD-UNet, a U-Net based segmentation framework built on a channel-disjoint multi-path dilated fusion (MDF) block deployed uniformly across all encoder and decoder stages, a residual formulation within each MDF block that stabilizes gradient flow without introducing pooling branches, and a task-driven distribution of three functionally distinct attention modules. We evaluate the proposed model through extensive experiments on the CVC-ClinicDB dataset comprising 612 endoscopic images and the ISIC-2018 dataset containing 2,594 dermoscopic images. During data augmentation, we apply geometric transformations, elastic deformations, and brightness adjustments; training is performed using a composite loss that combines Dice loss and binary cross-entropy.
The proposed method achieves a DSC of 0.9140 on CVC-ClinicDB and 0.9106 on ISIC-2018, yielding an improvement of approximately 2–3% over the baseline U-Net architecture. Ablation studies confirm that the channel-disjoint MDF design reduces the number of parameters relative to the baseline while improving sensitivity to small and low-contrast lesions, and that the targeted attention layout enhances boundary delineation.
MRD-UNet provides a practical balance between segmentation accuracy and computational efficiency. It outperforms baseline CNNs and performs comparably to heavier transformer-based models while using significantly fewer parameters. Cross-dataset evaluations demonstrate its ability to generalize within the same imaging modality. Notably, the lightweight MRD-UNet-S variant maintains competitive accuracy, showing the architecture’s suitability for resource-constrained medical image segmentation tasks.
Not applicable.
Musa Doğan, I. Ozkan· BMC Medical Imaging· 0 citations
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.· 2026 5th International Confe...· 0 citations
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.· Interdisciplinary Sciences C...· 0 citations
Liver tumors impose a significant global health burden, underscoring the urgent need for efficient and accurate diagnostic methods. Computer-assisted techniques, particularly deep learning-based segmentation models, have shown considerable promise in this domain. However, they continue to face persistent challenges in liver tumor segmentation, including severe background noise interference, indistinct lesion boundaries, and the difficulty of simultaneously improving segmentation accuracy while maintaining a balanced trade-off between Recall and Precision. To address these issues, this article proposes a novel residual “Enhancement-Calibration” U-Net architecture, termed REC-UNet. The model consists of two task-specific modules: a Residual Enhancement Module (REM) and a Calibration Module (CM). REM leverages residual connections and cross-dimensional attention to enhance tumor feature representation for accurate segmentation, thereby establishing a foundation for balancing Recall and Precision. CM further mitigates noise propagation from shallow to deep feature layers, refining segmentation precision while sustaining high levels of both Recall and Precision. Experiments on the LiTS2017 and MSD_Task08 liver tumor datasets demonstrate that REC-UNet achieves superior performance over mainstream models, with a 4.34% improvement in Dice and a 4.24% improvement in Intersection over Union (IoU) over the second-best model (VM-UNet) on LiTS2017. We further validate the model on an in-house clinical liver tumor Magnetic Resonance Imaging (MRI) dataset, where it attains a Dice score of 87.88% and an IoU of 86.91%, while maintaining a well-balanced trade-off between Recall and Precision. Importantly, REC-UNet achieves high overall segmentation accuracy across diverse lesion sizes and contrast conditions without relying on explicit size-stratified optimization. These results confirm the robust generalizability of REC-UNet and highlight its significant clinical value for computer-assisted liver tumor diagnosis.