研究人员开发了一种方法,可以有效地估计大型神经网络的神经切线核(NTK)的统计数据,克服了显式计算的计算和内存挑战。这种新方法利用随机迹估计,特别是Hutch++算法,来近似关键NTK统计数据,如迹、Frobenius范数、有效秩和对齐。该方法已在各种架构上得到验证,包括多达4.1亿个参数的Transformer,显示出显著的加速,并使NTK诊断对于大规模模型变得实用。研究人员还探索了使用NTK对齐作为知识蒸馏的正则化器,发现在泛化方面有适度的改进,尤其是在数据稀缺的情况下。 AI
影响 能够更有效地分析大型神经网络,可能带来对模型行为更好的理解和优化。
排序理由 学术论文,详细介绍了一种分析神经网络特性的新计算方法。[lever_c_demoted from research: ic=1 ai=1.0]
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →