Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Request for Baseline Training Code #14

Open
BubbleDirk opened this issue Dec 19, 2024 · 4 comments
Open

Request for Baseline Training Code #14

BubbleDirk opened this issue Dec 19, 2024 · 4 comments

Comments

@BubbleDirk
Copy link

I hope this message finds you well. I have been working on a project that builds upon your methodology and have successfully utilized the open-sourced code you provided. However, I encountered some challenges when attempting to replicate the baseline results reported in your paper. Specifically, the baseline performance on LA or Pancreas dataset I obtained (i.e., only use 10% labeled data for training) is significantly lower than the results you have described.

To ensure that my implementation aligns accurately with your experiments, I was wondering if you could share the original baseline training code. Having access to this would greatly aid in reproducing the baseline results and furthering my research based on your work.

Thank you very much for your time and assistance.

@himashi92
Copy link
Owner

All the reproducible hyperparameters and the model weights are mentioned in the repo.
To reproduce LA 10%, use this:
python train_cobionet_semi.py --dataset_name LA --labelnum 8 --lamda 0.7 --consistency 1.0 --mu 0.01 --t_m 0.4 --max_iteration 15000 --batch_size 4 --labeled_bs 2

@BubbleDirk
Copy link
Author

I am very grateful for your assistance. I'm trying to understand how you achieved the results shown in Figure 2(d), which is around 55.06%, using VNet with only 10% labeled data and no unlabeled data. Could you please point me to the file containing the corresponding training code?

@himashi92
Copy link
Owner

The code for this specific experiment is not in this repo. However, you can modify the code given here to train VNet using only 10% of LA data: https://github.com/himashi92/Co-BioNet/blob/main/code/train_cobionet_sup.py.

@BubbleDirk
Copy link
Author

I have indeed tried this, removing all modules and using only one vnet and the Dice supervised loss, with only random cropping for data augmentation. However, the results consistently remain around 20%, which has been puzzling me for a month. I am therefore writing to request your help. I would be extremely grateful if you could share the corresponding code.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants