arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2101.04844cs.LGcs.CVcs.NAmath.NA
收藏

Reproducing Activation Function for Deep Learning

Purdue University(普渡大学);Michigan State University(密歇根州立大学);Texas Tech University(德克萨斯理工大学)

Senwei Liang, Liyao Lyu, Chunmei Wang, Haizhao Yang

英文摘要:

We propose reproducing activation functions (RAFs) to improve deep learning accuracy for various applications ranging from computer vision to scientific computing. The idea is to employ several basic functions and their learnable linear combination to construct neuron-wise data-driven activation functions for each neuron. Armed with RAFs, neural networks (NNs) can reproduce traditional approximation tools and, therefore, approximate target functions with a smaller number of parameters than traditional NNs. In NN training, RAFs can generate neural tangent kernels (NTKs) with a better condition number than traditional activation functions lessening the spectral bias of deep learning. As demonstrated by extensive numerical tests, the proposed RAFs can facilitate the convergence of deep learning optimization for a solution with higher accuracy than existing deep learning solvers for audio/image/video reconstruction, PDEs, and eigenvalue problems. With RAFs, the errors of audio/video reconstruction, PDEs, and eigenvalue problems are decreased by over 14%, 73%, 99%, respectively, compared with baseline, while the performance of image reconstruction increases by 58%.

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

↑