MANGO: An Autodiff Neutrino Oscillation Engine for Differentiable Analysis Pipelines

見えない幽霊と、それらを動かす数学

宇宙は「ニュートリノ」と呼ばれる、小さく幽霊のような粒子で満たされていると想像してみてください。彼らは非常に内気で、1光年の厚さがある固体の鉛を、何にもぶつかることなく通り抜けることができます。今この瞬間も、何兆ものニュートリノがあなたの体を通り抜けていますが、全く気づかれることはありません。これらの粒子には3つの「フレーバー」（電子、ミュー、タウという、異なる色のようなものと考えてください）があり、広大な宇宙空間や地球を移動する際、魔法のようにその色を変えてしまう性質を持っています。この現象は「ニュートリノ振動」と呼ばれます。

科学者たちは、宇宙の根本的なルールを理解するために、数十年にわたりこれらの変身能力を持つ粒子を研究してきました。そのために、彼らは地下深くへと巨大な検出器を建設し、ニュートリノが相互作用するのを待っています。しかし、ニュートリノがどのような挙動を示したのかを解明するには、複雑なコンピュータ・シミュレーションを実行する必要があります。これらのシミュレーションは、タイムマシンのように機能し、ニュートリノが地球の核、マントル、そして大気を通過する際にどのように振る舞うべきかを予測します。

長い間、これらのシミュレーションは「一方通行の道」のようなものでした。初期条件（ニュートリノがどこから来たのか、どのくらいの速さで動いているのか）を入力すれば、答え（到着時にどのフレーバーになっているか）を得ることができました。しかし、もし単一の数値（例えば、地球の核の密度や、ニュートリノが入射する角度など）を少し変えたときに、なぜ答えが変わるのかを知りたいと思っても、シミュレーション全体を最初からやり直す必要がありました。数値をわずかに変えて、再度実行するのです。それは、ケーキに砂糖をひとつまみ増やしたらどうなるかを知りたいときに、そのたびに新しいケーキを焼き直して試しているようなものでした。確かに機能はしていましたが、遅くて不器用であり、実験自体の最適化をほぼ不可能にしていました。

新しいエンジン：MANGO

ここで「MANGO」（MANGO: A Neutrino Gradient Oscillator）が登場します。MANGOを単なる計算機ではなく、逆方向に計算を行うことができる、超スマートで自己認識を持つエンジンだと考えてください。コンピュータサイエンスの世界では、これは「自動微分」と呼ばれます。従来のツールは計算の結果しか示すことができませんが、MANটিは、その結果が「あらゆる材料」に対してどれほど敏感であるかを、一度の工程ですべて正確に伝えることができます。

この論文は、現代的なプログラミングフレームワークであるJAX上に構築された、新しいオープンソースのツールとしてMANGOを紹介しています。その最大の強みは、ニュートリノ物理学のプロセス全体を、単一の連続した数学の連鎖として扱うことです。通常、科学者はニュートリノがフレーバーを変える確率を計算して終了しますが、MANGOはその「数学の連鎖」を開いたままにします。これにより、コンピュータは最終的な結果の感度を、検出器の応答、データのグループ化の方法、さらには地球の物理的な形状に至るまで、最後まで遡って追跡することができるのです。

MANGOが実際に行っていること

研究者は、ニュートリノの確率の「微分（傾き）」を計算するという、特定の悩みを解決するためにMANGOを構築しました。かつて、これらの傾きを計算することは不可能であったか、あるいは、新しいシナリオごとに複雑でエラーの起きやすい数式を手動で書き出す必要がありました。MANGOは、あらゆる計算ステップを「微分可能」にすることで、このゲームのルールを変えました。

本論文が実証している内容は以下の通りです：

• 目に見えないものを見る： MANGOは、従来の数式では無視されていた要素、例えば検出器の正確な深さ、大気中でニュートリノが生成された高度、あるいは地球内部の特定の層の密度などを微調整した場合に、ニュートリノの確率がどのように変化するかを計算できます。
• 複雑な事象を扱う： 地球は滑らかな層を持つ完璧な球体ではなく、地殻、マントル、核といった異なる密度を持っています。MANGOは、この「層状」の地球（標準モデルであるPREMを使用）をモデル化でき、さらに、ニュートリノが新しい方法で物質と相互作用したり、量子的な「コヒーレンス（干渉性）」を失ったりするような、特殊な非標準物理学の理論さえも扱うことができます。
• 驚異的な速さ： 論文では、層状の地球モデルにおける369もの異なるパラメータ（密度、電子分率、およびシェルの半径）に対する感度計算が、シミュレーションを一度実行する場合のわずか2.5倍から3倍の時間で完了することを示しています。従来の方法では、これを行うために数百回の個別実行が必要でした。
• 極めて高い精度： 著者は、MANGOの結果が既存の確立されたツール（OscProbやNuFastなど）と、極めて小さな差（具体的には$10^{-9}$から$10^{-5}$程度の差）で一致することを検証しました。また、数学が非常に複雑になる領域である、ニュートリノが地球の核を通過する場合でも、数学が完璧に機能することを証明しました。

実世界でのテスト：地球断層撮影法（トモグラフィー）

その力を示すために、著者はMANGOを用いて地球の「トモグラフィー（断層撮影）」分析を行いました。これは、ニュートリノが地球の核を通過する様子を観察することで、地球の核の密度を突き止めようとする試みです。

MANGOを用いることで、彼らは単に密度を推測しただけでなく、その測定における不確実性がどの程度になるかを正確に計算しました。しかし、ここからが肝心な点です。彼らはその不確実性を「通じて」微分を行い、検出器の角度分解能（検出器が方向をどれだけ正確に識別できるか）を向上させた場合に、測定値がどのように変化するかを調べました。

わずか一度のパスで、MANGOは、検出器の角度測定能力を0.01向上させるだけで、地球の核の密度測定の精度が7.8%向上することを算出しました。これは、物理学、統計学、そして検出器のハードウェアという全プロセスを貫いて微分する必要があるため、以前は到達不可能だった「実験設計」の計算です。

できること、できないこと

MANGOが「何ではないか」についても注意が必要です。MANGOは、高エネルギーのニュートリノが地球によって吸収または破壊されるプロセス（減衰と呼ばれます）をシミュレートするためのツールではありません。そのためには、依然として他のツールが必要です。また、あらゆる物理問題を即座に解決する魔法の杖でもありません。それはあくまで「基盤」です。著者は、MANGOがコアとなる振動物理学を完璧に扱う一方で、フル密度の行列輸送や、他の大規模なソフトウェア・スイートが行うような完全な実験シミュレーションへと拡張することは、将来の課題であると明示しています。

結論

本論文は、MANGOを基礎的な転換点として提示しています。ニュートリノ物理学を、「結果を計算してから、手動で改善策を推測する世界」から、「コンピュータが、最高の答えを得るために実験、検出器、あるいはモデルをどのように微調整すべきかを教えてくれる世界」へと移行させます。これにより、ニュートリノ・シミュレーションという「ブラックボックス」が、透明で調整可能な機械へと変わり、科学者が実験を最適化し、これまでにない精度とスピードで地球内部を理解することを可能にします。著者はコードをオープンソースとして公開しており、科学コミュニティがこの新しい、微分可能な基盤の上に新たな研究を築いていくことを期待しています。