TY - GEN
T1 - A Novel Full-Scale Skip Connections Approach Based on U-Net for COVID-19 Lesion Segmentation in CT Images
AU - Wan, Yuchai
AU - Li, Yifan
AU - Jia, Shuqin
AU - Zhang, Lili
AU - Wang, Murong
AU - Liu, Ruijun
N1 - Publisher Copyright:
© The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd 2024.
PY - 2024
Y1 - 2024
N2 - In the post-pandemic era, as COVID-19 continues to spread, CT imaging is indispensable for diagnosing COVID-19. Utilizing computer vision techniques to segment the lesion regions in CT scans can assist doctors in efficient and accurate diagnosis. However, traditional CNN-based U-net segmentation models are more adept at extracting local information, lacking overall awareness of the data, and suffering from semantic loss in the upsampling and downsampling process. To tackle these concerns, we present a Transformer-based full-scale skip connections Unet model. By transforming the traditional CNN structure into a SwinTransformer structure, the model can focus more on the global information of the image, making the instance features more robust and informative. Additionally, we incorporate full-scale skip connections to facilitate the upsampling module to simultaneously access the spatial information from each downsampling module, reducing spatial information loss and improving the segmentation accuracy of the model. We trained and tested our model using an independent dataset of COVID-19 from Wuhan. Experimental results demonstrate that our model exhibits good segmentation capability for COVID-19 lesions and outperforms other methods in terms of average precision. Furthermore, we performed ablation experiments for validation. The effectiveness of the full-scale skip connections.
AB - In the post-pandemic era, as COVID-19 continues to spread, CT imaging is indispensable for diagnosing COVID-19. Utilizing computer vision techniques to segment the lesion regions in CT scans can assist doctors in efficient and accurate diagnosis. However, traditional CNN-based U-net segmentation models are more adept at extracting local information, lacking overall awareness of the data, and suffering from semantic loss in the upsampling and downsampling process. To tackle these concerns, we present a Transformer-based full-scale skip connections Unet model. By transforming the traditional CNN structure into a SwinTransformer structure, the model can focus more on the global information of the image, making the instance features more robust and informative. Additionally, we incorporate full-scale skip connections to facilitate the upsampling module to simultaneously access the spatial information from each downsampling module, reducing spatial information loss and improving the segmentation accuracy of the model. We trained and tested our model using an independent dataset of COVID-19 from Wuhan. Experimental results demonstrate that our model exhibits good segmentation capability for COVID-19 lesions and outperforms other methods in terms of average precision. Furthermore, we performed ablation experiments for validation. The effectiveness of the full-scale skip connections.
KW - COVID-19
KW - Full-scale Skip Connections
KW - Lesion Segmentation
KW - SwinTransformer
KW - U-Net
UR - https://www.scopus.com/pages/publications/85181978711
U2 - 10.1007/978-981-99-9109-9_23
DO - 10.1007/978-981-99-9109-9_23
M3 - 会议稿件
AN - SCOPUS:85181978711
SN - 9789819991082
T3 - Communications in Computer and Information Science
SP - 226
EP - 237
BT - Artificial Intelligence and Robotics - 8th International Symposium, ISAIR 2023, Revised Selected Papers
A2 - Lu, Huimin
A2 - Cai, Jintong
PB - Springer Science and Business Media Deutschland GmbH
T2 - 8th International Symposium on Artificial Intelligence and Robotics, ISAIR 2023
Y2 - 21 October 2023 through 23 October 2023
ER -