Boundary-Aware Transformer-UNet for Coronary Vessel Segmentation

  • Conference paper
  • First Online:
Advances in Guidance, Navigation and Control ( ICGNC 2022)

Part of the book series: Lecture Notes in Electrical Engineering ((LNEE,volume 845))

Included in the following conference series:

  • 256 Accesses

Abstract

Coronary CT angiography(CTA) has become the gold standard for diagnosing heart disease. Precisely segmentation of coronary stenosis is helpful for hemodynamic analysis. Neural Networks such as U-Net have ruled medical image segmentation field. However, Transformer-like encoder can focus on global information rather than focusing on local information with CNNs. Because of the complicated background of coronary angiography, methods above still do not work in certain images. In this paper, we proposed a Swin-Transformer based segmentation network called BATrans-Net with modules that can optimize boundary segmentation results. Experimental results on our coronary dataset show that our BATrans-Net performs more precisely than other networks.

This work was supported by the National Natural Science Foundation of China (under Grant 51807003).

This is a preview of subscription content, log in via an institution to check access.

Access this chapter

Subscribe and save

Springer+ Basic
EUR 32.99 /Month
  • Get 10 units per month
  • Download Article/Chapter or Ebook
  • 1 Unit = 1 Article or 1 Chapter
  • Cancel anytime
Subscribe now

Buy Now

Chapter
EUR 29.95
Price includes VAT (Germany)
  • Available as PDF
  • Read on any device
  • Instant download
  • Own it forever
eBook
EUR 459.03
Price includes VAT (Germany)
  • Available as EPUB and PDF
  • Read on any device
  • Instant download
  • Own it forever
Softcover Book
EUR 588.49
Price includes VAT (Germany)
  • Compact, lightweight edition
  • Dispatched in 3 to 5 business days
  • Free ship** worldwide - see info
Hardcover Book
EUR 588.49
Price includes VAT (Germany)
  • Durable hardcover edition
  • Dispatched in 3 to 5 business days
  • Free ship** worldwide - see info

Tax calculation will be finalised at checkout

Purchases are for personal use only

Institutional subscriptions

Similar content being viewed by others

