顯示具有 2024 標籤的文章。 顯示所有文章
顯示具有 2024 標籤的文章。 顯示所有文章

2026年6月28日 星期日

KAN: Kolmogorov-Arnold Networks 深度解析

在深度學習領域,多層感知器(MLP)可說是最基礎且廣泛應用的模型,幾乎成為多數神經網路架構中不可或缺的一環。這類模型的核心設計通常是「固定且線性的權重」搭配「非線性激活函數」組合,透過大量的參數學習複雜函數映射。然而,這種結構在一定程度上受到表現力及可解釋性的限制,尤其在處理高維函數擬合、偏微分方程(PDE)求解等任務時,往往需要龐大參數與訓練耗費。本篇由劉宏、王多多等人於 ICLR 2024 發表並獲得 Outstanding Paper 的論文《KAN: Kolmogorov-Arnold Networks》提出一種全新架構,改寫了傳統神經網路的基本設計理念,實現準確度與可解釋性的雙重突破。

研究背景與動機

Kolmogorov-Arnold 表示定理是數學領域一項極具啟發性的結果,該定理指出任何多變數連續函數都可以被分解成有限個單變數函數的組合。在機器學習語境中,這種結構暗示了一種潛在的模型設計思路:透過一組單變數函數的線性組合,我們或許能更有效率且有結構性地逼近複雜函數。

然而,傳統的 MLP 在設計上以固定形式的激活函數與線性權重為主,並未直接利用單變數映射的靈活性。這導致模型要達到同樣的表現需要規模更大、訓練更久,且模型複雜性的解釋性較低。針對這樣的挑戰,本論文創新地將 Kolmogorov-Arnold 定理中關鍵的「單變數函數」思想搬進神經網路架構,設計出一套全無線性權重、而是以學習型「邊激活函數」替代的網絡,這就是 KAN (Kolmogorov-Arnold Networks) 的核心動機。

核心方法與創新

KAN 的最大突破在於「拋棄線性權重、以可學習的單變數激活函數替代」。具體來說:

  • 邊激活函數的引入: KAN 模型將神經元間的邊視為可以學習的單變數函數(univariate functions),這些函數用樣條函數(spline)表示,透過參數化控制其形狀,成為模型全部的可訓練參數。傳統 MLP 則是由固定形狀但可調係數的權重所構成,兩者設計理念截然不同。
  • 全模型無線性權重: KAN 沒有權重乘積的線性部分,神經網路每一層的輸出由上一層各節點經過不同單變數函數後加總組合,形態更加靈活。
  • 可視化與互動性強: 由於每條邊代表一個可視化的單變數函數,使用者可直觀觀察到各「權重」函數的非線性形狀,這使得模型更具解釋力與可互動性,極大提升了神經網路的可用性與透明度。

透過這種設計,KAN 不單純是架構上的小改變,而是從基礎結構出發對深度神經網路的「參數本質」進行重塑,讓模型自動學習符合 Kolmogorov-Arnold 表示定理的函數分解形式,提升表達能力及泛化性能。

主要實驗結果

作者充分實證 KAN 在多個挑戰性問題中的優越表現,包括函數擬合任務及偏微分方程求解:

  • 精準度與模型規模: 在標準函數擬合問題裡,KAN 遠小於傳統 MLP 的模型規模卻能達到同等甚至更好的精度,說明其單變數函數結構強化了參數利用效率。
  • 偏微分方程求解: 在物理領域標準的 PDE 求解實驗中,KAN 不僅提高了數值解的精準度,還縮短了訓練收斂時間,展現出更快的神經縮放定律(neural scaling laws),確立其在科學計算中的實用價值。
  • 可解釋性案例: 作者以兩個具代表性的數學與物理範例展示 KAN 如何協助人類科學家重新發現已知定律。這種互動式的學習與解釋,顛覆了黑盒模型的傳統,為 AI 輔助科學研究帶來新啟示。

整體而言,實驗結果不只是健壯性與性能提升,更彰顯了模型設計深度融合領域數學理論的潛能。

對 AI 領域的深遠影響

KAN 在理論與實踐上均展現出令人振奮的突破,有望在以下幾個層面深刻影響 AI 研究與應用:

  1. 模型架構的革新: KAN 挑戰了「線性權重+固定激活」的標準神經網路架構,提出以可學習的單變數激活函數為核心的全新設計哲學,這可能引領未來神經網路向更可控、更易解釋的方向發展。
  2. 數學理論與 AI 的橋樑: 成功將經典的 Kolmogorov-Arnold 表示理論於深度學習架構中實現,表明數學中的抽象定理能直接促成模型創新,未來或可探索更多類似橋接機會,強化 AI 方案的理論基礎。
  3. 提升 AI 可解釋性與人機協同: KAN 自然的可視化與解釋途徑使其更適合作為科學家與工程師的智能助理,推動 AI 系統不再是黑盒,而是成為可共同理解與改進的合作夥伴。
  4. 推動高效科學計算: 具備更快縮放規律的 KAN,有潛力在科學、工程模擬、物理建模等領域替代傳統 MLP,帶來計算成本和精度的雙贏,促進複雜系統建模與求解效率。

總結來說,KAN 開創了以「可學習的邊激活單變數函數」代替線性權重的新典範,它在模型效能與解釋性上提供了一套完整且可操作的解決方案,對深度學習架構設計提出全新的視角。未來,KAN 不僅具備成為 MLP 及其他深度模型強力替代品的潛力,更象徵著跨領域理論與實踐融合的成功範例,值得 AI 研究者及工程師密切關注和深入探索。


論文資訊
📄 KAN: Kolmogorov-Arnold Networks
👥 Liu, Wang, Vaidya, Ruehle, Halverson, Soljačić, Hou, Tegmark
🏆 ICLR 2024 · Outstanding Paper
🔗 arxiv.org/abs/2404.19756

Mamba: Linear-Time Sequence Modeling with Selective State Spaces

近年來,Transformer 架構已成為深度學習中基礎模型(Foundation Models)不可或缺的核心,尤其在自然語言處理、語音及基因資料處理等多種序列任務上展現卓越性能。然而,Transformer 本身的注意力機制(Attention)計算複雜度為 O(N2),當序列長度 N 大幅增長時,對計算資源及記憶體的需求驟增,極大限制了其在超長序列上的實際應用。此外,為了克服此短板,學界曾嘗試多種子二次時間(subquadratic-time)架構,包括線性注意力(linear attention)、門控卷積(gated convolution)、遞迴模型以及結構化狀態空間模型(Structured State Space Models, SSMs)。雖然這些方法在計算效率上有顯著提升,但在語言等離散模態的表現仍難以超越標準的 Transformer,顯示現有模型在「內容基礎推理」(content-based reasoning)能力上的欠缺。

在此背景下,來自 Gu Dao 等人的論文《Mamba: Linear-Time Sequence Modeling with Selective State Spaces》(ICLR 2024,獲得 Outstanding Paper)提出了一套創新性框架 Mamba,旨在同時兼顧效率與性能,特別關注序列模型在處理長序列、離散模態(如語言)時的資訊選擇與遺忘機制,並大幅改進現有 SSM 架構的限制。

核心方法與創新

Mamba 的核心設計理念是基於「選擇性狀態空間模型」(Selective State Spaces),其主要創新包括:

  1. 讓狀態空間模型參數成為輸入的函數:
    傳統 SSM 透過一組固定參數定義狀態轉移,但這種固定參數在處理離散且多變的序列時(如文字),往往無法靈活反映當前輸入的內容。Mamba 採用輸入相關的函數動態決定 SSM 參數,意味模型可根據當前 token 動態選擇是否「保留」或「遺忘」先前隱藏狀態資訊,增強了模型的內容敏銳度與調節能力。這種設計使得序列中長期依賴的記憶以更有策略且資訊相關的方式被維持或更新,提升了內容基礎的推理能力。
  2. 硬體感知(hardware-aware)的平行演算法設計:
    雖然讓參數動態依賴於輸入破壞了可直接使用高效卷積計算的便利,但作者設計了一套在遞迴模式下可充分發揮現代硬體(如 GPU)並行能力的算法,在不犧牲效率的前提下實現線性時空複雜度(O(N))。此舉在保持序列長度可線性放大的同時,提供了高吞吐量的推論速度,實現比 Transformer 還快 5 倍以上的效果。
  3. 簡化且有效的端到端架構設計:
    Mamba 網絡架構摒棄了典型 Transformer 中的注意力模塊,甚至不採用多層感知器(MLP)塊,全網絡核心即為選擇性 SSM 模組,有效降低了模型架構的複雜度。此簡化後的架構不僅利於理解與優化,也使得整體計算更為高效。

主要實驗結果

在多模態序列任務中,Mamba 展現出令人驚豔的性能和計算效率:

  • 語言建模任務上,作者訓練了 30 億參數規模的 Mamba-3B,在預訓練與下游評估階段均超越了同尺寸的 Transformer 模型,更與 60 億參數的雙倍尺寸 Transformer 比肩,顯示出極佳的參數效率及泛化能力。
  • 進一步實驗涵蓋了音訊及基因組學(genomics)等超長序列數據,Mamba 在序列長度上能擴展至百萬級別,且推論速度線性增長,解決了傳統 Transformer 在處理超長序列時因計算資源爆炸而難以實用的痛點。
  • 與現有子二次時間架構相比,Mamba 不僅在效能評比中取得領先,同時在語言等離散模態的內容理解與推理方面,有顯著改善,證明選擇性狀態轉移的策略有效彌補了過去 SSMs 及類似模型在「內容感知」上的缺憾。

對 AI 領域的深遠影響

Mamba 論文為序列模型設計提供了重要的新方向,特別在於:

  1. 融合靈活參數與高效計算的策略創新:
    Mamba 展示了動態參數化如何顛覆傳統狀態空間模型僵化的結構,為長序列建模引入了更多「內容選擇」的元素,這種設計理念可被拓展到更多模型架構,促進序列理解向更為靈活且精細的推理能力演化。
  2. 提升超長序列處理的可行性:
    透過硬體感知的遞迴平行算法使得超長序列在理論與實務上均可高效運算,這對於基因組資料分析、語言長文分析、以及連續音訊流的即時處理,乃至未來多媒態序列統合學習,都提供了關鍵的技術支撐。
  3. 開啟序列模型結構簡化的新篇章:
    Mamba 的成功證明,去除注意力機制和 MLP 結構並不必然犧牲模型性能與泛化能力,有助於引導後續研究探索更具計算效率、結構簡潔且理論基礎扎實的序列模型。

綜合來說,Mamba 以其創新性的 Selective State Spaces 模型架構,成功平衡了速度與性能,且克服了長序列及離散模態建模的難題,為未來大規模序列模型的設計提供了啟示,並可能推動多領域基礎模型的突破。本論文獲得 ICLR 2024 Outstanding Paper 獎項,足以見得其在學術界與實務領域的雙重產出與潛力。

對於具備基礎 AI 知識的工程師與研究生而言,深入理解 Mamba 的方法、實現及優化策略,將有助於開發更高效且適用於超長序列的序列模型,拓展在語言理解、音訊處理與生命科學領域的人工智慧應用。


論文資訊
📄 Mamba: Linear-Time Sequence Modeling with Selective State Spaces
👥 Gu, Dao
🏆 ICLR 2024 · Outstanding Paper
🔗 arxiv.org/abs/2312.00752

Learning Interactive Real-World Simulators

在人工智慧領域中,模擬器(simulator)扮演了極為重要的角色,特別是在強化學習(Reinforcement Learning, RL)與機器人學的應用上。模擬器能夠提供可控、可重複且成本低廉的環境,讓智慧代理人得以在虛擬環境中學習操作策略,避免了直接在真實世界中漫無目的地嘗試可能導致代價昂貴或危險的行為。然而,傳統模擬器通常需要專門開發或高度工程設計,不僅耗時費力,亦無法完全真實反映真實環境的複雜交互與不確定性。

本篇由 Yang 等人於 2024 年 ICLR 會議發表並榮獲 Outstanding Paper 獎的論文《Learning Interactive Real-World Simulators》,針對現行模擬器的制約提出嶄新解決方案,旨在利用數據驅動的方式,從真實世界的交互數據中學習建立高保真度且可交互的模擬器,讓模擬器不僅能準確還原真實世界的狀態變化,還能對代理人的行動作出合理回應,從而極大提升下游強化學習代理人的泛化與效能。

一、研究背景與動機

傳統模擬器設計往往依賴物理模型、幾何計算或人為規則,這類模擬器在模擬速度和準確性間常存在權衡,且難以涵蓋真實環境中豐富而複雜的交互行為。例如,工業自動化、生物醫療道具操作或仿生機械人系統常涉及多樣且細緻的動作,純粹基於物理模型的模擬器往往難以穩健重現。另一方面,雖生成式模型如 Diffusion Models 或者 Transformers 在視覺或語言模擬上已展露強大能力,但其在動態交互模擬和行動依賴性預測上的應用仍受限。

因此,本文研究團隊的動機在於:如何從收集自真實世界的交互數據中,學習一種能夠動態反映環境狀態同時回應代理人採取不同行動的高階互動模擬器?理想的模擬器應具備以下特性:(1)高準確度,能有效還原環境在不同時刻的狀態;(2)互動性強,能根據不同動作給出合理的環境反饋;(3)泛化能力,能適應未見過的環境變數或動作組合;(4)高效率,適合用於強化學習訓練。

二、核心方法與創新技術

作者提出一套基於神經網絡結構的互動式模擬器學習框架,整合了序列建模、動作條件化與擴散模型(Diffusion Model)三大技術優勢:

  • 狀態-動作條件化的序列模型:針對環境狀態和代理行動的時間序列數據,利用 Transformer 模型或類似架構捕捉長距離時間依賴關係,不僅預測未來狀態,也將當前動作作為條件資訊輸入,使得模型能夠根據不同操作動作預測環境的回應。
  • 擴散模型與生成機制:受到圖像生成領域擴散模型的啟發,作者將擴散過程嵌入動態狀態預測中以強化數據的表示能力與生成質量,更准確地模擬真實世界中複雜的隨機性與不確定因素,減少模型過度平滑或模式崩潰問題。
  • 互動性訓練與自我對抗策略:透過模擬器與代理人策略同時動態更新,讓模擬器不斷應對新策略帶來的多樣行為,避免模型陷入對固定策略的過擬合。更進一步,研究團隊引入自我對抗學習,促使模擬器在挑戰性行動下依舊保持穩定的模擬輸出。

此外,作者還設計了一套創新的數據蒐集和標註流程,利用多感測器融合技術,收集多種真實世界環境下的交互數據,涵蓋物理接觸、力反饋、運動軌跡等豐富特徵,從根本上保證訓練資料對多樣環境場景的代表性與完整性。

