Mutual Distillation of Dual-Foundation Models for Semi-Supervised PET/CT Segmentation

· Source: Artificial Intelligence · Field: Technology & Digital — Artificial Intelligence & Machine Learning, Computer Vision & Pattern Recognition · Depth: Expert, quick

Summary

MuDuo, a novel mutual distillation framework, significantly advances semi-supervised PET/CT segmentation, a critical task for quantitative analysis and radiotherapy planning in oncology. This framework addresses the high annotation costs associated with medical imaging by leveraging both structural CT and metabolic PET imaging foundation models. Specifically, it employs SAM-Med3D for CT and SegAnyPET for PET, acting as modality-specific generalists. MuDuo distills knowledge from these models into a lightweight student network, effectively bridging the gap between task-specific precision and the segmentation priors of generalist foundation models. The approach eliminates the need for manual prompts and maximizes the utility of unlabeled data, achieving state-of-the-art performance on the AutoPET dataset with only 5 labeled cases.

Key takeaway

For research scientists developing medical image segmentation models, MuDuo offers a compelling strategy to overcome annotation bottlenecks. By integrating dual-foundation models via mutual distillation, you can achieve state-of-the-art performance on complex tasks like PET/CT segmentation using significantly fewer labeled examples. Consider adopting this framework to accelerate model development and deployment in resource-constrained medical imaging environments.

Key insights

Mutual distillation of dual-foundation models improves semi-supervised PET/CT segmentation with minimal labeled data.

Principles

Method

MuDuo uses mutual distillation to transfer knowledge from SAM-Med3D (CT) and SegAnyPET (PET) into a lightweight student network, eliminating manual prompts for segmentation.

In practice

Topics

Code references

Best for: Computer Vision Engineer, AI Scientist, Research Scientist

Related on AIssential

Open in AIssential →

Editorial summary, takeaway, and curation by AIssential. Original article published by Artificial Intelligence.