深度學習模型架構圖

神經網路架構圖產生器

把一段文字描述變成乾淨的神經網路架構圖。寫出各層、張量形狀、跳躍連接和重複模組,就能得到可用於論文、學位論文、海報和簡報的模型架構圖。

CNN 與 Transformer 架構圖張量形狀與跳躍連接無浮水印

AI 神經網路架構圖產生器

描述模型裡的各層、形狀和連接方式。

消耗 5 點數254 / 2000 字元

根據您的描述生成 PNG 架構圖。

預覽

神經網路架構圖範例

目標寬高比:16:9

Transformer 編碼器架構圖,含詞嵌入、位置編碼,以及由多頭注意力和前饋子層組成的 ×6 堆疊
CNN 架構圖,含三組卷積與池化階段、展平層和兩個全連接層,並標出張量形狀
U-Net 架構圖,含四級編碼器、1024 通道瓶頸層、解碼器,以及同解析度之間的灰色跳躍連接

以上為範例預覽。描述您的模型即可生成屬於您的架構圖。

神經網路架構圖範例

把滑鼠移到卡片上即可複製它的提示詞。每個範例都寫明了各層、形狀和連接方式,讓圖和模型對得上。

Transformer 編碼器架構圖,含詞嵌入、位置編碼,以及由多頭注意力和前饋子層組成的 ×6 堆疊

Prompt

生成一張 Transformer 編碼器的架構圖,自下而上排布。輸入詞元 → 詞嵌入(d_model = 512)加正弦位置編碼。然後在一個標註「×6」的圓角框裡畫一層編碼器:多頭自注意力(8 個頭)→ Add & Norm → 逐位置前饋網路(512 → 2048 → 512,ReLU)→ Add & Norm。每個子層外畫一條殘差箭頭匯入它的 Add & Norm。輸出標籤寫「編碼器輸出(batch, seq_len, 512)」。注意力層橙色,前饋層藍色,正規化層黃色。白色背景,乾淨的期刊風格。

Transformer 編碼器架構

CNN 架構圖,含三組卷積與池化階段、展平層和兩個全連接層,並標出張量形狀

Prompt

生成一張用於 10 類影像分類的 CNN 架構圖,從左到右用三維方塊排布。輸入層 32×32×3 → 卷積層 3×3、32 個濾波器、ReLU → 32×32×32 → 最大池化 2×2 → 16×16×32 → 卷積層 3×3、64 個濾波器 → 16×16×64 → 最大池化 2×2 → 8×8×64 → 卷積層 3×3、128 個濾波器 → 8×8×128 → 最大池化 2×2 → 4×4×128 → 展平 2048 → 全連接層 256、ReLU → 全連接層 10、softmax。每個形狀寫在對應方塊下方。卷積層藍色,池化層橙色,全連接層綠色,並配圖例。

CNN 影像分類器

U-Net 架構圖,含四級編碼器、1024 通道瓶頸層、解碼器,以及同解析度之間的灰色跳躍連接

Prompt

生成一張 U 形排布的 U-Net 分割架構圖。編碼器向下,每一級兩個 3×3 卷積層加 ReLU、使用 same 填充:256×256×64、128×128×128、64×64×256、32×32×512,各級之間用紅色 2×2 最大池化箭頭連接。瓶頸層:16×16×1024。解碼器向上,用綠色 2×2 上卷積箭頭:32×32×512、64×64×256、128×128×128、256×256×64。灰色水平箭頭標註「複製並拼接」,把每一級編碼器連到同尺寸的解碼器層。最後 1×1 卷積層 → 256×256×1 遮罩,sigmoid。輸入:256×256×1 影像。

U-Net 分割模型

LSTM 序列模型架構圖,按時間展開,含詞嵌入、兩層堆疊的 LSTM、隱藏狀態與細胞狀態,以及 sigmoid 輸出

Prompt

生成一張兩層 LSTM 情感分類器的深度學習模型架構圖,按時間步 t = 1, 2, …, T(T = 100)展開。最下面一列:輸入詞元 x1、x2、…、xT → 詞嵌入(128)。中間幾列:第 1 層 LSTM 單元和第 2 層 LSTM 單元,隱藏層維度 256,相鄰時間步之間用水平箭頭傳遞隱藏狀態 h 和細胞狀態 c。只有第 2 層最後一個時間步的隱藏狀態 h_T(256)向上進入全連接層 1 並經 sigmoid → 「正面 / 負面」。標出時間軸。乾淨、平面、白色背景。

LSTM 序列模型

圖神經網路架構圖,展示鄰居訊息聚合為節點更新、三層訊息傳遞、池化讀出和 MLP 預測

Prompt

生成一張分兩個分圖的圖神經網路架構圖。左圖:一個小圖,中心節點 v 和鄰居 u1、u2、u3;每個鄰居指向 v 的箭頭標註 m_uv = MLP(h_u, e_uv);一個標註「聚合」的求和符號;以及更新式 h_v(k+1) = UPDATE(h_v(k), Σ m_uv)。右圖畫流程:節點特徵(64)→ 第 1 層訊息傳遞(128)→ 第 2 層(128)→ 第 3 層(128)→ 全域平均池化 → MLP → 圖級預測。節點青色,訊息橙色。白色背景。

