本論文では、アナログ電子回路の基本原理であるキルヒホフの電流則に基づいた新しいクラスのニューラルネットワークモデル「KirchhoffNet」を提案している。
KirchhoffNetは、ノードと枝からなる有向グラフで表現される。各ノードには電圧が割り当てられ、ノード間の枝には非線形な電流-電圧特性を持つ学習可能なデバイスが接続される。キルヒホフの電流則に従って、各ノードの電圧の時間変化が記述される。
この構造は、連続時間モデルであるニューラルODEと情報伝播ネットワークであるメッセージパッシングニューラルネットワークの特性を併せ持つ。また、物理的な回路として実装可能であり、パラメータ数に関わらず1/fの時間で順伝播計算を完了できるという特徴がある。ここでfはハードウェアのクロック周波数を表す。
実験では、MNISTデータセットの分類タスクにおいて、従来の畳み込みニューラルネットワークと同等の98.86%の高精度を達成した。これは、KirchhoffNetが伝統的な層構造を持たない設計でありながら、高い表現力を持つことを示している。
今後の課題として、他のデータセットやタスクでの性能評価、物理回路実装に向けた検討などが挙げられる。
A otro idioma
del contenido fuente
arxiv.org
Ideas clave extraídas de
by Zhengqi Gao,... a las arxiv.org 05-07-2024
https://arxiv.org/pdf/2310.15872.pdfConsultas más profundas