Skip to main content
Advertisement
Browse Subject Areas
?

Click through the PLOS taxonomy to find articles in your field.

For more information about PLOS Subject Areas, click here.

  • Loading metrics

TGDBN: Topology-guided dual-backbone network for semi-supervised blastocyst segmentation in assisted reproductive technology

  • Yiming Li,

    Roles Conceptualization, Methodology, Software, Visualization, Writing – original draft

    Affiliations School of Biological Science and Medical Engineering, Beihang University, Beijing, China, Hefei Innovation Research Institute, Beihang University, Hefei, China

  • Hua Wang,

    Roles Investigation, Methodology, Software, Visualization, Writing – original draft

    Affiliations School of Biological Science and Medical Engineering, Beihang University, Beijing, China, Hefei Innovation Research Institute, Beihang University, Hefei, China

  • Jingfei Hu ,

    Roles Funding acquisition, Project administration, Supervision, Validation, Writing – review & editing

    jingfeihu@ahtcm.edu.cn (JH); jicongzhang@buaa.edu.cn (JZ)

    Affiliations School of Medical Informatics Engineering, Anhui University of Traditional Chinese Medicine, Hefei, China, State Key Laboratory of Cognitive Intelligence, Hefei, China

  • Jicong Zhang

    Roles Project administration, Resources, Supervision

    jingfeihu@ahtcm.edu.cn (JH); jicongzhang@buaa.edu.cn (JZ)

    Affiliations School of Biological Science and Medical Engineering, Beihang University, Beijing, China, Hefei Innovation Research Institute, Beihang University, Hefei, China

Abstract

In the field of assisted reproductive technology (ART), accurate segmentation of blastocyst tissues is crucial for evaluating embryo implantation potential. However, existing semi-supervised learning (SSL) methods often suffer from issues such as structural ambiguity, scale confusion, and insufficient utilization of unlabeled data, while the complex nested structure and inherent transparency of blastocyst images further exacerbate these challenges. To address these challenges, this study proposes a Topology-Guided Dual-Backbone Network (TGDBN), a novel semi-supervised segmentation framework featuring a single-model dual-head architecture: one head for pixel-level segmentation and the other for topology-guided structural optimization. Unlike traditional SSL methods that rely solely on feature consistency, TGDBN incorporates two core innovations within its topology-guided head: 1) Decoder Intra-Class Enhancement (DICE) module: Targeting the last three layers of the decoder (where fine-grained structural information is concentrated), this module adaptively strengthens class-specific “hard features” (e.g., low-contrast boundaries between TE and blastocoel) through category-channel mapping and Gaussian weighting; 2) Multi-Scale Feature Aggregation (MSFA) module: This module upsamples DICE-enhanced multi-scale features to the original image resolution, concatenates them to capture hierarchical structural cues, and feeds the aggregated features into a regression sub-layer to generate explicit topology-guided structural guidance. Experimental validation on a public blastocyst dataset demonstrates that TGDBN outperforms 11 state-of-the-art SSL methods. Visualization results confirm that the network effectively mitigates boundary ambiguity and improves intra-class consistency. Further tests on the left atrial (LA) dataset shows that TGDBN exhibits robust generalization capability. These results verify that TGDBN can effectively leverage unlabeled data to enhance blastocyst segmentation accuracy and structural continuity.

Introduction

In the field of assisted reproductive technology (ART), precise segmentation of blastocyst tissues, including inner cell mass (ICM), trophectoderm (TE), blastocoel and zona pellucida (ZP) is a critical step in evaluating embryo implantation potential, as it directly supports embryologists in assessing embryonic developmental quality [1]. However, existing semi-supervised learning (SSL) methods for blastocyst segmentation face three critical limitations: 1) structural ambiguity caused by the inherent transparency of embryonic tissues and Hoffman modulation contrast (HMC) optics (which obscure boundaries between adjacent tissues like TE and blastocoel) [2]; 2) scale confusion arising from significant depth-of-field variations in blastocyst images (leading to mismatched feature scales between superficial ZP and deep ICM) [3]; 3) insufficient exploitation of unlabeled data due to over-reliance on feature consistency constraints (failing to mine structural correlations in unannotated samples). These issues are further compounded by the complex nested structure of blastocysts, leading to suboptimal segmentation accuracy and poor boundary refinement.

