drosophila-brain-mlx
Kisame76 (@Kisame76)
公開されている Shiu らの leaky integrate-and-fire 全脳モデルを、MLX と自作 Metal カーネルで書き直したもの。掃引や対話的な実験を現実的にするのが狙いで、M4 Pro での実測として FlyWire v630(127,400 ニューロン・14,687,178 エッジ)を生物時間 1 秒あたり 0.29 秒、同じ機体の Brian2 参照実装は 62.6 秒と報告し、ベンチ JSON をリポジトリに同梱している。Shiu の run_exp の引数と parquet がそのまま通り、MaleCNS v1.0 も同じエンジンで動く。4 つの計算レーンはスパイク数の SHA-256 で一致を固定し、次数を保ったまま接続先を無作為化した対照コネクトームでは MN9 が一切発火しないという結果も示す。ただし定数は FlyWire に合わせられているため、2 つのデータセットのスパイク数は比較できないと作者は断っている。
確認した出典
上の情報は掲載日にこれらのページを読んで確認したものです。
- api.github.com https://api.github.com/repos/Kisame76/drosophila-brain-mlx
- raw.githubusercontent.com https://raw.githubusercontent.com/Kisame76/drosophila-brain-mlx/HEAD/README.md
- raw.githubusercontent.com https://raw.githubusercontent.com/Kisame76/drosophila-brain-mlx/HEAD/ATTRIBUTION.md
- raw.githubusercontent.com https://raw.githubusercontent.com/Kisame76/drosophila-brain-mlx/HEAD/bench/results.json
- raw.githubusercontent.com https://raw.githubusercontent.com/Kisame76/drosophila-brain-mlx/HEAD/bench/results_malecns.json