⛳

foldr の融合則が一般の融合則から導ける件

2023/12/14に公開

Haskell

圏論

tech

Haskell Advent Calendar 2023 の12日目の記事です。

『関数プログラミング入門』（オーム社）などを読んでいると foldr や foldl の融合則というのが出てきます。たとえば foldr の融合則は次の形です。

foldr の融合則

-- 有限リスト xs と関数 f, g, h について、
-- h (f x y) = g x (h y) のとき以下が成り立つ：

h (foldr f e xs) = foldr g (h e) xs

この法則を使うとたとえばリストを2回分走査する計算を1回分にまとめることができたりして便利なわけです。

-- 例
foldr f a . map g = foldr (f . g) a

が、この法則の見た目がややこしいですし、条件部分 h (f x y) = g x (h y) も意味がよく分からないと思っていました。

これがもっと一般的な、より短い形から導けるということを知ってびっくりしたので、学んだことを復習する意味で共有したいと思います。

一般の融合則

それがこの形です

圏論の言葉にはなりましたが、式自体はすごく簡潔になりました。むりやり可換図式にするとこんな感じでしょうか（本当は三角形にしたかった）。

\begin{CD} 0 @>!_{A}>> A \\ @VidVV @VVfV \\ 0 @>!_{B}>> B \end{CD}

この融合則で圏 C を具体的に用意してあげると foldr の融合則になるわけです。私は一般の融合則を後から知ったので、foldr の融合則がこんな当たり前っぽい式から導けるんだということにむちゃくちゃ驚きました。

F-代数

そのための準備としてまず F-代数というものを用意します。F-代数とは、圏 $C$ と関手 $F : C \to C$ について $X \gets F\,X$ の形をした射のことです。

また、F-代数の間の F-準同型というものを考えます。F-代数 $X \gets F\;X$ から $Y \gets F\;Y$ への F-準同型とは、射 $h : X \to Y$ であって以下を可換にするものです。

\begin{CD} X @<<< F\;X \\ @VhVV @VVF\,hV \\ Y @<<< F\;Y \end{CD}

このとき、F-代数を対象とし、F-準同型を射とする圏を考えることができます。この圏に始対象があるとき、それを F-始代数といいます。

リスト型 [A] を F-始代数として定義する

いま、対象 $A$ を固定して、対象 $X$ と射 $f$ に対して関手 $F$ を次のように定義します。

$F\;X = 1 + A \times X$
$F\;f = id_{1} + id_{A} \times f$

ただし $1$ は終対象、 $\times$ と $+$ は積と余積です。このとき、この F についての代数に始代数が存在することが示されます。都合上それを $[A] \gets 1 + A \times [A]$ と書くことにすると、始代数から他の代数への可換図式は次のように書けます。

\begin{CD} [A] @<[nil,\,cons]<< 1 + A \times [A] \\ @V(\!|e,f|\!)VV @VVid_{1} + id_{A} \times (\!|e,f|\!) V \\ X @<[e,\,f]<< 1 + A \times X \end{CD}

ここで $[A]$ から $X$ への射、つまり始対象からの射は $[e, f]$ に応じて1つに決まりますので、それを $(\![ e, \, f ]\!)$ という記法で表しています。

ここまで始代数をまるでリスト型やその上の関数かのように [A] や nil や cons と書いていますが、そう書いてよいことは後で分かります。いま、この図の表す等式を書き下してみます。

$(\!| e,\,f |\!) \circ [nil,\,cons] = [e,\,f] \circ (id_{1} + id_{A} \times (\!|e,f|\!))$

これは次の2つの式に分解されます。

$(\!| e,\,f |\!) \circ nil = e \circ id_{1}$
$(\!| e,\,f |\!) \circ cons = f \circ (id_{A} \times (\!|e,f|\!) )$

ここで、もとになっている圏 C をたとえば集合と写像の圏とします。すると nil, cons, e, f はそれぞれ次の形の写像になります。

$nil : 1 \to [A]$
$cons : A \times [A] \to [A]$
$e : 1 \to X$
$f : A \times X \to X$

ここで上の1番めの式より
　　 $(\!| e,\,f |\!) \circ nil = e$
一点集合 $1$ の元を $()$ として両辺に $()$ を適用すると、
　　 $((\!| e,\,f |\!) \circ nil) \; () = e \; ()$
左辺を計算して $nil \; ()$ を $\overline{nil}$ 、 $e \; ()$ を $\overline{e}$ と書くと
　　 $(\!| e,\,f |\!) \; \overline{nil} = \overline{e}$
となります。

