arXiv:2502.02756eess.IVcs.AI2025-02被引 3

用L1范数自适应加权,提升前列腺癌病灶检测分割精度

Adaptive Voxel-Weighted Loss Using L1 Norms in Deep Neural Networks for Detection and Segmentation of Prostate Cancer Lesions in PET/CT Images

  • 基于L1范数动态调整体素权重,缓解难易样本梯度失衡
  • 在380例PET/CT数据上,Dice系数提升至少4%,F1得分提高26%
  • 对大小不一的病灶均表现稳健,适合临床实际应用

由于病灶大小、活性、解剖位置差异及类间/类内不平衡,自动检测PSMA PET/CT中复发性前列腺癌极具挑战。传统深度学习损失函数常因背景体素或极端异常值主导梯度而优化不佳。为此,我们提出L1加权狄克斯特福尔损失(L1DFL),利用L1范数均衡体素间梯度幅值,根据分类难度自适应加权样本,实现预测校准并形成正确与错误预测的双峰分离。在380例PSMA PET/CT扫描数据上,训练了三种3D卷积网络(Attention U-Net、SegResNet、U-Net)和一个基于Transformer的UNETR模型,输入为PET与CT图像的拼接。同时微调SAM-Med3D基础模型并评估不同损失函数性能。在各架构下,L1DFL均优于狄克斯特损失(DL)和狄克斯特福尔损失(DFL),Dice相似系数至少提升4%;F1分数分别高出DL 6%、DFL 26%。DFL产生更多假阳性,DL对大病灶检测较差,而L1DFL实现平衡检测,在降低误报的同时保持高真阳性率。梯度均衡机制确保了对不同大小、体积和分布病灶的鲁棒性。代码已公开:https://github.com/ObedDzik/pca_segment.git。

原文摘要 · Abstract (English)

Accurate automated detection of recurrent prostate cancer in PSMA PET/CT scans is challenging due to heterogeneous lesion size, activity, anatomical location, and intra- and inter-class imbalances. Conventional deep learning loss functions often produce suboptimal optimization, as gradients are dominated by easy background voxels or extreme outliers. To address this, we propose L1-weighted Dice Focal Loss (L1DFL), which harmonizes gradient magnitudes across voxels using L1 norms to adaptively weight samples based on classification difficulty, resulting in well-calibrated predictions with a bimodal separation between correct and incorrect predictions. We trained three 3D convolutional networks (Attention U-Net, SegResNet, U-Net) and a transformer-based UNETR model on 380 PSMA PET/CT scans. PET and CT volumes were concatenated as input to the models. We also fine-tuned SAM-Med3D foundation model with the different loss functions and evaluated their performance. Across architectures, L1DFL consistently outperformed Dice Loss (DL) and Dice Focal Loss (DFL), achieving at least a 4% improvement in Dice Similarity Coefficient. F1 scores were higher by 6% and 26% compared to DL and DFL, respectively. While DFL produced more false positives and DL struggled with larger lesions, L1DFL achieved balanced detection, minimizing false detections while maintaining high true positive rates. The gradient harmonization mechanism ensured robustness across varying lesion sizes, volumes, and spread. The code is publicly available at: https://github.com/ObedDzik/pca_segment.git.

医学影像分割算法深度学习前列腺癌

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。