Although existing studies have endeavored to enhance segmentation structural continuity via topology-related tasks (e.g., Topology-constrained methods [4]), directly impose topological constraints on segmentation results through graph-based regularization, yet struggle with inadequate adaptation to complex nested structures of blastocysts. Input-embedded topology methods incorporate pre-extracted topological features (e.g., Euler characteristic, Betti number) as additional inputs to the segmentation network, with representative approaches such as [5]. Nevertheless, these methods exhibit weak dynamic interaction between topological features and image features, as the pre-extraction process of topological information is independent of the segmentation network training, leading to potential feature mismatch. Furthermore, a common shortcoming of these methods is the lack of a unified single-model framework to coordinate segmentation and topology learning, resulting in disjointed feature optimization and inefficient structural information utilization.

To address the above pain points, we propose a Topology-Guided Dual-Backbone Network (TGDBN), as shown in Fig 1. Distinct from existing dual-network or multi-branch designs, TGDBN adopts a single-model dual-head architecture that shares a common encoder but splits into two dedicated heads in the decoder stage: 1) Segmentation Head. Responsible for pixel-level blastocyst tissue segmentation, it maps encoder-extracted features to the category space via convolutional and Softmax layers, outputting tissue category probability maps; 2) Topology-Guided Head. Focused on structural guidance generation, it integrates two core modules to optimize segmentation from a topological perspective, and its output is used to constrain both labeled and unlabeled data learning.

thumbnail
Fig 1. The schematic diagram of our proposed TGDBN.

The network adopts a single-model dual-head architecture: the Segmentation Head outputs pixel-level tissue category predictions, while the Topology-Guided Head integrates DICE and MSFA Modules to generate structural guidance for segmentation optimization.

https://doi.org/10.1371/journal.pone.0355053.g001

The implementation logic of the Topology-Guided Head is as follows: First, to address the uneven structural features of blastocyst tissues across different depth-of-field regions, a Decoder Intra-Class Enhancement (DICE) Module is designed. This module targets the last three layers of the decoder (where the richest fine-grained structural information converges) and first binds feature channels to specific tissue categories via “category-channel mapping,” i.e., dividing each decoder layer’s feature map into C category-specific channel groups to avoid cross-category interference. Then it applies a Gaussian weighting strategy to calculate adaptive weights from the Segmentation Head’s probability maps, amplifying the contribution of “hard features” (e.g., blurred boundaries, low-contrast regions) while preserving reliable high-confidence features, effectively reducing intra-class feature variation. Second, to resolve scale confusion and capture multi-scale structural correlations (from local cell boundaries to global tissue topology), a Multi-Scale Feature Aggregation (MSFA) Module is proposed. This module first upsamples the DICE-enhanced multi-scale features (from the decoder’s last three layers) to the original image resolution via bilinear interpolation, then concatenates them along the channel dimension to integrate structural cues of different scales. The aggregated features are fed into a customized regression sub-layer (comprising two 3×3 convolutional blocks and a 1×1 convolution) to generate an explicit structural guidance map, which encodes the signed distance from each pixel to the nearest tissue topological structure to provide clear structural constraints for the segmentation task.

The main contributions of this study can be summarized as follows:

  • We propose a single-model dual-head architecture for blastocyst segmentation, which unifies pixel-level segmentation and topology-guided structural optimization in one framework. This design avoids feature disjointedness in multi-network methods and enables efficient interaction between segmentation and structural information.
  • Within the Topology-Guided Head, the DICE Module accurately enhances class-specific hard features through category-channel binding and Gaussian weighting, while the MSFA Module constructs a unified global-local structural representation by aggregating multi-scale features. The two modules work synergistically to improve the discriminative power of structural features and alleviate scale confusion.
  • On the public blastocyst dataset, TGDBN achieves a Dice coefficient of 91.78%, outperforming 11 mainstream SSL methods. Further tests on the left atrial (LA) dataset (medical images with structural features similar to blastocysts) confirm that the network exhibits robust generalization in segmenting structurally complex medical images, providing a transferable technical solution for clinically relevant segmentation tasks.

Related work

Core paradigms of semi-supervised learning and adaptability to medical images