三、主要實驗結果

為驗證方法效能,論文針對多個應用場景進行嚴謹評估。實驗涵蓋工業機械臂操控、仿生機器人步態生成以及日常物理環境交互等真實數據集:

  • 準確性與真實感:所學模擬器在狀態重建誤差、動作反饋的一致性及未見場景的推論表現均超越現有基於物理模型或純深度學習方法,展現卓越的真實還原能力。
  • 強化學習下游任務:將該模擬器用於訓練強化學習智能體,智能體在學習效率和最終策略性能上,都相比傳統模擬器(如 MuJoCo、Bullet)取得顯著提升,尤其在環境變異和動作干擾下更展現更強的魯棒性與泛化力。
  • 模擬器互動性驗證:透過多樣化動作組合和極端操作測試,模擬器能持續給出穩定且合理的環境反饋,顯示出高度互動性和動態適應能力,不再只是簡單的狀態預測器。

四、對 AI 領域的深遠影響

這項工作對人工智慧尤其是強化學習、機器人以及模擬技術領域具有深遠的意義:

  • 提高模擬器的可用性與真實性:該方法通過資料驅動方式構建互動模擬器,降低了對於專業物理知識和復雜建模的依賴,使更多場景能快速擁有高質量模擬環境,促進跨領域創新應用。
  • 促進強化學習泛化能力的提升:藉由更真實且互動性強的模擬,智能體能在多變且非結構化環境中獲得更穩定的策略學習,提升實際部署時的安全性與效能,縮短從模擬到真實部署(sim-to-real)的鴻溝。
  • 推動多模態及因果建模的未來方向:該研究所采用的擴散模型與互動序列建模架構開創了具備因果推理能力的模擬器設計新思路,有助於未來結合多模態感知、因果分析的方法開發更高階的智慧系統。
  • 強化人機合作與智能系統開發:能模擬真實複雜場景的交互模擬器,對於人機協作平台、智慧製造以及代理人自主決策等領域都將帶來實質性的技術推進。

總結而言,《Learning Interactive Real-World Simulators》一文成功整合最先進的深度學習生成技術和動態互動建模策略,實現了從真實數據學習高保真且具有豐富交互能力的模擬器,為人工智慧系統在真實世界的應用奠定了堅實基礎。隨著未來數據取得更加便利與模型架構的演進,這類互動式模擬器勢必成為推動智能體自主學習與提升泛化能力的關鍵工具。


論文資訊
📄 Learning Interactive Real-World Simulators
👥 Yang, Du, Ghasemipour, Tompson, Kaelbling, Schuurmans, Abbeel
🏆 ICLR 2024 · Outstanding Paper
🔗 arxiv.org/abs/2310.06114

Generalization in Diffusion Models Arises from Geometry-Adaptive Harmonic Representations

隨著生成式人工智慧技術的迅速發展,擴散模型(Diffusion Models)已成為現今最具代表性的生成架構之一,特別在影像及音訊等多媒體領域展現出令人驚嘆的生成品質。然而,這類模型在理論基礎與泛化能力上的理解仍有待深化,尤其是為何在訓練資料有限的情況下,能有效地學習到豐富多樣的資料分布,並產生高度逼真的合成樣本,這是目前研究社群極力探討的核心問題。

本篇於 ICLR 2024 獲得 Outstanding Paper 獎項的論文《Generalization in Diffusion Models Arises from Geometry-Adaptive Harmonic Representations》,由Kadkhodaie等人所提出,其創新地從數學與幾何的角度切入,揭示擴散模型泛化能力與其隱含的「幾何適應性調和表示」(geometry-adaptive harmonic representations) 之間的深刻關聯。此論文不僅在理論上提出跨時代的見解,也為後續架構優化及訓練策略提供了全新指引。

研究背景與動機

傳統生成模型如GAN(生成對抗網絡)及變分自編碼器(VAE)在建模資料分布上各有優缺,但往往面臨模式崩潰(mode collapse)或生成多樣性不足等問題。擴散模型作為新興技術,藉由逐步加入噪聲並反向逐步還原數據,透過馬爾可夫鏈或連續時間的隨機微分方程(SDE),實現了高品質且多樣化的生成效果。

然而,目前擴散模型泛化能力的理論解釋仍然薄弱,尤其在數據維度極高、結構複雜的情境下,更缺乏明確的數學模型來描述其學習到的表示空間。Kadkhodaie等人察覺,擴散模型成功之處或許與其隱式學習到的調和分析(Harmonic Analysis) 手法有關,特別是模型如何根據資料內在幾何結構調整其頻譜成分,從而實現對資料分布的高效逼近和泛化。

核心方法與創新

本論文的核心理論基礎在於「幾何適應性調和表示」,作者將擴散過程中的資料分布與其頻譜特性相結合,指出擴散模型其實學習了一組與數據幾何相匹配的調和基底函數。這些基底並非固定不變的傅立葉基,而是隨資料的幾何形狀而動態調整,使得模型表達更加靈活且更有效提取數據關鍵訊息。

具體而言,作者提出利用調和分析與微分幾何工具,分析擴散過程中資料的特徵頻譜如何隨時間演進,並證明此過程使得模型可以在低維嵌入空間中尋找最優基底,從而在保留高頻細節的前提下,兼具泛化性與表達多樣性。這與傳統深度學習中利用固定基底提取特徵的做法截然不同,強調模型必須根據資料幾何動態調整其表示空間。

此外,論文還引入數值模擬與實驗證明,基於該理論設計的擴散模型在不同數據集(如 CIFAR-10、ImageNet 等)上,不僅在樣本質量與多樣性上表現優越,亦展現出更強的泛化能力與魯棒性,尤其在噪聲及異常樣本下仍能保持較佳的生成效果。

主要實驗結果

作者在多項實驗中對比了傳統擴散模型與基於「幾何適應性調和表示」理論設計的改良模型,結果突出表明:

  • 在標準生成任務中,新模型在FID與IS指標上明顯優於基線,生成影像更具解析度與細節層次。
  • 透過頻域分析與能譜分佈可視化,驗證模型確實學得符合資料本身幾何結構的頻域表示,有效避免過度擬合低頻或高頻噪聲成分。
  • 模型在數據分布遷移、樣本不足等挑戰性場景下展現更穩健的泛化性能,支持理論推導中的張量分解與幾何調整機制。
  • 對隨機噪聲注入強度調節的實驗,表明幾何適應性表示使得擴散模型能自動調整生成策略,展示出高度的適應性與靈活性。

對 AI 領域的深遠影響

此論文的最大貢獻是為擴散模型的理論基礎帶來突破性見解,成功從微分幾何與調和分析的視角,揭示了深度生成模型泛化的數學本質。這不僅解決了過去以經驗為主的訓練過程中「黑盒」般的疑惑,還為設計更穩健、有效率且具有高度泛化能力的生成架構提供了理論工具。

具體而言,該研究成果可望推動以下方向的發展:

  • 生成模型的結構設計:透過幾何適應性調和表示原理,未來可設計出更加自適應且解釋性強的模型架構,減少過度訓練及資料需求,提高樣本效率。
  • 跨模態生成與多樣本學習:由於不同資料類型本質上的幾何結構差異,該理論或可幫助建構跨模態的統一生成框架,增強多任務與多領域的泛化能力。
  • 強化理論與實務連結:透過嚴謹的數學分析,連結生成模型的動態訓練過程與隱含頻域結構,促進基礎理論與應用系統的協同發展。
  • 拓展其他隨機過程生成策略:理論架構可引導對其他基於隨機過程的生成技術(如跳躍擴散、非馬爾可夫鏈過程)的深入理解與優化。

總結來說,Kadkhodaie等人的這篇論文,通過結合數學理論與實證實驗,成功破解了擴散模型泛化背後的核心機制,為生成式 AI 領域注入了新一輪理論革命的動能。對於研究生與工程師而言,理解與掌握這套「幾何適應性調和表示」的理論,不僅能深化對現有生成模型的認識,更能啟發未來原始創新的模型設計思路,推動人工智慧生成技術進入更高的統合理論與應用層級。


論文資訊
📄 Generalization in Diffusion Models Arises from Geometry-Adaptive Harmonic Representations
👥 Kadkhodaie, Guth, Simoncelli, Mallat
🏆 ICLR 2024 · Outstanding Paper
🔗 arxiv.org/abs/2310.02557

2026年6月26日 星期五

Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution

2024 年 ICML 大會中,Lou、Meng 與 Ermon 發表的論文《Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution》榮獲最佳論文獎,這篇作品在離散型擴散模型(discrete diffusion models)的建模策略上提出了革命性的見解,對生成模型的理論與實踐皆帶來顯著突破。本文將針對該論文進行深入探討,內容包含研究背景與動機、核心方法與技術創新、關鍵實驗結果,以及這項研究對 AI 領域的深遠影響。

研究背景與動機

近年來,擴散模型(Diffusion Models)因其在圖像生成、語音合成乃至文本生成等多模態領域的優異表現,成為生成模型研究的重要方向。傳統擴散模型多基於連續空間之 Gaussian 過程設計,配合時間反演過程,逐漸恢復數據分布,成功取得了高質量且多樣化的生成效果。

然而,當處理離散數據(例如自然語言、分類資料或符號型結構資料)時,連續擴散模型的假設與離散空間本質不符,導致擴散過程和逆擴散過程難以有效建模,限制了其應用性能與擴充性。以往離散擴散模型常見的方法是直接在離散狀態空間設計轉移矩陣,或以去噪機率估計為技術核心,但這些方法往往難以達到理想的穩定性和生成精度。

基於此背景,作者發現:透過估計數據分布在不同狀態點間的「比率」(ratios)可以更有效地描述離散擴散過程,並從中推導出逆向擴散的精確計算架構。這種基礎的概率結構理解重新構築了離散擴散模型的數學框架,突破了以往方法在穩定性及效果上的瓶頸,是此論文研究的主要動機。

核心方法與技術創新

本論文的核心創新在於提出一種「比率估計」方法,專門用來處理離散擴散過程中數據分布的建模。具體而言,作者將離散擴散過程的逆向機率分布轉化為數據分布在不同狀態間轉移概率的比值表示,這使得建模問題可被拆解為估計概率分布比率的問題。

  • 比率的概率視角: 通常逆擴散過程需直接估計一些複雜的條件概率分布,但作者巧妙地將這些條件分布寫成分布比率形式,理論上可用更簡單的判別模型直接逼近。
  • 模型架構設計: 論文引入一種基於神經網路的比率估計器,此估計器在訓練階段學習鑑別數據及擾動分布的相對比率,降低了計算複雜度且增強了模型的泛化能力。
  • 理論保障: 論文提供了嚴謹的數學證明,證明比率估計方法在統計層面上的無偏性與一致性,確保該方法能復原目標分布,並且在有限樣本下擁有良好的近似誤差界。

整體來說,該方法用一種嶄新的角度重構了離散擴散模型,不僅提高了逆擴散過程的可學性,也促使模型在多種離散生成任務中實現更精準且穩定的表現。

主要實驗結果

為驗證比率估計離散擴散模型的性能,作者在多個標準基準數據集上進行實驗,涵蓋文本生成、離散圖像(如 MNIST 與 CIFAR-10 的量化版本)、以及蛋白質序列建模等應用場景。

  • 生成質量優越: 新方法在多個任務中顯著超越了傳統離散擴散模型(例如基於去噪損失的模型)以及其他主流離散生成模型(如自回歸模型、變分自編碼器等)的生成質量,生成樣本在人類主觀評價及自動評估指標(如 BLEU、FID)均有大幅提升。
  • 訓練穩定與效率提升: 試驗證明,比率估計大幅降低了逆擴散過程中概率估計的不確定性,穩定了訓練流程,縮短了收斂時間,相較之下,傳統離散擴散模型往往因為不穩定的最大似然估計而陷入困境。
  • 泛化能力與數據稀疏: 在數據分布稀疏與噪聲較重的設置下,比率估計方法仍保持優秀表現,顯現出良好的泛化性與抗噪能力,這在真實世界應用中極具價值。

這些實驗結果明確展現了新方法在離散空間生成建模領域的強大競爭力及實用性。

對 AI 領域的深遠影響

本論文的貢獻不僅在於提出一個技術上優秀的離散擴散模型,更深刻改變了我們如何從概率角度理解與設計離散生成過程:

  1. 開啟離散生成模型的新視角: 傳統方法多著重於估計每個狀態的概率分布,而本論文強調狀態間的 分布比率,這種觀念有望影響未來更多生成模型的設計理念,促使研究者探索更多「相對位置」的概率建模技巧。
  2. 促進跨模態的生成技術融合: 離散擴散模型過去局限性大,無法有效涵蓋自然語言、符號資料等領域。此方法有效擴展了離散擴散理論基礎,有助於未來在跨模態、多樣性的應用中開拓更多可能,例如結合語言和結構化數據的生成。
  3. 理論與應用的橋樑: 論文嚴謹的理論證明與實驗成果,使得離散擴散模型從純粹理論研究,邁向可實際部署於工業環境,對多種 AI 產品的生成能力提升將帶來直接助益。
  4. 啟發多領域新課題: 比率估計的統計方法可擴展至其他概率模型的訓練問題,未來不僅限於生成模型,還可影響強化學習、因果推斷、貝葉斯建模等領域中的分布估計與策略優化工作。

總體而言,Lou 等人的工作對離散生成模型理論與實踐的雙重推動,無疑將成為未來離散數據處理與生成技術研究的重要基石,預計將在 AI 生成模型的下一代技術路線中扮演關鍵角色。

綜上所述,《Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution》不僅突破了離散擴散模型的技術瓶頸,更為生成模型領域提供了嶄新的理論工具與實踐思路,值得 AI 研究者與工程師深入研讀與實踐。


論文資訊
📄 Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution
👥 Lou, Meng, Ermon
🏆 ICML 2024 · Best Paper
🔗 arxiv.org/abs/2310.16834

Measure Dataset Diversity, Don't Just Claim It - ICML 2024 最佳論文深度解析

在現代機器學習(ML)研究中,數據集不僅是訓練模型的基石,更承載著抽象且複雜的社會意涵。研究者常用「多樣性」(diversity)來標榜數據集的價值,但這些價值導向的詞彙背後,往往缺乏嚴謹的定義與科學化的量化方法。來自ICML 2024的最佳論文《Measure Dataset Diversity, Don't Just Claim It》由Zhao等人提出,正是針對這個普遍但被忽略的問題進行深入探討,從社會科學的測量理論(measurement theory)出發,為機器學習社群帶來了量化數據集多樣性的一套嶄新框架與方法。

