Galton–Watson extinction criterion: is the least root of in , and
OpenGaltonWatson.extinction_criterionThis is the extinction criterion for the Galton–Watson branching process.
Let be a probability distribution on (the offspring distribution) with and with finite mean
and let
be its probability generating function.
On a probability space let be independent random variables, each with law for all ; is the number of children of the -th individual of generation . The Galton–Watson process is
where an empty sum is (so once the population dies out it stays extinct). Let
be the extinction probability. Then:
- is the smallest root of the equation in ;
- if and only if .
So the population survives forever with positive probability exactly in the supercritical case . The hypothesis only excludes the degenerate process in which every individual has exactly one child (then for all , so although ).
This is the basic dichotomy of branching-process theory (subcritical and critical processes die out, supercritical ones survive with positive probability), and the fixed-point characterization of is the standard tool for computing extinction probabilities in population genetics, epidemic models, branching random walks and the exploration of random graphs.
Formalization Note The offspring law is a PMF ℕ. The mean is taken in ; finiteness is a hypothesis and the comparison is made there. The generating function is the real series (Lean's convention gives ), and "smallest root in " is IsLeast of . The array is mutually independent as a family indexed by pairs , each is measurable with for every , and is any function satisfying the recursion above (which determines it). The extinction probability is the real-valued measure of the event .
import Mathlib open MeasureTheory ProbabilityTheory
namespace GaltonWatson
/-- Galton–Watson extinction criterion (Athreya–Ney, Ch. I §5 Thm 1; Grimmett–Stirzaker Thm 5.4.5;
Durrett PTE §4.3.4). The offspring law is `p`, the offspring numbers `ξ n i` (child count of the
`i`-th individual of generation `n`) are iid with law `p`, `Z 0 = 1` and
`Z (n+1) = ∑_{i < Z n} ξ n i`. Then the extinction probability `q = P(∃ n, Z n = 0)` is the
smallest root of `f s = s` in `[0,1]`, where `f s = ∑ p_k s^k`, and `q = 1 ↔ m ≤ 1`. -/
theorem extinction_criterion
{Ω : Type*} [MeasurableSpace Ω] (P : Measure Ω) [IsProbabilityMeasure P]
(p : PMF ℕ) (hp1 : p 1 < 1) (hm : ∑' k : ℕ, (k : ENNReal) * p k ≠ ⊤)
(ξ : ℕ → ℕ → Ω → ℕ) (hξ_meas : ∀ n i, Measurable (ξ n i))
(hξ_indep : iIndepFun (fun ni : ℕ × ℕ => ξ ni.1 ni.2) P)
(hξ_law : ∀ n i k, P (ξ n i ⁻¹' {k}) = p k)
(Z : ℕ → Ω → ℕ) (hZ0 : ∀ ω, Z 0 ω = 1)
(hZ : ∀ n ω, Z (n + 1) ω = ∑ i ∈ Finset.range (Z n ω), ξ n i ω) :
IsLeast {s : ℝ | s ∈ Set.Icc 0 1 ∧ ∑' k : ℕ, (p k).toReal * s ^ k = s}
(P.real {ω | ∃ n, Z n ω = 0}) ∧
(P.real {ω | ∃ n, Z n ω = 0} = 1 ↔ ∑' k : ℕ, (k : ENNReal) * p k ≤ 1) := by sorry
end GaltonWatson