arXiv Computer Vision

ThreshGuide: Class-Aware Labeled-Guided Thresholding for Semi-Supervised 3D Abdominal Multi-Organ Segmentation

Hugging Face Trending Papers
Aug 12

CoDiR: Confidence-Guided Diffusion Refinement for Semi-Supervised Histopathology Segmentation

Semi-supervised histopathology segmentation is challenging due to scarce annotations and unreliable pseudo-labels in ambiguous gland regions. To address this problem, we propose Confidence-Guided Diffusion Refinement (CoDiR), a semi-supervised framework that combines a Mean Teacher segmentation model with diffusion-based pseudo-label refinement.

arXiv Computer Vision
Sep 2

Semi-Supervised Biomedical Image Segmentation via Diffusion Models and Teacher-Student Co-Training

The paper presents a semi‑supervised biomedical image segmentation method that uses a diffusion‑based teacher–student framework. The teacher is pretrained via unsupervised diffusion reconstruction and then co‑trained with a student, leveraging supervised labels and cross pseudo‑supervision on unlabeled data. A multi‑round extension generates multiple stochastic reconstructions to further refine pseudo‑labels, achieving competitive or superior results on several 2D and 3D biomedical datasets, especially when labels are scarce.

By Luca Ciampi, Gabriele Lagani, Giuseppe Amato, Fabrizio Falchi
arXiv Computer Vision
Aug 31

Anatomy-Aware Promptable Segmentation with Online Interactive Training for AUTOPET V

The paper introduces an anatomy-aware, promptable segmentation model for whole-body lesion detection in FDG and PSMA PET/CT scans, tailored for the AUTOPET V challenge. The approach builds on nnU-Net, employing a two-stage training process: an initial pre-training phase for strong baseline segmentation and an online interactive phase that refines predictions using scribble prompts. Anatomical context is integrated via organ supervision with a shared head predicting both lesions and organs, reducing false positives, while a tracer classifier directs studies to either a combined FDG+PSMA model or a PSMA-specific model. Cross-validation results show that organ-supervised training yields the most stable performance, the interactive stage consistently improves Dice scores, and PSMA-specific training delivers the best tracer-wise results.

By Pablo Lozano-Jimenez, Sergio Romero-Tapiador, Ruben Tolosana
arXiv Computer Vision
Aug 31

Report Supervision

The paper introduces Report Supervision (R‑Super), a framework that uses radiology reports to supervise tumor segmentation models. By incorporating loss functions that align segmentation outputs with report‑derived tumor counts, sizes, and locations, R‑Super improves detection and segmentation performance. Experiments on kidney and pancreatic tumors show up to a 15% increase in F1‑Score and DSC compared to mask‑only training, outperforming methods like CLIP and multi‑task learning.

By Pedro R. A. S. Bassia, Wenxuan Li, Jakob Wasserthal, Jieneng Chen, Xinze Zhou, Zheren Zhu, Chuntung Zhuanga, Sergio Decherchi, Andrea Cavalli, Kang Wang, Yang Yang, Alan Yuille, Zongwei Zhou
arXiv Machine Learning
Aug 27

Unsupervised Anatomical Feature Learning via Diffusion Models: Enhanced Medical Image Segmentation with Denoising Diffusion Probabilistic Models

The paper introduces an unsupervised approach to medical image segmentation by training a Denoising Diffusion Probabilistic Model (DDPM) on 21 unlabeled abdominal CT scans to learn anatomical features. The encoder weights from the DDPM are transferred to a U‑Net for downstream segmentation on the BTCV multi‑organ dataset, resulting in a significant Dice score improvement for liver segmentation from 0.75 to 0.93. In low‑data regimes, diffusion‑pretrained models retain robust performance, achieving high Dice scores even with only 10% of labeled data.

By Akshat G, Divyansh Gupta, Shaleen Bhatnagar, Shilpa Ankalaki, Tusar Kanti Mishra
Hugging Face Trending Papers
Sep 2

Progressive Pseudo-Label Optimization for Point-Supervised Change Detection

The paper introduces a two‑stage framework for point‑supervised change detection that leverages SAM2 priors to generate object‑aware candidate masks from sparse point annotations. In Stage I, a mask selection strategy converts generic segmentation outputs into reliable change pseudo‑labels, followed by a lightweight CNN refinement module with an uncertainty‑aware loss to enhance boundary quality. Stage II employs a teacher‑student self‑training loop, where the teacher is updated via exponential moving average and periodically refreshes pseudo‑labels, creating a closed‑loop optimization that alternates between pseudo‑label refinement and model re‑optimization. Experiments on WHU‑CD, LEVIR‑CD, and SYSU‑CD show the method surpasses prior weakly supervised approaches and competes with several fully supervised methods.

arXiv Computer Vision
Sep 3

Generalizable Brain Tumor Segmentation with Self-Training and Tumor-Aware Deformations

The paper introduces a method for generalizable brain tumor segmentation in the BraTS 2026 Challenge. It builds on the nnU-Net framework with a large residual encoder, adding semi‑supervised learning via pseudo‑labels and a tumor‑aware deformable augmentation that locally deforms lesions while preserving surrounding anatomy. The approach improves Dice and NSD scores across all tumor regions compared to labeled‑only baselines, demonstrating the complementary benefits of self‑training and the proposed augmentation.

By Henrique Zan Grande, Jeovane Honorio Alves, Rayson Laroca, Andre Gustavo Hochuli