圖神經網路的訊息傳遞

多模態模型架構圖,ViT 影像編碼器和 Transformer 文字編碼器各投影到 512 維,再在融合 MLP 頭中匯合

Prompt

生成一張有兩條平行分支的多模態模型架構圖。上分支:影像 224×224×3 → 切成 16×16 的圖塊(196 個詞元)→ ViT-B/16 影像編碼器、12 層、寬度 768 → [CLS] 向量 768 → 線性投影 512。下分支:文字,最多 77 個詞元 → Transformer 文字編碼器、12 層、寬度 512 → [EOS] 向量 512 → 線性投影 512。兩條分支匯入融合頭:拼接(1024)→ MLP 1024 → 512、ReLU → N 類、softmax。影像分支紫色,文字分支綠色,融合部分灰色。

多模態融合模型

如何畫神經網路架構圖

1

寫出模型架構

按順序列出各層:輸入層、隱藏層、輸出層各自的類型、規模和輸出形狀。每條跳躍連接、每個重複模組和每條分支都要寫出來。

2

定好視覺編碼

選擇排布方向、每種層類型的顏色,以及模組畫成平面方框還是三維體塊。

3

生成並核對

生成圖之後,逐一核對每個標籤和維度是否與程式碼一致,必要時修改提示詞再生成。

神經網路架構圖的標註慣例

  • 張量形狀標籤:在每個模組後面印出輸出形狀,全圖只用一種格式,例如影像用 H×W×C,序列用 (batch, seq_len, d_model)。
  • 跳躍連接:殘差連接和 U-Net 的跳躍連接畫成弧線或虛線箭頭,並匯入一個可見的合併節點,相加標 ⊕,拼接則標註「concat」。
  • 重複模組:只畫一份,放在方括號或外框裡並標出 ×N,例如六層 Transformer 編碼器寫「×6」,不要疊六份一樣的。
  • 按層類型配色:為卷積層、池化層、正規化層、注意力層和全連接層各分配一種顏色,並加上圖例。不要只靠顏色區分,標籤上仍然要寫出每一層的名稱。

怎樣在提示詞裡把架構寫準確

把模型寫成帶箭頭的有序鏈條,每一步後面跟上形狀:「輸入層 32×32×3 → 卷積層 3×3、32 個濾波器 → 32×32×32 → 最大池化 2×2 → 16×16×32」。工具會照著您寫的畫,所以「一個很深的 CNN」這種模糊提示只會得到模糊的圖。

說明排布方向(從左到右、自下而上,還是 U 形)、哪些連接跨層、分支在哪裡匯合、哪些模組會重複。措辭要緊的地方,就把標籤原文用引號給出。畫 CNN 架構圖時,請說明要三維體塊還是平面方框。

與 PlotNeuralNet、draw.io 和 NN-SVG 的比較

PlotNeuralNet 是採用 MIT 授權的專案,為架構圖生成 LaTeX/TikZ 程式碼,並提供可選的 Python 介面,適合本來就用 LaTeX 寫作、希望把圖當程式碼管理的作者。NN-SVG 是採用 MIT 授權的參數化工具,可匯出 FCNN、LeNet 和 AlexNet 風格的 SVG,模型剛好符合這幾種風格時最合用。draw.io 是採用 Apache 2.0 授權的免費通用繪圖工具,有網頁版和桌面版,每個圖形都能手動控制。

當架構是那些工具沒有模板的類型(例如多模態模型或圖模型),而您又不想寫 TikZ、不想一個個方框擺位時,本工具最適合快速拿到一版精修的初稿。它的輸出是 PNG 影像,不是精確的向量幾何。最近核對時間:2026 年 9 月 15 日,依據 PlotNeuralNet(github.com/HarisIqbal88/PlotNeuralNet)、NN-SVG(github.com/alexlenail/NN-SVG)和 draw.io(github.com/jgraph/drawio)的官方儲存庫。

發表前的核對要點

  • 數一數圖裡的模組,和模型定義對照,包括標了 ×N 的重複堆疊。
  • 重算形狀:步長 2 的 2×2 池化會把高和寬各減半,4×4×128 展平後是 2048。
  • 檢查每條跳躍連接的起點和終點是否落在對應的層和解析度上。
  • 逐字讀每個標籤;影像模型可能把「LayerNorm」之類的詞寫錯,或把數字調換。
  • 讓圖裡的名稱與方法部分和程式碼裡的名稱保持一致。

神經網路架構圖常見問題

它根據一段文字描述畫出模型架構圖:CNN、Transformer、U-Net、RNN 與 LSTM 模型、圖神經網路、自編碼器和多模態模型。您可以要求標出張量形狀、跳躍連接、用 ×N 標記的重複模組、注意力模組,以及按層類型區分的顏色圖例。

畫出你的模型架構圖

描述各層、形狀和連接方式,為下一篇論文或報告生成乾淨的模型架構圖與實驗架構圖。