[ENH] roadmap of probability distributions to implement

[fkiraly]

It would be great to have a basic set of probability distributions implemented.

Umbrella issue for implementing sktime probability distributions.

Recipe: use the extension_templates/distribution.py extension template.
Examples:

• Normal, for de-novo implementations or manual interfaces
• Fisk, for interfacing scipy distributions - this is much easier than using the full template

High priority:

• laplace - [ENH] Laplace distribution #19
• empirical incl delta - [ENH] empirical distribution #25
• multivariate normal

mid priority:

• t-distribution - [ENH] Student's t-distribution #49
• Cauchy (special case of t) - [ENH] Student's t-distribution #49
• mixture composition - [ENH] mixture of distributions #26
• truncation, compositor

low priority:

• uniform continuous [ENH] Uniform Continuous distribution #223
• beta [ENH] Beta Distribution #298
• chi-squared [ENH] Chi-Squared Distribution #217
• exponential [ENH] Exponential distribution #325
• logistic [ENH] Logistic distribution #241
• log-normal [ENH] Log-normal probability distribution #218
• Weibull [ENH] Weibull distribution #242

lower priority:

• alpha [ENH] Alpha distribution #356
• binomial
• burr III
• burr XII
• erlang
• f
• fatigue-life
• generalized Pareto
• gamma [ENH] Gamma Distribution #355
• geometric
• half-cauchy
• half-normal
• half-logistic
• levy
• log-gamma
• log-laplace
• negative binomial
• pareto
• skellam
• truncated normal
• truncated pareto

list of many more (lowest priority)
https://docs.scipy.org/doc/scipy/reference/stats.html#probability-distributions - can be interfaced via _ScipyDist adapter easily!
https://en.wikipedia.org/wiki/File:ProbOnto2.5.jpg

Mirrors sktime/sktime#4518
(for high and mid priority)

Contributions can be made to either repository, and should be copied over to the other once approved/merged, until the modules are merged into one.

[bhavikar04]

Hi, I'm interested in taking this up. Would you say priority of the distributions aligns with the difficulty in implementation? I'd like to do either multivariate normal or uniform continuous.

[fkiraly]

Hmmm, I'd say it is currently actually the opposite way. That is, the remaining low priority ones are easier to get started with, than the remaining high priority ones - simply since the easy higher priority ones are already done.

So, uniform continuous then? Parameterized by lower and upper.

I don't have a reference for energy and squared norm integrals, but these should not be too difficult to obtain. Let me know if you need input there, we can always start with the more common methods.

[an20805]

Hey @fkiraly, I have implemented uniform continuous distribution in my local branch. How do I proceed further?
I would also love to implement other distributions.

[fkiraly]

@an20805, nice! Let's not duplicate then, @bhavikar04 - how about beta?

The next step would be making a pull request to this repository, and a review cycle, then merge.

[fkiraly]

Re energy, for $X, Y\sim Unif(a, b)$, I get:

$\mathbb{E}[|X - y|] = |y - \frac{b+a}{2} |$ if $y$ lies outside $[a, b]$,
and $\mathbb{E}[|X - y|] = \frac{(b-y)^2}{2(b-a)}+ \frac{(a-y)^2}{2(b-a)}$ if inside,

and

$\mathbb{E}[|X - Y|] = \frac{1}{3} (b-a)$ - double checking appreciated.

[bhavikar04]

In that case I'll take up log normal distribution then.

[fkiraly]

Pinging @Alex-JG3 and @ivarzap who most recently implemented distributions, in case you have any general starter advice.

[bhavikar04]

Hey,

So I'm a little unsure on what the energy will be for the log normal distribution and can't find much online, is there any literature you can point me to?

[fkiraly]

@bhavikar04, Appendix A.2 of "evaluating forecasts with scoringRules" has a few explicit formulae for the energy, including the log-normal distribution. The expression is hard to track in implementation, so I would advise comparing against the Monte-Carlo default if you implement it.

I would also suggest you try it on paper, there's a good chance of errors in rare calculations like these.
Further, Wolfram Alpha might also help. Whereas, ChatGPT and the like typically produce plausible garbage.

[bhavikar04]

Hey thank you so much, I'll try to chalk out a suitable implementation soon. ChatGPT was humble enough to admit it doesn't know enough ;)

[fkiraly]

Yes, I admit I also tried as computing integrals can get tiring: https://xkcd.com/2117/
Wolfram is not bad, it makes sense to double check though. As said, there is a default Monte Carlo implementation, so if you set the number of samples high, the matrices should be similar.

[sukjingitsit]

I would like to work on implementing the chi-square distribution. To confirm, we have to follow the template of Laplace and Normal, where we implement the
mean, var, pdf, cdf, logpdf and ppf alongside the energy, right?
To characterise chi-square, I assume, as standard practice, we will use the degrees of freedom, right?

[sukjingitsit]

The current implementation of ppf wraps a scipy function directly due to lack of a closed mathematical form. Similarly, while cross-energy can be mathematically derived, self-energy is difficult to solve (nor could I find literature on it) in a closed form, the best options for that is integration or sampling. Thus, energy hasn't been implemented yet

[malikrafsan]

Ahh, I see, thank you so much for your guidance @fkiraly ! Then my PRs are ready to be reviewed. I would very much love to hear your feedback. However, if you have any reference on energy formula of those two distribution, then I would still very like to implement it, thank you so much!

[fkiraly]

if you have any reference on energy formula of those two distributiom

have you checked in the paper above? If not there, one would have to derive it.

[malikrafsan]

Do you mean this paper?

Appendix A.2 of "evaluating forecasts with scoringRules" has a few explicit formulae for the energy, including the log-normal distribution.

Yes, I have checked the paper but I cannot find the formula. I can only find CRPS and CDF formulas. Does CRPS mean the energy? If so, I think I misunderstood your previous statements

[fkiraly]

Yes, CRPS is closely releated, it is the cross-term minus half the self-term (compare definitions).

The unfortunate bit about the paper is that it only gives CRPS, but not the self-term or cross-term in isolation. However, it should not be too hard to back these out, using that shifting the distribution location by a constant leaves the self-term unchanged, but not the cross-term.

[fkiraly]

More precisely, a useful formula to use is

$$\lim_{y \rightarrow \infty} \mbox{CRPS}(y) - y = -\mathbb{E}[X] - \frac{1}{2}\mathbb{E}|X - X'|,$$

i.e., you can obtain the cross-term via taking a limit, if you know the expressions for CRPS and the expectation already.

(the equation follows from observing that the absolute value in $\mathbb{E}\left|y - X \right|$ disappears in the limit)

[sukjingitsit]

I would like to work on the Pareto distribution if possible

[fkiraly]

all yours, @sukjingitsit!