Embryo image segmentation serves as a crucial support for computational embryology and assisted reproductive technology, and its performance directly impacts the accuracy of embryo quality assessment, developmental potential prediction, and clinical decision-making. However, this field has long been constrained by the scarcity of high-quality annotated data. Medical image annotation relies on the professional knowledge of embryologists, which is not only time-consuming, labor-intensive and costly, but is also inevitably subject to subjective differences. This annotation bottleneck forms a sharp contradiction with the demand for large-scale annotated data by deep learning models [6,7]. Semi-supervised learning provides an effective solution to this contradiction by collaboratively using a small amount of annotated data and a large amount of unannotated data for model training. First, consistency regularization, which explores potential structural information in unannotated data by forcing the model to output consistent predictions for different perturbed versions of inputs; second, pseudo-labeling, which uses the model’s high-confidence prediction results on unannotated data as pseudo-labels to iteratively supplement supervision signals for model optimization; third, entropy minimization, which encourages the model to generate low-entropy (high-confidence) predictions for unannotated data to improve the clarity of decision boundaries [8]. However, these general paradigms face significant challenges when adapting to embryo images, making it difficult to effectively address domain-specific issues such as low contrast, blurred structures, and complex spatiotemporal dynamic changes.

Research progress and existing challenges in semi-supervised segmentation of embryo images

In response to the characteristics of embryo images, researchers have made domain-specific improvements based on general semi-supervised learning frameworks, forming various targeted methods. Topology-aware semi-supervised learning integrates biological constraints and topological information of embryo development into the model, significantly improving the recognition accuracy of inner and outer membranes by ensuring that the segmentation results comply with topological rules such as cell contact relationships and layered structures. A 3D live embryo cell segmentation method proposed in 2024 further combines synthetic dataset expansion and topological constraints to effectively solve the core problem of blurred cell boundaries [9]. The geometric constraint integration strategy encodes prior knowledge of embryo morphology (such as cell size and shape regularity) in the loss function, making the segmentation results more consistent with the conservative developmental patterns of human embryos [9]. Multi-modal fusion and self-supervised pre-training integrate multi-modal data such as phase-contrast microscopy and fluorescence microscopy, explore general features through cross-modal contrastive learning, reduce the dependency on annotated data and improve model generalization ability [10]. The combination of adaptive contrast enhancement and semi-supervised frameworks realizes end-to-end joint optimization of image preprocessing and segmentation tasks, allowing for the targeted enhancement of key features without manual adjustment of parameters [8]. The introduction of multi-scale feature fusion and attention mechanisms improves the perception of blurred edges by capturing multi-level information from local details to global context, among which the hybrid architecture of Transformer and CNN performs outstandingly, capable of modeling both local intensity changes and long-range dependencies [11]. Generative data augmentation technology, especially diffusion model-based methods, can synthesize highly realistic and diverse embryo images, effectively supplementing the sample gap of rare cell types and developmental stages. A 2024 study used conditional diffusion models to generate embryo images at specific stages of development, significantly alleviating the problem of unbalanced data distribution [1214]. Uncertainty-aware methods are specifically designed for blurred boundary regions, adjusting pseudo-labeling strategies by quantifying prediction confidence to avoid the propagation and amplification of incorrect annotations. A 2024 study showed that this method can improve the Dice coefficient of cell boundary segmentation by approximately 5–7% [7,15,16]. Dynamic reweighting strategies alleviate the problem of class imbalance in the early stages of embryo development by adjusting class weights and spatial position weights [11]. Despite the certain progress made by existing methods, most of them simply adapt general deep learning frameworks to embryo image tasks, failing to fully explore the inherent structural features of embryo images. Due to the transparent nature of embryo cells and the requirement of no staining, the images have inherent problems such as weak intensity differences between cells and the background, as well as between cells themselves, and unclear boundary definitions, leading to limited segmentation accuracy and robustness. How to deeply integrate the structural features of embryo images, enhance tissue-specific feature expression, and combine topological structure guidance to fully exploit the intrinsic value of data has become a key research direction to improve model performance.

Method

This section provides a detailed introduction to the proposed network, with a focus on its dual-backbone dual-head architecture and the core modules within the Topology-Guided Head, as well as the corresponding loss functions.

Overall network architecture

The network adopts a non-shared weight dual-backbone design, as illustrated in Fig 2. Each backbone processes the input image X independently and includes two functional heads: a Segmentation Head (Seg-Head) for pixel-level segmentation and a Structure Guidance Head (SG-Head, i.e., TG-Head) for structural information modeling. The key role of the dual-backbone structure is to enforce consistency constraints (pixel consistency and boundary consistency) between the two branches, with a consistency mask guiding the effective application of these constraints. Ground truth data includes pixel-level segmentation label Y and structural label S.

thumbnail
Fig 2. Overall dual-backbone dual-head architecture of the network.

