The role of data augmentation and attention mechanisms in UNet and ConvNeXt architectures for optimizing breast tumor segmentation

数据增强和注意力机制在UNet和ConvNeXt架构中优化乳腺肿瘤分割的作用

阅读:1

Abstract

This study conducts a comprehensive analysis of various configurations of the UNet + ConvNeXt Tiny architecture for breast tumor segmentation. We assess the influence of data augmentation, skip connections, attention mechanisms, and dropout rates on segmentation performance utilizing the BUSI dataset, subsequently conducting cross-dataset evaluation on the BUS-UCLM dataset. Our findings indicate that the integration of attention mechanisms markedly improves metrics including F1 score, precision, IoU, recall, and Dice coefficient. Data augmentation alone did not consistently enhance model performance and occasionally caused instability, however its integration with attention processes frequently resulted in significant improvements. Dropout was helpful in mitigating overfitting, with a rate of 0.5 consistently yielding enhanced generalization in both in-domain and cross-domain assessments, especially when combined with strategically positioned attention modules. Models using attention mechanisms and a dropout rate of 0.5 attained the optimal equilibrium between accuracy and recall, emphasizing the necessity of balancing regularization intensity with architectural intricacy. Nevertheless, our results underscore the potential for certain augmentation techniques-particularly those that employ aggressive geometric transformations-to distort anatomical structures in ultrasound images, resulting in a decline in the quality of segmentation. This highlights the necessity for meticulous, domain-specific selection of augmentation strategies in medical imaging. This study emphasizes the necessity for meticulous selection of architectural elements and regularization techniques to enhance medical image segmentation and offers direction for future investigations in resilient deep learning applications for clinical imaging.

特别声明

1、本页面内容包含部分的内容是基于公开信息的合理引用;引用内容仅为补充信息,不代表本站立场。

2、若认为本页面引用内容涉及侵权,请及时与本站联系,我们将第一时间处理。

3、其他媒体/个人如需使用本页面原创内容,需注明“来源:[生知库]”并获得授权;使用引用内容的,需自行联系原作者获得许可。

4、投稿及合作请联系:info@biocloudy.com。