Consistency regularization and pseudo labeling-based semi-supervised methods perform co-training using the pseudo labels from multi-view inputs. However, such co-training models tend to converge early to a consensus, degenerating to the self-training ones, and produce low-confidence pseudo labels from the perturbed inputs during training. To address these issues, we propose an Uncertainty-guided Collaborative Mean-Teacher (UCMT) for semi-supervised semantic segmentation with the high-confidence pseudo labels. Concretely, UCMT consists of two main components: 1) collaborative mean-teacher (CMT) for encouraging model disagreement and performing co-training between the sub-networks, and 2) uncertainty-guided region mix (UMIX) for manipulating the input images according to the uncertainty maps of CMT and facilitating CMT to produce high-confidence pseudo labels. Combining the strengths of UMIX with CMT, UCMT can retain model disagreement and enhance the quality of pseudo labels for the co-training segmentation. Extensive experiments on four public medical image datasets including 2D and 3D modalities demonstrate the superiority of UCMT over the state-of-the-art. Code is available at: https://github.com/Senyh/UCMT.
Comparative Benchmarking of Failure Detection Methods in Medical Image Segmentation: Unveiling the Role of Confidence Aggregation
German Cancer Research Center (DKFZ)(德国癌症研究中心);Medical Faculty Heidelberg, Heidelberg University(海德堡大学海德堡医学院);Helmholtz Imaging(亥姆霍兹成像);Heidelberg University Hospital(海德堡大学医院)
East China Normal University(华东师范大学);Huashan Hospital, Fudan University(复旦大学附属华山医院)
Medical image segmentation is one of the fundamental problems for artificial intelligence-based clinical decision systems. Current automatic medical image segmentation methods are often failed to meet clinical requirements. As such, a series of interactive segmentation algorithms are proposed to utilize expert correction information. However, existing methods suffer from some segmentation refining failure problems after long-term interactions and some cost problems from expert annotation, which hinder clinical applications. This paper proposes an interactive segmentation framework, called inte…
CAD: Confidence-Aware Adaptive Displacement for Semi-Supervised Medical Image Segmentation
College of Information and Intelligence, Hunan Agricultural University(湖南农业大学信息与智能科学学院);School of Computer Science and Engineering, University of New South Wales(新南威尔士大学计算机科学与工程学院)
A Chebyshev Confidence Guided Source-Free Domain Adaptation Framework for Medical Image Segmentation
Harbin Institute of Technology at Shenzhen(哈尔滨工业大学(深圳));The Peng Cheng Laboratory(鹏城实验室);Guangdong Provincial Key Laboratory of Aerospace Communication and Networking Technology(广东省航空航天通信与网络技术重点实验室)
The University of British Columbia(不列颠哥伦比亚大学);Harvard Medical School(哈佛医学院)
Fully convolutional neural networks (FCNs), and in particular U-Nets, have achieved state-of-the-art results in semantic segmentation for numerous medical imaging applications. Moreover, batch normalization and Dice loss have been used successfully to stabilize and accelerate training. However, these networks are poorly calibrated i.e. they tend to produce overconfident predictions both in correct and erroneous classifications, making them unreliable and hard to interpret. In this paper, we study predictive uncertainty estimation in FCNs for medical image segmentation. We make the following co…
From Few to More: Scribble-based Medical Image Segmentation via Masked Context Modeling and Continuous Pseudo Labels
National Engineering Laboratory for Integrated Aero-Space-Ground-Ocean Big Data Application Technology, School of Computer Science and Engineering, Northwestern Polytechnical University(集成空天地海大数据应用技术国家工程实验室,计算机科学与工程学院,西北工业大学)
Foundation Model-guided Iteratively Prompting and Pseudo-Labeling for Partially Labeled Medical Image Segmentation
Department of Biomedical Engineering, Columbia University(哥伦比亚大学生物医学工程系);Department of Radiation Oncology, Columbia University Irving Medical Center(哥伦比亚大学欧文医学中心放射肿瘤科);Data Science Institute, Columbia University(哥伦比亚大学数据科学研究所);Herbert Irving Comprehensive Cancer Center, Columbia University(哥伦比亚大学赫伯特·欧文综合癌症中心);Department of Radiation Oncology, Icahn School of Medicine at Mount Sinai(西奈山伊坎医学院放射肿瘤科)
University of Trento(特伦托大学);Fondazione Bruno Kessler(布鲁诺·凯斯勒基金会);Facebook Reality Labs(Facebook现实实验室)
Pseudo-LiDAR-based methods for monocular 3D object detection have received considerable attention in the community due to the performance gains exhibited on the KITTI3D benchmark, in particular on the commonly reported validation split. This generated a distorted impression about the superiority of Pseudo-LiDAR-based (PL-based) approaches over methods working with RGB images only. Our first contribution consists in rectifying this view by pointing out and showing experimentally that the validation results published by PL-based methods are substantially biased. The source of the bias resides in…
FPL+: Filtered Pseudo Label-based Unsupervised Cross-Modality Adaptation for 3D Medical Image Segmentation
School of Mechanical and Electrical Engineering, University of Electronic Science and Technology of China(电子科技大学机械与电气工程学院);Shanghai AI laboratory(上海人工智能实验室);West China Hospital, Sichuan University(四川大学华西医院);West China Biomedical Big Data Center, West China Hospital, Sichuan University(四川大学华西医院华西生物医学大数据中心);SenseTime Research(商汤科技研究院)
Enhancing Dual Network Based Semi-Supervised Medical Image Segmentation with Uncertainty-Guided Pseudo-Labeling
School of Artificial Intelligence, Guilin University of Electronic Technology, Guilin, China, 541004(桂林电子科技大学人工智能学院);The Laboratory for Imagery, Vision and Artificial Intelligence, École de Technologie Supérieure, Montreal, Canada, H3C 1K3(影像、视觉与人工智能实验室);College of Medicine, Taibah University, Al Madinah, Saudi Arabia, 42353(塔布大学医学院);Department of Radiology, Prince Mohammed Bin Abdulaziz Hospital, Ministry of National Guard Health Affairs, Al Madinah, Saudi Arabia, 42324(王子穆罕默德·本·阿卜杜勒阿齐兹医院放射科);College of Computer Science and Engineering, Taibah University, Madinah, Saudi Arabia, 42353(塔布大学计算机科学与工程学院);College of Computer Science and Engineering, Jeddah University, Jeddah, Saudi Arabia, 23445(吉达大学计算机科学与工程学院)
SAM Carries the Burden: A Semi-Supervised Approach Refining Pseudo Labels for Medical Segmentation
Institute of Medical Informatics, Universität zu Lübeck(吕贝克大学医学信息学研究所);EchoScout GmbH(EchoScout有限公司);University Hospital Schleswig-Holstein(石勒苏益格-荷尔斯泰因大学医院)
本文提出利用 SAM 的提示分割能力为医学图像半监督学习生成精化伪标签,在儿科腕骨和牙科牙齿分割上显著提升 Dice 分数,并优于 nnU-Net 和 mean teacher 等现有方法。
The Hong Kong University of Science and Technology(香港科技大学);The Chinese University of Hong Kong(香港中文大学);The Hong Kong University of Science and Technology Shenzhen Research Institute(香港科技大学深圳研究院)
Generalizing the medical image segmentation algorithms to unseen domains is an important research topic for computer-aided diagnosis and surgery. Most existing methods require a fully labeled dataset in each source domain. Although some researchers developed a semi-supervised domain generalized method, it still requires the domain labels. This paper presents a novel confidence-aware cross pseudo supervision algorithm for semi-supervised domain generalized medical image segmentation. The main goal is to enhance the pseudo label quality for unlabeled images from unknown distributions. To achieve…
Witnessing the success of deep learning neural networks in natural image processing, an increasing number of studies have been proposed to develop deep-learning-based frameworks for medical image segmentation. However, since the pixel-wise annotation of medical images is laborious and expensive, the amount of annotated data is usually deficient to well-train a neural network. In this paper, we propose a semi-supervised approach to train neural networks with limited labeled data and a large quantity of unlabeled images for medical image segmentation. A novel pseudo-label (namely self-loop uncer…
Dual Cross-image Semantic Consistency with Self-aware Pseudo Labeling for Semi-supervised Medical Image Segmentation
School of Biomedical Engineering & State Key Laboratory of Advanced Medical Materials and Devices, ShanghaiTech University(生物医学工程学院及先进医疗材料与器件国家重点实验室,上海科技大学);Lingang Laboratory(临港实验室);Department of Radiology, Stanford University(放射科,斯坦福大学)
Recent semi-supervised learning methods use pseudo supervision as core idea, especially self-training methods that generate pseudo labels. However, pseudo labels are unreliable. Self-training methods usually rely on single model prediction confidence to filter low-confidence pseudo labels, thus remaining high-confidence errors and wasting many low-confidence correct labels. In this paper, we point out it is difficult for a model to counter its own errors. Instead, leveraging inter-model disagreement between different models is a key to locate pseudo label errors. With this new viewpoint, we pr…
Cross Domain Object Detection via Multi-Granularity Confidence Alignment based Mean Teacher
National University of Defense Technology (NUDT)(国防科技大学);College of System Engineering, National University of Defense Technology (NUDT)(国防科技大学系统工程学院);College of Electronic Science and Technology, NUDT(国防科技大学电子科学学院);College of Meteorology and Oceanography, NUDT(国防科技大学气象海洋学院)
Bidirectional Uncertainty-Aware Region Learning for Semi-Supervised Medical Image Segmentation
Key Laboratory of Intelligent Computing \& Signal Processing (ICSP), Ministry of Education, School of Artificial Intelligence, Anhui Uniersity, Hefei, 230601, China;Anhui Provincial Key Laboratory of Multimodal Cognitive Computation, School of Computer Science;Technology, Anhui University, Hefei, 230601, China;Institute of Artificial Intelligence, Hefei Comprehensive National Science Center, Hefei, 230026, China
Peking University(北京大学);Deepwise AI Lab(深睿医疗人工智能实验室);Microsoft Research Asia(微软亚洲研究院);The University of Hong Kong(香港大学)
As an important upstream task for many medical applications, supervised landmark localization still requires non-negligible annotation costs to achieve desirable performance. Besides, due to cumbersome collection procedures, the limited size of medical landmark datasets impacts the effectiveness of large-scale self-supervised pre-training methods. To address these challenges, we propose a two-stage framework for one-shot medical landmark localization, which first infers landmarks by unsupervised registration from the labeled exemplar to unlabeled targets, and then utilizes these noisy pseudo l…
Semi-supervised learning (SSL) can reduce the need for large labelled datasets by incorporating unlabelled data into the training. This is particularly interesting for semantic segmentation, where labelling data is very costly and time-consuming. Current SSL approaches use an initially supervised trained model to generate predictions for unlabelled images, called pseudo-labels, which are subsequently used for training a new model from scratch. Since the predictions usually do not come from an error-free neural network, they are naturally full of errors. However, training with partially incorre…
TABNet: A Triplet Augmentation Self-Recovery Framework with Boundary-Aware Pseudo-Labels for Medical Image Segmentation
School of Information Science and Technology, Northwest University(信息科学与技术学院,西北大学);Institute of Biomedical Engineering, Department of Engineering Science, University of Oxford(生物医学工程研究所,工程科学系,牛津大学)
Deep learning models, e.g. supervised Encoder-Decoder style networks, exhibit promising performance in medical image segmentation, but come with a high labelling cost. We propose TriSegNet, a semi-supervised semantic segmentation framework. It uses triple-view feature learning on a limited amount of labelled data and a large amount of unlabeled data. The triple-view architecture consists of three pixel-level classifiers and a low-level shared-weight learning module. The model is first initialized with labelled data. Label processing, including data perturbation, confidence label voting and unc…
Dual-Decoder Consistency via Pseudo-Labels Guided Data Augmentation for Semi-Supervised Medical Image Segmentation
Fuzhou University(福州大学);Fujian Medical University Union Hospital(福建医科大学附属协和医院);Fujian Medical Ultrasound Research Institute(福建省医学超声研究所);Macao Polytechnic University(澳门理工大学)
Nanjing University of Science and Technology(南京理工大学);University of Notre Dame(圣母大学)
We aim to quantitatively measure the practical usability of medical image segmentation models: to what extent, how often, and on which samples a model's predictions can be used/trusted. We first propose a measure, Correctness-Confidence Rank Correlation (CCRC), to capture how predictions' confidence estimates correlate with their correctness scores in rank. A model with a high value of CCRC means its prediction confidences reliably suggest which samples' predictions are more likely to be correct. Since CCRC does not capture the actual prediction correctness, it alone is insufficient to indicat…
Diff-CL: A Novel Cross Pseudo-Supervision Method for Semi-supervised Medical Image Segmentation
School of Mathematical Science(数学科学学院);Academy for Multidisciplinary Studies(交叉学科研究院);International Information and Software Institute(国际信息与软件学院);Dalian University of Technology(大连理工大学)
Semi-supervised Medical Image Segmentation Method Based on Cross-pseudo Labeling Leveraging Strong and Weak Data Augmentation Strategies
Hangzhou Dianzi University(杭州电子科技大学);Zhejiang University of Finance and Economics(浙江财经大学);Shenzhen Institute of Advanced Technology, Chinese Academy of Sciences(中国科学院深圳先进技术研究院);Shenzhen Children's Hospital(深圳市儿童医院);Shenzhen Research Institute of Big Data(深圳市大数据研究院)
Baidu Inc.(百度公司);The University of Hong Kong(香港大学);Purdue University(普渡大学);University of Electronic Science and Technology of China(电子科技大学);Shenzhen Eye Hospital, Jinan University(暨南大学附属深圳眼科医院)
In medical image segmentation, it is often necessary to collect opinions from multiple experts to make the final decision. This clinical routine helps to mitigate individual bias. But when data is multiply annotated, standard deep learning models are often not applicable. In this paper, we propose a novel neural network framework, called Multi-Rater Prism (MrPrism) to learn the medical image segmentation from multiple labels. Inspired by the iterative half-quadratic optimization, the proposed MrPrism will combine the multi-rater confidences assignment task and calibrated segmentation task in a…
Pseudo Label-Guided Data Fusion and Output Consistency for Semi-Supervised Medical Image Segmentation
college of physics and information engineering, University of Fuzhou University(福州大学物理与信息工程学院);Faculty of Applied Science, University of Macao Polytechnic University(澳门理工大学应用科学学院);University of Toronto University(多伦多大学)
Hebei University of Technology(河北工业大学);University of Oxford(牛津大学)
Deep learning based semi-supervised learning (SSL) methods have achieved strong performance in medical image segmentation, which can alleviate doctors' expensive annotation by utilizing a large amount of unlabeled data. Unlike most existing semi-supervised learning methods, adversarial training based methods distinguish samples from different sources by learning the data distribution of the segmentation map, leading the segmenter to generate more accurate predictions. We argue that the current performance restrictions for such approaches are the problems of feature extraction and learning pref…
University College London(伦敦大学学院);NAAMII;Redev Technology(Redev科技)
We present our novel deep multi-task learning method for medical image segmentation. Existing multi-task methods demand ground truth annotations for both the primary and auxiliary tasks. Contrary to it, we propose to generate the pseudo-labels of an auxiliary task in an unsupervised manner. To generate the pseudo-labels, we leverage Histogram of Oriented Gradients (HOGs), one of the most widely used and powerful hand-crafted features for detection. Together with the ground truth semantic segmentation masks for the primary task and pseudo-labels for the auxiliary task, we learn the parameters o…
Karlsruhe Institute of Technology(卡尔斯鲁厄理工学院);University Medicine Essen(埃森大学医学院)
Producing densely annotated data is a difficult and tedious task for medical imaging applications. To address this problem, we propose a novel approach to generate supervision for semi-supervised semantic segmentation. We argue that visually similar regions between labeled and unlabeled images likely contain the same semantics and therefore should share their label. Following this thought, we use a small number of labeled images as reference material and match pixels in an unlabeled image to the semantics of the best fitting pixel in a reference set. This way, we avoid pitfalls such as confirm…
Recent advances in domain adaptation show that deep self-training presents a powerful means for unsupervised domain adaptation. These methods often involve an iterative process of predicting on target domain and then taking the confident predictions as pseudo-labels for retraining. However, since pseudo-labels can be noisy, self-training can put overconfident label belief on wrong classes, leading to deviated solutions with propagated errors. To address the problem, we propose a confidence regularized self-training (CRST) framework, formulated as regularized self-training. Our method treats ps…
Leveraging Labelled Data Knowledge: A Cooperative Rectification Learning Network for Semi-supervised 3D Medical Image Segmentation
School of Mechanical Engineering and Automation, Northeastern University(东北大学机械工程与自动化学院);Australian Institute for Machine Learning, University of Adelaide(阿德莱德大学澳大利亚机器学习研究所);Centre for Vision, Speech and Signal Processing (CVSSP), University of Surrey(萨里大学视觉、语音与信号处理中心)
Boosting Semi-Supervised Medical Image Segmentation via Masked Image Consistency and Discrepancy Learning
School of Information and Communication Engineering, Beijing University of Posts and Telecommunications(北京邮电大学信息与通信工程学院);School of Artificial Intelligence, Beijing University of Posts and Telecommunications(北京邮电大学人工智能学院);School of Computer Science and Engineering, Southeast University(东南大学计算机科学与工程学院)
Computer Vision Laboratory, ETH Zurich(苏黎世联邦理工学院计算机视觉实验室)
Supervised deep learning-based methods yield accurate results for medical image segmentation. However, they require large labeled datasets for this, and obtaining them is a laborious task that requires clinical expertise. Semi/self-supervised learning-based approaches address this limitation by exploiting unlabeled data along with limited annotated data. Recent self-supervised learning methods use contrastive loss to learn good global level representations from unlabeled images and achieve high performance in classification tasks on popular natural image datasets like ImageNet. In pixel-level …
Mutual Evidential Deep Learning for Medical Image Segmentation
Key Laboratory of High Confidence Software Technologies (MOE) School of Computer Science Peking University(高可信软件技术重点实验室(MOE)计算机科学学院,北京大学);College of Computer and Information Science School of Software Southwest University(计算机与信息科学学院 软件学院 西南大学);Department of Computer and Information Science University of Macau(计算机与信息科学系 澳门大学)
ProCNS: Progressive Prototype Calibration and Noise Suppression for Weakly-Supervised Medical Image Segmentation
Southern University of Science and Technology(南方科技大学);University of Hong Kong(香港大学);Jiaxing Research Institute, Southern University of Science and Technology(南方科技大学嘉兴研究院)