Each backbone outputs pixel probability via Seg-Head and structural probability via SG-Head. Consistency constraints are applied between the two backbones to optimize model learning.

https://doi.org/10.1371/journal.pone.0355053.g002

Backbone and Seg-Head

The segmentation head (Seg-Head) is implemented as a lightweight fully convolutional module to map high-resolution decoder features to pixel-level category predictions. Seg-Head contains convolutional blocks (3×3 convolution + BN + ReLU) and a 1×1 convolution layer. After the final convolution, a softmax activation is applied across the category dimension to generate the pixel-level probability map , where the probability of a pixel (i,j) belonging to category c is given by:

(1)

where denotes the logit value for category c at pixel (i,j) output by the final 1×1 convolutional layer. The softmax normalization ensures for all pixels.

SG-Head (topology-guided head)

The nested anatomical structure of blastocysts-where the ICM is enclosed by the TE, which itself surrounds the blastocoel, all within the ZP-presents a natural topological prior that should be preserved during segmentation. While existing SSL methods enforce pixel-level consistency, they often fail to capture such structural regularity, leading to anatomically implausible predictions (e.g., discontinuous TE layers or misaligned tissue boundaries). To address this, we propose an explicit topological guidance mechanism that translates biological nesting into a learnable geometric representation, enabling the model to maintain anatomical consistency even when labeled data are scarce.

The SG-Head (as shown in Fig 3) is dedicated to generating structural guidance information, integrating two core modules: Decoder Intra-Class Enhancement (DICE) Module and Multi-Scale Feature Aggregation (MSFA) Module. Its input is the last three layers of the decoder (High-scale: , Mid-scale: , Low-scale: ), and its output is the structural probability map , which encodes the relevance of each pixel to tissue boundaries.

thumbnail
Fig 3. Detailed structure of SG-Head.

The DICE module enhances multi-scale decoder features using Gaussian functions. The MSFA module aggregates enhanced features and generates via regression.

https://doi.org/10.1371/journal.pone.0355053.g003

Decoder intra-class enhancement (DICE) module.

The DICE module enhances class-specific “hard features” (e.g., low-contrast boundaries) from the decoder’s last three multi-scale layers (High: , Mid: , Low: ) while avoiding cross-category interference. It first splits each scale’s decoder feature map (where k = 1,2,3 denotes scale, is total channels) into C category-specific groups:

(2)

where and denote the start and end channel indices of the c -th category-specific feature group at the k -th scale, respectively. For index assignment: when , and (with representing the base number of channels per group); when c = C, and (covering all remaining channels). This design ensures no channels are discarded due to the potential indivisibility of (total channels of ) by C (number of tissue categories), while achieving effective feature isolation between different categories.

Next, adaptive weights for each category group are generated via a simplified smooth function applied to the Seg-Head’s probability map (interpolated to match ‘s resolution), which uses a sharp central peak (for medium-confidence ambiguous regions) and flat baseline (for reliable regions) to balance enhancement and preservation, expressed as:

(3)

Here, denotes the Sigmoid activation function, whose mathematical definition is given by , u represents the input probability value (e.g., the category probability output by the Segmentation Head) and ranges within the interval [0,1].

Finally, each category group is element-wise multiplied by to get enhanced groups , which are concatenated to form the final enhanced feature map for each scale.

The category-channel mapping in DICE explicitly aligns feature groups with specific tissue types (ICM, TE, blastocoel, ZP), ensuring that structural enhancement respects biological layering. This design reflects the inherent nesting of blastocyst tissues, allowing the network to strengthen features that define inter-tissue boundaries while preserving intra-tissue continuity.

Multi-scale feature aggregation (MSFA) module.

The Multi-Scale Feature Aggregation (MSFA) module aggregates multi-scale enhanced features (High-scale F1, Mid-scale F2, Low-scale F3) to capture hierarchical structural cues for blastocyst tissue segmentation. First, it performs resolution alignment by bilinearly upsampling Mid-scale F2 and Low-scale F3 to match the High-scale resolution (), yielding upsampled features F2,up and F3,up. Next, it conducts attention-weighted fusion, where channel attention weights () calculated via a Sigmoid map are assigned to F1, F2,up, and F3,up to adaptively adjust their feature importance, resulting in weighted features as expressed in:

(4)

