Optimization plays a crucial role in a vast range of problems in science, engineering, and data analytics including the training of machine-learning models and performing statistical inference. Algorithm unrolling accelerates the iterative optimization process by learning efficient iterative updates. We propose a method that improves alternating optimization for nonnegative matrix factorization (NMF). Unlike conventional unrolling, which treats a fixed number of iterations as a parametric network, our approach learns a single update step near optimum. Furthermore, unlike most existing methods that assume a fixed sensing matrix, our model can factorize data generated through previously unseen sensing matrices. Experimental results demonstrate that a model trained on randomly generated artificial data successfully decomposes spectral data.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Unrolled Neural Adaptive Alternating Gradient Descent for NMF

  • Toshimitsu Aritake

摘要

Optimization plays a crucial role in a vast range of problems in science, engineering, and data analytics including the training of machine-learning models and performing statistical inference. Algorithm unrolling accelerates the iterative optimization process by learning efficient iterative updates. We propose a method that improves alternating optimization for nonnegative matrix factorization (NMF). Unlike conventional unrolling, which treats a fixed number of iterations as a parametric network, our approach learns a single update step near optimum. Furthermore, unlike most existing methods that assume a fixed sensing matrix, our model can factorize data generated through previously unseen sensing matrices. Experimental results demonstrate that a model trained on randomly generated artificial data successfully decomposes spectral data.