Research
Weakly-Supervised Lung Nodule Segmentation via Training-Free Guidance of 3D Rectified Flow
arXiv:2604.08313v1 Announce Type: new Abstract: Dense annotations, such as segmentation masks, are expensive and time-consuming to obtain, especially for 3D medical images where expert voxel-wise labe
arXiv:2604.08313v1 Announce Type: new Abstract: Dense annotations, such as segmentation masks, are expensive and time-consuming to obtain, especially for 3D medical images where expert voxel-wise labeling is required. Weakly supervised approaches aim to address this limitation, but often rely on attribution-based methods that struggle to accurately capture small structures such as lung nodules. In this paper, we propose a weakly-supervised segmentation method for lung nodules by combining pretrained state-of-the-art rectified flow and predictor models in a plug-and-play manner. Our approach uses training-free guidance of a 3D rectified flow model, requiring only fine-tuning of the predictor using image-level labels and no retraining of the generative model. The proposed method produces improved-quality segmentations for two separate predictors, consistently detecting lung nodules of varying size and shapes. Experiments on LUNA16 demonstrate improvements over baseline methods, highlighting the potential of generative foundation models as tools for weakly supervised 3D medical image segmentation.
Related
- SD-FSMIS: Adapting Stable Diffusion for Few-Shot Medical Image Segmentation
- Adapting Foundation Models for Annotation-Efficient Adnexal Mass Segmentation in Cine Images
- Component-Adaptive and Lesion-Level Supervision for Improved Small Structure Segmentation in Brain MRI
- Flemme: A Flexible and Modular Learning Platform for Medical Images
- Hierarchical Feature Learning for Medical Point Clouds via State Space Model
Source: arXiv cs.CV | 2026-04-10