References

  1. Cao, H., et al.: Swin-unet: Unet-like pure transformer for medical image segmentation. ar**v preprint ar**v:2105.05537 (2021)

  2. Carion, N., Massa, F., Synnaeve, G., Usunier, N., Kirillov, A., Zagoruyko, S.: End-to-end object detection with transformers. In: Vedaldi, A., Bischof, H., Brox, T., Frahm, J.-M. (eds.) ECCV 2020. LNCS, vol. 12346, pp. 213–229. Springer, Cham (2020). https://doi.org/10.1007/978-3-030-58452-8_13

    Chapter  Google Scholar 

  3. Chen, J., Lu, Y., Yu, Q., Luo, X., Adeli, E., Wang, Y., Lu, L., Yuille, A.L., Zhou, Y.: Transunet: Transformers make strong encoders for medical image segmentation. ar**v preprint ar**v:2102.04306 (2021)

  4. Chen, T., Hu, X., **ao, J., Zhang, G.: BPFINet: boundary-aware progressive feature integration network for salient object detection. Neurocomputing 451, 152–166 (2021)

    Article  Google Scholar 

  5. Dosovitskiy, A., Beyer, L., Kolesnikov, A., Weissenborn, D., Zhai, X., Unterthiner, T., Dehghani, M., Minderer, M., Heigold, G., Gelly, S., Uszkoreit, J., Houlsby, N.: An image is worth 16x16 words: Transformers for image recognition at scale. CoRR abs/2010.11929 (2020), http://dblp.uni-trier.de/db/journals/corr/corr2010.htmlabs-2010-11929

  6. Gu, Z., et al.: Ce-net: context encoder network for 2d medical image segmentation. IEEE Trans. Med. Imaging 38(10), 2281–2292 (2019)

    Article  Google Scholar 

  7. Jia, D., Zhuang, X.: Learning-based algorithms for vessel tracking: a review. Comput. Med. Imaging Graph. 89, 101840 (2021)

    Article  Google Scholar 

  8. Lin, A., Chen, B., Xu, J., Zhang, Z., Lu, G.: Ds-transunet: Dual swin transformer u-net for medical image segmentation. ar**v preprint ar**v:2106.06716 (2021)

  9. Lin, C., Xu, C., Luo, D., Wang, Y., Tai, Y., Wang, C., Li, J., Huang, F., Fu, Y.: Learning salient boundary feature for anchor-free temporal action localization. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. pp. 3320–3329 (2021)

    Google Scholar 

  10. Liu, J.J., Hou, Q., Cheng, M.M., Feng, J., Jiang, J.: A simple pooling-based design for real-time salient object detection. In: Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition. pp. 3917–3926 (2019)

    Google Scholar 

  11. Liu, Z., Lin, Y., Cao, Y., Hu, H., Wei, Y., Zhang, Z., Lin, S., Guo, B.: Swin transformer: Hierarchical vision transformer using shifted windows. ar**v preprint ar**v:2103.14030 (2021)

  12. Long, J., Shelhamer, E., Darrell, T.: Fully convolutional networks for semantic segmentation. In: Proceedings of the IEEE conference on computer vision and pattern recognition. pp. 3431–3440 (2015)

    Google Scholar 

  13. Qin, X., Fan, D.P., Huang, C., Diagne, C., Zhang, Z., Sant’Anna, A.C., Suarez, A., Jagersand, M., Shao, L.: Boundary-aware segmentation network for mobile and web applications. ar**v preprint ar**v:2101.04704 (2021)

  14. Ronneberger, O., Fischer, P., Brox, T.: U-net: convolutional networks for biomedical image segmentation. In: Navab, N., Hornegger, J., Wells, W.M., Frangi, A.F. (eds.) MICCAI 2015. LNCS, vol. 9351, pp. 234–241. Springer, Cham (2015). https://doi.org/10.1007/978-3-319-24574-4_28

    Chapter  Google Scholar 

  15. Schlemper, J., Oktay, O., Chen, L., Matthew, J., Knight, C.L., Kainz, B., Glocker, B., Rueckert, D.: Attention-gated networks for improving ultrasound scan plane detection. CoRR abs/1804.05338 (2018), http://dblp.uni-trier.de/db/journals/corr/corr1804.htmlabs-1804-05338

  16. Shen, H., Wang, R., Zhang, J., McKenna, S.J.: Boundary-aware fully convolutional network for brain tumor segmentation. In: Descoteaux, M., Maier-Hein, L., Franz, A., Jannin, P., Collins, D.L., Duchesne, S. (eds.) MICCAI 2017. LNCS, vol. 10434, pp. 433–441. Springer, Cham (2017). https://doi.org/10.1007/978-3-319-66185-8_49

    Chapter  Google Scholar 

  17. Silva, J.L., Menezes, M.N., Rodrigues, T., Silva, B., Pinto, F.J., Oliveira, A.L.: Encoder-decoder architectures for clinically relevant coronary artery segmentation. ar**v preprint ar**v:2106.11447 (2021)

  18. Wang, R., Chen, S., Ji, C., Fan, J., Li, Y.: Boundary-aware context neural network for medical image segmentation. ar**v preprint ar**v:2005.00966 (2020)

  19. Wang, W., **e, E., Li, X., Fan, D.P., Song, K., Liang, D., Lu, T., Luo, P., Shao, L.: Pyramid vision transformer: A versatile backbone for dense prediction without convolutions. ar**v preprint ar**v:2102.12122 (2021)

  20. **ao, T., Liu, Y., Zhou, B., Jiang, Y., Sun, J.: Unified perceptual parsing for scene understanding pp. 418–434 (2018)

    Google Scholar 

  21. **ao, X., Lian, S., Luo, Z., Li, S.: Weighted res-unet for high-quality retina vessel segmentation. In: 2018 9th international conference on information technology in medicine and education (ITME). pp. 327–331. IEEE (2018)

    Google Scholar 

  22. Zhao, H., Shi, J., Qi, X., Wang, X., Jia, J.: Pyramid scene parsing network. In: Proceedings of the IEEE conference on computer vision and pattern recognition. pp. 2881–2890 (2017)

    Google Scholar 

  23. Zhou, Z., Rahman Siddiquee, M.M., Tajbakhsh, N., Liang, J.: UNet++: a nested u-net architecture for medical image segmentation. In: Stoyanov, D., et al. (eds.) DLMIA/ML-CDS -2018. LNCS, vol. 11045, pp. 3–11. Springer, Cham (2018). https://doi.org/10.1007/978-3-030-00889-5_1

    Chapter  Google Scholar 

  24. Zhu, X., Cheng, Z., Wang, S., Chen, X., Lu, G.: Coronary angiography image segmentation based on PSPNet. Comput. Methods Program. Biomed. 200, 105897 (2021)

    Article  Google Scholar 

Download references

Author information

Authors and Affiliations

Authors

Corresponding author

Correspondence to ** **ao .

Editor information

Editors and Affiliations

Rights and permissions

Reprints and permissions

Copyright information

© 2023 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.

About this paper

Check for updates. Verify currency and authenticity via CrossMark

Cite this paper

**ao, J., Cao, J., Hu, X., Jiang, H., Chen, T., Wang, S. (2023). Boundary-Aware Transformer-UNet for Coronary Vessel Segmentation. In: Yan, L., Duan, H., Deng, Y. (eds) Advances in Guidance, Navigation and Control. ICGNC 2022. Lecture Notes in Electrical Engineering, vol 845. Springer, Singapore. https://doi.org/10.1007/978-981-19-6613-2_300

Download citation

Publish with us

Policies and ethics

Navigation