行列積とアダマール積の使い分け ミキサーとふるいで教える
こんにちは。ゆうせいです。
ディープラーニングの研修で、必ず質問が出る場面があります。
「なぜここは掛け算の記号が違うのですか」
コードを見ると、@ と * が混在しています。どちらも掛け算に見えますが、まったく別の演算です。そして、この違いを理解しているかどうかで、モデルの構造の読み方が変わります。
この記事では、講師がこの区別をどう教えるかを、例え、数式、演習まで含めて整理します。
結論
二つの演算は、次のように役割が分かれています。
| 演算 | 記号 | 役割 | 例え |
|---|---|---|---|
| 行列の積 | @ | 情報をすべて混ぜ合わせて新しい特徴を作る | ミキサー |
| アダマール積 | * | 情報を残すか消すか通すかを決める | ふるい |
この例えは、単なる語呂合わせではありません。計算の構造そのものから導かれます。
ミキサーは、入れたものが混ざり合って別のものになります。行列の積では、和を取る操作が入るため、複数の入力が一つの出力に混ざります。
ふるいは、通すか通さないかを決めるだけで、粒そのものは変えません。アダマール積では、位置ごとに独立して掛け算するため、情報が混ざりません。
研修では、この構造の違いを先に示してから、AIのどこで使われるかに進んでください。順序を逆にすると、暗記になります。
定義と記号
行列の積
が
、
が
のとき、積
の各要素は次のように計算されます。
総和の記号があることが要点です。 について足し合わせるため、複数の値が一つにまとまります。
アダマール積
と
が同じ形のとき、アダマール積
の各要素は次のように計算されます。
総和がありません。同じ位置どうしを掛けるだけです。
記号は で表します。要素ごとの積、要素積、element-wise productとも呼ばれます。
記号の対応
| 数式 | Python | 別名 |
|---|---|---|
| @ または matmul | 行列積、内積型の積 | |
| * または mul | アダマール積、要素積 |
なお、数学の教科書では が行列積を意味しますが、プログラムでは * が要素ごとの積になります。この不一致が、混乱の原因になります。研修では最初に指摘しておいてください。
数値で違いを確認する
同じ二つの行列で、両方を計算してみます。
行列の積
各要素を、定義に従って計算します。
1行1列目です。
1行2列目です。
2行1列目です。
2行2列目です。
まとめると次のようになります。
アダマール積
同じ位置どうしを掛けるだけです。
何が違ったか
結果を並べます。
| 位置 | 行列の積 | アダマール積 |
|---|---|---|
| 1行1列 | 19 | 5 |
| 1行2列 | 22 | 12 |
| 2行1列 | 43 | 21 |
| 2行2列 | 50 | 32 |
行列の積の19という値には、 の1行目の全要素と、
の1列目の全要素が関与しています。四つの数字が混ざって19になりました。
アダマール積の5という値には、 と
しか関与していません。他の要素は一切影響しません。
ここが、ミキサーとふるいの違いです。
形の規則という決定的な差
計算そのものより、形の規則を先に押さえたほうが理解が速い場合があります。
| 演算 | 入力の形 | 出力の形 |
|---|---|---|
| 行列の積 | ||
| アダマール積 |
行列の積は、内側の次元が一致していれば計算でき、出力の形が変わります。次元を変換する操作です。
アダマール積は、形が同じでなければ計算できず、出力の形も変わりません。次元を変えない操作です。
この違いだけで、役割が説明できます。
特徴を作り出すには、次元の変換が必要です。だから行列の積を使います。
情報を選別するには、形を保ったまま各位置の値を調整すればよい。だからアダマール積を使います。
AIのどこで使われるか
行列の積が使われる場所
| 場面 | 数式 | 何をしているか |
|---|---|---|
| 全結合層 | 入力の全要素を混ぜて新しい特徴を作る | |
| Q、K、Vの生成 | 埋め込みを三つの役割に変換する | |
| 注意スコアの計算 | 単語どうしの関連度を測る | |
| 情報の集約 | 重みに応じて情報を混ぜ合わせる | |
| 出力層 | 隠れ状態を語彙数の得点に変換する |
いずれも、次元が変わっています。そして、複数の入力が一つの出力に寄与しています。
アダマール積が使われる場所
| 場面 | 数式 | 何をしているか |
|---|---|---|
| LSTMのゲート | 記憶を残すか消すかを決める | |
| GRUの更新 | 前の状態と新しい状態を混ぜる比率を決める | |
| ドロップアウト | ランダムに一部を0にする | |
| レイヤー正規化 | 正規化後の尺度を次元ごとに調整する | |
| SwiGLUなどのゲート機構 | 片方が他方の通過量を制御する | |
| 誤差逆伝播 | 活性化関数の微分を各位置に掛ける |
いずれも、形が変わっていません。そして、各位置が独立して処理されています。
LSTMで両方が並ぶ例
最も教えやすいのが、LSTMの忘却ゲートです。
まず、ゲートの値を作ります。
ここは行列の積です。前の状態と今の入力を混ぜ合わせ、シグモイド関数で0から1の値にします。
次に、そのゲートを適用します。
ここはアダマール積です。作ったゲートの値を、記憶の各次元に掛けます。
流れを整理すると、次のようになります。
| 段階 | 演算 | 目的 |
|---|---|---|
| ゲートを作る | 行列の積 | どこを通すべきかを判断する |
| ゲートを適用する | アダマール積 | 判断に従って通す量を決める |
判断はミキサーで作り、実行はふるいで行う。この対比が、両者の役割を最もよく表しています。
ゲートの値の意味
シグモイド関数の出力は、0から1の範囲に収まります。
具体的に確認します。
この値をアダマール積で掛けると、次のような意味になります。
| ゲートの値 | 効果 |
|---|---|
| 1に近い | ほぼそのまま通す |
| 0.5前後 | 半分程度に弱める |
| 0に近い | ほぼ消す |
ふるいの目の粗さを、次元ごとに0から1で調整している、と説明できます。
数値例で確認します。記憶の状態とゲートが次のようになっているとします。
アダマール積を計算します。
1番目と4番目はほぼ保持され、2番目はほぼ消え、3番目は半減しました。次元ごとに独立して扱われていることが確認できます。
誤差逆伝播で両方が現れる
もう一つ、両者が同時に出てくる重要な場面があります。
全結合層の誤差逆伝播では、次の式が現れます。
括弧の中は行列の積です。次の層から来た誤差を、重みの転置を使って前の層の次元へ戻しています。
括弧の外はアダマール積です。活性化関数の微分を、各位置に掛けています。
なぜここでアダマール積なのか。活性化関数は、各要素に独立して適用されるためです。
番目の出力は
番目の入力にしか依存しません。したがって微分も位置ごとに独立し、掛け算も位置ごとになります。
この点は、連鎖律から導けます。研修で扱う際は、なぜ和が現れないのかを問いかけると、理解が定着します。
計算量の違い
実務上の感覚として、押さえておくべき差があります。
の行列どうしの場合を比較します。
行列の積では、出力の各要素に 回の乗算と加算が必要です。要素は
個あります。
計算量は です。
アダマール積では、各要素に1回の乗算だけです。
計算量は です。
具体的な数値で確認します。 の場合を比べます。
千倍の差があります。アダマール積は、行列の積に比べて圧倒的に軽い演算です。
ゲート機構が多用される理由の一つが、ここにあります。制御のために大きな計算コストを払わずに済みます。
つまずきやすい点
研修で必ず補足すべき三点を挙げます。
1 ブロードキャストによる無言の失敗
最も危険な落とし穴です。
アダマール積は形が一致していなければ計算できない、と説明しました。しかし実際のライブラリでは、形が違っても自動的に拡張されて計算が通ってしまう場合があります。これをブロードキャストといいます。
たとえば、 の配列と
の配列を掛けると、
の結果が返ります。エラーは出ません。
意図した計算と違っていても、プログラムは動き続けます。学習が進まない原因を探して、何日も費やすことになりかねません。
対策は、計算の前後で形を確認する習慣です。研修では、実際に形を出力させる演習を必ず入れてください。
2 数学とプログラムで記号の意味が違う
数学の教科書では、 は行列の積を指します。しかしプログラムでは、* は要素ごとの積です。
論文の数式をコードに書き写すとき、この不一致が誤りを生みます。論文中に があればアダマール積、なければ行列の積、という読み方を伝えてください。
3 交換法則が成り立つかどうか
アダマール積は、順序を入れ替えても結果が変わりません。
一方、行列の積は一般に順序を入れ替えられません。
先ほどの数値例で確認します。
先ほどの とは、まったく違う結果になりました。
この違いも、混ぜるか選別するかという性質から説明できます。混ぜる順序は結果を変えますが、選別は順序に依存しません。
講師向けの進め方
推奨する順序
| 段階 | 内容 | 時間 |
|---|---|---|
| 1 | ミキサーとふるいの例えを提示する | 5分 |
| 2 | 2行2列の数値で両方を手計算させる | 15分 |
| 3 | 形の規則を表で整理する | 5分 |
| 4 | なぜ役割が分かれるかを、総和の有無から説明する | 10分 |
| 5 | LSTMのゲートで両方が並ぶ例を示す | 10分 |
| 6 | ブロードキャストの落とし穴を実演する | 10分 |
| 7 | 演習と解説 | 15分 |
合計70分程度です。
山場は2番
手計算を省略しないでください。定義を読むだけでは、総和の有無が体感されません。
2行2列であれば、全員が数分で計算できます。同じ二つの行列から違う結果が出ることを、自分の手で確認させることが重要です。
6番は必ず実演する
ブロードキャストは、口頭で説明しても危機感が伝わりません。実際に形の違う配列を掛けて、エラーが出ずに違う形の結果が返る様子を見せてください。
「これは動きます。しかし、あなたが意図した計算ではありません」
この一言が、後の実務での事故を防ぎます。
想定質問と回答例
質問。なぜ行列の積のほうがよく使われるのですか。
回答。新しい特徴を作り出せるからです。アダマール積は形を変えられないため、単独では表現力が上がりません。ただし、制御の役割ではアダマール積のほうが適しています。役割が違うため、どちらが優れているという話ではありません。
質問。アダマール積は学習するパラメータを持ちますか。
回答。持つ場合と持たない場合があります。レイヤー正規化のスケールパラメータは学習されます。ドロップアウトのマスクは、学習ではなくランダムに生成されます。LSTMのゲートは、その値を作る行列の積の側にパラメータがあります。
質問。ゲートの値はなぜ0から1なのですか。
回答。通過量を表すためです。0で遮断、1で全通過という直感的な意味を持たせるために、シグモイド関数が使われます。もし負の値を許すと、符号が反転してしまい、ふるいという解釈が成り立たなくなります。
質問。行列の積で情報が混ざると、元の情報は失われますか。
回答。一般には失われます。ただし、残差接続を併用すると、元の情報を別経路で保持できます。トランスフォーマーが深い層を積めるのは、この仕組みがあるためです。
演習問題
問題1
次の二つの行列について、行列の積とアダマール積をそれぞれ求めてください。
解答を示します。
行列の積を計算します。
1行1列目です。
1行2列目です。
2行1列目です。
2行2列目です。
アダマール積を計算します。
注目すべきは、 の扱いです。アダマール積では1行2列目が0になりますが、行列の積では10になりました。0であっても、他の要素の寄与によって出力は0になりません。混ぜる操作と選別する操作の違いが、はっきり現れています。
問題2
の行列と
の行列について、次の問いに答えてください。
(1) 行列の積は計算できますか。できる場合、出力の形は何ですか。 (2) アダマール積は計算できますか。
解答を示します。
(1) 計算できます。内側の次元が6で一致しているためです。出力の形は です。
(2) 計算できません。アダマール積は形が完全に一致している必要があります。
補足します。ブロードキャストが働く条件は、各次元が一致しているか、どちらかが1であることです。この例では4と6、6と3が対応し、いずれも条件を満たさないため、エラーになります。
問題3
次の式のうち、下線部にあたる演算がアダマール積であるものをすべて選んでください。
ア における
イ
における
ウ
における
エ
における
解答はイとエです。
解説します。イはLSTMのゲート適用であり、記憶の各次元に通過量を掛けています。エは活性化関数の微分を各位置に掛ける操作で、位置ごとに独立しています。
アは全結合層の変換です。入力の全要素を混ぜて出力を作るため、行列の積です。ウは注意の重みに従ってバリューを集約する操作で、複数の位置の情報を足し合わせるため、行列の積です。
ウを誤って選ぶ受講者が多く出ます。重みを掛けるという表現から、要素ごとの積を連想するためです。「足し合わせているか」を判断基準にするよう指導してください。
問題4
の行列どうしについて、行列の積とアダマール積の計算量の比を求めてください。
の場合の具体的な値も示してください。
解答を示します。
行列の積は 、アダマール積は
です。比は次のようになります。
の場合、512倍になります。
補足します。この差があるため、ゲート機構を追加しても計算コストの増加は限定的です。一方、行列の積を一つ追加すると、影響は大きくなります。モデルの設計で、どこにパラメータを置くかを考える際の判断材料になります。
振り返り
受講者に、次の問いで締めてください。
- 二つの演算のうち、次元を変えられるのはどちらですか
- 総和が現れるのはどちらですか
- 交換法則が成り立つのはどちらですか
- LSTMで、ゲートの値を作るのはどちらの演算ですか
- 形が違ってもエラーが出ない場合があるのは、どちらですか
すべて答えられれば、この単元は理解できています。
まとめと次の学習ステップ
行列の積は、総和を伴うため情報が混ざります。次元を変換し、新しい特徴を作り出します。ミキサーの役割です。
アダマール積は、位置ごとに独立して掛け算します。形を変えず、通過量を調整します。ふるいの役割です。
そして、この二つはしばしば連携します。ゲートの値をミキサーで作り、それをふるいとして適用する。LSTMもGRUも、現代のゲート付き活性化関数も、この組み合わせでできています。
次の学習ステップとしては、テンソルの形の追跡に進むとよいでしょう。実際のモデルでは、バッチ次元や系列長の次元が加わり、形が複雑になります。どの演算でどう形が変わるかを追えるようになれば、論文の実装を読む力が大きく伸びます。
あわせて、ブロードキャストの規則を体系的に押さえておくことをおすすめします。便利な仕組みですが、意図しない計算を許してしまう側面もあります。規則を知ったうえで使うか、明示的に形を揃えるかを、方針として決めておいてください。
セイ・コンサルティング・グループでは新人エンジニア研修のアシスタント講師を募集しています。
投稿者プロフィール


