arXiv reaDer
StitchNet: 事前トレーニング済みのフラグメントからニューラル ネットワークを構成する
StitchNet: Composing Neural Networks from Pre-Trained Fragments
複数の事前トレーニング済みニューラル ネットワークからフラグメント (1 つまたは複数の連続したネットワーク レイヤー) をつなぎ合わせる新しいニューラル ネットワーク作成パラダイムである、StitchNet を提案します。 StitchNet を使用すると、バックプロパゲーション トレーニングによる従来のモデル作成プロセスで必要とされる大規模な計算およびデータ要件なしで、高性能ニューラル ネットワークを作成できます。特定の精度のニーズと計算リソースの制約に合わせて調整された特定のタスク用のネットワークを構成する際に、これらのフラグメントの選択を効率的にガイドするための互換性対策として、Centered Kernel Alignment (CKA) を活用します。次に、これらのフラグメントをつなぎ合わせて、コンピューティング リソースとデータ要件のほんの一部で、従来のトレーニング済みネットワークに匹敵する精度でニューラル ネットワークを作成できることを示します。最後に、この新しいパラダイムによって可能になる、斬新なオンザフライのパーソナライズされたモデル作成と推論アプリケーションを探ります。
We propose StitchNet, a novel neural network creation paradigm that stitches together fragments (one or more consecutive network layers) from multiple pre-trained neural networks. StitchNet allows the creation of high-performing neural networks without the large compute and data requirements needed under traditional model creation processes via backpropagation training. We leverage Centered Kernel Alignment (CKA) as a compatibility measure to efficiently guide the selection of these fragments in composing a network for a given task tailored to specific accuracy needs and computing resource constraints. We then show that these fragments can be stitched together to create neural networks with comparable accuracy to traditionally trained networks at a fraction of computing resource and data requirements. Finally, we explore a novel on-the-fly personalized model creation and inference application enabled by this new paradigm.
updated: Mon Jul 17 2023 09:06:53 GMT+0000 (UTC)
published: Thu Jan 05 2023 08:02:30 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト