pymc-devs/pymc

Add rewrite for Mixture when `comp_dists` can be "fused"

Open

#6803 aperta il 29 giu 2023

Vedi su GitHub
 (0 commenti) (1 reazione) (0 assegnatari)Python (1902 fork)batch import
feature requesthelp wantedlogprob

Metriche repository

Star
 (7926 star)
Metriche merge PR
 (Merge medio 11g 6h) (12 PR mergiate in 30 g)

Descrizione

Description

The following distributions are equivalent:

import pymc as pm

pm.Mixture.dist(w=[0.5, 0.5], comp_dists=[pm.Normal.dist(-1), pm.Normal.dist(1)])
pm.Mixture.dist(w=[0.5, 0.5], comp_dists=pm.Normal.dist([-1, 1]))

But the second one is more efficient, because the logp is vectorized among a single batched Normal.

We could add a rewrite in the logprob_rewrites to convert the former to the second, so that users are not penalized from using the first form (which may be more intuitive for some).

Actually that sort of rewrite stack([rv1, rv2]) -> rv3 could be useful in many places in the logprob submodule

Guida contributor

Direzione di ricerca
Indaga il modulo logprob rewrites nella codebase di PyMC (probabilmente in pymc/logprob/rewrites.py o simile). L'obiettivo è aggiungere una riscrittura che rilevi quando i comp dists di una Mixture sono tutti istanze della stessa distribuzione (ad esempio, Normal) con parametri diversi, e li sostituisca con una singola distribuzione in batch. Esamina le riscritture esistenti per individuare i pattern. Implementa la trasformazione e aggiungi i test corrispondenti per garantire la correttezza. Verifica se eventuali PR o commenti collegati forniscono indicazioni; attualmente nessuno.
Tech stack
python
Dominio
backend
Tipo issue
Funzionalità
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-3 ore
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Nuova
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
Familiarity with PyMC's Mixture distributionUnderstanding of logp rewrites in PyMC
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
30