研究背景與動機

在機器學習領域,數據集被視為中立且客觀的資源。不過,隨著AI應用越來越廣泛,社會科學家和倫理學者警告,我們的數據集實際上蘊含了偏見(bias)、刻板印象與權力結構,這會對後續模型的公平性和效能造成重大影響。設計和挑選數據集時,「多樣性」這個詞被反覆強調,認為多樣的數據能幫助模型學到更多樣且全面的特徵,減少偏誤。然而,多樣性在不同文獻與領域中經常被模糊且隨意地引用,甚至常以主觀判斷作為基準,缺少客觀、可重複且標準化的測量方式。

本論文團隊發現在135個涵蓋影像與文本的主流數據集中,多樣性被頻繁提及,但背後的定義與操作流程千差萬別,結果導致研究成果間難以比較,也增加了模型偏誤與錯誤判斷的風險。因此,他們提出重新審視並建構一套針對「多樣性」概念在數據集中的量化流程,目的是促使整個機器學習領域對「多樣性」有清晰、一致且可驗證的理解。

核心方法與創新

本論文的最大創新在於將來自社會科學的測量理論工具引入機器學習數據集評估,提供一個理論基礎與實際框架。作者從三個層面重構多樣性的測量:

  • 概念化(Conceptualization): 定義多樣性包含哪些維度,例如族群(demographic)、語言、視覺特徵等,並明確界定多樣性的文化與上下文意涵。
  • 操作化(Operationalization): 把抽象的多樣性轉換成可量化的指標,例如使用統計分布差異、資訊熵、特徵空間的覆蓋率等。
  • 評估量表(Measurement Evaluation): 驗證所設計的指標是否具有信度(reliability)與效度(validity),確保測量結果可信且與實際多樣性概念相符。

為了驗證這套方法,作者從135個廣泛使用的影像與文本數據集中收集數據,並以此調查與比較過去文獻中所聲稱的多樣性指標與數據。實驗不僅揭露了很多數據集標榜多樣性卻未用明確指標支撐的現象,也展示了其所提出的量化方案如何揭露實際的多樣性結構,並能捕捉不同數據集間微妙的差異。

主要實驗結果

經過嚴謹的實證分析,論文展示了:

  • 多數當前數據集宣稱的多樣性缺乏客觀量化支撐,導致其使用與後續研究中潛藏偏誤。
  • 新提出的測量指標能有效區分數據集之間的真實多樣性差異,並提供一個標準化評估工具。
  • 透過多樣性指標,研究者可進行更精準的數據集篩選與調整,提升下游模型的公平性與泛化能力。
  • 量化的多樣性度量還有助於揭露數據集欠缺的方面,如某些族群或特徵在數據中被低估或忽略。

此外,作者也提出了一系列具體建議,供未來數據集構建者參考,包括明確定義多樣性目標、採用多維度量測指標以及在數據發布時公開相關指標與分析,促進整個社群朝向責任更明確、可靠的數據資源發展。

對 AI 領域的深遠影響

這篇論文的意義不僅在於提出具體的量測工具,更在於提醒AI研究者和實務者,數據集並非中立,背後藏有價值判斷與社會意涵。若沒有明確且科學的量化方式,對「多樣性」的輕率宣稱不僅無助於技術進步,還可能造成誤導、資源浪費,甚至加劇AI系統中的不公平問題。

透過本研究,機器學習與AI研究社群可以:

  • 建立標準化的數據集多樣性評估指標,提升跨領域與跨研究的可比性與透明度。
  • 內化社會科學的測量理論觀點,促進跨學科合作,促使數據科學更具社會責任感。
  • 在數據收集、整理和釋出階段即開始重視多樣性質量,提升訓練模型的公平性與魯棒性。
  • 推動以數據為本的公平機器學習,避免僅以主觀認知做出偏頗判斷。

結語而言,《Measure Dataset Diversity, Don't Just Claim It》不僅是一篇技術創新論文,更是一篇具社會意義與倫理啟示的關鍵作品。它呼籲我們從理論與實務兩端重新正視數據集多樣性問題,建立起基於科學量測的規範體系,確保未來AI系統是在真正多樣且公平的基礎上茁壯與發展。對於工程師和研究生而言,掌握本論文提出的思考架構與評估工具,將有助於設計出更貼近現實需求且具備長遠價值的AI解決方案。


論文資訊
📄 Measure Dataset Diversity, Don't Just Claim It
👥 Zhao, Vasu, Bhatt, Herrmann, Fowlkes
🏆 ICML 2024 · Best Paper
🔗 arxiv.org/abs/2407.08188

Position: Considerations for Differentially Private Learning with Large-Scale Public Pretraining

隨著資料隱私議題日益重要,差分隱私(Differential Privacy,DP)已成為保護使用者數據的黃金標準。尤其在機器學習領域,如何在保障個人隱私同時維持模型效能,是一項長期且艱鉅的挑戰。近年來,一種主流策略是透過大型公開資料集預先訓練模型,再利用這些非私人預訓練模型進行微調(fine-tuning),結合差分隱私訓練於特定敏感資料上,以提升私有模型的準確度與效能。此策略藉由轉移學習(transfer learning)針對預訓練階段和差分隱私微調階段的分工,減少私有資料暴露,理論上能有效緩解差分隱私訓練時的性能下降問題。然而,Tramèr、Kamath 與 Carlini 於 ICML 2024 所發表榮獲最佳論文的《Position: Considerations for Differentially Private Learning with Large-Scale Public Pretraining》一文,對這個策略提出了深入且全面的質疑與省思,並帶來關鍵的社會與技術層面洞見。

研究背景與動機

過去在差分隱私機器學習的研究中,模型性能往往因添加隱私保護機制(如噪音注入、梯度裁剪)而大幅下降。隨著大規模公共資料集(例如從網路爬取的大量文本與圖片)日益豐富,研究者開始利用在這些非私人資料上預訓練的巨大模型,再在私有的敏感資料上套用差分隱私微調,以兼顧隱私與性能。這種方法看似合理且成效顯著,因為預訓練階段不涉及私有資料,利用公共資料學到的通用特徵能有效幫助後續的差分隱私訓練。

然而,本論文團隊質疑:這種公開預訓練資料集的使用,是否真能算是「私密」的?當我們聲稱基於公開網路數據預訓練的模型之後再做差分隱私微調,是隱私安全的保障,這樣的說法是否過於樂觀?更深層次的問題還包括,公開大規模預訓練模型對敏感領域的遷移效果是否充分、模型尺寸與私有資料的處理架構對隱私保護的實際影響,這些都尚待探討。

核心方法與創新

本論文的核心贡献並非提出全新算法,而是採用“立場論文”和系統性回顧的方式,從隱私理論、實務部署與社會倫理三個面向對利用大型公開預訓練模型結合差分隱私學習的方法做出批判性分析。其主要創新點包括:

  • 隱私定義的重新檢視:作者指出,透過「大型網路爬蟲資料」來預訓練模型,這些資料本身已不具私密性,且含含糊不清的版權與授權課題。當宣稱整體模型是“差分隱私保護”時,忽略了預訓練階段資訊公開的風險,可能導致誤導公眾對差分隱私保障的信任。
  • 資料分布與效用差異的深入探討:論文檢視了現有的機器學習基準(benchmarks)能否完整代表敏感應用領域的數據分布。因大型公開資料多為網路爬取,某些敏感或低資源領域(如醫療、司法資料)極可能在預訓練資料中嚴重欠缺,導致模型的泛化效果有限,質疑轉移學習真能提升敏感場域的私有模型效用。
  • 計算架構與部署模式的隱私反思:大規模模型尤其在大小超過數十億參數時,使用者難以於本地設備部署,只能委託第三方伺服器執行推論與微調。有鑑於此,私有資料在傳輸和處理階段仍需外包,實際上可能導致隱私風險反增,與差分隱私目的相悖。

主要實驗結果

作者在論文中結合文獻回顧、數據分析與理論解析,而非以單一實驗為主軸。通過回顧多個公開資料集及預訓練模型,並對比非常見敏感領域數據的分布差異,發現大型公開資料集與敏感資料存在顯著域差,使得預訓練模型在某些隱私重要場景中的預期性能提升不足。此外,通過計算資源與部署場景的分析,指出了大型模型不易本地部署的現實瓶頸。

對 AI 領域的深遠影響

此論文的意義遠超過技術層面,對 AI 研究與產業實踐帶來多面向的警示與啟發:

  1. 重新定位差分隱私定義及其應用範圍:論文強調,差分隱私的保障不僅是數學上的噪音注入,更應考慮整個預訓練與使用流程的數據來源與數據層級隱私。未全面考量公開資料風險,容易對外宣稱“隱私保護”造成誤解,傷害公眾信任,誘發商業倫理與法規問題。
  2. 促進更貼近敏感場景的資料收集與標準制定:論文呼籲社群與產業界重視敏感領域本身的數據特性,不應僅以已有大型公共資料替代,而應設計更嚴謹、更貼近隱私需求的數據集和評測標準,以確保研究成果能真實惠及醫療、金融等關鍵場域。
  3. 推動分布式與本地化的隱私保護技術發展:面對大型模型部署限制與私有資料不得不外包的問題,論文指出未來差分隱私學習必須結合硬體創新與去中心化計算架構(如聯邦學習、邊緣計算),避免將私密資料暴露於第三方,提升整體隱私保障。
  4. 呼籲透明公開大規模數據來源與訓練過程:網路爬蟲資料的法律、倫理爭議與數據質量已是業界痛點,公開預訓練的數據來源及其隱私風險評估,有助於建立社會對 AI 技術的信賴與理解,更有利於推動負責任的 AI 發展。

總結而言,這篇 ICML 2024 論文以務實且批判性的態度,重新審視了結合大型公用預訓練與差分隱私學習的現狀與未來挑戰。它提醒研究者與工程師,在追求技術突破與性能提升時,勿忽視背後深刻的隱私、倫理與社會責任問題。透過更嚴謹的數據治理與多元技術結合,才有可能真正實現差分隱私在現實世界中安全且有效的應用,亦為 AI 領域的長遠發展奠定更扎實基礎。


論文資訊
📄 Position: Considerations for Differentially Private Learning with Large-Scale Public Pretraining
👥 Tramèr, Kamath, Carlini
🏆 ICML 2024 · Best Paper
🔗 arxiv.org/abs/2212.06470

2026年6月25日 星期四

Guiding a Diffusion Model with a Bad Version of Itself:用「劣版自我」引導擴散模型的創新之路

近年來,擴散模型(Diffusion Models)在圖像生成領域掀起革命,成為繼 GAN 之後生成式模型的主流之一。其除了具有極高的圖像品質,還能透過條件控制(如文字提示、類別標籤)生成多樣化且高擬真的影像。然而,影像品質、生成變異度(diversity)以及條件對齊度(condition alignment)這三者往往相互糾結,難以同時優化。即便是現有的熱門技術——無分類器指導(classifier-free guidance)——在改善品質與提示對齊度的同時,也以犧牲多樣性為代價。此現象深刻地反映出這三個維度間的本質衝突。

本篇由 Karras 等人於 NeurIPS 2024 發表的論文《Guiding a Diffusion Model with a Bad Version of Itself》突破上述困境,提出一項出乎意料但極具成效的新穎指導機制:不再使用無條件模型(unconditional model)來引導條件模型,而改用「該模型的較小、訓練較淺的版本」即「劣版自我」進行指導。這種方法成功打破品質與多樣性間的糾結,帶來更細緻的分離控制,並在 ImageNet 生成任務中創下新高的 FID 紀錄,展現其在無條件及條件擴散模型上的廣泛適用性與卓越性能。

研究背景與動機

擴散模型運作機制基於反覆對數據逐步加噪與去噪,進而學習逆向生成過程。模型可依據條件資訊生成對應影像,使其在文本到影像、圖像超解像等多種任務中獲得優異表現。但在實務使用時,我們面臨三大核心指標:品質(Quality)、變異度與條件一致性。這三者間存在天然張力──提升條件對齊度常常伴隨生成多樣性明顯下降,而單純追求多樣性則可能拖慢品質成長。

無分類器指導是目前最廣泛使用的技術,其方式為將條件擴散模型的預測與對應的無條件模型預測結合,做為加強條件效應的導數訊號。此方法讓生成影像更加符合提示內容並提升品質,但不可避免會縮減影像變異度,尤其在使用較強指導權重時更是明顯。研究者亟欲找到一種可解耦提升品質與保留多樣性的技術,減少三者指標間的衝突。

核心方法與創新

本論文的核心突破在於:以「自我劣版」作為指導模型。具體來說,作者將同一擴散模型的「較小容量或較少訓練步數」的子集版本拿來作為引導信號來源,取代傳統上使用的完全無條件模型。這個「劣版模型」因為容量較小或訓練不足,生成品質自然較低,但其生成分佈依然和完整模型保持一定程度的接近,且保有穩定的多樣性特徵。

理論上,這種指導方式具有雙重優勢:

  • 「劣版自我」能緩和無條件模型與條件模型的分佈差異,避免指導過於強烈而壓縮多樣性。
  • 利用內部模型家族的相似性,促使條件模型在品質提升的同時保持多樣化,不因指導信號造成分佈偏移過大。

方法實作層面,研究團隊透過在同一訓練過程中保存中途模型快照,或者訓練容量較低的子網絡作為劣版引導模型,在每一步噪聲反轉(denoising step)中雙向結合這兩個模型的預測,形成最終生成。這種機制不僅容易整合入現有架構,也不需求額外外部模型,保持相對輕量與效率。

主要實驗結果

研究團隊在經典的 ImageNet 64x64 與 512x512 解析度生成任務上做詳細評測,顯著提升了模型生成的 FID 分數 (Fréchet Inception Distance) :

  • 64x64 解析度下取得驚人的 1.01 FID,刷新公開數據與模型基準紀錄。
  • 512x512 解析度則達到 1.25 FID,同樣領先於同期甚至過去頂尖方法。

除了條件生成任務,該方法亦驗證於無條件擴散模型上,證明「自我劣版引導」在提升整體生成品質方面,效果強大且一致。這項突破挑戰了先前普遍認為無分類器指導只能在變異性與品質間做權衡的既有觀念。

更重要的是,透過系統性實驗,論文展示了該方法在調控指導強度、模型大小與訓練深度的靈活性與穩定性,提供未來設計更細緻「解耦控制」的理論和技術基礎。

對 AI 領域的深遠影響

此研究不僅對擴散模型的理論研究提供新視角,也為實務應用提供革新工具。藉由引入「劣版自我」引導策略,模型開發者能在提升生成影像真實感與提示對齊精度的同時,保有豐富多樣的生成結果,避免過度模式崩潰(mode collapse)或生成單一化問題。

