Work

Boosting liver segmentation accuracy with UNet++-ViT: integrating autoencoder and hybrid loss function

Year: 2026

Type: article

Abstract: The presence of noise and intensity variations in medical images poses significant challenges to traditional segmentation methods. In this paper, we propose an enhanced UNet-based framework, UNet++-ViT, incorporating an autoencoder for image denoising and a hybrid loss function combining Tversky Loss and Focal Loss to improve segmentation performance. This paper introduces UNet++-ViT, a robust segmentation framework that integrates a Vision Transformer (ViT) for global contextual understanding, an autoencoder for effective denoising, and a hybrid Tversky-Focal loss to address class imbalance and refine boundary delineation. We evaluate the approach on the LiTS17 liver CT dataset and compare against the models using single loss functions. The proposed method demonstrates superior performance, achieving a Dice coefficient of 0.994, Jaccard index of 0.957, surpassing the leading model SwinUNet by 3.3 % & nnUNet by 4.1% in Dice score along with improved recall and specificity, outperforming other baseline approaches. These results confirm the effectiveness of combining denoising and hybrid loss strategies for robust and precise liver segmentation.

Source: Network Modeling Analysis in Health Informatics and Bioinformatics

DOI: 10.1007/s13721-025-00697-w

Authors: Rashmita Kumari Mohapatra, Lochan jolly, Sarada Prasad Dakua

Institutions: Thakur College of Engineering and Technology, Hamad Medical Corporation, Qatar University

Language: English

Topic: Medical Image Segmentation Techniques

Subfield: Computer Vision and Pattern Recognition

Field: Computer Science

Domain: Physical Sciences

SDG: Good health and well-being

Open Access status: hybrid

Estimated APC paid: $3,290