quartz-research-note/content/The w-calculus a synchronous framework for the verified modelling of digital signal processing algorithms.md
松浦 知也 Matsuura Tomoya 72368b208c
All checks were successful
Build / build (push) Successful in 1m54s
modified frontmatter depending on timestamp
2024-02-08 18:01:44 +09:00

5.7 KiB
Raw Blame History

date
2023-12-06T20:20:54+09:00

#paper #programming-language

https://dl.acm.org/doi/10.1145/3471872.3472970

first author: Emilio Jesús Gallego Arias

概要

ラムダ計算にfeedという1サンプルフィードバックを導入する形式を取る

mimiumselfを使って1サンプルフィードバックができる、というのと近いちょうどmimiumと同じカンファレンスで発表された

ただし対象は線形時不変システムの形式化で、異なるトポロジーのフィルターが最終的に同一のものであるのが証明できたりすると嬉しいよね、みたいなモチベだと理解してる(なので、項同士の加算は定義されているが、乗算は項と定数のみに制限されている)

Coqで証明が実装されている https://github.com/ejgallego/mini-wagner-coq/

よく見たらこのAuthorのEmilio Jesús Gallego Arias、Coq本体の開発をゴリゴリやってる人だ強いわけだ

実装

最終的にはA:意味論としてわかりやすいDenotational、B:有限なメモリと計算結果の再利用を考慮したImperativeなインタプリタがOCamlで、C:MetaOCamlを使って最適化したコードの3つが出てくる。

A:Denotationalなインタプリタ

  • machineがある時刻のサンプルを計算し返却する関数
  • machine_nが指定した時刻までのすべてのサンプルを計算しリストとして返却する関数

以上2つの関数を相互再帰させている 毎サンプルごとに全てのサンプルのhistoryを再生成してたりするので著しく非効率的

let rec machine_n : type a. int  a expr  env  a list = 
	fun step e env  
		if step < 0 
			then [] 
			else machine step e env :: machine_n (step-1) e env 
and machine : type a . int  a expr  env  a = 
	fun step e env  match e with 
		| Var (idx, id)  lookup env id idx 
		| Lam e  fun hist  machine step e (hist :: env) 
		| App (ef, ea)  let a_hist = machine_n step ea env in 
			machine step ef env a_hist 
		| Feed e  let e_hist = machine_n (step-1) (Feed e) env in 
			machine step e (e_hist::env) 
		|...

ただし、リポジトリのCoqのコードでは別に相互再帰で実装されてるわけではない

Definition exprI n (pI : I n) : I n.+1 :=
  fix eI k hk {Γ t} e {struct e} :=
  match e in expr Γ t return env k Γ -> tyI k t with
  | var v Γ idx      => fun Θ => nth (I0 k _) (hnth (hist0 k _) v Θ) idx
  | add _ _  e1 e2   => fun Θ => eI k hk e1 Θ + eI k hk e2 Θ
  | mul _ _  c  e    => fun Θ => c *: eI k hk e Θ
  | pair _ _ _ e1 e2 => fun Θ => col_mx (eI k hk e1 Θ) (eI k hk e2 Θ)
  | proj m _ i e     => fun Θ => \col__ (eI k hk e Θ) i ord0
  | lam m _ _ e      => fun Θ v => eI k hk e (v, Θ)
  | app _ _ _ ef ea  => fun Θ =>
                        let efI : tyI k (tfun _ _) := eI k hk ef Θ in
                        let eaI : hist k _ :=
                            accI (fun k hk => eI k hk ea) hk Θ in
                        efI eaI
  | feed tn Γ e      => fun Θ =>
                        let prev :=
                            accF (fun k hk => pI k hk _ _ (feed e)) hk Θ in
                        eI k hk e (prev, Θ)
  end.

(* Need to prove |acc_feed n.+1| =  acc_norm n ?? *)

(* We build our step-indexed representation *)
Fixpoint exprIn n {struct n} : I n.+1 :=
  match n return I n.+1 with
  | 0    => fun k hk Γ t e (Θ : env k Γ) =>
            exprI (@IE0 _) hk e Θ
  | n.+1 => fun k hk Γ t e (Θ : env k Γ) =>
            exprI (@exprIn n) hk e Θ
  end.

そしてCoqで証明されているのはここまでMetaOCamlのReal-World exampleが欲しいよ

B:Imperativeなインタプリタ

この実装が多分現在のmimiumの実装に近い

let heap = ... (* pre-allocated *) 
let rec imachine : expr  env  value = 
	fun e env  match e with 
		| Var (idx, id)  
			let p = ptr_of id env in 
				lookup heap p idx 
		| Lam e  
			fun ptr  imachine e (ptr :: env) 
		| App (p,ef,ea)  
			let va = imachine ea env in 
				shift_one heap ptr va; 
				imachine ef env ptr 
		| Feed (ptr,e)  
			let v = imachine e (ptr::env) in 
				shift_one heap ptr v; 
				v 
		|... 
		
let rec repeat : int  (unit  'a)  'a = 
	fun n c  
		if n = 0
			then c () 
			else (c (); repeat (n-1) c) 

let eval e env n = repeat n (fun ()  imachine e env)

これAppの引数はpじゃなくてptrだよな

		| App (p,ef,ea) → 
			let va = imachine ea env in 
				shift_one heap ptr va; 
				imachine ef env ptr 

これ、Rustで改めて簡単に実装してみて思ったけど、heapの中のデータ構造がどうなってるか初めにどうやってどのサイズのヒープを確保すべきかが一番重要なのにそこに触れられていないのでは、、、

C:MetaOCamlでの多段階インタプリタ

let rec eval : type a. env  a expr  a code = 
	fun g e  match e with 
		| Cst r  let module M = Lift_array(Lift_float) in M.lift r 
		| Var id  
			.< Array.(unsafe_get .~(heap) .~(List.nth g id)) >. 
		| Idx (vec, id)  
			.<  let v = .~(eval g vec) in 
				let i = .~(eval g id) in 
					Array.(unsafe_get v i) >. 
		| Add (e1,e2)  
			.<  let v1 = .~(eval g e1) in 
				let v2 = .~(eval g e2) in 
					WArray.map2 (+.) v1 v2 >. 
		| Lam e  
			.< fun p  .~(eval (.<p>.::g) e)>. 
		| App (p, f, e)  
			.< let v = .~(eval (.<p>.::g) e) in
				 shift_one v Array.(unsafe_get .~(heap) p); .~(eval g f) p >. 
		|...

A:DenotationalとB:Imperativeが等しい計算を行なっていることの形式的証明はOpen Questionとなっている。