arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2409.14174cs.LGmath.STstat.TH

基于组件的深度 ReLU 网络草图方法

Component-based Sketching for Deep ReLU Nets

School of Science, China Jiliang University(中国计量大学理学院)

Di Wang, Shao-Bo Lin, Deyu Meng, Feilong Cao

AI总结:

针对深度网络优化与泛化不一致问题,提出基于深度网络组件的草图方法,将训练转化为线性经验风险最小化问题,理论证明其达到几乎最优泛化误差界,实验显示在降低训练成本的同时提升了泛化性能。

摘要:

深度学习在数据挖掘和人工智能领域产生了深远影响,其在众多实际应用中的突破性成就和创新算法设计理念尤为突出。然而,它存在优化与泛化之间的不一致性问题,因为遵循偏差-方差权衡原则实现良好泛化更倾向于欠参数化网络,而确保基于梯度的算法有效收敛则需要过参数化网络。为解决这一问题,我们针对各种任务开发了一种基于深度网络组件的新型草图方法。具体而言,我们利用具有特定效能的深度网络组件构建了一个体现深度网络优势的草图基。随后,我们基于所构建的基,将深度网络训练转化为线性经验风险最小化问题,成功避免了迭代算法复杂的收敛性分析。所提出的基于组件的草图方法的有效性通过理论分析和数值实验得到了验证。理论上,我们证明了所提出的基于组件的草图方法在浅层网络近似饱和函数时提供了几乎最优的速率,并且也实现了几乎最优的泛化误差界。在数值实验上,我们证明了与现有的基于梯度的训练方法相比,基于组件的草图方法具有更优的泛化性能,同时降低了训练成本。

英文摘要:

Deep learning has made profound impacts in the domains of data mining and AI, distinguished by the groundbreaking achievements in numerous real-world applications and the innovative algorithm design philosophy. However, it suffers from the inconsistency issue between optimization and generalization, as achieving good generalization, guided by the bias-variance trade-off principle, favors under-parameterized networks, whereas ensuring effective convergence of gradient-based algorithms demands over-parameterized networks. To address this issue, we develop a novel sketching scheme based on deep net components for various tasks. Specifically, we use deep net components with specific efficacy to build a sketching basis that embodies the advantages of deep networks. Subsequently, we transform deep net training into a linear empirical risk minimization problem based on the constructed basis, successfully avoiding the complicated convergence analysis of iterative algorithms. The efficacy of the proposed component-based sketching is validated through both theoretical analysis and numerical experiments. Theoretically, we show that the proposed component-based sketching provides almost optimal rates in approximating saturated functions for shallow nets and also achieves almost optimal generalization error bounds. Numerically, we demonstrate that, compared with the existing gradient-based training methods, component-based sketching possesses superior generalization performance with reduced training costs.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