Finally, it executes concatenation and regression. The weighted features , , and are concatenated into a single aggregated feature map , which is then fed into a regression layer composed of 3×3 convolution blocks and a 1×1 convolution. This regression layer applies Sigmoid activation to map the output to the interval [0,1], generating the structural probability map as formulated in

(5)

Intuitive role of topology guidance in semi-supervised learning.

Topology guidance provides an additional source of anatomical regularization that complements conventional consistency-based SSL. For unlabeled images, where no ground truth masks are available, the model must rely on internal consistency between differently augmented views. However, pixel-wise consistency alone can perpetuate structural errors (e.g., fragmented tissues or physically impossible layer relationships). Our topology-guided head introduces a geometric prior: it enforces that predictions should conform to the nested organization of blastocyst tissues. By requiring the two backbones to produce structurally coherent probability maps, where signed distances to boundaries evolve smoothly and layers remain properly nested. The model learns to extrapolate anatomical from limited labeled examples to the unlabeled data. This mechanism acts as a form of “structural common sense”, reducing the reliance on large labeled datasets while improving segmentation plausibility.

Loss function

The total loss of the network combines a supervised loss for labeled data and an unsupervised consistency loss.

Supervised Loss ().

For labeled data (X, Y, S), includes three components: Pixel cross entropy (CE) loss (): Aligns with Y:

(6)

where is the one-hot label of the i-th pixel for category c.

Structural mean-square error (MSE) loss (): Aligns with structural ground truth S:

(7)

where and are values of and B at the i-th pixel.

Structure consistency loss (): ensures consistency between of the two backbones ():

(8)

The weighted supervised loss is:

(9)

where , (determined via cross-validation).

Unsupervised Consistency Loss ().

For unlabeled data , enforces consistency between the two backbones:

(10)

Total Loss with Dynamic Self-Distillation Weight.

The final total loss integrates the supervised loss (from labeled data) and unsupervised self-distillation loss (from unlabeled data), leveraging the independent weight design of B0 and B1 to achieve synergistic learning:

(11)

where follows the setting of the previous work [17].

Experiments and results

Datasets

To validate the effectiveness and generalization of the proposed method, two publicly available datasets with different medical imaging modalities and structural characteristics are used for evaluation.

The first dataset is a human blastocyst microscopic image dataset [18], which serves as the primary testbed for blastocyst segmentation tasks. These images were acquired using Hoffman modulation contrast optics, a specific type of phase-contrast microscopy commonly used in time-lapse embryo culture systems to visualize transparent specimens without staining. It comprises 249 high-resolution blastocyst images paired with manually annotated ground truth (GT) masks, where annotations for key blastocyst components-inner cell mass and blastocoel-are provided by the Pacific Center for Reproductive Medicine. For this dataset, 190 images of the data are used for training, 9 images for validation, and 50 images for testing. For semi-supervised learning, 10% and 50% of the data in the training set are used for training, respectively.

The second dataset is the left atrium 3D magnetic resonance imaging (MRI) dataset [19]. Specifically, these are gadolinium-enhanced MRI (GE-MRI) scans, which provide high soft-tissue contrast for atrial structures. As a widely used benchmark for semi-supervised segmentation performance evaluation [17,2026], it contains 100 labeled 3D GE-MRI scans of the left atrium. To ensure comparability with the existing literature, we follow the data splitting protocol established in [24,26].

Implementation details

To alleviate overfitting and improve the generalization of the model to diverse input variations, a suite of data augmentation strategies is applied during training, including random scaling (to simulate different imaging magnifications), random rotation (to account for arbitrary image orientations), and adaptive adjustments to brightness and contrast (to mimic real-world variations in imaging conditions). For training hyperparameters, a carefully tuned configuration is adopted: the training process runs for a total of 2000 epochs to ensure sufficient convergence; the initial learning rate is set to to balance fast convergence and stable parameter updates; the weight decay of is introduced to regularize the model and prevent excessive parameter growth. For blastocyst segmentation, four standard metrics from prior literature [18] are adopted: Accuracy, Recall, Dice Coefficient, and Jaccard Index. The U-Net architecture [27] is selected as the backbone. For LA segmentation, the evaluation focuses on metrics suitable for 3D structural analysis: Dice Coefficient, Jaccard Index (for overlap assessment), average surface distance (ASD), and 95% Hausdorff Distance (95HD) [26] (for boundary proximity measurement). Here, the VNet [28] serves as the backbone.

Comparison with state-of-the-art methods