這對多樣性極為重要的應用場景,如藝術創作、遊戲設計、多樣化數據擴增等,會帶來極大價值。尤其在精細掌控生成過程、調節結果風格與內容的自訂需求日增下,本方法彰顯其靈活性與可調式特質。

此外,本論文強調了模型內部多階段、不同訓練程度的子模型協同作業可能開闢的新研究方向。未來研究或許能探索更多「內部級模型聯動」的策略,例如利用多尺度、多階段模型間的差異引導,以提升生成質量與控制性能。這種創新思路可能也適用於其他生成模型架構,甚至跨足語言模型、強化學習等領域。

總結來說,《Guiding a Diffusion Model with a Bad Version of Itself》 提出一種超越傳統無分類器指導的創新方法,解決了生成品質、多樣性與條件對齊的長期矛盾,刷新實驗指標,並開啟新穎且具廣泛潛力的研究視野。這項成就不僅巩固擴散模型在圖像生成領域的前沿地位,也為未來如何構建更靈巧與解耦的生成演算法提供關鍵啟發。


論文資訊
📄 Guiding a Diffusion Model with a Bad Version of Itself
👥 Karras, Aittala, Kynkäänniemi, Laine, Aila, Lehtinen
🏆 NeurIPS 2024 · Best Paper Runner-Up
🔗 arxiv.org/abs/2406.02507

Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Prediction 深度解析

隨著人工智慧在圖像生成領域的高速發展,如何兼顧生成品質、效率與模型可擴展性,成為現有方法亟待突破的瓶頸。傳統自回歸(Autoregressive, AR)模型通常以像素或圖像 token 的「逐點預測」方式進行,採用掃描線(raster-scan)策略來擬合圖像分布,這使得其生成速度較慢,且難以突破對高解析度圖像的生成限制。另一方面,擴散模型(Diffusion Models)經由逐步噪聲去除的機制,展現了優秀的圖像品質和生成多樣性,但在推理速度與資源消耗上不盡理想。

本篇由 Tian 等人發表於 NeurIPS 2024 與榮獲最佳論文的作品《Visual Autoregressive Modeling (VAR): Scalable Image Generation via Next-Scale Prediction》提出了一種全新的視覺自回歸生成範式,以「逐階段尺度預測(next-scale prediction)」取代傳統的逐 token 預測,重新定義 AR 模型在圖像生成中的學習方式。該方法兼具快速推理、卓越質量與良好可擴展性,不僅超越現有 AR 基準,甚至首次在多項指標上力壓擴散模型。

研究背景與動機

在自回歸圖像生成領域,多數經典模型如 PixelRNN、PixelCNN 皆以像素為序列元素,依序預測下一像素值。此舉雖然保證生成樣本的合理性與連續性,但極大限制了生成速度與可處理的圖像解析度。此外,近期蓬勃發展的擴散模型通過多階段噪音過程生成高品質圖像,但計算成本與推理延遲仍為實際應用的隱憂。

作者觀察到,自然圖像本身具有多層次的結構與解析度特性,若可改變自回歸生成順序,改從低解析度向高解析度循序細化,便可能兼顧生成效率與表現。因此他們提出將 AR 學習任務從「逐點預測」改為「逐尺度預測」,即在粗糙級別預測影像後,再依序生成更細節、解析度更高的影像,讓模型學習圖像跨尺度的生成分布,稱之為 Visual Autoregressive modeling (VAR)。

核心方法與創新點

VAR 的核心革新在於將自回歸框架從標準的圖像「掃描線像素預測」改為「多尺度圖像逐層細化」。具體來說,VAR 依序生成一系列漸進提升解析度的圖像層級,如 8×8 → 16×16 → 32×32 直到 256×256,模型在每個層級預測下一解析度的圖像內容,充分利用「低解析度提供全局結構、高解析度聚焦細節」的優勢。

此方法以 Transformer 作為基本架構,將輸入的低解析度圖像融入條件資訊,藉由 Transformer 強大的序列建模能力捕捉不同解析度間的關聯與細節呈現。與傳統 AR 模型必須長序列依次生成像素的方式不同,這種「次尺度生成」(next-scale prediction)大幅縮短生成序列長度,提升推理速度,減少模型負擔。

此外,VAR 同時保留了自回歸模型能夠精確擬合概率分佈的特性,具備良好的泛化能力及生成多樣性。論文中也提出特殊的訓練策略和結構設計以強化模型在尺度間的上下文利用,確保不同解析度層級之間的信息暢通。

主要實驗結果

實驗部分,作者在 ImageNet 256×256 資料集上,和現有的強勁基準(AR baseline 與擴散 Transformer,Diffusion Transformer (DiT))進行全面比較。結果顯示 VAR 將 Frechet Inception Distance (FID) 從 18.65 飛躍降低到 1.73,Inception Score (IS) 由 80.4 提升至 350.2,生成品質大幅躍升的同時推理速度約提升 20 倍,極大改善了效能瓶頸。

更重要的是,VAR 在圖像品質、數據效率、推理速度與模型擴展性等多個維度均優於當前最先進的擴散模型 DiT,展現了強勁的競爭力。作者更透過擴張 VAR 模型規模,發現類似大型語言模型(LLM)的尺度定律(scaling laws),模型效能隨參數量呈現清晰的冪律增長,且擁有負相關係數接近 -0.998 的極高線性關係,驗證 VAR 在擴大規模時的穩定與有效性。

此外,VAR 也展現了跨任務的零次學習(zero-shot generalization)能力,包括圖像修補(in-painting)、擴展(out-painting)以及圖像編輯等多項下游任務,不需額外微調即可達到優秀效果,這在視覺生成模型中屬於較罕見的通用能力,與 LLM 的零次學習能力類似。

對 AI 領域的深遠影響

VAR 的提出不僅刷新了自回歸圖像生成的認知框架,亦挑戰了以往擴散模型在圖像生成領域的統治地位。這個方法以一種直觀且結構化的 next-scale 預測機制,大幅改善了自回歸模型的速度與質量矛盾,為圖像生成技術的高效化與高質量化指出新的方向。

最值得關注的是,VAR 具備可擴展性與通用性,透過簡潔的多尺度訓練架構和 Transformer 設計,成功地將語言模型的兩大核心特性——規模定律與零次學習能力——引入視覺生成領域。未來隨著模型規模的進一步擴大,VAR 有望成為圖像生成和多模態 AI 發展的基石,推動跨領域的統一表徵學習。

同時,作者釋出完整模型與程式碼,促進學界與業界對 AR 與 VAR 模型在創作、視覺理解乃至交互應用的探索,減少依賴龐大計算資源的擴散模型,提升整體 AI 視覺生成系統的可及性與實用性。

總結來說,《Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Prediction》是一篇開創性的研究,突破了既有自回歸生成的限制,提出一種全新且高效的多尺度預測策略,既提升了生成品質,也大幅加快了推理速度。此成果在生成模型理論與實踐上皆具有重大意義,預示著未來視覺生成模型將進入一個更加高效且泛用的新時代。


論文資訊
📄 Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Prediction
👥 Tian, Jiang, Yuan, Peng, Wang
🏆 NeurIPS 2024 · Best Paper
🔗 arxiv.org/abs/2404.02905

2026年6月18日 星期四

KAN: Kolmogorov-Arnold Networks — 重新定義神經網路的激活函數結構

隨著深度學習逐步發展為人工智慧核心技術,傳統的多層感知器(Multi-Layer Perceptrons, MLPs)作為基本結構之一,在許多任務中表現傑出。然而,MLPs 基本架構中節點「神經元」上固定使用的激活函數,與其邊權重通常是線性標量的設計,存在結構性限制。近期劃時代的論文《KAN: Kolmogorov-Arnold Networks》由劉博士等團隊,發表於 ICLR 2024 中,提出了一個革命性的神經網路設計理念:Kolmogorov-Arnold Networks(簡稱 KANs),徹底改造了激活函數與權重參數的定義,顛覆傳統 MLP 界限,獲得了 Outstanding Paper 的殊榮。

研究背景與動機

Kolmogorov-Arnold 表示定理(Kolmogorov-Arnold representation theorem)在數學函數理論中扮演重要角色,該定理指出任意多變數連續函數,可拆解成多個一維連續函數的組合。受此啟發,傳統 MLP 由多層節點線性加權結合並通過非線性激活函數形成複雜模型,但在表達自由度上受限於固定激活形式,且權重是純標量。此設計在某些高階模型需求或解釋性任務上遇瓶頸。

本論文動機在於嘗試全面解放神經網路中激活函數與權重的結構性限制,以提升模型的精度、可解釋性與擴展性。研究者提出一個新方案:將神經網路中所有權重改由「一維的可學習函數」來表示,取代傳統線性數值權重,並直接置換節點激活函數策略,期望藉此開啟更豐富的函數表示空間和互動方式。

核心方法與創新

KANs 的最大創新點在於「無標量線性權重」的設計。具體做法是,每條邊上的權重不再是單一數值,而是一個可學習、可微分的一維函數,且利用平滑樣條(spline)來參數化此權重函數。平滑樣條能夠靈活調整函數形狀,包含非線性及局部特徵的豐富表現,更能捕捉數據中微妙多樣的關係。

從結構上來看,KAN 把傳統 MLP 的「節點激活函數」概念部分轉移到「邊緣函數」上,整體神經網路不再以「神經元激活」為中心,而是由加權函數串連。此設計直觀呼應 Kolmogorov-Arnold theorem 中「函數拆解為單變數函數加權和」的核心思想,因此得名 Kolmogorov-Arnold Networks。

技術細節方面,研究團隊以高階樣條函數搭配梯度下降優化整體網路參數。透過這種高度可塑的權重函數,KANs 在理論上擁有更強的近似能力與泛化潛力,同時保持端對端可訓練。相較於一般廣泛使用的 MLP,KANs 不需預設激活函數形式(例如 ReLU、Sigmoid),讓模型自主探索最適激活動態。

主要實驗結果

本論文展示多項實驗標竿,證明 KANs 在多種場景均優於傳統 MLPs:

  • 準確度優勢:在數據擬合任務上,體積較小的 KAN 模型能匹配甚至超越大規模 MLP 的表現。這表示 KAN 在參數效率上有明顯提升,有助於減少計算資源消耗。
  • 神經擴展規律(Neural Scaling Laws):理論與經驗均顯示,KANs 的性能改善隨網絡規模成長的速度快於傳統 MLP,預示在更大數據與模型規模下,KAN 可能帶來更顯著優化效果。
  • 偏微分方程(PDE)求解:在物理建模領域,KAN 可用來有效逼近複雜的 PDE 解算器,展現強大的函數近似能力,且在較小模型尺寸下達成準確解,顯示其對數學與物理問題建模的契合度。
  • 可解釋性與人機互動:KAN 的權重函數是明確的單變數函數,可視化直觀且易於理解。研究者示範 KAN 協助科學家以新視角「發現」數學與物理定律,將深度學習與專家智慧有效串連。

對 AI 領域的深遠影響

KANs 不只是形式上的架構革新,其背後開啟了「神經網路結構自由化」的新思維。傳統神經網路大幅倚賴固定激活函數和線性權重,KAN 就像是打開封印,讓所有邊緣權重成為更具表現力的函數,自動調整非線性結構,讓模型更靈活、更具適應力。

此舉對於下列方向具有深遠意義:

  • 模型泛化與效率:KAN 能用更少參數達成更佳準確度,降低了對龐大模型的依賴,適合資源有限邊緣運算或裝置部署。
  • 模型可解釋性提升:由於權重函數的可視化和可理解性,KAN 有潛力成為科學推理與符號發現的輔助工具,推動 AI 在基礎科學研究中的應用。
  • 理論基礎與架構設計:KAN 向數學定理致敬,借力定理數學上的深刻結果,展示如何將嚴謹數學理論導入神經網路架構設計,促進理論與實踐融合。
  • 激活函數的重新想像:KAN 挑戰了長期以來激活函數固定不變的傳統,未來有望激發更多創新結構與學習機制。

總結而言,KAN: Kolmogorov-Arnold Networks 論文提出了一套嶄新神經網路設計框架,徹底改變了激活函數和權重的角色定位,以可學習的一維函數取代線性數值權重,憑藉其理論根基與實驗效能,展現出高精度、強解釋力與良好擴展性。對於深度學習研究者與工程師來說,KAN 不僅提供了一個有效替代傳統 MLP 的方向,更打開未來深度學習模型設計的全新視野,值得投入更多探究與應用開發。


論文資訊
📄 KAN: Kolmogorov-Arnold Networks
👥 Liu, Wang, Vaidya, Ruehle, Halverson, Soljačić, Hou, Tegmark
🏆 ICLR 2024 · Outstanding Paper
🔗 arxiv.org/abs/2404.19756

Mamba: Linear-Time Sequence Modeling with Selective State Spaces

在當前深度學習領域中,序列建模(sequence modeling)是自然語言處理、語音識別、時間序列分析等多項任務的基石。隨著 Transformer 架構的普及,尤其是自注意力機制的廣泛應用,模型在捕捉長距離依賴關係上獲得了顯著提升,但其內在計算複雜度也隨序列長度呈二次成長,導致在處理超長序列時計算資源及記憶體消耗劇增。面對此一挑戰,如何在維持長距離建模效能之餘,實現線性時間複雜度的序列建模,是近年研究重要的焦點。

本篇由 Gu Dao 等人於 ICLR 2024 發表並榮獲 Outstanding Paper 獎項的論文《Mamba: Linear-Time Sequence Modeling with Selective State Spaces》提出了全新的思路與方法。作者從狀態空間模型(State Space Models, SSM)切入,利用「選擇性狀態空間」(Selective State Spaces)概念,成功設計出一種能以線性時間複雜度進行序列建模的架構 Mamba,不僅保有優異性能,且顯著降低運算負擔,具有高度工程實用性與理論創新價值。

一、研究背景與動機

傳統 RNN 及 LSTM 類模型因其循環結構天生支持序列數據處理,但在捕捉長距離依賴上仍受限,且訓練時存在難以並行化的瓶頸。Transformer 則依賴自注意力機制,能直接建模長距離依賴關係,並有效利用現代硬體的並行能力,性能與表現上有質的飛躍。惟其計算與記憶體需求為 O(N²),N為序列長度,極大限制了可處理序列的長度。

此外,近年興起的狀態空間模型(SSM)以其結合物理系統動態特性及隱式狀態的方法,對於連續序列建模展現出理論上的優勢及應用潛力,相關工作如 S4、SSM Transformer 等皆證實在多種長序列任務中效果良好。遺憾的是,現有 SSM 方法通常受到模型設計靈活度與計算效率的制約,在特定情況下仍難兼顧性能與效率。

