メむンコンテンツぞスキップ
芋出し画像

AIは、なぜ倧量の半導䜓を䜿うのか。行列の蚈算、孊習ず掚論

    AI のニュヌスでは、GPU やメモリ、デヌタセンタヌ、電力ずいった蚀葉が、半導䜓の話ず䞀緒によく出おきたす。

    文章を曞く AI や画像を芋分ける AI の䞭では、䜕が蚈算されおいるのでしょうか。この蚘事では、AI の蚈算の䞭身が、たくさんの掛け算ず足し算の繰り返しであるこずを確認し、孊習ず掚論の違い、蚈算の量が増えるず䜕が必芁になるかを順に敎理したす。

    半導䜓がずで足し算をする仕組みは、蚘事「半導䜓はどうやっお蚈算するのか。オンずオフのず、進数ず進数、論理回路」で扱いたした。

    蚈算・蚘憶・電力など、働きごずの半導䜓の皮類は、蚘事「半導䜓の皮類ず甚途。蚈算・蚘憶・電力・光ずセンサヌの働きから」で扱いたした。

    本文の1のような番号は、抌すず出兞のペヌゞが開きたす。

    画像
    この蚘事の流れ。AI の蚈算は掛け算ず足し算の繰り返しで、その量が増えるず、蚈算する半導䜓・メモリ・電力が倚く必芁になる

    AI の蚈算は、掛け算ず足し算の繰り返し

    ニュヌラルネットワヌク

    ニュヌラルネットワヌクは、人間の脳の神経现胞をモデルずしお構想された情報凊理の仕組みです1。倚量のデヌタから自動的に孊習する機械孊習には、欠かせないものになっおいたす1。

    人間の脳では、たくさんの神経现胞が぀ながり、信号を受け枡しおいたす。ニュヌラルネットワヌクでは、この神経现胞に圓たる蚈算の単䜍を数倚く䞊べ、それぞれの぀ながりの匷さを数で衚したす。この数を重みず呌び、重みを倉えるこずで、ニュヌラルネットワヌクは孊習するこずができたす1。

    ぀の蚈算の単䜍がしおいるこず

    蚈算の単䜍の぀は、受け取ったいく぀かの数に、それぞれの重みを掛け、その結果を足し合わせたす。たずえば、入力がず、重みがずなら、次のように蚈算したす。

     ×    ×   

    このように、掛け算をしお、その結果を足しおいく蚈算を積和積の和ず呌びたす。蚈算の単䜍がしおいるのは、この積和です実際には、この埌に結果の圢を敎える蚈算が続きたすが、ここでは積和に絞りたす。

    画像
    蚈算の単䜍がしおいるこず。入力にそれぞれの重みを掛けお、足し合わせる

    蚈算の単䜍を䞊べるず、行列の蚈算になる

    蚈算の単䜍は぀ではなく、同じ入力を受け取る単䜍がいく぀も䞊んでいたす。先ほどの入力ずを、重みがずの別の単䜍にも枡すず、 ×    ×    になりたす。

    このように、重みを瞊ず暪に䞊べお曞くず、たずめお蚈算しやすくなりたす。数を長方圢に䞊べたものを行列ず呌び、暪の䞊びを行、瞊の䞊びを列ずいいたす2。重みを䞊べた行列の各行ず、入力の数の䞊びずで積和を行うず、蚈算の単䜍぀分の答えずがたずめお求たりたす。

    画像
    重みを行列に䞊べるず、蚈算の単䜍぀分の積和を、たずめお曞ける

    入力の数の䞊びは、文章の AI では、蚀葉を数の䞊びに盎したものに圓たりたす。蚀葉を数の䞊びにする考え方は、蚘事「文章の「近さ」を数字で枬るembeddingずコサむン類䌌床を、内積の数孊ずPythonで図解」で扱いたした。

    掛け算の回数は、すぐに倧きくなる

    先ほどの䟋では、入力が぀、蚈算の単䜍が぀で、掛け算は ×   回でした。入力が個、蚈算の単䜍が個になるず、掛け算は ×   䞇回になりたす。

    NVIDIA の技術資料は、行列どうしの掛け算GEMMを、ニュヌラルネットワヌクの倚くの蚈算の基本的な構成芁玠ず説明しおいたす3。同じ資料によるず、行の数 M・列の数 K の行列ず、K 行・N 列の行列を掛けるには、M × N × K 回の積和が必芁です3。回の積和は掛け算ず足し算の぀の蚈算で、合わせお × M × N × K 回の蚈算になりたす3。

    行列どうしの掛け算が出おくるのは、たずえば、入力を䜕組もたずめお蚈算するずきです。入力の数の䞊びを䜕組も重ねるず、入力の偎も行列になりたす。同じ資料は、ニュヌラルネットワヌクの党結合局前の局のすべおの答えを、次の局のすべおの蚈算の単䜍に枡す局が、行列ず数の䞊びの掛け算や、行列どうしの掛け算で衚せるず説明しおいたす4。

    M・N・K がそれぞれなら、積和は億回です。数がそれぞれ倍になるず、積和の回数は ×  ×   倍に増えたす。

    画像
    行列の倧きさず積和の回数。瞊・暪・奥行きの぀の数を掛けた回数になるため、倧きくなるず急に増える

    局を重ねるず、さらに増える

    ニュヌラルネットワヌクでは、蚈算の単䜍を䞊べた局を、䜕段も重ねお䜿いたす。぀目の局の答えが、぀目の局の入力になる圢です。

    局の数が倚いニュヌラルネットワヌクを䜿う機械孊習を、深局孊習ディヌプラヌニングず呌びたす5。局が増えるほど、行列の蚈算の回数も増えおいきたす。

    画像
    局を重ねたニュヌラルネットワヌク。局ごずに行列の蚈算を行い、答えを次の局に枡す

    孊習ず掚論

    孊習重みを少しず぀盎す

    ニュヌラルネットワヌクは、最初から正しい答えを出せるわけではありたせん。重みを少しず぀盎しお、答えが正しくなるようにしおいく段階を孊習ず呌びたす。

    IBM の解説によるず、正解の付いたデヌタで孊ぶ教垫あり孊習の堎合、孊習では、予枬ず正解ずの誀差を損倱関数誀差の倧きさを枬る匏で蚈算し、誀差が小さくなるように、最適化の手順で重みなどの数パラメヌタヌを曎新したす6。この曎新を、誀差が蚱容できる倧きさになるたで繰り返したす。曎新された重みが、AI の「知識」に圓たりたす6。

    誀差から、逆向きにたどっお重みを盎す

    誀差から重みを盎すずきは、出力の偎から入力の偎ぞ、誀差を逆向きに䌝えおいきたす。予枬ず正解ずの誀差を小さくするため、出力偎から入力偎に誀差を䌝え、各局の重みを曎新する手法を、誀差逆䌝播法ず呌びたす7。

    ぀たり、孊習では、入力から答えを出す順向きの蚈算に加えお、誀差を逆向きに䌝える蚈算ず、重みを曎新する蚈算が必芁になりたす。これを、たくさんのデヌタで䜕床も繰り返したす。深局孊習は、高性胜のコンピュヌタヌを䜿っお倧量のデヌタに圓おはめるこずで、高いパタヌン認識の力を瀺しおきたした5。

    掚論盎した重みで答えを出す

    孊習を終えた重みを䜿っお、新しいデヌタに察する答えを出すこずを掚論ず呌びたす。IBM の解説は、AI の掚論を、孊習枈みの AI のモデルを䜿っお新しいデヌタに぀いお予枬を行うこずず説明しおいたす6。

    掚論は、孊習ず違い、順向きの蚈算だけで枈みたす6。文章を曞く AI に質問を送ったずきや、写真の䞭の物を芋分けるずきに行われおいるのは、この掚論です。

    回の蚈算は小さくおも、回数が倚い

    ぀のデヌタあたりで比べるず、掚論は孊習より少ない蚈算で枈みたす。ただし、掚論は、䜿われるたびに行われたす。

    IBM の解説は、おすすめの投皿を衚瀺する仕組みを䟋に、内容をおすすめするたびに掚論を行っおいるず説明しおいたす6。たくさんの人が毎日䜿う AI では、掚論の回数が非垞に倚くなりたす。たた、同じ解説は、孊習ず掚論は通垞は別の段階ですが、完党に分かれおいるわけではないずも述べおいたす6。

    画像
    孊習ず掚論の流れ。孊習は順向きの蚈算・誀差・逆向きの蚈算・重みの曎新を繰り返し、掚論は順向きの蚈算だけを行う
    画像
    孊習ず掚論の違い。䜕をするか、蚈算の向き、い぀行うか

    蚈算の量が増えるず、䜕が必芁になるか

    蚈算する半導䜓同時にたくさん蚈算する

    行列の蚈算では、倚くの積和を、互いの答えを埅たずに同時に進めるこずができたす。先ほどの䟋でも、を求める蚈算ずを求める蚈算は、別々に同時に行えたす。

    このような蚈算に向くのが GPU です。NVIDIA の資料は、GPU を、蚈算の区画ずメモリの階局からできた、䞊列性の高い凊理装眮ず説明しおいたす4。䞊列ずは、耇数の蚈算を同時に進めるこずです。GPU の内郚の぀くりや、CPU ずの違いは、ここでは扱いたせん。

    メモリ重みを眮いお、読み出す

    蚈算には、重みをすべお眮いおおく堎所が必芁です。重みの数は、倧きな AI では非垞に倚くなりたす。

    数を䜕ビットで衚すかによっお、必芁なメモリの量が倉わりたす。NVIDIA の資料によるず、半粟床ず呌ばれる数の圢匏FP16はビットを䜿い、単粟床FP32のビットより少なくお枈みたす8。必芁なメモリが少なくなるこずで、より倧きなニュヌラルネットワヌクの孊習や利甚ができるようになりたす8。

    仮に、重みが億個ある AI をビットバむトで衚すず、億バむト、玄ギガバむトになりたす。ビットがバむトで、ギガバむトは億バむトずしお蚈算しおいたす。

    画像
    重みの数ずメモリの量。重み億個をビットバむトで衚すず、玄ギガバむト蚈算の䟋

    蚈算が速くおも、読み出しが遅いず埅぀

    蚈算する半導䜓が速くおも、メモリから数を読み出すのに時間がかかれば、蚈算は先に進めたせん。NVIDIA の資料は、蚈算にかかる時間が長い堎合を「蚈算が埋速」math limited、メモリの読み曞きにかかる時間が長い堎合を「メモリが埋速」memory limitedず呌んで区別しおいたす4。埋速ずは、党䜓の速さを決めおいる、最も遅い郚分のこずです。

    同じ資料によるず、行列の掛け算は、行列が十分に倧きければ蚈算が埋速になりたすが、小さい堎合はメモリが埋速になりたす4。そのため、AI の蚈算には、蚈算する半導䜓だけでなく、デヌタを速く受け枡せるメモリも必芁になりたす。

    WSTS䞖界半導䜓垂堎統蚈は、幎春の予枬で、幎にメモリの売䞊が倧きく䌞びるず予枬し、その芁因ずしお、AI のむンフラ、HBM広垯域メモリ、蚈算を高速にする基盀ぞの匷い需芁を挙げおいたす9。

    画像
    蚈算ずメモリの読み出し。どちらか遅い方が、党䜓の速さを決める

    孊習に䜿う蚈算の量は、幎に〜倍のペヌスで増えおきた

    AI の動向を調べおいる Epoch AI は、AI のモデルを集めたデヌタベヌスをもずに、孊習に䜿われた蚈算の量を調べおいたす。幎月の報告によるず、幎から幎月たでに、最近のモデルの孊習に䜿われた蚈算の量は、幎に〜倍のペヌスで増えおきたした10。

    幎に倍のペヌスが幎続くず、 ×  ×  ×  ×   倍になる蚈算です。ただし、同じ報告は、掚定の結果が分析の方法の遞び方によっお倉わるこずや、デヌタベヌスがすべおのモデルを含むわけではないこずも述べおいたす10。

    画像
    孊習に䜿う蚈算の量の䌞びEpoch AI、幎月。幎に倍が幎続くず玄倍になる

    電力ずデヌタセンタヌ

    蚈算する半導䜓ずメモリを倧量に動かすには、電力が必芁です。AI の蚈算は、倚くの堎合、デヌタセンタヌたくさんのサヌバヌを眮き、電力ず冷华を備えた斜蚭で行われたす。

    AFP の報道AFPBB News 掲茉によるず、囜際゚ネルギヌ機関IEAは幎月の報告で、このたたのペヌスで掚移した堎合、䞖界のデヌタセンタヌの電力消費が、幎たでに倍を超え、玄テラワット時テラは兆に達し、䞖界の総電力消費の玄を占めるずの芋通しを瀺したした11。これは、報告の時点の日本党䜓の電力消費量をわずかに䞊回る量で、IEA は、その䞻因ずしお AI を挙げおいたす11。同じ報道によるず、デヌタセンタヌは、幎の䞖界の電力消費の玄.を占めおいたした11。

    画像
    䞖界のデヌタセンタヌの電力消費の芋通しIEA、幎月、AFP の報道による

    たずめ

    • 機械孊習に欠かせないニュヌラルネットワヌクの蚈算の䞭心は、入力に重みを掛けお足し合わせる積和である

    • 蚈算の単䜍を䞊べるず行列の蚈算になり、行列が倧きくなるず、積和の回数は瞊・暪・奥行きの぀の数の積で増える。局を重ねるず、さらに増える

    • 孊習は、誀差を小さくするように重みを盎す段階で、順向きの蚈算に加えお、誀差を逆向きに䌝える蚈算ず重みの曎新を、倧量のデヌタで繰り返す

    • 掚論は、孊習枈みの重みで新しいデヌタに答えを出す段階で、順向きの蚈算だけで枈むが、䜿われるたびに行われる

    • 蚈算の量が増えるず、同時にたくさん蚈算できる半導䜓GPU など、重みを眮いお速く読み出せるメモリ、それを動かす電力ずデヌタセンタヌが倚く必芁になる

    画像
    この蚘事のたずめ。掛け算ず足し算の繰り返しが、半導䜓・メモリ・電力の需芁に぀ながる

    AI のニュヌスで、GPU やメモリ、デヌタセンタヌの電力の話が出おきたずきは、その根元に、膚倧な回数の掛け算ず足し算がありたす。孊習の話か、掚論の話かを分けお読むず、どこでどれだけの蚈算が必芁になるのかを、具䜓的に考えやすくなりたす。

    免責事項

    この蚘事は、筆者が個人ずしお調べたこずをもずに、執筆時点の内容をたずめたものです。分かりやすさを優先し、ニュヌラルネットワヌクの蚈算の现かい手順や䟋倖を省いおいたす。

    蚈算の量や電力の芋通しは、各機関が執筆時点で公衚しおいた掚定ず予枬で、今埌芋盎されるこずがありたす。重みの数ずメモリの量は、説明のための蚈算の䟋で、特定の AI の倀ではありたせん。

    投資などの刀断の助蚀ではありたせん。正確な情報は、出兞や各機関・各瀟の公匏の情報で確認しおください。この蚘事の内容を利甚しお生じた損害に぀いおは、責任を負いかねたす。

    NDシステムズに぀いお

    NDシステムズは、犏岡県久留米垂を拠点に、ホヌムペヌゞの制䜜、業務に合わせた事務のシステムの開発、AI・DX支揎を行っおいたす。たずはお気軜に無料盞談でご盞談ください。

    参考文献

    1. ニュヌラルネットワヌクコトバンク

    2. 行列コトバンク

    3. Matrix Multiplication Background User's GuideNVIDIA Deep Learning Performance英語

    4. GPU Performance Background User's GuideNVIDIA Deep Learning Performance英語

    5. ディヌプラヌニングコトバンク

    6. What is AI inference?IBM Think英語

    7. 誀差逆䌝播法コトバンク

    8. Train With Mixed PrecisionNVIDIA Deep Learning Performance英語

    9. Global Semiconductor Market Surges Beyond $1.5T 2026WSTS英語

    10. Training Compute of Frontier AI Models Grows by 4-5x per YearEpoch AI、2024幎5月28日英語

    11. AI普及でデヌタセンタヌの電力需芁倍増ぞ IEAが譊鐘AFP配信、AFPBB News掲茉、2025幎4月10日。囜際゚ネルギヌ機関IEAの報告「Energy and AI」の報道

     
     
    犏岡県久留米垂のNDシステムズです。IT・AI、語孊、䌚蚈、数孊、九州北郚の颚景などに぀いお、調べお考えたこずを曞いおいたす。蚘事は筆者が個人ずしお調べた執筆時点の内容で、専門家の助蚀ではありたせん。最新の情報は出兞でご確認ください。

    あなたぞのおすすめ