A novel deep learning model DDU-net using edge features to enhance brain tumor segmentation on MR images
作者:
Highlights:
• We propose DDU-net to improve the performance by making the network focus on the learning of edge features.
• We use a novel edge extraction algorithm to ensure each pixel on one edge comes from the same part of the tumor.
• We propose an improved CE loss function to solve the serious class imbalance problem in the learning of edges.
• We introduce a regularization loss to encourage the predicted segmentation map to align with the boundary of ground truth.
摘要
•We propose DDU-net to improve the performance by making the network focus on the learning of edge features.•We use a novel edge extraction algorithm to ensure each pixel on one edge comes from the same part of the tumor.•We propose an improved CE loss function to solve the serious class imbalance problem in the learning of edges.•We introduce a regularization loss to encourage the predicted segmentation map to align with the boundary of ground truth.
论文关键词:Brain tumor,Automatic segmentation,Dual stream,DDU-net,Edge extraction
论文评审过程:Received 24 December 2020, Revised 29 August 2021, Accepted 21 September 2021, Available online 28 September 2021, Version of Record 30 September 2021.
论文官网地址:https://doi.org/10.1016/j.artmed.2021.102180