跳到主要內容
AI News HubLIVE
站內改寫1 分鐘閱讀

S-GAI:基於譜幾何的Sigmoid MLP初始化方法——從資料集幾何到網路權重

文章摘要

經典萬能逼近定理證明了Sigmoid多層感知機的表達能力,但未規定如何初始化權重以編碼資料分佈的幾何結構。本文提出S-GAI,一種針對單隱層Sigmoid MLP的譜幾何感知初始化框架。該框架利用SVD為每個類別估計均值、主方向和譜尺度,透過能量閾值選擇保留方向,並每個方向用兩個Sigmoid門表示。這些類別特定的門構成共享的隱層,直接從訓練集初始化。實驗表明,S-GAI初始化的MLP比Xavier初始化具有更豐富的隱藏狀態,在完全訓練下達到相近的最終精度;凍結隱層時,只訓練輸出層仍優於凍結隨機門。

來源arXiv Machine Learning作者: Yi-Shan Chu
S-GAI:基於譜幾何的Sigmoid MLP初始化方法——從資料集幾何到網路權重
回報錯誤

更正管道尚未開通,可先複製下方文章資訊留存。

查看更正說明
直接讀正文

經典萬能逼近定理早已證明,Sigmoid多層感知機(MLP)具有強大的表達能力,但如何為網路權重賦予初始值,使其能夠編碼資料分佈的幾何結構,這一直是一個懸而未決的問題。來自研究者Yi-Shan Chu的最新工作提出了S-GAI(Spectral Geometry-Aware Initialization),一種針對單隱層Sigmoid MLP的譜幾何感知初始化框架,旨在從資料集的幾何特性出發,為網路提供更優的起點。

S-GAI的核心思想源於將Sigmoid單元視為平滑的半空間門控。傳統方法常依賴手工指定的平面幾何,而S-GAI則轉向從影像資料中估計的類別譜幾何。具體而言,對於每個類別,該方法透過奇異值分解(SVD)計算均值、主方向以及譜尺度。隨後,利用能量閾值篩選出最重要的方向,每個保留方向由兩個Sigmoid門表示。這些類別特定的門組合成一個共享的隱藏層,其權重直接由訓練集初始化,從而將資料幾何結構嵌入網路。

為了驗證所估計的譜類別幾何是否已具備判別性,研究者還設計了一個基於SVD的子空間分類器作為非神經網路的幾何參考。實驗在MNIST、Fashion-MNIST以及更具挑戰性的CIFAR-10資料集上進行。結果顯示,採用S-GAI初始化的MLP,其隱藏狀態從一開始就比Xavier初始化包含更多資訊;在完整訓練後,兩者能達到相近的最終準確率。特別地,當隱藏層被凍結、僅訓練輸出層時,S-GAI初始化的網路效能顯著優於使用隨機門的凍結隱層,這有力證明了S-GAI能夠有效地將類別譜幾何嵌入到MLP的初始權重中。這一工作為利用資料幾何指導神經網路初始化提供了新思路,未來可能擴充套件到深層網路和其他啟用函式。此外,S-GAI的計算開銷主要集中在訓練前的SVD分解,實際訓練過程中不增加額外負擔,具有較好的實用潛力。該研究成果已在arXiv上釋出,程式碼預計將開源,有望推動相關領域的進一步探索。

展開要點與分析

文章情報

研究者進階

要點

  • S-GAI利用SVD從影像資料中估計類別的譜幾何資訊,用於初始化Sigmoid MLP的隱層。
  • 每個類別的主方向透過能量閾值篩選,每個方向由兩個Sigmoid門表示,構建共享隱層。
  • 在MNIST、Fashion-MNIST和CIFAR-10上,S-GAI初始化使MLP的隱藏狀態更豐富,全訓練時精度與Xavier初始化相當。
  • 凍結隱層時,僅訓練輸出層效能優於隨機門初始化,表明S-GAI有效嵌入了類別幾何結構。

要點與分析由自動化流程生成,可能有誤,請結合原始來源核實。