From Table 1, compared to other semi-supervised methods, it is clear that in the ICM binary segmentation task with 10% labeled data, TGDBN achieves an Accuracy of 97.62%, a Dice coefficient of 91.78%, and a Jaccard index of 86.33%, all of which are the optimal results in this data partition. These metrics reflect a high level of spatial overlap between TGDBN’s segmentation results and ground truth, indicating its ability to precisely delineate the ICM. When the labeled data ratio increases to 50%, TGDBN still performs exceptionally well.

thumbnail
Table 1. Comparison of performances of semi-supervised learning methods with U-Net as the backbone for binary segmentation. “*” and “” denote significance levels of and , respectively, from a two-sided paired t-test when comparing our best model with others. Best results are highlighted in bold.

https://doi.org/10.1371/journal.pone.0355053.t001

The qualitative visualization results in Fig 4 further confirm the superiority of TGDBN. Its predicted segmentation masks closely match the ground truth labels, while other semi-supervised methods show scattered and less coherent results. Notably, TGDBN excels in handling the boundaries of blastocyst components, where errors from competing methods are mainly concentrated in low-contrast and small-scale regions. These qualitative findings collectively validate the effectiveness of our proposed method, especially in its robust boundary delineation capability.

thumbnail
Fig 4. Visualization results of our method and other state-of-the-art methods in ICM segmentation task.

https://doi.org/10.1371/journal.pone.0355053.g004

Ablation study

Table 2 displays the ablation study outcomes for the ICM dataset, where our proposed TGDBN maintains consistent superiority over all baseline models. The “B” baseline delivers acceptable performance across all evaluation metrics but remains relatively underwhelming compared to other variants. Even when augmented with individual functional modules, such as “DICE” in the “B + D” variant, or “MSFA” in the “B + M” variant, the performance sees incremental improvements, yet still fails to match TGDBN. This confirms that the integrated design of TGDBN, which combines pixel-level details with topological structural guidance, is the key to achieving more precise and robust ICM segmentation results.

thumbnail
Table 2. Ablation study results on ICM dataset. “*” and “” denote significance levels of and , respectively, from a two-sided paired t-test when comparing our best model with others. Best results are highlighted in bold.

https://doi.org/10.1371/journal.pone.0355053.t002

The qualitative visualization results in Fig 5 further highlight the value of topological structure guidance in TGDBN. Our method’s segmentation outputs closely align with the ground truth, effectively preserving and accurately depicting the inherent topological architecture of the ICM. By contrast, variants lacking this integrated guidance struggle to capture such structural nuances, underscoring TGDBN’s unique advantage in leveraging topological cues for high-fidelity segmentation.

thumbnail
Fig 5. Visualization results of ablation study in ICM segmentation task.

https://doi.org/10.1371/journal.pone.0355053.g005

Discussions

Our experiments demonstrate that explicit topological guidance substantially improves semi-supervised segmentation, particularly under low-label regimes. The signed distance representation used in our structural probability map directly encodes the layered organization of blastocyst tissues, providing a continuous geometric signal that is both differentiable and biologically interpretable. Unlike topological invariants (e.g., Betti numbers) or graph-based constraints, our approach integrates topological reasoning into the feature learning process, allowing dynamic interaction between pixel-level segmentation and structural optimization. This design not only mitigates boundary ambiguity but also enables the model to leverage unlabeled data more effectively by enforcing anatomical consistency across predictions.

To further verify the generalization of our method, we conducted validation on the LA dataset-a benchmark with anatomical structural similarities to blastocysts. Table 3 compares semi-supervised segmentation methods for LA binary tasks, all using V-Net as the backbone, and our TGDBN stands out with exceptional performance. Whether under 10% or 20% labeled data ratios, TGDBN maintains strong competitiveness. This consistent performance across both low and high labeled data scenarios confirms its capability to mine valuable information from unlabeled data, thereby ensuring accurate LA segmentation. Fig 6 provides qualitative comparisons of LA segmentation results between TGDBN and other semi-supervised methods, offering direct visual evidence of TGDBN’s superiority. This visual validation not only aligns with the quantitative metrics in Table 3 but also demonstrates TGDBN’s stronger ability to capture the unique features of LA images. It further corroborates the performance insights discussed earlier, serving as additional proof of our proposed method’s effectiveness.

thumbnail
Table 3. Comparison of performances of semi-supervised learning methods with V-Net as the backbone for binary segmentation on LA dataset. “*” and “” denote significance levels of and , respectively, from a two-sided paired t-test when comparing our best model with others. Best results are highlighted in bold.

