Project-MONAI / Project-MONAI/MONAI
Feature Request: Add SparseDice, SparseClDice, CenterlineDice Losses and 3D TransUNet
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 8.7k
- Forks
- 1.6k
- Avg merge
- 5d 1h
- Merged PRs (30d)
- 20
Description
Hello MONAI team,
I would like to request the addition of several features to MONAI that could benefit the community for 3D medical image segmentation tasks:
SparseDiceLoss – A Dice loss variant that supports sparse labels, useful for multi-class segmentation with incomplete annotations.
SparseClDiceLoss (Centerline Dice for sparse data) – Combines clDice and sparse label support for tubular structures such as vessels, fibers, or centerlines.
CenterlineDiceLoss (clDiceLoss) – Measures topological agreement between predicted and ground-truth structures by comparing their soft skeletons; very useful for thin elongated objects in 2D/3D.
3D TransUNet – Extend the current 2D TransUNet to full 3D, to handle volumetric medical images directly.
These additions would greatly help users working on complex 3D segmentation tasks, especially with sparse annotations and tubular structures.
Thank you for considering this feature request!
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Review MONAI's existing 2D TransUNet and loss implementations first, then separate the request into sparse-label losses, centerline Dice, and 3D TransUNet. Define completion against the stated needs: sparse annotations, topological agreement for tubular structures, and direct volumetric medical-image handling; the issue does not name files or tests.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python, pytorch
- Domain
- computer-vision, machine-learning
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 20/100