DenUnet: enhancing dental image segmentation through edge and body fusion

DOI: https://doi.org/10.1007/s11042-024-19513-0
IF: 2.577
2024-06-09
Multimedia Tools and Applications
Abstract:Accurate tooth segmentation is of paramount importance in oral healthcare because it provides critical positional data for clinical diagnosis, orthodontic treatment, and surgical procedures. Despite the widespread use of convolutional neural networks (CNNs) in image segmentation, its limitations in collecting complete global context and long-range interactions are acknowledged. Although vision transformers show promise in understanding larger contextual information, they struggle to manage the spatial complexities of medical images. To tackle these issues, the proposed DenUnet leverages the strengths of both CNNs and transformers. It introduces a dual-branch encoder that simultaneously extracts edge and body information and a double-level fusion module for integrating multi-scale features. To ensure the fine fusing of edge and body information derived from the two mentioned encoders, we propose a local cross-attention feature fusion module to enhance feature fusion with accurate blending losses. Experimental results underscore the superior efficacy of DenUnet in comparison to existing methods. We achieved 95.4% accuracy and 92.7% F1-score on the DNS dataset, which is particularly evident in its ability to adeptly handle irregular boundaries of dental datasets. Code is available at https://github.com/Omid-Nejati/DenUnet
computer science, information systems, theory & methods,engineering, electrical & electronic, software engineering
What problem does this paper attempt to address?