cvlab-epfl / iter_unc

Official code for "Enabling Uncertainty Estimation in Iterative Neural Networks" (ICML 2024)
https://www.norange.io/projects/unc_iter/
MIT License
5 stars 0 forks source link

Remote Sensing Example #1

Open nilsleh opened 5 days ago

nilsleh commented 5 days ago

Hi, Thanks for providing the code to your paper. I was wondering if there is an ETA for the remote sensing example you provide in your paper. We are trying to add your method to our UQ library. Our background is Earth Observation data and we are planning a new release soonish, so we would like include your method and create a tutorial notebook showcasing the method for a remote sensing task. Cheers and thanks again!

nikitadurasov commented 5 days ago

Hey @nilsleh, thanks for the interest in our work!

Sure, I think it would be great adding our approach to the library. How do you think the tutorial would look like?

We were thinking about sharing pretrained model with some input images showcasing confident / uncertain cases (similar to the teaser image in the paper).

I'm not sure about time, but let's say we'll try to figure it out within a couple of weeks?

Let me know if it sounds good!

Best, Nikita

nilsleh commented 4 days ago

Thanks for your reply, the question was really out of curiosity and didn't mean to pressure time-wise! So personally, there would be two aspects that we like to consider for a notebook:

We tried doing that with the ZigZag MNIST Example (adding some explanation and mathematical details to the notebook is still on our ToDO List). So for the remote sensing example with the Iterative Method, we would:

Generally we try to structure the notebooks as follows, for example in this SWAG example:

From our perspective, we would only need to have details about the dataset and training scheme you used (some script similar to the ones you have provided for ZigZag MNIST or the regression cases here, would be helpful), so we can try to reproduce it, and then we can add the notebook. Of course, if you like and have any time availability, we'd also be happy if you want to add some details to a notebook or provide feedback/suggestions etc. Thanks in advance :)

nikitadurasov commented 1 day ago

Hey @nilsleh,

Sounds good! The issue with the code release for remote sensing experiments is that my colleague responsible for them recently left the lab. However, I've obtained the code from him. We have two options now: I can share the model architecture, training pipeline, data, etc., with you so you can re-implement it. Meanwhile, I'll prepare the code to add to this repository. I believe that's the best option; please email me at my university address :)

Additionally, I could release the MNIST experiments for the ICML paper (similar to what you have for Zigzag). It's a good idea too.

Best, Nikita