MIC-DKFZ / MedNeXt

[MICCAI 2023] MedNeXt is a fully ConvNeXt architecture for 3D medical image segmentation.
https://arxiv.org/pdf/2303.09975
Apache License 2.0
345 stars 26 forks source link

MedNeXt fails to converge for my specific task #19

Closed YankaiChen0308 closed 6 months ago

YankaiChen0308 commented 6 months ago

I have encountered difficulties while attempting to solely utilize the MedNeXt architecture for image segmentation tasks without employing nnUnet's training process. I have observed that the model fails to converge for my specific task, with the Dice coefficient hovering around 0.5. I am seeking insights on the possible reasons behind this issue and would greatly appreciate it if you could share your experiences in this regard. My speculation is that the problem may stem from improper network initialization. Could you kindly advise on how weight initialization should be carried out for MedNeXt? Thank you for your assistance.