Project-MONAI / Project-MONAI/MONAI

Feature Request: Add SparseDice, SparseClDice, CenterlineDice Losses and 3D TransUNet

Open
#8,731 1 comment 0 reactions 0 assignees View on GitHub

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

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. 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

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.