nyukat / GMIC

An interpretable classifier for high-resolution breast cancer screening images utilizing weakly supervised localization
https://doi.org/10.1016/j.media.2020.101908
GNU Affero General Public License v3.0
168 stars 48 forks source link

Training #16

Closed cyh-0 closed 3 years ago

cyh-0 commented 3 years ago

Hi there,

Thanks for sharing the code. I have a question in terms of the training part. Given that the GMIC can be trained end-to-end, so the f_g and f_l would be updated simultaneously during training. But, at the early stage, the saliency maps could make mistake and cause the retrieve_roi function to extract incorrect patches (i.e., background), would that affect the convergence of the local module?

Cheers

seyiqi commented 3 years ago

Hi @Ch1kara ,

Yes, when the global module is weak in localizing true ROI, the local module might learn from possibly incorrect patches. In our experiment, we found that as long as the global module can provide non-trivial localization, the local module will eventually learn to classify using noisy patch sets that contain irrelevant inforamtion. Hope this helps.

Best,