Algolzw / image-restoration-sde

Image Restoration with Mean-Reverting Stochastic Differential Equations, ICML 2023. Winning solution of the NTIRE 2023 Image Shadow Removal Challenge.

Home Page:https://algolzw.github.io/ir-sde/index.html

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to determine the max_sigma

h7eona opened this issue · comments

Hi Algolzw,
I wonder you how to determine the max_sigma and surrogate differentiable function (20).
Could you explain in detail if it's decided by something?

Hi, we just manually chose the max_sigma. And in our Refusion paper, experiments show that setting the max_sigma to be 30 or 50 would good to most tasks.

Alright, Thank you so much!
I have one more question. How to determine the surrogate differentiable function (20)?

Hi, the surrogate differentiable function is a tricky choice. I just tried many times on my draft and found it is the best function for solving the SDE.

Thanks for the details. The more I read the paper, the more questions I have. One more thing I'm wondering:

When inferring, I understand that the Markov chain iterates the desired output for a number of timesteps, but when training, I wonder if you are training to generate x(t) as x(0) after degrading x(0) to that timestep and adding noise, or x(t) as x(t-1)?

In training, we generate x(t) from x(0) using this function, which is based on the forward-SDE solution (Equation (6) in the paper).

Thank you! But, I mean I wonder the backward process. Is it right that the score network train the x(t) to x(t-1)?

Thank you! But, I mean I wonder the backward process. Is it right that the score network train the x(t) to x(t-1)?

Yes if you use maximum likelihood loss.