Project-MONAI / Project-MONAI/MONAI

Add sigmoid/softmax interface for `AsymmetricUnifiedFocalLoss`

Open
#8,603 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

Is your feature request related to a problem? Please describe.
It seems that AsymmetricUnifiedFocalLoss has not yet been updated to the latest interface like FocalLoss. It is a general loss function and we should make it easier for everyone to use. We can also allow multiple classes segmentation since it only supports binary one accordingly

Describe the solution you'd like
Follow the interface of FocalLoss. Also, we don't need to reinvent the wheel. I am sure we can use the available loss function to replace the implementation.

Describe alternatives you've considered
No

Additional context
No

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

Start in monai/losses/unified_focal_loss.py at the implementation linked in the issue, and compare the current AsymmetricUnifiedFocalLoss entry points with the FocalLoss interface. Done means the asymmetric unified focal loss exposes sigmoid and softmax usage and supports multiclass segmentation while reusing an available loss implementation where appropriate.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
machine-learning
Issue type
Feature
Difficulty
4/5
Estimated time
3-5 days
Activity status
Quiet
Clarity
Mostly clear
Newbie friendliness
48/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.