arXiv reaDer
適応展開のためのマトリョーシカ表現
Matryoshka Representations for Adaptive Deployment
学習した表現は、最新のMLシステムの中心的なコンポーネントであり、多数のダウンストリームタスクに対応します。このような表現をトレーニングする場合、各ダウンストリームタスクの計算および統計上の制約が不明であることがよくあります。このコンテキストでは、固定容量の表現は、目前のタスクに過剰に対応することも、不十分に対応することもできます。これにより、次のように質問することになります。さまざまな計算リソースを使用して複数のダウンストリームタスクに適応できる柔軟な表現を設計できますか?私たちの主な貢献は、マトリョーシカ表現学習(MRL)です。これは、さまざまな粒度で情報をエンコードし、単一の埋め込みがダウンストリームタスクの計算上の制約に適応できるようにします。 MRLは、既存の表現学習パイプラインを最小限に変更し、推論および展開中に追加のコストを課しません。 MRLは、独立してトレーニングされた低次元表現と少なくとも同じくらい正確で豊富な粗い表現から細かい表現を学習します。学習したマトリョーシカ表現の柔軟性により、次のことが可能になります。(a)同じレベルの精度でImageNet-1K分類の埋め込みサイズを最大14倍小さくする。 (b)ImageNet-1Kおよび4Kでの大規模な検索のための最大14倍の実世界のスピードアップ。 (c)ロングテールの数ショット分類で、最大2%の精度の向上。すべて、元の表現と同じくらい堅牢です。最後に、MRLがさまざまなモダリティ(ビジョン(ViT、ResNet)、ビジョン+言語(ALIGN)、言語(BERT))にわたってWebスケールのデータセット(ImageNet、JFT)にシームレスに拡張されることを示します。 MRLコードと事前トレーニング済みモデルは、https://github.com/RAIVNLab/MRLでオープンソース化されています。
Learned representations are a central component in modern ML systems, serving a multitude of downstream tasks. When training such representations, it is often the case that computational and statistical constraints for each downstream task are unknown. In this context rigid, fixed capacity representations can be either over or under-accommodating to the task at hand. This leads us to ask: can we design a flexible representation that can adapt to multiple downstream tasks with varying computational resources? Our main contribution is Matryoshka Representation Learning (MRL) which encodes information at different granularities and allows a single embedding to adapt to the computational constraints of downstream tasks. MRL minimally modifies existing representation learning pipelines and imposes no additional cost during inference and deployment. MRL learns coarse-to-fine representations that are at least as accurate and rich as independently trained low-dimensional representations. The flexibility within the learned Matryoshka Representations offer: (a) up to 14x smaller embedding size for ImageNet-1K classification at the same level of accuracy; (b) up to 14x real-world speed-ups for large-scale retrieval on ImageNet-1K and 4K; and (c) up to 2% accuracy improvements for long-tail few-shot classification, all while being as robust as the original representations. Finally, we show that MRL extends seamlessly to web-scale datasets (ImageNet, JFT) across various modalities -- vision (ViT, ResNet), vision + language (ALIGN) and language (BERT). MRL code and pretrained models are open-sourced at https://github.com/RAIVNLab/MRL.
updated: Wed Jun 01 2022 00:03:14 GMT+0000 (UTC)
published: Thu May 26 2022 04:33:56 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アソシエイト