MIC-DKFZ / nnUNet

Apache License 2.0
5.97k stars 1.77k forks source link

Customizing training parameters for stronger focus on structural information #2536

Closed code8865 closed 6 days ago

code8865 commented 1 month ago

Hi there,

I’m currently working with “intentional manipulated” CT data. This manipulation can cause voxel regions to appear unique. I would like the network to "reinterpret" these areas based on learned structural information, by assigning the appropriate labels in line with the expected anatomy.

nnUNet’s default setup handles such problems reasonably well. However, I’m looking for ways to optimize the network training so that it focuses even more on structural consistency or spatial relationships.

Does anyone know methods (e.g. another loss function) to focus more on these structural information when dealing with unexpected data deviations?

Thanks a lot in advance!

ykirchhoff commented 3 weeks ago

Hi @code8865,

I am not sure I understand what you are exactly trying to do here, could you maybe give me some more details?

Best, Yannick