Optimizing TEM Image Segmentation: Advancements in DRU-Net Architecture with Dense Residual Connections and Attention Mechanisms
Received: 1 June 2024 | Revised: 18 June 2024 | Accepted: 20 June 2024 | Online: 27 June 2024
Corresponding author: Nagalingam Rajeswaran
Abstract
This study introduces an innovative enhancement to the U-Net architecture, termed Modified DRU-Net, aiming to improve the segmentation of cell images in Transmission Electron Microscopy (TEM). Traditional U-Net models, while effective, often struggle to capture fine-grained details and preserve contextual information critical for accurate biomedical image segmentation. To overcome these challenges, Modified DRU-Net integrates dense residual connections and attention mechanisms into the U-Net framework. Dense connections enhance gradient flow and feature reuse, while residual connections mitigate the vanishing gradient problem, facilitating better model training. Attention blocks in the up-sampling path selectively focus on relevant features, boosting segmentation accuracy. Additionally, a combined loss function, merging focal loss and dice loss, addresses class imbalance and improves segmentation performance. Experimental results demonstrate that Modified DRU-Net significantly enhances performance metrics, underscoring its effectiveness in achieving detailed and accurate cell image segmentation in TEM images.
Keywords:
DRU Net, U-Net, dense connections, attention blocks, SEM imagesDownloads
References
M. Gamarra, E. Zurek, H. J. Escalante, L. Hurtado, and H. San-Juan-Vergara, "Split and merge watershed: A two-step method for cell segmentation in fluorescence microscopy images," Biomedical Signal Processing and Control, vol. 53, Aug. 2019, Art. no. 101575.
L. Heinrich et al., "Whole-cell organelle segmentation in volume electron microscopy," Nature, vol. 599, pp. 141–146, Nov. 2021, Art. no. 7883.
W. Yin et al., "A petascale automated imaging pipeline for mapping neuronal circuits with high-throughput transmission electron microscopy," Nature Communications, vol. 11, no. 1, Oct. 2020, Art. no. 4949.
O. Ronneberger, P. Fischer, and T. Brox, "U-Net: Convolutional Networks for Biomedical Image Segmentation," in Medical Image Computing and Computer-Assisted Intervention – MICCAI 2015, Munich, Germany, 2015, pp. 234–241.
B. Tian and W. Wei, "Research Overview on Edge Detection Algorithms Based on Deep Learning and Image Fusion," Security and Communication Networks, vol. 2022, no. 1, 2022, Art. no. 1155814.
S. Yin, H. Li, D. Liu, and S. Karim, "Active contour modal based on density-oriented BIRCH clustering method for medical image segmentation," Multimedia Tools and Applications, vol. 79, no. 41, pp. 31049–31068, Nov. 2020.
N. Micallef, D. Seychell, and C. J. Bajada, "Exploring the U-Net++ Model for Automatic Brain Tumor Segmentation," IEEE Access, vol. 9, pp. 125523–125539, 2021.
S. Das, S. Bose, Ritu, R. Jain, and M. Rout, "Light-UNet++: A Simplified U-NET++ Architecture for Multimodal Biomedical Image Segmentation," in 2023 IEEE International Conference on Contemporary Computing and Communications (InC4), Apr. 2023, vol. 1, pp. 1–5.
P. Harsh, R. Chakraborty, S. Tripathi, and K. Sharma, "Attention U-Net Architecture for Dental Image Segmentation," in 2021 International Conference on Intelligent Technologies (CONIT), Hubli, India, Jun. 2021, pp. 1–5.
R. Yousef et al., "U-Net-Based Models towards Optimal MR Brain Image Segmentation," Diagnostics, vol. 13, no. 9, Jan. 2023, Art. no. 1624.
P. K. Gadosey et al., "SD-UNet: Stripping down U-Net for Segmentation of Biomedical Images on Platforms with Low Computational Budgets," Diagnostics, vol. 10, no. 2, Feb. 2020, Art. no. 110.
V. Morath et al., "Semi-automatic determination of cell surface areas used in systems biology," Frontiers in Bioscience-Elite, vol. 5, no. 2, pp. 533–545, Jan. 2013.
Z. Ji et al., "U-Net_dc: A Novel U-Net-Based Model for Endometrial Cancer Cell Image Segmentation," Information, vol. 14, no. 7, Jul. 2023, Art. no. 366.
M. Ghaffari, A. Sowmya, and R. Oliver, "Automated Brain Tumour Segmentation Using Cascaded 3D Densely-Connected U-Net," in Brainlesion: Glioma, Multiple Sclerosis, Stroke and Traumatic Brain Injuries, Lima, Peru, 2021, pp. 481–491.
C. E. Akbaş and M. Kozubek, "Condensed U-Net (Cu-Net): An Improved U-Net Architecture for Cell Segmentation Powered by 4x4 Max-Pooling Layers," in 2020 IEEE 17th International Symposium on Biomedical Imaging (ISBI), Iowa City, IA, USA, Apr. 2020, pp. 446–450.
D. L. Pham, C. Xu, and J. L. Prince, "Current Methods in Medical Image Segmentation1," Annual Review of Biomedical Engineering, vol. 2, pp. 315–337, Aug. 2000.
S. T. Acton and N. Ray, Biomedical Image Analysis: Segmentation. Cham, Switzerland: Springer International Publishing, 2009.
Z. Linlin, H. Lu, D. Hong, and F. Huijie, "Multi-active contour cell segmentation method based on U-Net network," presented at the 12th International Conference on Image and Signal Processing (ICISP), 2020.
C. Zotti, Z. Luo, A. Lalande, and P.-M. Jodoin, "Convolutional Neural Network With Shape Prior Applied to Cardiac MRI Segmentation," IEEE Journal of Biomedical and Health Informatics, vol. 23, no. 3, pp. 1119–1128, May 2019.
Z. Qu, X. Tao, F. Shen, Z. Zhang, and T. Li, "Investigating Shift Equivalence of Convolutional Neural Networks in Industrial Defect Segmentation," IEEE Transactions on Instrumentation and Measurement, vol. 72, pp. 1–17, 2023.
J. Zhang, Y. Xie, Y. Wang, and Y. Xia, "Inter-Slice Context Residual Learning for 3D Medical Image Segmentation," IEEE Transactions on Medical Imaging, vol. 40, no. 2, pp. 661–672, Oct. 2021.
D. Cheng and E. Y. Lam, "Transfer Learning U-Net Deep Learning for Lung Ultrasound Segmentation." arXiv, Oct. 05, 2021.
X. Dong et al., "Automatic multiorgan segmentation in thorax CT images using U-net-GAN," Medical Physics, vol. 46, no. 5, pp. 2157–2168, 2019.
Z. Lou, W. Huo, K. Le, and X. Tian, "Whole Heart Auto Segmentation of Cardiac CT Images Using U-Net Based GAN," in 2020 13th International Congress on Image and Signal Processing, BioMedical Engineering and Informatics (CISP-BMEI), Chengdu, China, Oct. 2020, pp. 192–196.
N. Li and K. Ren, "Double attention U-Net for brain tumor MR image segmentation," International Journal of Intelligent Computing and Cybernetics, vol. 14, no. 3, pp. 467–479, Jan. 2021.
W. Shi and H. Liu, "Modified U-Net Architecture for Ischemic Stroke Lesion Segmentation and Detection," in 2019 IEEE 4th Advanced Information Technology, Electronic and Automation Control Conference (IAEAC), Sep. 2019, vol. 1, pp. 1068–1071.
H. Seo, C. Huang, M. Bassenne, R. Xiao, and L. Xing, "Modified U-Net (mU-Net) With Incorporation of Object-Dependent High Level Features for Improved Liver and Liver-Tumor Segmentation in CT Images," IEEE Transactions on Medical Imaging, vol. 39, no. 5, pp. 1316–1325, Feb. 2020.
"TEM Dataset - Computer Vision Group, Freiburg." [Online]. Available: https://lmb.informatik.uni-freiburg.de/resources/datasets/tem.en.html.
Downloads
How to Cite
License
Copyright (c) 2024 M. Nagaraju Naik, Nagajyothi Dimmita, Vijayalakshmi Chintamaneni, P. Srinivasa Rao, Nagalingam Rajeswaran, Amar Y. Jaffar, Fahd M. Aldosari, Wesam N. Eid, Ayman A. Alharbi
This work is licensed under a Creative Commons Attribution 4.0 International License.
Authors who publish with this journal agree to the following terms:
- Authors retain the copyright and grant the journal the right of first publication with the work simultaneously licensed under a Creative Commons Attribution License that allows others to share the work with an acknowledgement of the work's authorship and initial publication in this journal.
- Authors are able to enter into separate, additional contractual arrangements for the non-exclusive distribution of the journal's published version of the work (e.g., post it to an institutional repository or publish it in a book), with an acknowledgement of its initial publication in this journal.
- Authors are permitted and encouraged to post their work online (e.g., in institutional repositories or on their website) after its publication in ETASR with an acknowledgement of its initial publication in this journal.