Multi-StudentNet: a Semi-supervised Deep-Learning Framework for Endometrial Segmentation in Transvaginal Ultrasound Images

Semi-supervised learning (SSL) is a promising solution for the problem of insufficient medical labeled data. However, it is still a challenging task to segment transvaginal ultrasound (TVUS) images because of their poor contrast and speckle noise. In this study, we propose Multi-StudentNet, a semi-supervised deep-learning framework for endometrial segmentation in TVUS images. This method employs multiple student models to effectively integrate labeled and unlabeled data, improving the reliability and performance of segmentation while significantly reducing the need for extensive manual labeling. By leveraging the weights of multiple models, our framework facilitates feature sharing, which enhances model robustness. Specifically, Multi-StudentNet achieves a Dice coefficient (DSC) of 0.81 and a specificity of 0.99. Our multi-student models consistently outperform single-student models across various conditions, including normal, polyp, and cancer cases. Extensive experiments demonstrate that Multi-StudentNet achieves state-of-the-art performance in both accuracy and robustness.