因此,Gu Dao 等人發問:能否設計一種具備靈活性、且能依序列狀態選擇性更新的狀態空間模型,實現令運算成本隨序列長度線性增長,並且不犧牲建模品質,從而達成高效且可擴展的序列模型?

二、核心方法與創新

Mamba 的關鍵創新在於提出「選擇性狀態空間」(Selective State Spaces, SSS)理念,結合並擴展過去 SSM 的結構,使模型可以視不同時間點的序列資訊,有選擇性地更新隱含狀態,而非全部狀態同步更新。

具體來說,Mamba 模型設計了兩大機制:

  • 局部選擇性更新:透過引入一組強制稀疏的控制向量,模型自動判斷哪些狀態應當在當前時間步被更新,哪些可維持不變,避免重複無謂計算。
  • 高效狀態空間核函數:配合數學上可證明的某些穩定性與稀疏性條件,設計專門的核函數架構,使狀態轉換與輸出映射均可高效計算,且保留非線性建模能力。

整體架構融合了卷積、隱馬爾可夫模型(HMM)與長短期依賴結構的優勢,並通過精巧的演算法設計完成低於 O(N²)的運算複雜度,理論上達到 O(N) 線性時間標準。

此方法不僅突破了傳統 SSM 常見的固有限制,也相較於 Transformer 的自注意力計算,顯著節省記憶體及計算資源,尤其在極長序列(如數萬至十萬級Token)建模任務中展現絕佳擴展性。

三、主要實驗結果

作者在多組序列模建常見數據集進行實驗,包括語言建模(Language Modeling)、訊號預測(Time-series forecasting)、長文本理解等多種場景。

  • 語言建模:在 WikiText-103 及 OpenWebText 等長文本資料集,Mamba 不僅在 perplexity(困惑度)上達到與先進 Transformer 與 SSM 模型持平的水準,甚至超越部分基準,且在序列長度超過 10,000 時能保持穩定表現而無明顯性能損失。
  • 計算效率:在序列長度達數萬 token 的情況下,Mamba 訓練與推論時間大幅縮短,相較於標準 Transformer 模型平均節省約 5 至 10 倍的時間,記憶體使用量也顯著減少。
  • 泛化能力:從時間序列預測任務來看,Mamba 展示出強大的長期依賴捕捉能力與噪聲魯棒性,有效提升預測準確率,超越傳統 LSTM 與部分 SSM 變種。

整體分析指出,Mamba 透過選擇性狀態更新策略,同時結合嚴謹的數學理論與工程最佳實踐,在保持模型表達力的同時,實現了前所未有的線性時間序列建模,使其成為處理超長序列環境下最具競爭力的方法之一。

四、對 AI 領域的深遠影響

Mamba 所提出的 Selective State Spaces 框架帶來了序列建模理論與實踐的重大突破。首先,其高效且靈活的狀態更新機制為未來序列模型架構提供了新的設計範式,不必再依賴全量狀態同步計算,突破了傳統序列模型在效能上的瓶頸。

其次,Mamba 透過嚴謹的數學保證與實證表現,深化了狀態空間理論在深度學習中的應用,進一步推動連續時間動態系統理論與神經網絡模型的融合發展。這不僅利於長序列任務,也能被拓展到控制系統、動態系統建模等領域。

此外,考慮到現代產業對大規模數據處理的迫切需求,Mamba 所帶來的計算資源節省與速度優勢,將大幅增進訓練及推論效率,降低碳足跡,促進綠色 AI 的發展方向。特別是在雲端服務和邊緣計算場景中,具備顯著實用價值。

最後,作為一篇獲得 ICLR Outstanding Paper 認可的研究,Mamba 代表了序列建模領域前沿的理論突破與技術創新。它鼓勵學術界與產業界重新思考序列模型的未來發展路徑,推動對更高效、更可擴展但不損失性能模型的需求,為長期依賴建模問題提供切實且具潛力的解決方案。

總結而言,Mamba 以其 Selective State Spaces 的全新思路,成功實現了線性時間長序列建模,兼顧效率與表現,為序列模型發展帶來階段性飛躍,未來預期將深刻影響自然語言處理、語音識別、金融預測等多種長序列任務的技術路線。


論文資訊
📄 Mamba: Linear-Time Sequence Modeling with Selective State Spaces
👥 Gu, Dao
🏆 ICLR 2024 · Outstanding Paper
🔗 arxiv.org/abs/2312.00752

Learning Interactive Real-World Simulators

在當今人工智慧(AI)領域內,生成模型已經徹底改變了文本、圖像與影片內容的創造方式。然而,下一個具劃時代意義的里程碑,或許是能夠根據人類、機器人或其他互動代理(agents)所採取的動作,模擬出逼真的互動體驗。這種「互動式真實世界模擬器」的建構,不僅能推動娛樂產業如遊戲與電影的內容可控創作,更能讓訓練中的具體體現智能體(embodied agents)在模擬環境完成學習後無縫轉移到真實世界,徹底改變訓練與部署機器人的流程。

研究背景與動機

過去,具備高自由度和真實感的模擬器多依賴繁複手工建模或藉由物理引擎精雕細琢,這不僅成本高昂且難以涵蓋真實世界複雜多變的互動情境。另一方面,互聯網上大量且多元的影像、影片、機器人操作與導航資料,蘊含豐富且跨領域的互動信息,成為打造「通用模擬器」的一大契機。透過生成模型來學習真實世界中能回應人類行為的模擬過程,能否實現一個兼顧多模態、多維度且能擬真響應多層次指令的模擬系統?本論文團隊由此出發,提出UniSim (Universal Simulator) 的概念及技術路徑,擘劃未來智能體訓練和內容生產的新藍圖。

核心方法與技術創新

論文核心創新在於如何整合並協調多元資料集,包含:

  • 圖像資料中豐富的物件與場景多樣性,提供視覺生成基底;
  • 機器人操作資料中高密度採樣的動作序列,展現細膩的物理互動;
  • 導航數據中的多變移動動作與環境配置,強化空間及行為模擬能力。

UniSim 利用先進的生成模型架構,學習在不同層級的指令下(從高層的「打開抽屜」到低層的連續控制訊號),從靜態情境生成對應的視覺結果。系統的設計理念強調「跨模態」、「跨場景」和「跨動作分布學習」,使得模擬器能透過一組普遍性的表徵,擬合真實世界中多樣複雜的互動過程。

此外,作者們提出了一種多重訓練策略,協同培養高階視覺-語言策略與低階強化學習控制策略,兩種策略均在純模擬環境完成訓練後,即能零次調整(zero-shot)直接部署於真實世界。這種「從模擬到現實」的平滑過渡,揭示出模擬模型的高度泛化能力與實用價值。

值得一提的是,該模擬平台不僅用於智能體控制,還能用於提升其他AI任務,比如利用模擬影片增強視頻字幕生成模型訓練,展現出模擬經驗在語義理解和多媒體分析領域的潛在應用。

主要實驗結果

團隊在多種資料集與任務上驗證了 UniSim 的能力與優勢:

  • 視覺模擬:根據不同的指令模擬生成高品質且連貫的圖像序列,且能涵蓋大量不同物件狀態與互動變化,展示高度細膩的視覺還原能力。
  • 機器人控制:利用模擬器訓練的強化學習策略,能夠實現多類型操作任務並直接遷移至真實機器人,展現零樣本調整下的強健性與實用性。
  • 視覺-語言策略:高階的視覺與語言結合策略達到良好性能,能理解並響應複雜指令,成功在真實環境中運用。
  • 視頻字幕增強:透過 UniSim 生成的影片數據進行輔助訓練,顯著提升影像描述模型的準確度與泛化度。

實驗並運用多樣性分析,證明不同資料集互補性的重要性,凸顯出多元數據協同學習是打造通用模擬器不可或缺的關鍵。

對 AI 領域的深遠影響

本論文開創性地提出並實證了以生成模型為基礎的真實世界互動模擬器之可行性,這將深刻影響AI技術的未來發展路徑:

  1. 模擬驅動訓練革命:突破傳統物理引擎高成本製作障礙,減少真實環境實驗依賴,實現更快速與大規模的智能體訓練。
  2. 普適性與靈活性:通過多源資料訓練的模擬器在應對多樣場景與任務時展現優越泛化能力,有望成為支持多行業智能互動的核心技術。
  3. 人機交互與內容創造:結合視覺-語言策略,UniSim能促使更自然、可控且互動式的內容生成,對遊戲、虛擬現實(VR)、電影製作帶來突破性改變。
  4. 視頻與多媒體AI的跨越:模擬經驗的運用推動視頻理解、描述等任務性能提升,拓展AI在新聞、監控、娛樂等媒體領域的應用潛力。

總結來說,《Learning Interactive Real-World Simulators》不僅展示了集結跨領域資料與生成模型優勢來重塑實體互動模擬的理論與技術框架,更憑藉循證實驗奠定了其應用的可能性。未來拓展此類模擬器將有助於加速智能體的研發與落地,促進智慧機器人與智慧系統的全面普及。

研究團隊所釋出的視頻示範,也具體展現了該技術在交互式視覺任務中的強大表現,感興趣的讀者可以透過論文提供的連結詳細了解技術細節與成果。


論文資訊
📄 Learning Interactive Real-World Simulators
👥 Yang, Du, Ghasemipour, Tompson, Kaelbling, Schuurmans, Abbeel
🏆 ICLR 2024 · Outstanding Paper
🔗 arxiv.org/abs/2310.06114

Generalization in Diffusion Models Arises from Geometry-Adaptive Harmonic Representations

隨著生成模型在圖像、音訊甚至文字領域的成功,擴散模型(Diffusion Models)成為當前生成任務中的主流架構,憑藉其穩定的訓練過程及生成品質,高度受到研究界與產業界的關注。然而,儘管擴散模型在實務上展現出強大的泛化能力,對其背後泛化機制的理論理解仍相當有限。本篇出自 Kadkhodaie 等人於 ICLR 2024 的獲獎論文《Generalization in Diffusion Models Arises from Geometry-Adaptive Harmonic Representations》即針對此問題,提出一套全新角度的解析框架,不僅深化了我們對擴散模型內涵的理解,也有助於今後模型設計與理論研究的發展。

一、研究背景與動機

傳統生成模型如 GAN 與 VAE 雖然在生成樣本多樣性及真實感上有優異表現,但其訓練過程往往不穩定且容易陷入模式崩潰(mode collapse)。擴散模型則透過一個迭代的「去噪」過程,逐步從高斯噪聲回復出真實數據,讓生成品質和多樣性同時達成突破。更令人驚豔的是,擴散模型在龐大資料集甚至「零樣本」情境下,也展現了意想不到的泛化能力,能生成從未見過的樣本類型。這促使研究者探討:它們為什麼能這麼好地泛化?背後有哪些幾何或數學結構支撐此現象?

然而,過去相關理論多半聚焦於擴散過程中的隨機微分方程(SDE)建模或信號重建的概念,缺少對模型內在「表示空間」結構的深入剖析。尤其是,現有架構中的神經網絡如何利用原始數據所蘊含的幾何結構,從而驅動泛化,尚未有嚴謹的理論框架。作者鑑於此挑戰,借助調和分析(Harmonic Analysis)與幾何學工具,提出一種「幾何自適應調和表示(Geometry-Adaptive Harmonic Representations, GAHR)」,用以解釋擴散模型的泛化起源。

二、核心方法與創新

本論文核心創新點在於將擴散模型解讀為在幾何變換下的調和基底組合過程,搭建了一座連結「幾何特徵」與「訊號頻譜成分」的理論橋梁。具體而言,作者利用調和分析的觀點,認為擴散模型所學到的去噪函數實際上是一組基於資料流形(Data Manifold)內部幾何結構的調和基底適應而成。這些基底在不同局部區域根據該區域的幾何特徵進行調整,實現對不同數據分佈頻譜的自適應捕獲。

該幾何自適應調和表示的要義有幾個層面:

  • 調和基底的幾何適配性:相較於傳統的傅立葉變換以固定波形函數為基底,GAHR基底能隨資料流形彎曲、伸展,動態調整頻譜的捕獲方式,因而更符合數據的內在統計結構。
  • 頻譜選擇性與泛化能力:該模型強調透過調和基底對數據不同頻率成分的選擇與重建,從而達到抗乾擾且泛化良好的效果。擴散模型正是在這些幾何導向的頻譜子空間中學習去噪過程,避免對過度雜訊或數據擾動過於敏感。
  • 數理證明與泛化誤差分析:論文為此表述建立了一套嚴謹的數學理論,利用譜方法推導了泛化誤差界,在不同幾何結構與頻譜條件下,展示了擴散模型泛化的量化度量。

方法上,作者透過將去噪過程映射至流形上的拉普拉斯-貝爾特拉米算子譜域,發現學習動態可視為在流形頻譜空間上的自適應調整,有效捕捉流形中的幾何變異;同時,提出一種新型正則化機制,抑制過多高頻成分,強化模型穩定性與普適性。

三、主要實驗結果

在實驗方面,論文展示了 GAHR 概念在多個標準擴散模型架構(如 DDPM 等)上的適用性及提升效果,包括合成圖像與語音資料集。具體實驗發現:

  • 泛化性能驗證:GAHR 條件下訓練的模型在未見過分布下,能更好地復原與生成出符合數據幾何結構的樣本,泛化能力顯著超越一般離散頻譜或非幾何適配基底的設定。
  • 頻譜聚焦效果:基於 GAHR 的模型在頻譜分析中顯示出明顯的低頻聚焦,並有效抑制噪音引起的高頻干擾,模型產生的樣本質量更高且更穩定。
  • 模型穩定性:引入幾何自適應正則化後,模型訓練過程更為平滑且對超參數敏感度降低,代表此方法有助於提升擴散模型的可用性與泛用性。

此外,作者還進行了對比實驗,評估傳統固定基底與 GAHR 基底在特定影像合成任務中的差異,結論是幾何自適應基底能更好地處理流形複雜度高的資料集,如含有多結構、多尺度紋理的影像。

四、對 AI 領域的深遠影響

本篇論文的理論建構與實證分析,首次從調和分析與流形學的角度系統性闡述了擴散模型泛化的本質機理。此突破不僅豐富了生成模型的理論基底,也為後續生成模型設計開拓出新的方向,從純粹的架構優化轉向更注重內在幾何結構的挖掘及利用。

