L20|混合精度、風格轉換與神經元胞自動機(Mixed Precision)

零基礎提示

  • 這一課不用額外數學
  • Part 2 是大學程度:中三同學先求看懂概念,不用每行程式都會
  • 字幕:離線網站選「中英」(電腦翻譯,大意正確)
  • 看不懂的詞查 白話小辭典

用混合精度與 Hugging Face Accelerate 加速訓練;Johno 示範風格轉換(內容損失與 Gram 矩陣)和受生命遊戲啟發的神經元胞自動機。 (Speed up training with mixed precision and Hugging Face Accelerate; Johno demonstrates style transfer (content loss and Gram matrices) and Neural Cellular Automata inspired by the Game of Life.)

項目內容
所屬Part 2:Deep Learning Foundations to Stable Diffusion
影片YouTube
對應書本章節無對應章節(Part 2 以 notebook 為主)
原始頁面Lessons/lesson20
難度(F3 視角)★★★★★ 大學程度(F3 只求理解概念)

影片

本課重點

  1. 混合精度:大部分運算改用 16 位元浮點數(記憶體減半、速度更快),關鍵部分保留 32 位元以免精度不足。寫成 MixedPrecision callback,也可用 Accelerate 函式庫。
  2. 更快的資料載入:用自訂 collation 函數,在 GPU 上一次處理整批資料的增強,加快速度。
  3. 風格轉換:用預訓練網路不同層的特徵:內容損失比較中間層特徵,保留圖片內容;風格損失用 Gram 矩陣比較特徵之間的相關性,抓出筆觸和色彩。兩者相加就能把照片畫成某種畫風。
  4. 神經元胞自動機:受康威生命遊戲啟發:每個格子只看鄰居來決定下一步。先用固定 filter,再改用神經網路學出規則,用風格損失訓練出會自己長出紋理的系統。
  5. 技巧:環狀補邊(circular padding)讓圖案在邊界接續;梯度正規化讓訓練更穩定。

關鍵概念

英文術語中文一句話解釋
Mixed precision混合精度16 位元與 32 位元浮點數混用
AccelerateAccelerateHugging Face 的訓練加速函式庫
Style transfer風格轉換保留照片內容、換成另一種畫風
Content loss內容損失比較中間層特徵,保留內容
Gram matrixGram 矩陣特徵之間的相關性,代表風格
Cellular automata元胞自動機格子依鄰居狀態決定下一步的系統

認識關聯圖與重點圖

L20_關聯圖_202611041526

練習題

  1. 32 位元浮點數每個數字佔 4 bytes,16 位元佔 2 bytes。一個有 1 億個參數的模型,兩種格式各佔多少記憶體(以 MB 計,1 MB ≈ 100 萬 bytes)?
  2. 風格轉換中,內容損失和風格損失各負責什麼?
  3. 康威生命遊戲裡,每個格子怎樣決定下一步?這和神經元胞自動機有什麼關係?

資源連結

英文延伸資源(零基礎可以先略過,點開看)

相關筆記

Built with LogoFlowershow