ResNet-34 with Lightweight Decoder for Accurate and Efficient Segmentation of Fetal Brain MRI

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision & Pattern Recognition, Medical Imaging AI · Depth: Expert, quick

Summary

A novel deep learning model, combining a ResNet-34 encoder with a lightweight decoder utilizing multi-layer perceptron (MLP) modules, has been developed for accurate and efficient segmentation of fetal brain tissues in MRI. This model addresses challenges like fetal motion, low tissue contrast, and anatomical variability, particularly for complex structures such as white matter and brainstem. Its design enhances anatomical boundary preservation and mitigates segmentation errors from motion artifacts. Computational efficiency is achieved through reduced parameter count and bilinear upsampling. Validated on the FeTA 2021 dataset using 5-fold cross-validation, the model achieved an average Accuracy of 97.37%, a mean Dice Similarity Coefficient (DSC) of 90.33%, mean Intersection over Union (IoU) of 86.93%, and Precision of 90.83%. It outperforms baseline architectures like UNet and DeepLabV3+, making it suitable for real-time clinical integration.

Key takeaway

For Machine Learning Engineers developing fetal brain MRI segmentation solutions, this model provides a compelling architecture. Its ResNet-34 encoder and lightweight MLP decoder achieve high accuracy (97.37% Accuracy, 90.33% DSC) with efficient inference. You should evaluate this design for integrating into real-time clinical workflows, especially where computational resources are constrained or rapid diagnosis is critical. This approach offers a robust alternative to traditional UNet-based models.

Key insights

A ResNet-34 encoder with a lightweight MLP decoder achieves accurate and efficient fetal brain MRI segmentation, outperforming baselines.

Principles

Method

A ResNet-34 encoder is paired with a lightweight decoder using MLP modules for adaptive feature refinement, trained on the FeTA 2021 dataset via 5-fold cross-validation.

In practice

Topics

Best for: Computer Vision Engineer, AI Scientist, Machine Learning Engineer, Research Scientist

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.