また2番目の式の両辺に (x, xs) を適用して、
　　 $((\!| e,\,f |\!) \circ cons) \; (x, xs) = (f \circ (id_{A} \times (\!|e,f|\!) )) \; (x, xs)$
左辺と右辺をそれぞれ計算して
　　左辺
　　 $= ((\!| e,\,f |\!) \circ cons) \; (x, xs)$
　　 $= (\!| e,\,f |\!) \; (cons \; (x, xs))$
　　右辺
　　 $= f \; ((id_{A} \times (\!|e,f|\!)) \; (x, xs)$
　　 $= f \; (x, (\!|e,f|\!) \; xs)$
左辺 = 右辺の形に戻して
　　 $(\!| e,\,f |\!) \; (cons \; (x, xs)) = f \; (x, (\!|e,f|\!) \; xs)$

まとめると

となりますが、これを foldr の定義

foldr f e [] = e
foldr f e (x:xs) = f x (foldr f e xs)

と見比べると、 $(\!|e,f|\!)$ は foldr f e と同じ意味だと分かります。同様に、 $\overline{nil}$ は [] と、 $cons$ は : と、 $[A]$ は [a] と同じ意味ということになります。

foldr の融合則を導く

さきほどのF-代数の圏に融合則を適用します。次の可換図式で、

\begin{CD} [A] @<[nil,\,cons]<< 1 + A \times [A] \\ @V(\!|e,f|\!)VV @VVid_{1} + id_{A} \times (\!|e,f|\!) V \\ X @<[e,\,f]<< 1 + A \times X \\ @V h VV @VVid_{1} + id_{A} \times h V \\ Y @<[d,\,g]<< 1 + A \times Y \\ \end{CD}

一番上の $[A] \gets 1 + A \times [A]$ は始対象ですから、融合則は $h \; \circ \; !_{[e, f]} \; = \; !_{[d, g]}$ の形になります。 $!_{[e, f]}$ はつまり $(\!|e,f|\!)$ のことなので、

$h \circ (\!|e,f|\!) = (\!|d,g|\!)$

となります。また、下側の四角形の可換性を書き下すと次のようになります。

$h \circ [e, f] = [d, g] \circ (id_{1} + id_{A} \times h)$

これは次の2つの式に分解されます。

$h \circ e = d \circ id_{1}$
$h \circ f = g \circ (id_{A} \times h)$

1番の式から $d = h \circ e$ であることが分かるので、融合則は結局

$h \circ (\!|e,f|\!) = (\!|h \circ e,g|\!)$

となります。前と同様に集合と写像の圏だと思って両辺に xs を適用すると
左辺
　　 $= (h \circ (\!|e,f|\!)) \; xs$
　　 $= h ((\!|e,f|\!) \; xs)$
右辺
　　 $= (\!|h \circ e,g|\!) \; xs$
左辺 = 右辺の形に戻して
　　 $h \; ((\!|e,f|\!) \; xs) = (\!|h \circ e,g|\!) \; xs$

2番の式でも同様に両辺に (x, y) を適用すると
左辺
　　 $= (h \circ f) \; (x, y)$
　　 $= h \; (f \; (x, y))$
右辺
　　 $= (g \circ (id_{A} \times h)) \; (x, y)$
　　 $= g \; ((id_{A} \times h) \; (x, y))$
　　 $= g \; (x, h \; y)$
左辺 = 右辺の形に戻して
　　 $h \; (f \; (x, y)) = g \; (x, h \; y)$

まとめると

となりますが、これは $(\!|e,f|\!)$ が foldr f e だったことを思い出すと、冒頭の foldr の融合則

h (foldr f e xs) = foldr g (h e) xs
-- ただし以下を満たすとき
h (f x y) = g x (h y)

と同じ意味であることが分かります。

まとめ

翻って考えると、foldr の融合則の条件 h (f x y) = g x (h y) は h が準同型という意味だったんですね。実のところまだよく分かってないのですが、前よりはイメージがつかめた気がします。

参考文献

圏論勉強会：第7回 @ワークスアプリケーションズ（中村晃一）

「代数的データ型」以降のページで、上の内容のほとんどがより丁寧に書かれています。本記事でやったのはリストの catamophism の融合則が foldr の融合則と一致することを具体的に確かめたことだけです。

『関数プログラミング入門』（オーム社）
『Algorithm Design With Haskell』（Cambridge University Press）

foldr の融合則についてはこの本で知りました。