动态调整损失权重,提升乳腺超声分割精度
Adaptive Real-Time Multi-Loss Function Optimization Using Dynamic Memory Fusion Framework: A Case Study on Breast Cancer Segmentation
- 用历史损失值动态调节多损失权重
- 在乳腺超声数据上提升分割指标
- 适合需要自适应优化的医学图像任务
深度学习在众多应用中表现出色,尤其在同时优化多个标准时使用多损失函数效果显著。然而,损失函数的选择与加权对模型性能影响大,手动调参效率低且缺乏灵活性。本文提出动态记忆融合框架,实现多损失函数的实时自适应惩罚。该框架利用历史损失数据,在训练过程中动态调整各损失权重,并引入辅助损失以增强早期性能。此外,提出类别平衡骰子损失,优先关注少数类。在乳腺超声数据集上的实验表明,该框架在多种评估指标上均提升了分割表现。结果证明,该方法能动态聚焦关键标准,适应变化环境。代码已公开于GitHub。
原文摘要 · Abstract (English)
Deep learning has proven to be a highly effective tool for a wide range of applications, significantly when leveraging the power of multi-loss functions to optimize performance on multiple criteria simultaneously. However, optimal selection and weighting loss functions in deep learning tasks can significantly influence model performance, yet manual tuning of these functions is often inefficient and inflexible. We propose a novel framework called dynamic memory fusion for adaptive multi-loss function penalizing in real-time to address this. This framework leverages historical loss values data to dynamically adjust the weighting of multiple loss functions throughout the training process. Additionally, this framework integrates an auxiliary loss function to enhance model performance in the early stages. To further research horizons, we introduce the class-balanced dice loss function, designed to address class imbalance by prioritizing underrepresented classes. Experiments on breast ultrasound datasets demonstrate that the framework improves segmentation performance across various metrics. These results demonstrate the effectiveness of our proposed framework in ensuring that the model dynamically adjusts its focus to prioritize the most relevant criteria, leading to improved performance in evolving environments. The source code for our proposed methodology is publicly available on GitHub.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。