https://doi.org/10.1371/journal.pone.0355053.t003

thumbnail
Fig 6. Visualization results on the LA dataset.

https://doi.org/10.1371/journal.pone.0355053.g006

Conclusions

This study tackles two fundamental challenges in semi-supervised medical image segmentation: achieving sufficient accuracy under low-labeling conditions and effectively leveraging topological structure information. Taking TGDBN as the solution, its effectiveness was validated through two typical medical segmentation tasks. In the ICM segmentation task, based on a dataset containing 249 microscopic images, TGDBN achieved significantly improved performance compared to baselines under 10%–50% labeled data ratios. Qualitative results also showed that it could accurately capture the topological structure of the ICM and avoid the scattered missegmentation issues common in other methods. In the LA segmentation task, using a dataset of 100 3D MRI scans and V-Net as the backbone network, TGDBN still exhibited excellent performance with 10%–20% labeled data, effectively mitigating problems such as boundary blurriness and regional missegmentation to a certain extent. The advantages of TGDBN stem from its integrated design of “pixel-level constraints + topological guidance” ,which enables efficient extraction of information from unlabeled data. It is compatible with 2D/3D multimodal medical data and demonstrates strong generalization. However, further verification and research are required to explore its performance under more limited labeling conditions and reduce its computational resource overhead.

