Progressive Uncertainty-Guided Evidential U-KAN for Trustworthy Medical Image Segmentation
2510.08949v1
eess.IV, cs.CV
2025-10-14
Авторы:
Zhen Yang, Yansong Ma, Lei Chen
Abstract
Trustworthy medical image segmentation aims at deliver accurate and reliable
results for clinical decision-making. Most existing methods adopt the evidence
deep learning (EDL) paradigm due to its computational efficiency and
theoretical robustness. However, the EDL-based methods often neglect leveraging
uncertainty maps rich in attention cues to refine ambiguous boundary
segmentation. To address this, we propose a progressive evidence uncertainty
guided attention (PEUA) mechanism to guide the model to focus on the feature
representation learning of hard regions. Unlike conventional approaches, PEUA
progressively refines attention using uncertainty maps while employing low-rank
learning to denoise attention weights, enhancing feature learning for
challenging regions. Concurrently, standard EDL methods suppress evidence of
incorrect class indiscriminately via Kullback-Leibler (KL) regularization,
impairing the uncertainty assessment in ambiguous areas and consequently
distorts the corresponding attention guidance. We thus introduce a
semantic-preserving evidence learning (SAEL) strategy, integrating a
semantic-smooth evidence generator and a fidelity-enhancing regularization term
to retain critical semantics. Finally, by embedding PEUA and SAEL with the
state-of-the-art U-KAN, we proposes Evidential U-KAN, a novel solution for
trustworthy medical image segmentation. Extensive experiments on 4 datasets
demonstrate superior accuracy and reliability over the competing methods. The
code is available at
\href{https://anonymous.4open.science/r/Evidence-U-KAN-BBE8}{github}.
Ссылки и действия
Дополнительные ресурсы: