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 AI
Sep 18

Optimal Transport Metric Learning for Feature Alignment in Partially Supervised Segmentation

The paper proposes a two‑stage learning framework for multi‑organ segmentation that handles partially annotated datasets and domain shifts. First, the model learns accurate segmentations from available annotations to build robust feature representations. Second, it introduces learnable organ prototypes and a Sinkhorn‑triplet loss to enforce organ‑wise feature consistency across datasets, keeping embeddings of the same organ close while separating different organs, even when annotations are missing.

By Dakini Mallam Garba, Salim Abdou Daoura
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 Computer Vision
Oct 15

Selective Prediction Forgetting for Adaptation of Weakly Supervised Localization of Histology Images

arXiv:2603.12468v2 Announce Type: replace Abstract: Under strong cross-domain shifts, weakly supervised object localization (WSOL) models often lose their capacity to discriminate and develop a stron...

By Alexis Guichemerre, Banafsheh Karimian, Soufiane Belharbi, Natacha Gillet, Nicolas Thome, Pourya Shamsolmoali, Mohammadhadi Shateri, Luke McCaffrey, Eric Granger
arXiv Machine Learning
Nov 2

Enhancing Medical Lesion Segmentation via Several Simple Modules

The paper presents three lightweight modules to improve medical lesion segmentation, especially in data‑scarce and computationally constrained settings. The modules include a DT‑Morph Combined Loss for better topological and boundary optimization, a 3D Group Prompt that captures domain‑universal knowledge through volumetric image‑mask and text prompts, and a Many‑to‑One Conditional Segmentation Head that supports fast incremental learning with minimal computational overhead. Ablation studies on abdominal and extra‑abdominal datasets confirm the effectiveness of these components for both organ and lesion segmentation tasks.

By Ruichen Li, Zhao Yao