References

  1. 1. Enderasn AC, Schlafke SJ. The fine structure of the blastocyst: Some comparative studies. John Wiley & Sons, Ltd; p. 29–59.
  2. 2. Chow DJX, Tan TCY, Upadhya A, Lim M, Dholakia K, Dunning KR. Viewing early life without labels: optical approaches for imaging the early embryo†. Biol Reprod. 2024;110(6):1157–74. pmid:38647415
  3. 3. Wang Z. Computer vision guided embryo biopsy. 2015.
  4. 4. Li X, Wen C, Wang L, Fang Y. Topology Constrained Shape Correspondence. IEEE Trans Vis Comput Graph. 2021;27(10):3926–37. pmid:32406841
  5. 5. Xu J, Collins MD, Singh V. Incorporating User Interaction and Topological Constraints within Contour Completion via Discrete Calculus. Proc IEEE Comput Soc Conf Comput Vis Pattern Recognit. 2013;2013:1886–93. pmid:33994765
  6. 6. Chumachenko O, Riazanovskiy K. Semi-supervised Segmentation of Medical Images. Electron Control Syst. 2024;3(81):22–9.
  7. 7. Fang J. A semi-supervised segmentation method for microscopic hyperspectral pathological images based on multi-consistency learning. Front Oncol. 2024;14:1396887. pmid:38962265
  8. 8. Syed S, Anderssen KE, Stormo SK, Kranz M. Weakly supervised semantic segmentation for MRI: exploring the advantages and disadvantages of class activation maps for biological image segmentation with soft boundaries. Sci Rep. 2023;13(1):2574. pmid:36781947
  9. 9. Li Z, Huang Z, Yan H. 3D Cellular Segmentation of Live Embryos via Topologically and Biologically Boundary-aware Semi-supervised Learning. In: 2024 IEEE International Conference on Systems, Man, and Cybernetics (SMC). 2024. p. 3312–7.
  10. 10. Hu X, Zeng D, Xu X, Shi Y. Semi-supervised Contrastive Learning for Label-Efficient Medical Image Segmentation. Medical Image Computing and Computer Assisted Intervention – MICCAI. Springer International Publishing; 2021. p. 481–90.
  11. 11. Reyes AA, Paheding S, Deo M, Audette M. Gabor filter-embedded U-Net with transformer-based encoding for biomedical image segmentation. In: Li X, Lv J, Huo Y, Dong B, Leahy RM, Li Q, editors. Multiscale multimodal medical imaging. Springer Nature Switzerland; p. 76–88.
  12. 12. Kingma DP, Rezende DJ, Mohamed S, Welling M. Semi-supervised learning with deep generative models. Adv Neural Inform Process Syst. 2014;27.
  13. 13. Li D, Yang J, Kreis K, Torralba A, Fidler S. Semantic Segmentation with Generative Models: Semi-Supervised Learning and Strong Out-of-Domain Generalization. In: 2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR). p. 8296–307.
  14. 14. Rosnati M, Ribeiro FDS, Monteiro M, de Castro DC, Glocker B. Analysing the effectiveness of a generative model for semi-supervised medical image segmentation. arXiv preprint arXiv:221101886. 2022. https://doi.org/arXiv:221101886
  15. 15. Liu J, Desrosiers C, Zhou Y. Semi-supervised medical image segmentation using cross-model pseudo-supervision with shape awareness and local context constraints. In: Wang L, Dou Q, Fletcher PT, Speidel S, Li S, editors. Medical Image Computing and Computer Assisted Intervention – MICCAI. Springer Nature Switzerland; 2022. p. 140–50.
  16. 16. Szentimrey Z, Al-Hayali A, de Ribaupierre S, Fenster A, Ukwatta E. Semi-supervised learning framework with shape encoding for neonatal ventricular segmentation from 3D ultrasound. Med Phys. 2024;51(9):6134–48. pmid:38857570
  17. 17. Tarvainen A, Valpola H. Mean teachers are better role models: Weight-averaged consistency targets improve semi-supervised deep learning results. Adv Neural Inform Process Syst. 2017;30.
  18. 18. Saeedi P, Yee D, Au J, Havelock J. Automatic Identification of Human Blastocyst Components via Texture. IEEE Trans Biomed Eng. 2017;64(12):2968–78. pmid:28991729
  19. 19. Xiong Z, Xia Q, Hu Z, Huang N, Bian C, Zheng Y, et al. A global benchmark of algorithms for segmenting the left atrium from late gadolinium-enhanced cardiac magnetic resonance imaging. Med Image Anal. 2021;67:101832. pmid:33166776
  20. 20. Zhang Y, Yang L, Chen J, Fredericksen M, Hughes DP, Chen DZ. Deep Adversarial Networks for Biomedical Image Segmentation Utilizing Unannotated Images. In: Medical Image Computing and Computer Assisted Intervention, MICCAI. Springer International Publishing; 2017. p. 408–16.
  21. 21. Yu L, Wang S, Li X, Fu C-W, Heng P-A. Uncertainty-Aware Self-ensembling Model for Semi-supervised 3D Left Atrium Segmentation. In: Medical Image Computing and Computer Assisted Intervention, MICCAI 2019. Springer International Publishing; 2019. p. 605–13.
  22. 22. Vu T-H, Jain H, Bucher M, Cord M, Perez P. ADVENT: Adversarial Entropy Minimization for Domain Adaptation in Semantic Segmentation. In: 2019 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR). 2019. p. 2512–21.
  23. 23. Wu L, Li J, Wang Y, al e. R-drop: Regularized dropout for neural networks. Adv Neural Inform Process Syst. 2021;34:10890–905.
  24. 24. Luo X, Chen J, Song T, Wang G. Semi-supervised Medical Image Segmentation through Dual-task Consistency. AAAI. 2021;35(10):8801–9.
  25. 25. Verma V, Kawaguchi K, Lamb A, Kannala J, Solin A, Bengio Y, et al. Interpolation consistency training for semi-supervised learning. Neural Netw. 2022;145:90–106. pmid:34735894
  26. 26. Xu Z, Wang Y, Lu D, Luo X, Yan J, Zheng Y, et al. Ambiguity-selective consistency regularization for mean-teacher semi-supervised medical image segmentation. Med Image Anal. 2023;88:102880. pmid:37413792
  27. 27. Ronneberger O, Fischer P, Brox T. U-Net: Convolutional Networks for Biomedical Image Segmentation. In: MICCAI; 2015.
  28. 28. Milletari F, Navab N, Ahmadi S-A. V-Net: Fully Convolutional Neural Networks for Volumetric Medical Image Segmentation. In: 2016 Fourth International Conference on 3D Vision (3DV). 2016. p. 565–71.
  29. 29. Wang Y, Xiao B, Bi X, Li W, Gao X. MCF: Mutual Correction Framework for Semi-Supervised Medical Image Segmentation. In: 2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR). 2023. p. 15651–60.
  30. 30. Yang L, Qi L, Feng L, Zhang W, Shi Y. Revisiting Weak-to-Strong Consistency in Semi-Supervised Semantic Segmentation. In: 2023 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR). 2023. p. 7236–46.
  31. 31. Zeng Q, Xie Y, Lu Z, Lu M, Zhang J, Zhou Y, Xia Y. Consistency-Guided Differential Decoding for Enhancing Semi-Supervised Medical Image Segmentation. IEEE Trans Med Imag. 2025;44(1):44–56. pmid:39088492