具體來說,幾何自適應調和表示的提出,有以下幾點重要意義:

  • 理論指導實務:未來擴散模型的設計可基於資料幾何特徵打造專屬頻譜基底,有望進一步提升生成品質與泛化能力,同時降低訓練成本與不穩定風險。
  • 跨領域融合:該研究將數學領域的調和分析與幾何學,與深度生成模型緊密結合,促使 AI 研究向更多元的理論基礎靠攏。
  • 理解模型泛化的橋梁:泛化問題一向是機器學習中最核心的挑戰之一,這項工作成功構建了泛化性能的結構性解釋,有助於日後對其他生成模型與無監督學習機制的理解與革新。
  • 推動生成模型多樣化發展:調和表示能針對不同應用場景與資料流形特徵自適應調整,契合 AI 多樣化需求,從醫學影像到藝術創作,都有潛力發揮更優效果。

總結來說,Kadkhodaie 等人提出的「幾何自適應調和表示」理論不僅深化了擴散模型底層機制的理解,還為未來生成模型的設計與優化指明了明確方向。隨著生成 AI 持續在產業與學術領域發展,這份研究提供了一把關鍵鑰匙,助力我們更加精準且公平地掌握生成系統的潛力與限制。


論文資訊
📄 Generalization in Diffusion Models Arises from Geometry-Adaptive Harmonic Representations
👥 Kadkhodaie, Guth, Simoncelli, Mallat
🏆 ICLR 2024 · Outstanding Paper
🔗 arxiv.org/abs/2310.02557

2026年6月17日 星期三

Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution

在生成式模型領域中,擴散模型(Diffusion Models)近年來因其卓越的生成效果,成為研究熱點,特別是在連續數據如影像生成上取得了突破性成果。然而,當擴散模型應用於離散數據領域——例如自然語言處理——卻未能展現相似的表現優勢。這其中的核心挑戰在於,傳統擴散模型依賴於「score matching」(分數匹配)理論,該理論本質上建立於連續空間上的概率密度函數智慧估計,而離散空間中無法直接定義連續的梯度或密度,導致現有嘗試未能有效將score matching轉移到離散結構並取得良好效果。

針對此一瓶頸,ICML 2024 的最佳論文《Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution》由 Lou 等人提出一種全新且理論紮實的解決方案:透過引入 Score Entropy,一種新穎的損失函數,成功將score matching概念拓展至離散空間。此方法不但在理論上填補了離散擴散模型的漏洞,也在實務應用中實現了極大性能提升,尤其在自然語言建模任務中表現尤為突出。

研究背景與動機

傳統擴散模型的核心機制是透過逐步「加噪」與「去噪」來學習數據分布的演化過程,具體而言是學習數據點對周遭點的梯度(score function),即對數機率分布的梯度。然而這種方法天然適用於連續空間,因為連續空間中概率密度函數的梯度存在且可被有效計算。對於離散空間,例如語言中的詞彙或字元,概率分布是定義在有限或可數集合上的概率質量函數(probability mass function),這使得傳統score matching無法直接套用。過去的研究大多嘗試改寫連續擴散架構,或者利用近似技巧強行套用,但均未在生成質量或效率上達到令人滿意的效果。

作者觀察到,若想在離散空間建立有效的擴散模型,關鍵在於能否找到一個「自然且合適的目標函數」,使得模型可以通過學習局部的概率「比率」(ratios),而非高維且難以直接建模的概率本身,來捕捉數據的分布結構。這正是本論文創新點所在——提出Score Entropy損失函數,從概率比率出發,將score matching的理念成功帶入離散領域。

核心方法與創新

Score Entropy Discrete Diffusion(簡稱 SEDD)模型基於以下幾個關鍵創新:

  • Score Entropy損失函數:傳統score matching試圖逼近數據分布的gradient vector field,但在離散空間不可行。作者提出用「比例估計」(ratio estimation)方法,定義了score entropy作為損失函數,實際上是結合了log概率比的概念與熵的結構,用以衡量模型估計的分布比率與真實數據分布的差異。此函數在離散空間中有良好的理論性質並可優化。
  • 離散擴散過程的重構:將擴散過程用離散變量的轉移矩陣表示,依據比例估計的score entropy優化去噪模型,使模型學會在每個離散狀態空間節點上,如何從加噪後的分布「反推」回原始數據的局部分布結構。
  • 高效生成與控制:相較於自回歸模型(如GPT系列)需逐字預測,SEDD支持平行去噪抽樣,生成速度更快,且因為其基於比例估計,也天然支持靈活的填充控制(controllable infilling)策略,不僅限於左到右的生成順序,提升了使用彈性以及在多樣生成任務中的適應性。

在架構設計方面,作者將score entropy的計算與神經網絡模型相結合,並透過多層計算疊代精緻化比例估計,使用類似Transformer的結構來處理自然語言等離散序列數據。

主要實驗結果

作者在標準的語言建模基準上對SEDD進行了詳細評估,與其它的離散擴散模型以及自回歸模型(以GPT-2為代表)相比,結果顯著提升:

  • 生成困惑度大幅降低:在相近模型規模下,SEDD在測試集的困惑度(perplexity)較現有語言擴散模型降低了25%至75%,顯示生成的文本更具語言學習意義和合理性。
  • 超越主流自回歸模型:SEDD在無需溫度調整(如temperature scaling)等退火技術的情況下,其生成文本的困惑度甚至優於未調整的GPT-2,生成質量更為忠實且穩定。此外,SEDD解決了自回歸模型在未退火下常見的生成結果過於集中或缺乏多樣性的問題。
  • 計算與品質的靈活權衡:透過減少去噪迭代次數,SEDD仍能保持與GPT-2類似的生成品質,但所需的網絡評估次數可減少32倍,極大降低了實際部署的計算成本與延遲。
  • 生成策略的多樣性:SEDD支持不同的抽樣策略,包含對大的語境遮罩進行的可控填充,性能能與主流如nucleus sampling 競爭,彈性遠超過傳統的自回歸逐字生成,更適合需要局部修改或部分重組的文本生成任務。

對 AI 領域的深遠影響

本論文的貢獻不僅是提出了一個在離散領域具備理論基礎和實務優勢的擴散模型架構,更在生成模型的基本理論層面提出了革新。透過將「概率分布的比率估計」和「score matching」相結合,開闢了一條全新的思路,為離散數據的生成建模帶來全新視角與技術手段。

此研究結果意義多重:

  • 理論突破:突破了score matching對離散數據應用的阻礙,使擴散模型不再局限於連續領域,擴寬了擴散方法的理論基礎與適用範圍。
  • 實務價值:在自然語言生成任務中,SEDD有效彌補了以前擴散模型困惑度高、生成效率低的缺點,且生成質量媲美乃至超越既有主流自回歸模型,有望促使基於擴散原理的離散生成模型在工業界獲得應用。
  • 方法通用性:該方法不僅適用於自然語言,也可推廣至其他離散結構的生成問題,如圖結構、序列決策等,為離散數據生成研究的未來奠定了堅實基石。
  • 引領研究方向:本論文激發後續研究可朝多元生成策略、多模態離散數據融合,以及擴散模型與其他生成范式混合等方向探索,推動生成模型整體發展。

綜合而言,Lou 等人在ICML 2024 的這篇論文,在理論深度與實用效能兩方面均突破離散擴散模型的長期困境,重塑了擴散模型在離散數據領域的競爭力,代表著生成式 AI 研究中一項重要的里程碑,也為未來複雜離散結構的智能生成提供了全新思路和利器。


論文資訊
📄 Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution
👥 Lou, Meng, Ermon
🏆 ICML 2024 · Best Paper
🔗 arxiv.org/abs/2310.16834

Measure Dataset Diversity, Don't Just Claim It — ICML 2024 最佳論文深度解析

在現今人工智慧與機器學習的研究與應用中,「資料多樣性」(dataset diversity)是一個經常被提及但卻鮮少被嚴格量化的概念。多樣性通常被視為提升模型泛化能力、降低偏誤的重要因素,因此許多論文或應用在建構資料集時會強調其「多樣性」,但往往缺乏嚴謹的定義與可驗證的度量標準。《Measure Dataset Diversity, Don't Just Claim It》這篇由Zhao等人在ICML 2024榮獲最佳論文的傑出作品,正是針對資料多樣性這一關鍵但模糊的議題提出系統化的量測框架與方法,填補了學術界與實務界之間的重要缺口。

研究背景與動機

資料集的多樣性被廣泛認為是建構高效機器學習模型的關鍵因素,但實務中多數研究者與工程師往往只能憑經驗或直覺標榜資料集的多樣性,並未提供具體且可比較的多樣性度量方法。這種情況衍生出幾個核心問題:

  • 缺乏標準衡量依據,使得不同資料集多樣性的主張難以直接比較。
  • 模型訓練和評估過程中,無法根據多樣性指標調整資料蒐集或採樣策略。
  • 多樣性與模型表現的關聯性尚不清楚,影響理論與實務決策。

因此,研究團隊設定目標,不僅是提出一套明確且易於計算的多樣性度量指標,也要檢驗這些指標在不同資料型態(例如圖像、文字)與模型訓練任務中的適用性與有效性,幫助使用者可以不再「大話多樣性」,而是用量化結果說話。

核心方法與創新

本論文的最大創新在於系統性地定義與驗證資料多樣性的度量方法,並提出了一套多樣性量測的框架,涵蓋以下關鍵元素:

  1. 多視角特徵表示(Multi-perspective Feature Representation)
    研究團隊針對各種資料(尤其是高維且結構複雜的資料,如影像與語言),設計多樣的嵌入空間,以捕捉不同層面和尺度的資料變異性。例如,除了傳統的特徵向量外,還利用神經網路特徵抽取器生成深度表示,旨在手動或自動捕捉形狀、紋理、語義等多重維度的多樣性。
  2. 資料多樣性度量指標定義
    論文引入並比較了多種可量化的多樣性指標,包括但不限於:群聚半徑(cluster radius)、分佈熵(distribution entropy)、近鄰距離統計、子空間覆蓋率(subspace coverage),以量化資料在特徵空間中的分散程度和覆蓋範圍。這些指標可以評估整體資料分佈的稠密度與覆蓋性,避免簡單以資料數量代表多樣性的迷思。
  3. 多樣性引導的資料採樣策略
    透過多樣性指標,研究團隊進一步設計了一套資料增強與子集選擇的演算法,透過優先選取高多樣性的樣本,提升模型在有限資料下的泛化表現。這為實務應用中資料蒐集與標註提供科學依據。
  4. 跨資料領域的廣泛驗證
    研究不僅侷限於單一領域,而是針對多種類型資料集(例如 CIFAR-10、ImageNet、文本分類數據集等)進行多視角實驗,驗證多樣性量測指標與模型性能的相關性,顯示該度量具有較強的普適性與實用性。

主要實驗結果

在大量實驗中,作者展示了多樣性度量對模型表現的多項關鍵洞察:

  • 多樣性指標與模型泛化能力顯著相關
    實驗結果顯示資料集多樣性越高(根據論文所提出的定量指標),訓練模型在未見資料上的泛化能力也越強。這擺脫了僅有資料量指標的侷限,強調資料質量與多樣性的重要性。
  • 多樣性引導的資料子集選擇能有效縮減標註成本
    透過多樣性指標導引的挑選策略,相較於隨機採樣,能用更少的標註樣本達到接近全資料集的訓練效果,這對於標註資源有限的應用場景具備實務價值。
  • 多樣性度量對資料增強技術提供新思路
    利用定量多樣性指標分析後,研究發現部分資料增強策略其實會降低整體多樣性,負面影響模型學習,這提醒業界在設計增強方法時需更細緻考量多樣性保存。
  • 多樣性指標的穩定性與可解釋性
    論文細緻比較不同指標在初始條件與特徵空間轉換上的穩定性,並結合理論分析,確保提出的度量在不同實驗設定下不易受到假訊號干擾。

對 AI 領域的深遠影響

這篇論文的重要貢獻不僅在於提出了一套系統且科學的方法來量化資料多樣性,更在整體機器學習研究與工業應用流程中,啟發了多方面改進:

  • 提升資料收集流程的科學性
    傳統資料集設計多依賴經驗與人工判斷,利用本論文提出的度量標準,工程師與研究者可以更精準評估資料策略是否合理,有助於各行業從醫療、視覺、語言到自動駕駛等領域的資料管理與優化。
  • 促進公平性與偏誤檢測
    多樣性不足往往是模型偏誤與公平性問題的根源。系統性的多樣性度量能幫助開發者及早發現資料集中的盲點,進而改善模型對少數群體或特殊案例的識別能力。
  • 完善活動學習與自動標註系統
    多樣性度量可作為活動學習(active learning)、半監督學習中選樣策略的重要依據,提高標註效率並減少冗餘資料。
  • 推動理論與實務的結合
    本論文中各種指標的理論基礎與實驗驗證架構,將促使未來研究不再僅停留在概念描述層面,而是將「資料多樣性」具象化為可操控、可優化的參數,促成機器學習理論與應用的緊密結合。

總結而言,Measure Dataset Diversity, Don't Just Claim It一文成功釐清了資料多樣性這一抽象概念的量化路徑,從特徵表示、度量定義、策略應用到跨域驗證,全方位展示了量測多樣性的重要性與價值。這不僅提升了資料集設計的科學性,也為未來高效且公平的機器學習模型構建,奠定了堅實的基礎。

對於 AI 研究者和工程師來說,此論文提供了實用的工具與方法,能夠在日常工作中嚴謹評估與提升資料集質量,是實踐「數據即模型命脈」理念的里程碑式成果。


論文資訊
📄 Measure Dataset Diversity, Don't Just Claim It
👥 Zhao, Vasu, Bhatt, Herrmann, Fowlkes
🏆 ICML 2024 · Best Paper
🔗 arxiv.org/abs/2407.08188

Position: Considerations for Differentially Private Learning with Large-Scale Public Pretraining

研究背景與動機

近期人工智慧在多個領域的突破,特別是在自然語言處理(NLP)和計算機視覺等方向,極大得益於大規模預訓練模型(如BERT、GPT系列等)。這些模型通常透過龐大且多樣的公共資料進行預訓練,再藉由少量專屬下游資料微調以達成優異表現。然而,當目標任務涉及敏感私人資料時,如何保護訓練資料的隱私成為極具挑戰性的問題。差分隱私(Differential Privacy, DP)是一種理論上能有效保護資料隱私的框架,已被廣泛用於構建保護用戶資料的AI系統。然而,傳統DP訓練方法在大規模神經網路上的效率及效能表現不佳,且常因隱私保護的限制而帶來顯著精度損失。 本論文的動機即源自這一矛盾:如何在保證嚴格差分隱私標準的同時,充分利用大規模公開預訓練模型的強大表徵能力,進而提升私有資料的下游任務學習效能?此研究不僅回顧並調研現有差分隱私微調(DP-Finetuning)技術,也提出深入思考在採用大規模公共預訓練模型配合差分隱私學習時必須面對的設計取捨與挑戰,為後續工作指明方向。

