Skip to main content
QUICK REVIEW

[論文レビュー] A Solution for Large Scale Nonlinear Regression with High Rank and Degree at Constant Memory Complexity via Latent Tensor Reconstruction

Sándor Szedmák, Anna Cichońska|arXiv (Cornell University)|May 4, 2020
Tensor decomposition and applications参考文献 31被引用数 4
ひとこと要約

本稿では、高次多項式および高ランクテンソルを伴う大規模非線形回帰のための潜在的テンソル再構成(LTR)手法を提案する。本手法は、ランク1のテンソル成分の段階的勾配ベース学習により、定常的メモリ複雑度を達成する。このアプローチにより、線形時間複雑度と最先端手法(FM や KRR)に比べ優れた精度を実現する大規模データセット上での複雑な非線形関数の効率的かつスケーラブルな学習が可能になる。

ABSTRACT

This paper proposes a novel method for learning highly nonlinear, multivariate functions from examples. Our method takes advantage of the property that continuous functions can be approximated by polynomials, which in turn are representable by tensors. Hence the function learning problem is transformed into a tensor reconstruction problem, an inverse problem of the tensor decomposition. Our method incrementally builds up the unknown tensor from rank-one terms, which lets us control the complexity of the learned model and reduce the chance of overfitting. For learning the models, we present an efficient gradient-based algorithm that can be implemented in linear time in the sample size, order, rank of the tensor and the dimension of the input. In addition to regression, we present extensions to classification, multi-view learning and vector-valued output as well as a multi-layered formulation. The method can work in an online fashion via processing mini-batches of the data with constant memory complexity. Consequently, it can fit into systems equipped only with limited resources such as embedded systems or mobile phones. Our experiments demonstrate a favorable accuracy and running time compared to competing methods.

研究の動機と目的

  • 高次多項式度および高ランクを伴う大規模データセットからの極めて非線形的かつ多次元関数の学習という課題に対処すること。
  • トレーニング中に定常的メモリ複雑度を維持することにより、従来のテンソルベース手法のメモリおよび計算上のボトル neck を克服すること。
  • リソース制約のある環境(例:モバイルデバイスや組み込みシステム)に適した、ミニバッチ処理を用いたオンライン学習を効率的に行えるようにすること。
  • FM における対称的因子分解にとどまらず、一般の非対称非線形関数へも多項式モデルの適用範囲を拡張すること。
  • 合成データおよび実世界のデータセット(特にマルチラベル画像アノテーションタスクを含む)において、強力な性能を示すこと。

提案手法

  • 連続関数が多項式によって近似可能であるという事実を活用し、非線形関数学習をテンソル再構成問題に再定式化すること。
  • 各 $ \ell_d^{(t)} $ が線形形式である線形形式による因子分解を用いて多項式を表現する:$ f(\mathbf{x}) = \sum_{t=1}^{n_t} \prod_{d=1}^{n_d} \ell_d^{(t)}(\mathbf{x}) $。
  • サンプル数、次数、ランク、入力次元に関して線形時間複雑度を達成するため、勾配ベース最適化アルゴリズムを用いてテンソル成分を段階的に学習すること。
  • ミニバッチ形式で実装することで、定常的メモリ複雑度を維持し、オンラインおよびスケーラブルな学習を可能にすること。
  • 適切な損失関数およびテンソル構造を用いて、分類、マルチビュー学習、ベクトル値出力へとフレームワークを拡張すること。
  • 深層ニューラルネットワークに類似した階層的非線形相互作用をモデル化するためのマルチレイヤード定式化を統合すること。

実験結果

リサーチクエスチョン

  • RQ1大規模データから高次・高ランク非線形関数を学習する際、テンソルベース手法が定常的メモリ複雑度を達成できるか。
  • RQ2提案された潜在的テンソル再構成手法は、FM や KRR といった既存手法と比較して、精度および効率性においてどの程度優れているか。
  • RQ3多項式次数に関して線形時間複雑度を維持しつつ、1,000万例を超えるデータセットへどの程度スケーリングできるか。
  • RQ4非対称で複雑な非線形関数において、対称的多項式モデル(例:FM)よりも性能を上回るか。
  • RQ5スパarsity が極めて高い実世界のマルチラベル予測タスク(例:画像アノテーション)に、本手法は効果的に適用可能か。

主な発見

  • 高ノイズレベル、高次多項式度、多数の変数を伴う合成データにおいて、LTR は FM や TFFM よりも顕著に高い予測精度を達成する。
  • 小規模データセット(5,000例)では、パラメータ空間が大きいため LTR や TFFM は KRR よりも性能が劣るが、大規模データセットでは LTR がより優れたスケーラビリティを示す。
  • 実験により、多項式次数に関して線形時間複雑度を示すことが確認され、1,000万件のサンプルを含む実験で KRR よりも優れた性能を発揮する。
  • 実世界のマルチラベル画像データセット(Corel5k, Espgame, Iaprtc12)において、それぞれ F1 スコア 42.7%、27.2%、37.0% を達成し、MBRM や TagProp、FastTag よりも優れた性能を示す。
  • トレーニング中に定常的メモリ複雑度を維持するため、モバイルデバイスや組み込みシステムなどリソース制限のあるシステムへのデプロイが可能になる。
  • LTR アルゴリズムは、10^7例の規模ですら多項式次数に比例して実行時間が線形に増加するため、競争力あるトレーニング時間を達成する。一方、KRR はスケーリングが著しく劣る。

より良い研究を、今すぐ始めましょう

論文の読解から最終レビューまで、研究時間を劇的に削減しましょう。

クレジットカード登録不要

このレビューはAIが作成し、人間の編集者が確認しました。