search

bowang-lab / BLEEP

Spatially Resolved Gene Expression Prediction from H&E Histology Images via Bi-modal Contrastive Learning
Apache License 2.0
56 stars 10 forks source link

The reproducibility of this paper #12

Closed Hanminghao closed 11 hours ago

Hanminghao commented 14 hours ago

Hello author, first of all thank you for your outstanding work. I had a problem with poor test set results while experimenting entirely with your code and data. Specifically, the model is running on a single A6000 48GB GPU. I have set the batch size to 512 and the learning rate to 1e-3. Apart from not using distributed training, my setup is the same as yours. In addition, the best model filtered by the validation set appears at epoch 38, with a val loss of 3.38. However, the model is only showing performance around 0.02 and 0.03 on the metrics of mean correlation highly expressed genes and mean correlation highly variable genes. I really look forward to your receiving my questions. Thank you. image image

Hanminghao commented 11 hours ago

After I changed to distributed training, I got results similar to the manuscript. image