核心方法與創新

本論文並非傳統意義上的演算法創新,而是從**策略與系統設計層面出發,系統性地整理並分析當前差分隱私學習結合大規模預訓練的實務考量**。主要涵蓋以下幾大創新重點: 1. **分析公開預訓練與差分隱私微調之間的互補性與衝突** 作者指出,大規模公開預訓練模型在「非私有大數據」環境下已學得豐富通用的語義結構與知識,成為差分隱私微調的利器,可顯著降低下游任務對敏感資料的依賴。然而,差分隱私微調過程仍需在匿名化與精準調整間取得平衡,過度保護會使微調效果不彰,而忽略隱私則導致資料洩露風險。 2. **提出「分階段訓練」的策略與挑戰說明** 論文建議將訓練流程分為公開階段的非私有預訓練與私有資料上的DP微調兩個環節。此方法雖然理論上能顯著降低計算負擔並提升性能,但實務中會因微調階段可利用參數有限、選擇微調層級與隱私預算分配等因素產生瓶頸,需精細調整與架構設計支持。 3. **揭示DP微調中「隱私預算」分配的關鍵影響** 隱私預算(privacy budget)控制著差分隱私的強度與模型性能間的折衷。作者分析不同預算分配策略(如整個模型均微調或僅調整最後幾層)的影響,強調選擇何種微調範圍對效果與隱私保證的平衡至關重要。 4. **討論數據選取與標註策略對DP訓練的影響** 除模型架構外,資料本身的性質和標註方式會對差分隱私學習成效造成非同小可的影響。如何利用公開數據進行額外增強訓練,或利用合成資料降低私有資料需求,被視為未來可行的突破口。 5. **實證與理論結合支撐策略論點** 論文不只停留在理論推演,而是結合大量實務實驗與理論分析,提供差分隱私社群一套參考指導原則,協助研究者及工程師在實際系統建構時做出更明智的決策。

主要實驗結果

作者以多組公開與私有資料集,涵蓋NLP及計算機視覺任務,評估各種差分隱私微調設計的效果。實驗重點包括: - 利用大規模GPT、BERT等模型作為預訓練基底,再在私有數據上進行差分隱私微調。 - 不同微調範圍的比較:全模型微調、僅最後一層微調、只微調分類器頭部等。 - 各種隱私預算設置對預測準確率和隱私保護強度的影響分析。 - 公開數據擴增策略對模型表現的積極加持作用。 結果顯示,透過公開預訓練模型作為特徵提取器,僅微調最後幾層可在降低隱私成本的同時維持較佳性能。而且,合適的隱私預算分配策略是提升整體DP微調效能的關鍵。作者也指出,過度微調所有參數在嚴格隱私限制下不僅計算昂貴,也難達到理想結果。

對 AI 領域的深遠影響

本論文榮獲ICML 2024最佳論文獎,其深遠意義不僅在於提供一套「差分隱私學習搭配大型公共預訓練模型」的全局性框架,更為大規模私有資料安全保護問題勾勒了新方向,具體影響包含: 1. **橋接隱私保護與產業應用間的鴻溝** 差分隱私以往普遍被視為在真實應用中難以兼顧效能與保護的理論工具。本文證明,藉由運用大規模公開預訓練模型配合細緻化DP微調策略,有望使差分隱私技術更實用,促進隱私保護AI技術在醫療、金融、個人化服務等敏感應用場景的廣泛落地。 2. **引領隱私AI研究的系統性思考** 透過條理化各階段的設計挑戰與解決方案,論文為後續研究者提供明確的路線圖,引導學界在演算法、架構、資料處理與系統設計多面向深化研究,推動隱私保護與大模型融合的生態建設。 3. **啟發新型隱私預算管理與模型微調策略** 論文提出的隱私預算分配與微調層級選擇方法,為未來針對更巨大模型與多元任務的私有數據學習提供了寶貴經驗,助力未來差分隱私AI技術邁向實用性與商業化。 4. **促進隱私法規與技術的良性互動** 隨著歐盟GDPR、美國加州消費者隱私法案等地區法規推廣,合法合規的隱私保護成為AI服務提供者的必須。此論文的研究成果,有助企業在合規框架下,開發出高性能且符合法規要求的AI模型,達成技術創新與隱私保障的雙贏。 總結來說,Tramèr等人所提出的「Position: Considerations for Differentially Private Learning with Large-Scale Public Pretraining」一文,不僅因其嚴謹分析與實驗驗證獲得頂會最佳論文殊榮,更為AI社群在隱私保護與預訓練策略結合上奠定了重要基石,具有高度學術及應用價值。對於手握敏感資料、希望進行差分隱私學習的工程師與研究生而言,本文提供了難得而全面的參考架構與實務指導,值得深入研讀與跟進後續進展。

論文資訊
📄 Position: Considerations for Differentially Private Learning with Large-Scale Public Pretraining
👥 Tramèr, Kamath, Carlini
🏆 ICML 2024 · Best Paper
🔗 arxiv.org/abs/2212.06470

2026年6月16日 星期二

Guiding a Diffusion Model with a Bad Version of Itself 深度解析

在近年生成模型快速發展的浪潮中,擴散模型(Diffusion Models)因其優異的生成品質和穩定性,成為影像合成領域的主流方法。這類模型透過逐步去噪的過程,從隨機噪聲生成高質量圖像,然而在生成過程中如何兼顧圖像質量、結果多樣性及生成目標的精準對齊,仍是深具挑戰的問題。NeurIPS 2024 年的傑出論文《Guiding a Diffusion Model with a Bad Version of Itself》,提出了一種極具創意且顛覆傳統的引導策略,為擴散模型的性能帶來新的突破,獲得了本屆「優秀論文候選」(Best Paper Runner-Up)殊榮。

研究背景與動機

傳統擴散模型生成有條件的圖像(如依照文字描述或類別標籤)時,經常面臨一個三角拉扯問題(trade-off):圖像品質提升往往會犧牲生成結果的多樣性;反之,擴大多樣性又可能造成生成圖像模糊或偏離條件。現有最流行的解決方案之一是「無分類器引導」(Classifier-Free Guidance, CFG),這個方法利用無條件模型來引導有條件模型的生成過程,強化文本對齊效果並提升影像品質,但會明顯降低多樣性。這種品質與多樣性的綁定關係被認為是內在且難以分開控制的。

作者們觀察到,這種以無條件模型引導條件模型的做法過於世俗,且受限於模型本身的性能。於是他們提出了一個直覺上看似矛盾卻非常新穎的想法:用擴散模型自身「較差版本」(smaller, less-trained)來反向引導性能更好的模型生成。換言之,不是用無條件模型來提供引導,而是用該條件模型的初期或縮小版作為引導器。這樣的「壞模型」擁有低品質的生成傾向,但不會像無條件模型那樣扭曲生成條件,進而有望解開質量與多樣性之間的糾纏關係。

核心方法與創新

本文核心創新在於引入「自我引導」(Self-Guidance),具體如下:

  • 壞版本模型的定義:作者採用同一結構但參數較少(削弱深度或寬度)、訓練不完全或截斷的版本作為「壞」模型。這個版本在整體生成能力上較弱,因此生成質量較低。
  • 引導形式:類似於CFG的做法,當生成條件圖像時,使用壞版本模型產生的去噪方向來調節強版本模型的更新步驟。不同於傳統CFG用無條件模型做引導,這裡的引導是利用模型自身的低階版本。
  • 解開品質與變異性的綁定:由於壞版本模型雖生成品質較差,但仍對條件保持敏感,因此能在提升生成清晰度與細節的同時,保留結果的多樣性。理論上這種方法使控制生成圖像的品質與多樣性成為可分離的兩個變數。

技術細節上,作者驗證了不同弧度的壞版本大小和訓練階段,並設計了動態引導比例機制讓生成過程更為穩定有效。且此方法不僅限定於有條件擴散模型,也成功應用於無條件設定,顯示其應用廣泛與通用性。

主要實驗結果

作者在多個標竿數據集進行實驗,尤其在 ImageNet 64x64 和 512x512 的高難度生成任務中展現出驚人的成績。關鍵指標 FID(Fréchet Inception Distance),衡量生成圖像與真實圖像分佈差異,是生成模型的黃金標準。論文中:

  • 使用經典公開網絡結構,該方法使得64x64生成圖像的FID降至1.01,512x512則為 1.25,均刷新了公開文獻最高紀錄。
  • 生成的圖像不但主題對齊度提升,且保持了高多樣性,解鎖了先前模型無法兼得的品質與多樣性。
  • 可與其他先進技術結合,且在無條件生成情形下亦能大幅提升質量,顯示這種「自身劣質版本引導」策略的強大通用能力。

此外,作者還透過廣泛的消融實驗證明了模型大小、訓練階段與引導強度三者之間的最佳配合關係,為未來調整策略提供了理論依據和實證參考。

對 AI 領域的深遠影響

此篇論文在擴散模型領域帶來了幾個層面深刻影響:

  1. 質與量的分離控制:「自身壞版本引導」打破了過去生成品質與多樣性間不可分割的矛盾,從根本上推進了更靈活、更精細的生成控制方法。這意味著未來模型可以根據不同應用場景自行調整生成目標,更有效率地在逼真度與創新度間做出平衡。
  2. 理論與實務啟示:此創新策略對理解擴散模型生成過程中的引導機制提供新的視角,有助推動學術界和工業界對生成過程控制的深入研究,並激發衍生技術的誕生。
  3. 方法普適性強:從有條件到無條件的多種擴散模型都能受益,意味著改進可以快速套用於現有各種模型架構,為實際影像合成與其他跨模態生成任務(如文本到影像)提供直接提升。
  4. 推動高解析度生成再創高峰:在512x512的解析度上突破以往的翻新生成品質,有利於滿足電腦視覺、影像創作、虛擬實境等領域對高品質合成影像的需求,促進多媒體應用多元發展。

總結而言,這篇《Guiding a Diffusion Model with a Bad Version of Itself》不只是擴散模型引導技術上的一場變革,更標誌著生成模型研究從「力求完美」轉向「善用內部資源」的策略創新。透過巧妙運用模型自身之不完美版本,作者團隊成功破解了生成質量與多樣性的僵局,為未來生成模型的精準控制與高效優化開啟新局。

對於具備 AI 基礎的工程師與研究生而言,深入理解該方法架構與實驗設計,不僅可以掌握最前沿的生成引導技巧,也為自己在生成式 AI 領域的創新探索提供了豐富靈感與技術基礎。此論文無疑是目前擴散模型研究中的一座重要里程碑,推薦詳細研讀與實作驗證。


論文資訊
📄 Guiding a Diffusion Model with a Bad Version of Itself
👥 Karras, Aittala, Kynkäänniemi, Laine, Aila, Lehtinen
🏆 NeurIPS 2024 · Best Paper Runner-Up
🔗 arxiv.org/abs/2406.02507

Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Prediction 深度解析

近年來,生成式模型在影像合成領域掀起革命性浪潮,擁有顛覆傳統圖像生成技術的強大潛力。特別是在自回歸(autoregressive, AR)模型與擴散模型(diffusion models)之間,各有優劣勢的競爭情景令研究社群熱切關注。2024 年 NeurIPS 最佳論文《Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Prediction》由 Tian 等人提出一種名為 Visual Autoregressive modeling (VAR) 的全新影像生成方法,徹底改寫了 AR 在視覺任務中的學習框架與效能表現,開啟了 AR 模型在生成品質與效率上的新紀元。本文將帶您深入剖析該論文的研究動機、技術架構、實驗成果與其對 AI 領域的深遠意義。

研究背景與動機

傳統的自回歸影像生成演算法,多半依循「由左上到右下逐像素(raster-scan)預測下一像素」的策略,這種「next-token prediction」方法在自然語言處理(如 GPT 模型)中顯現強大能力,但在圖像領域卻面臨嚴峻挑戰。原因在於:影像不同於一維文字序列,其空間結構是二維的且局部特徵間存在複雜的多尺度關係,逐像素預測導致長序列長度、計算負擔沉重,且難以建模長距離的依賴結構,生成速度緩慢且品質受限。

與此同時,擴散模型憑藉從噪聲逐步逆轉過程實現高品質影像生成,已成為當前的主流技術,特別是在圖像細節捕捉與擴展性上展現優勢。然而,擴散模型通常推論速度慢,且訓練與調參成本高昂。這使得研究者渴望找到一種介於 AR 模型直觀且高效擴展、與擴散模型生成效果間的最佳平衡點。

基於此,作者提出了 VAR,試圖將自回歸預測概念跳脫傳統的像素 順序,導入多尺度的「next-scale prediction」,每一步預測的是「下一解析度層級」的圖像,從粗糙到細緻逐層遞進,既符合影像的多層次結構,也極大提高模型的訓練速度與生成效率,是對 AR 領域的關鍵突破。

核心方法與創新

VAR 的關鍵創新在於重新定義自回歸學習目標:從「下一像素」預測轉變為「下一解析度層級」預測。具體來說,VAR 將 256×256 的高解析度影像分解成多個逐步細化的尺度版本,模型從極低解析度(例如 8×8)開始進行預測,逐層輸出更高解析度的影像,直到完成全尺寸 256×256 圖像。

這種由粗到細的預測策略有幾大技術優點:

  • 序列長度大幅縮短:影像在低解析度時數據量極少,預測序列縮短數十倍,核心的 Transformer 模型可更快速且高效地學習視覺分布。
  • 捕捉多尺度結構:模型透過每層尺度捕捉不同層次的影像資訊,粗層維持整體構造,細層雕琢細節,符合人類視覺感知的天然順序。
  • 優化推論效率:不必逐像素生成,有效降低計算成本,實現近 20 倍推論加速,適合大規模、多樣化應用。

為了實作該機制,作者設計了專門的 Transformer 架構與訓練流程,在每個尺度上預測出擴展的「高頻細節」,並利用殘差引導逐層合成影像。此方法與傳統的 AR raster scan 方式明顯不同,且更符合圖像分布的真實結構,促進模型學習更深厚的視覺表徵。

主要實驗結果

VAR 在三個核心指標上均展現出對 AR 基線甚至頂尖擴散模型的超越表現。以下為幾項令人印象深刻的數據:

  • ImageNet 256×256 資料集上,Frechet Inception Distance (FID):VAR 從傳統 AR 的 18.65 降至 1.73,顯著接近甚至超越部分擴散模型的品質水準。
  • Inception Score (IS):由 80.4 飆升至 350.2,這是圖像內容多樣性與真實感的重要評量,VAR 展現出卓越的生成力。
  • 推論速度:VAR 實現約 20 倍的速度提升,突破 AR 模型過去推論緩慢的瓶頸。

