像素擴散Transformer參數化新方法MFlow提出,並配套ConvRef…

論文AI 評分 66/100arXiv cs.AI
AI 聚合

本條為 AI 依據下方公開信源自動整理生成的摘要,不構成轉載,可能存在偏差,請以原文為準。整理者:貝果科技 AI 資訊助手。

論文重新審視像素空間擴散模型中用乾淨圖像預測(x0-prediction)替代速度預測(u-prediction)的做法,提出自適應參數化方法MFlow:它在數據的主方向上學習在x0與u兩種預測之間插值,並與擴散訓練端到端聯合學習。研究發現,偏好的參數化方式會隨數據表示與模型容量系統性變化,模型容量增大時更傾向速度預測。作者在ImageNet、UCF-101影片與大規模文生圖任務上評測MFlow,生成表現相當或更好。論文還指出,像素擴散Transformer相對真實圖像存在持續的高頻缺失,為此提出輕量卷積模組ConvRefiner,在凍結擴散主幹的前提下用對抗損失做後置訓練,少量額外訓練即可恢復高頻統計量並保持FID。

全文梳理

AI 摘要依據下方信源原文自動整理,非原文轉載

參數化選擇的再審視

像素空間擴散近期重新受到關注,原因在於它繞開了有損的潛空間瓶頸和單獨訓練的自編碼器。此前JiT的做法是把常用的速度預測換成乾淨圖像預測,從而讓普通擴散Transformer直接在像素空間生成。這篇論文對這一選擇提出疑問:兩種預測方式未必只能二選一。作者提出的MFlow讓模型在數據的主方向上學習兩種預測之間的插值比例,該比例不是人工設定,而是與擴散訓練一起端到端學出來。論文給出的觀察是,哪種參數化更合適並非固定結論,它會隨數據表示形式和模型容量發生系統性偏移,容量越大越偏向速度預測。這一發現意味着,像素擴散的設計空間比此前單點方案更寬。

多場景評測結果

作者把MFlow放到三類任務上檢驗:ImageNet圖像生成、UCF-101影片生成,以及大規模文生圖。論文的表述是在這些不同設定下取得相當或更好的生成表現,覆蓋靜態圖像、影片與文本條件生成三類場景。對讀者而言,值得關注的是這套自適應參數化沒有綁定在單一數據模態或單一規模上,而是在差異較大的任務中都能用。論文同時給出33頁、23張圖、15張表的篇幅,說明評測覆蓋面較廣,具體數值需查閱原文表格。

高頻缺失與ConvRefiner

論文另一個獨立發現是:像素擴散Transformer不管採用哪種參數化,生成的圖像相對真實圖像都存在持續的高頻成分不足。這不是參數化選擇能解決的問題,因此作者單獨設計了ConvRefiner——一個輕量卷積模組。它的訓練方式是在擴散主幹完全凍結的前提下,用對抗損失做後置訓練,只增加少量訓練量。效果上,ConvRefiner恢復了生成圖像的高頻統計特性,同時FID保持不變。這一做法的意義在於把「主幹生成質素」與「高頻細節補足」拆成兩步,避免為修補細節而重新訓練整個擴散模型。

為什麼值得看

想了解像素空間擴散能否替代潛空間方案、以及參數化選擇如何隨模型規模變化,這篇給出了可復現的設計取捨與補救模組。

文生圖論文

信源1 家

  1. [1]arXiv cs.AI一手信源Rethinking the design space of Vanilla Pixel Diffusion Transformers

關鍵事實

  • 論文提出MFlow,一種在數據主方向上學習在x0預測與u預測之間插值的自適應參數化,並與擴散訓練端到端學習。[1]

  • 研究發現偏好的參數化隨數據表示和模型容量系統性變化,模型容量增大時更傾向速度預測。[1]

  • MFlow在ImageNet、UCF-101影片與大規模文生圖任務上評測,生成性能相當或更好。[1]

  • 論文指出像素擴散Transformer無論採用何種參數化,相對真實圖像都存在持續的高頻缺失。[1]

  • ConvRefiner是輕量卷積模組,在凍結擴散主幹的情況下用對抗損失後置訓練,少量額外訓練即可恢復高頻統計量並保持FID。[1]

相關 · 論文

本頁內容由 AI 自動聚合公開信源生成,僅供了解行業動態參考,不構成任何投資或決策建議。如需引用請以原文出處為準。