核心概念
POMONAG 透過整合多目標反向擴散引導、擴展元資料集和改進的性能預測器,在保持高準確性的同時,生成在計算資源和速度上都高效的神經網路架構。
摘要
本文提出了一個名為POMONAG的創新框架,用於生成在多目標優化下的神經網路架構。POMONAG 同時考慮了準確性、參數數量、乘法累加運算(MACs)和推理延遲等多個目標,並將其整合到反向擴散引導過程中。為了增強POMONAG的性能和適應性,作者進行了以下改進:
- 顯著擴展了元資料集的規模和多樣性,以提高性能預測器的能力。
- 採用了帕累托前沿過濾和拉伸技術,有效地平衡了多個目標之間的權衡。
- 優化了性能預測器的架構和訓練方法,提高了預測準確性。
實驗結果表明,POMONAG 在 NASBench201 和 MobileNetV3 搜索空間上,都能生成在準確性和效率方面都優於現有最先進方法的神經網路架構。與傳統方法相比,POMONAG 只需要訓練一個最優架構,大幅降低了計算開銷。這些結果突出了POMONAG在生成帕累托最優架構方面的有效性,可以適應各種部署場景和資源限制。
统计
在 NASBench201 搜索空間上,POMONAG的POMONAGAcc配置的平均準確率為81.89%,優於DiffusionNAG的77.83%。
在 MobileNetV3 搜索空間上,POMONAG的POMONAGEff配置將參數數量減少了56%,MACs減少了67%,推理延遲減少了44%,相比DiffusionNAG有顯著提升。
POMONAG的POMONAGAcc配置在 MobileNetV3 搜索空間上的平均準確率為89.96%,比DiffusionNAG提高了24%,同時參數數量減少了10%。
引用
"POMONAG 同時考慮了準確性、參數數量、乘法累加運算(MACs)和推理延遲等多個目標,並將其整合到反向擴散引導過程中。"
"POMONAG 只需要訓練一個最優架構,大幅降低了計算開銷。"
"POMONAG 在生成帕累托最優架構方面的有效性,可以適應各種部署場景和資源限制。"