Gastric cancer is a serious health threat and pathological images are an important criterion in its diagnosis. These images can help doctors accurately identify cancerous regions and provide important evidence for clinical decision-making. Thanks to the remarkable achievements of deep learning technology in the field of image processing, an increasing number of superior image segmentation models have emerged. The Swin-Unet model has achieved great success in the field of image segmentation. However, when applied to the image segmentation of gastric cancer pathological section data, the segmentation boundary appears jagged. We have put forth two potential solutions. Initially, we devised an attention connection module to supplant the skip connections within the model, thereby enhancing the model’s predictive precision. Subsequently, we engineered a prognostic processing unit that inputs the model’s predictive outcomes and employs a Conditional Random Field (CRF) for further predictive computations. The enhanced model increases the DSC by 2% and decreases the HD by 17%. Additionally, the issue of jagged boundaries in prediction results has been better optimized. We conducted comparative and ablation experiments, and the results showed that our improved method increased the accuracy of the model’s predictions and reduced the jaggedness of the results at the boundary.
|