Deepreg: Support multiple loss functions

Created on 8 Nov 2020  路  8Comments  路  Source: DeepRegNet/DeepReg

Subject of the feature

What are you trying to do and how would you want to do it differently?

I'd like to use multiple loss functions together, e.g. dice loss + cross-entropy.
But it is not possible for the moment I guess.

If the feature request is approved, would you be willing to submit a PR? _(Help can be
provided if you need assistance submitting a PR)_

Yes

architecture feature

Most helpful comment

It allows multiple loss functions by specifying a list under each loss type: "image", "label", "regularization".
The way to do so is using a dashed sign (-). A config file example is:

loss:
  "image"
      - "name": "ssd"
         "weight": 0.5
      - "name": "lncc"
         "weight": 0.5
  "label"
       "name": "cross-entropy"
       "weight": 1.0
  "regularization":
        "name": "bending"
        "weight": 0.1
...

All 8 comments

I have had a vague idea for this now that I've worked on the config docs. @acasamitjana are you working on this right now? Docs ticket: #525

No, I haven't started yet, so go ahead. I'm happy to help if needed.

This issue is doable now with the registry :) Will any of you be still interested and available? @acasamitjana @NMontanaBrown

I'd close this if the other issue #512 is open and is going to use registry...?

This issue is doable now with the registry :) Will any of you be still interested and available? @acasamitjana @NMontanaBrown

I'd close this if the other issue #512 is open and is going to use registry...?

This issue is doable now with the registry :) Will any of you be still interested and available? @acasamitjana @NMontanaBrown

@NMontanaBrown I thought the multiple losses here mean:
You can apply two different losses to images at the same time, like LNCC loss and SSD ;)

Ah yes, I misread your comment as double not doable. I will not be able to work for this for the next couple of weeks actively, as I will be busy with some other things. So if they want to, someone else should pick this up, but otherwise I will return to it after a couple of weeks :) @mathpluscode

Ah yes, I misread your comment as double not doable. I will not be able to work for this for the next couple of weeks actively, as I will be busy with some other things. So if they want to, someone else should pick this up, but otherwise I will return to it after a couple of weeks :) @mathpluscode

Sure, I might handle it myself. It's a small change :)

It allows multiple loss functions by specifying a list under each loss type: "image", "label", "regularization".
The way to do so is using a dashed sign (-). A config file example is:

loss:
  "image"
      - "name": "ssd"
         "weight": 0.5
      - "name": "lncc"
         "weight": 0.5
  "label"
       "name": "cross-entropy"
       "weight": 1.0
  "regularization":
        "name": "bending"
        "weight": 0.1
...
Was this page helpful?
0 / 5 - 0 ratings

Related issues

NMontanaBrown picture NMontanaBrown  路  5Comments

YipengHu picture YipengHu  路  6Comments

mathpluscode picture mathpluscode  路  6Comments

mathpluscode picture mathpluscode  路  6Comments

mathpluscode picture mathpluscode  路  6Comments