【ITニュース解説】Drawing Quantum Diagrams with Matplotlib on Android Termux

Androidスマートフォンやタブレットといったモバイルデバイス上で、量子計算の複雑な概念を視覚的に表現する試みは、近年注目されている。現代のモバイル端末は高性能な画面と、Termuxという強力なターミナル環境を備えており、これらを活用することで、まるでパソコンで作業しているかのように、高品質な量子状態の図を直接生成することが可能になる。

通常、量子計算のプログラミングには、IBMが開発したQiskitという優れたライブラリが使われる。Qiskitは量子コンピュータのシミュレーションから実際の量子デバイスの操作まで幅広い機能を提供する。しかし、このQiskitをAndroidのTermux環境に直接インストールしようとすると、大きな問題に直面する。特に問題となるのが、Qiskitの一部のコンポーネント、例えば高速なシミュレータであるqiskit-aerなどが、Rustというプログラミング言語で書かれており、そのコードをデバイス上で動かすために「Rustコンパイラ」という特別なソフトウェアが必要となる点だ。Rustコンパイラは、Rustで書かれたプログラムを、皆さんのAndroidデバイスが理解できる形に変換する役割を担う。ところが、モバイルデバイスのARM64アーキテクチャ上でRustコンパイラを動作させ、Qiskitの巨大なコードベースをコンパイルしようとすると、非常に多くの計算資源とメモリを消費する。このプロセス中に、デバイスのメモリが足りなくなり、処理が途中で停止してしまう「Out-Of-Memory (OOM) killer」と呼ばれる現象が頻繁に発生し、インストールの試みが失敗に終わることがほとんどである。Qiskitの全機能をモバイル環境で動かすのは、今のところ非常に困難な状況だと言える。

そこで、この記事では、Qiskitの全機能ではなく、本当に必要な部分だけを抜き出して利用するという、賢いアプローチを提案している。目的としているのは、量子ビットの状態を3次元の球体で視覚的に表現する「Bloch球（ブロッホきゅう）」と呼ばれる図を描くことだ。Bloch球は、量子ビットが「0」と「1」の重ね合わせ状態や、その間の無限の状態にあることを直感的に理解するのに非常に役立つ。このBloch球の描画機能は、Qiskitの中でも主にPythonだけで書かれた部分に存在しており、特に重いRustコンポーネントとは直接関係がない。

この軽量なアプローチを実現するために、以下の主要なツールと手順を用いる。まず、Termux環境に、Pythonとその科学技術計算ライブラリであるNumPy、そしてグラフ描画ライブラリであるMatplotlibをインストールする。NumPyは、行列の計算など、量子計算の基礎となる数学的操作を行うために不可欠だ。Matplotlibは、Bloch球のような複雑な図形をきれいに描画するために使われる。

次に、Qiskitの公式GitHubリポジトリから、特定のバージョンのソースコードをダウンロードする。ただし、ダウンロードするのはQiskitの全ファイルではなく、Bloch球の描画に必要なqiskit/visualization/bloch.pyという特定のファイルだけだ。このファイルをTermuxの指定された場所に配置する。これにより、巨大なQiskitを丸ごとインストールする手間と資源を省きつつ、必要な機能だけを利用できるようにする。

そして、Pythonスクリプトを作成する。このスクリプトは、ダウンロードしたbloch.pyファイルをPythonの内部で動的に読み込む工夫が施されている。具体的には、Qiskitのモジュール構造を模倣するような処理をPython内部で行い、あたかもQiskitがインストールされているかのようにbloch.pyを利用できるようにする。この仕組みにより、システムはQiskitの他の重い部分に依存することなく、Bloch球クラスをインスタンス化して利用できる。

