2022-10-11

[]強化学習による、より高速な行列乗算アルゴリズム発見

AIアルゴリズムを考える時代がきたようだ

Deepmind は Alphabet(Google)の子会社AlphaGoを作ったところ

深層強化学習アルゴリズムDeep Q-Network(DQN)を作っておりDQNを使うといろんなゲーム人間よりいいスコアを出すことができる

今回は行列乗算のアルゴリズムに深層強化学習を使ってよりよいアルゴリズム作成を行った

行列乗算は、そうした基本的タスクの1つであり、ニューラルネットワークから科学計算ルーチンまで、多くの系で見られる。機械学習を用いたアルゴリズム自動発見によって、人間直感を超え、人間設計した現在最良のアルゴリズム凌駕する見通しが得られる。しかし、アルゴリズム発見手順の自動化は、アルゴリズム存在し得る空間が巨大であるため難解である

~~~

AlphaTensorは、多くの行列サイズについて、最先端の複雑さを凌駕するアルゴリズム発見した。中でも特に重要なのは、有限体における4 × 4行列場合で、AlphaTensorのアルゴリズムはこの行列において、50年前に発見されたStrassenの2レベルアルゴリズムに、我々の知る限りでは発見以来初めて改良を加えた。

Nature 610, 47–53 (2022)

記事への反応(ブックマークコメント)

ログイン ユーザー登録
ようこそ ゲスト さん