epiforecasts / EpiNow2

Estimate Realtime Case Counts and Time-varying Epidemiological Parameters
https://epiforecasts.io/EpiNow2/dev/
Other
112 stars 31 forks source link

Locally stationary GPs #765

Open seabbs opened 1 week ago

seabbs commented 1 week ago

Extend the current GP functionality to be locally stationary by using a patch based approach of stationary Gps with the GP parameters being related with hierarchical distributions. The main issue with this approach is what to do at the boundary between patches. In the first instance I think it makes sense do nothing and see how it goes but if/when that fails I think extending patches to overlap and using a sigmoid or similar to transition from one patch to another might be a good way to go.

As a feature this would support better fitting to retrospective data when the lengthscale has changed over time. It might also be a useful forecasting model in some formulations if you tweak the hierarchical priors to have long lengthscales (as this makes the prior model for future patches that they have long lenghtscales (i.e slower change over time with less variance).

seabbs commented 1 week ago

I plan to work on this in the next few months