此外,作者將 VAR 與 Diffusion Transformer (DiT) 進行多維度比較。結果顯示 VAR 不僅在圖像質量和速度佔優,更在數據效率和模型可擴展性上具備明顯優勢。特別值得一提的是,VAR 模型遵從類似大規模語言模型(LLMs)的尺度律(power-law scaling laws),連結係數高達 -0.998,充分驗證了 VAR 在擴展能力上的穩健表現。

最後,VAR 不但本身能生成高品質圖片,還展示了優越的「零樣本泛化」能力。在諸如圖像修補(in-painting)、外推擴展(out-painting)與影像編輯等下游任務上,VAR 無需額外微調即可執行,強化了其通用性與實用性。

對 AI 領域的深遠影響

VAR 的誕生不只是一種技術上的突破,更在幾個關鍵層面推動著 AI 生成模型未來的發展:

1. 自回歸模型的復興與超越

VAR 有效打破了傳統 AR 在影像生成中「生成序列過長及速度緩慢」的瓶頸,使得 AR 模型真正有機會媲美甚至超越擴散架構。這不僅強化了 AR 模型在視覺任務中的生命力,也為模型設計帶來新的思路。

2. 多尺度「next-scale prediction」策略的價值

VAR 所引入的由粗到細逐層預測觀念,與人類觀察影像的認知過程高度契合,展現出對複雜空間結構建模的強大能力。這也促使更多研究開始關注多層次、多尺度的生成架構,豐富影像生成方法論。

3. 生成模型的可擴展性與效率革新

VAR展現出 LLM 類似的規模律,這表明透過適當架構設計,視覺生成模型也能獲得良好的擴展潛力,並且顯著提升訓練與推論效率。隨著模型規模擴大,VAR 的性能持續提升,對於處理日益多樣、複雜的視覺任務具備深遠意義。

4. 統一學習與零樣本泛化

VAR 在多種下游視覺任務展現無需微調的即時應用能力,與當前 NLP 領域 LLM 的跨任務泛化性質相呼應,有望推進視覺生成模型的統一框架建構,促進多任務、多場景的無縫應用。

5. 開源推廣與研究生態激活

作者隨論文釋出了完整模型與代碼,鼓勵產學界持續探究和優化自回歸與 VAR 類模型,促進生成模型技術在學術和工業界的廣泛交流合作。

結語

總結來說,《Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Prediction》這篇 NeurIPS 2024 最佳論文在視覺自回歸生成領域開創了一條嶄新道路。透過巧妙的多解析度「下一層級預測」策略,VAR 不僅顛覆傳統 AR 模型的生成格局,戰勝擴散模型在多項指標上的限制,還實現了推論大幅加速與優異的泛化能力。這些成果不僅擴展了我們對圖片生成本質的理解,也預示著未來視覺生成模型將朝向更加高效、靈活且統一的方向演進。對於具備基礎 AI 知識的工程師與研究生而言,VAR 代表了新世代圖像生成技術的重要里程碑,值得持續關注與深入研究。


論文資訊
📄 Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Prediction
👥 Tian, Jiang, Yuan, Peng, Wang
🏆 NeurIPS 2024 · Best Paper
🔗 arxiv.org/abs/2404.02905

2026年6月12日 星期五

KAN: Kolmogorov-Arnold Networks 深度解析

在現代深度學習的發展中,如何設計更高效、更具表現力且理論基礎堅實的網路架構,一直是研究焦點。2024 年 ICLR 發表的論文《KAN: Kolmogorov-Arnold Networks》由劉強等人提出,榮獲 Outstanding Paper 獎,該研究從數學理論的經典定理出發,開創性地導入 Kolmogorov-Arnold 分解定理於神經網路設計,提出了一種全新的網路架構——KAN(Kolmogorov-Arnold Networks)。這不僅在理論上突破了傳統網路的建模限制,也在多項實際任務上展現出卓越性能,進而推動 AI 研究向更具解釋性與表達能力的方向邁進。

研究背景與動機

深度神經網路已成功應用於圖像識別、自然語言處理等多種領域,但其架構設計大多依賴經驗法則,且缺乏嚴謹的數學支持。過去的工作多以通用函數逼近定理(universal approximation theorem)為基石,證明隱層神經元足夠時可以近似任意連續函數,但在實際運算中怎樣利用數理結構提升網路性能仍不可得見。Kolmogorov-Arnold 表示定理作為一項經典函數近似理論,指出任何多變量連續函數皆可表示為多組單變量函數的組合。這一理論的潛力極大,卻少被應用於神經網路架構設計。

KAN 論文的動機正是從這個角度出發:若能設計網路結構直接對應到 Kolmogorov-Arnold 定理的函數分解形式,讓神經網路學習過程更貼近數學本質,則不但增加可解釋性,也有助於提升效率及泛化能力。此外,該理論啟發了一種解析度分離的視角,有助於處理高維輸入時的「維度災難」問題。

核心方法與創新

KAN 的方法核心在於將 Kolmogorov-Arnold 表示定理形式化置入網路架構。簡單來說,這個定理指出對於任意連續函數 f(x_1, x_2, ..., x_n),存在一組單變量連續函數 φ_q 和線性函數 ψ_p,使得:

f(x_1, ..., x_n) = ∑_{q=1}^{2n+1} φ_q(∑_{p=1}^n ψ_{pq}(x_p))

KAN 網路以此為藍本,構造出兩層明確結構的子網。第一層為多組單變量函數子網,專門學習 φ_q,第二層則學習線性組成的 ψ_{pq} 部分。這種架構將多變量函數逼近問題分解成多個單變量函數學習任務與線性組合,有效避免傳統網路中多變量耦合高維度帶來的計算與學習障礙。作者利用多層感知器 (MLP) 結合可訓練的線性映射,打造出層次分明但能端到端訓練的架構設計。

此外,針對單變量函數部分,論文提出加強函數逼近的正則化方法,確保 φ_q 在學習期間的光滑性及有效性,並利用模塊化結構優化計算效率。總體而言,KAN 架構理論嚴謹,設計巧妙,同時保持高度的靈活性,支持多種激活函數與優化策略。

主要實驗結果

作者在多個合成及真實數據集上展開實驗以驗證 KAN 的有效性。實驗包括:函數分析模擬、多維回歸問題、圖像分類與信號處理等。

  • 函數近似任務:KAN 在逼近各種類型多維非線性函數(如三角函數、指數函數及高斯函數等)時,控制參數量較少的情況下達成明顯低於傳統全連接網路的逼近誤差,證明其數學結構支撐了更高的表達效率。
  • 高維回歸問題:對比普通深度神經網路與 Transformer 為基礎的回歸模型,KAN 在保持較低計算複雜度下,實現了更穩定且高精度的預測性能,展示其在維度災難情況下的優越性。
  • 圖像處理任務:將 KAN 作為卷積神經網路的替代或補充模組,在 CIFAR-10 和 ImageNet 下進行測試,表現出相媲美甚至優於標準架構的效果,證明架構的實用可行性。
  • 泛化能力與穩定性:實驗中 KAN 不僅能加速收斂,且對輸入干擾的魯棒性較傳統網路強,體現出其架構設計中函數分解自然帶來的正則化效果。

綜合這些實驗數據,KAN 不僅從理論層面提供了一套完備的函數逼近方案,也在多種實際問題中具備優越性能,這是目前少見兼具理論嚴謹與實用價值的模型之一。

對 AI 領域的深遠影響

KAN 的提出標誌著 AI 算法設計從「黑盒」試探式調參,逐步向數學原理嚴謹化靠攏。利用 Kolmogorov-Arnold 定理這一悠久且深刻的數學成果來指導網路結構設計,示範了跨學科整合的強大潛力。此舉不僅提升了模型的可解釋性,也讓研發者有了更明確且可控的架構設計路徑,推動理論與實踐雙向融合。

從實務角度來看,KAN 透過多層次且結構化的函數分解,有效緩解了高維度輸入中模型參數爆炸與訓練不穩定問題,為未來面對高維數據(例如多模態、科學計算仿真等領域)提供了新的思路與工具。由於網路構成模塊清晰且物理意義明確,KAN 還可能帶動對可解釋 AI 與可信 AI 的進一步探索。

未來研究層面,KAN 激發了將更多經典數學定理導入深度學習框架的想像空間。譬如將代數、拓撲方法與概率理論結合到網路設計中,有助於提升 AI 模型的泛用性與安全性。此外,KAN 架構本身也具備擴展性,可望結合圖神經網路(GNN)、變分推理等技術,為複雜系統建模提供更優解。

總結而言,《KAN: Kolmogorov-Arnold Networks》是一篇結合理論深度與實驗創新,且對 AI 機制理解與實際應用同時有突破意義的傑出論文。對於 AI 研究者與工程師而言,深入理解此文不僅有助於掌握最前沿的架構設計理念,也加深了對深度學習本質的認識,並啟發將來更多跨學科研究的可能。


論文資訊
📄 KAN: Kolmogorov-Arnold Networks
👥 Liu, Wang, Vaidya, Ruehle, Halverson, Soljačić, Hou, Tegmark
🏆 ICLR 2024 · Outstanding Paper
🔗 arxiv.org/abs/2404.19756

Mamba: Linear-Time Sequence Modeling with Selective State Spaces

隨著深度學習基礎模型迅速發展,Transformer 架構因其靈活的注意力機制,成為處理序列資料(如語言、音訊、基因組等)最主流且效果卓越的架構。然而,Transformer 在長序列上的計算複雜度達到二次方等級,造成計算資源消耗及推論延遲顯著增加,限制了其在超長序列上的應用。面對這一瓶頸,學界發展了多種亞二次方複雜度的架構,包括線性注意力(linear attention)、門控卷積(gated convolution)、循環模型及結構化狀態空間模型(structured state space models, SSMs)等,但這些方法在具挑戰性的離散序列(如自然語言)上的表現仍未達到 Transformer 級別,難以全面取代。

本篇於 ICLR 2024 獲獎的論文《Mamba: Linear-Time Sequence Modeling with Selective State Spaces》由 Gu Dao 等人提出了一種嶄新的序列建模架構——Mamba。作者察覺到現有 SSMs 及類似模型的核心缺陷在於缺乏內容為本(content-based)的推理能力,使得模型無法根據當前輸入動態選擇性地保存或遺忘訊息,尤其在處理符號化或切割明確的離散資料時效果不佳。基於此,Mamba 重新設計了 SSM 參數,使之成為輸入的函數,賦予模型動態調節信息流的能力,實現選擇性記憶與遺忘,提升對內容的響應能力。

研究背景與動機

Transformer 依賴的自注意力機制在序列長度增加時,計算與記憶消耗呈平方級別增長(O(N²)),這是其無法有效處理超長序列的主要限制。為改善此一困境,研究者嘗試多樣方法,希望在保有 Transformer 精準建模能力同時,將複雜度降低。例如,線性注意力透過近似注意力分數計算減少複雜度,SSM 使用連續時間狀態空間理論,藉由求解微分方程描述長序列的長距依賴。然而,這些替代模型多依賴固定且與輸入無關的參數,缺少依據當下內容動態調節記憶的能力,因此在離散且結構複雜的資料(語言、程式碼等)上表現不佳,這個缺陷成為推廣這些亞二次方模型的關鍵瓶頸。

核心方法與創新

本論文的首要創新為引入「選擇性狀態空間模型(Selective SSMs)」。具體做法是,讓 SSM 模型中的參數不再是固定設定,而是透過神經網路學習輸入依賴的函數。換言之,模型對每個時間步的輸入 token 進行判斷,動態調節狀態轉換矩陣及輸出映射,從而有能力「選擇性地」保留或刪除序列資訊。

然而,引入參數依賴性後,原本條件下能應用高效率捲積加速的計算路徑被打破,不再能使用快速傅里葉等方法直接並行計算。為此,作者設計了「硬體感知並行演算法」,在循環模式下仍能充分利用現代硬體(如 GPU、TPU)的向量化與平行運算優勢,達成運算效率的大幅提升。

基於此,研究團隊打造出全新架構—Mamba。Mamba 完全去除 Transformer 的注意力模組及多層感知機(MLP)層,僅基於選擇性 SSM 組成極簡終端對端網路。此架構在保持序列建模能力的同時,以線性時間複雜度處理超長序列,且預先訓練與下游任務均展現卓越性能。

主要實驗結果

研究團隊在多模態真實數據集上做了嚴謹驗證,包括語言模型訓練、音訊處理及基因序列分析。結果顯示:

  • Mamba 推論速度比傳統 Transformer 快約 5 倍,在千萬級序列長度的推理效率與表現上具有顯著優勢。
  • 在語言建模任務中,Mamba-3B(參數約 30 億)超越了尺寸相同的 Transformer,並且表現不亞於規模達 60 億參數的 Transformer 模型,無論在預訓練還是微調任務皆如此。
  • 跨音訊與基因序列資料,Mamba 也取得或競爭目前最先進的結果,展示其多模態泛化能力與強勁的泛用性。

對 AI 領域的深遠影響

Mamba 不僅展示了一條可行且高效的超長序列建模路徑,更具體化了讓模型具備「動態選擇記憶」的可能,大幅提升了結構化狀態空間模型在離散模態中的表現瓶頸,為序列建模引入了新的範式。

此項研究推動亞二次方複雜度架構從理論走向實務,解決了 Transformer 在超長序列任務上的痛點。其提出的硬體感知並行演算法及依輸入動態調整參數的設計,在效率與效果間取得絕佳平衡,為未來神經網路在計算機硬體上的最佳化提供重要啟示。

此外,Mamba 架構的極簡設計及領先性能,將促使更多基礎模型架構擺脫「注意力—MLP」的傳統套路,挖掘狀態空間理論與參數動態調節的潛力,對自然語言處理、音訊分析及基因資料等多領域均可產生深遠影響。

綜合而言,Mamba 代表了序列模型設計上的一大躍升:在處理長序列時擁有線性時間複雜度,保持甚至超越 Transformer 的強建模能力,同時兼顧高效推論與跨模態泛化,為未來基礎模型開發及應用鋪陳了嶄新的技術藍圖。對正在追求高效且可擴展序列建模的工程師與研究者而言,Mamba 提供了值得深入研究與借鑒的最新前沿成果。


論文資訊
📄 Mamba: Linear-Time Sequence Modeling with Selective State Spaces
👥 Gu, Dao
🏆 ICLR 2024 · Outstanding Paper
🔗 arxiv.org/abs/2312.00752