スクリプトの次の重要な部分は、NumPyを使った量子状態の計算である。量子計算では、「量子ゲート」と呼ばれる操作によって量子ビットの状態が変化する。例えば、Pauli-Xゲートは量子ビットの0と1を反転させ、Hadamardゲートは重ね合わせ状態を作り出す。これらの量子ゲートは、数学的には「ユニタリー行列」と呼ばれる特別な行列で表現される。スクリプトでは、NumPyの行列計算機能を使って、初期状態にこれらのゲートを順番に適用し、その結果として得られる量子状態を計算する。Bloch球は3次元の座標で量子状態を表すため、計算で得られた量子状態をBloch球上のx、y、z座標に変換する関数も用意されている。

最後に、Matplotlibを使って、これらの計算結果をBloch球として描画する。スクリプトは、Pauli-Xゲート適用前後の状態、Hadamardゲート適用前後の状態、そしてPauli-Zゲート適用前後の状態を、それぞれBloch球として描画し、それらを一つの大きな図の中に並べて表示する。これにより、各量子ゲートが量子状態にどのような変化をもたらすのかを、視覚的に明確に理解できる。生成された図は、最初はPNG形式で保存される。

Matplotlibが出力するPNG形式の画像は、非常に高解像度で高品質だが、その分ファイルサイズが大きくなりがちだ。特にウェブサイトなどで利用する場合には、ファイルサイズを小さくしたいというニーズがある。そこで、Termuxにインストールしたcwebpというツールを使って、生成されたPNG画像を「WebP（ウェッピー）」という形式に変換する。WebP形式は、画質をほとんど損なうことなく、ファイルサイズを劇的に削減できる優れた画像形式だ。記事の例では、1.4MBあったPNG画像が、WebPに変換することでわずか128KBにまで削減され、約91%のファイルサイズ削減が実現している。これにより、高品質な図を高速にウェブで公開したり、他のアプリケーションで利用したりすることが容易になる。

この一連の作業は、Androidデバイス上で数秒以内に完了し、出版品質と呼べるような、非常に鮮明で美しい量子状態図を生成できる。これは、モバイル環境のリソース制約を巧みに回避し、必要な機能だけを効率的に利用するという、システムエンジニアリングの観点からも非常に優れたアプローチだと言える。

作業中に発生しやすい問題とその解決策も提示されている。例えば、「Matplotlibの3D描画モジュールが見つからない」というエラーは、Matplotlibのインストール方法に起因する。pip install matplotlibではなく、Termuxのパッケージマネージャであるpkgを使ってpkg install python-matplotlibとインストールすることで、必要なすべての依存関係が適切にインストールされ、この問題は解決する。また、Termuxはグラフィカルユーザーインターフェース（GUI）を持たないため、MatplotlibがGUIウィンドウを開こうとして失敗することがある。これに対しては、plt.switch_backend('Agg')というコードをPythonスクリプトに追加することで、描画結果を画面に表示せず、直接ファイルに保存する「非対話型」の描画モードに切り替えることができ、問題なく図を生成できるようになる。図のラベルが画像の外にはみ出てしまうような場合には、plt.savefig関数の引数にbbox_inches="tight"を追加することで、余白を自動調整し、すべての要素が収まるようにする。さらに、数式をきれいに表示するためにLaTeXという専門的な組版システムが必要だと思われがちだが、Matplotlibにはmathtextという内蔵のパーサーがあるため、特に別途LaTeXをインストールする必要はなく、数式をそのまま記述できる。

この手法は、Qiskitの本格的なシミュレーション機能（例えばqiskit-aerを使った大規模な量子回路シミュレーション）をTermux上で実行するものではない。そのような高度なシミュレーションには、PRoot DebianやUbuntuのような、より完全なLinux環境をTermux内に構築するアプローチが推奨される。これは、Androidのlibc（標準Cライブラリ）と、一般的なLinuxディストリビューションのglibcとの互換性の違いに起因する。しかし、Bloch球の描画という特定の目的においては、この記事で紹介されている軽量なアプローチが、はるかにシンプルで効率的である。

モバイルデバイスの可能性を最大限に引き出し、量子計算という最先端の分野に触れるための、実践的な一歩となるだろう。