可変幅トランスフォーマーによる効率的なモデルスケーリング

Variable-Width Transformers

トランスフォーマーの幅を非均一に調整

2026-06-16 中級 arXiv
LLMMoETransformer
  • トランスフォーマーのモデルサイズ拡大が進展をもたらしている。
  • 本研究では、層ごとに異なる幅を持つ新しいアーキテクチャを提案。
  • 非均一な幅の割り当てがリソース最適化に寄与することを示した。
トランスフォーマーモデルスケーリング非均一幅

トランスフォーマーにおけるモデルサイズの拡大は、言語モデルの進展に大きく寄与していますが、従来のアーキテクチャは層ごとに一定の幅を維持しています。本研究では、層の深さに応じて幅を調整する新しいアーキテクチャを提案し、言語モデリングの損失において従来の均一なベースラインを上回る成果を示しました。このアプローチは、計算資源の最適化にも寄与するため、トランスフォーマーに関心のある研究者や実務者にとって興味深い内容です。

トランスフォーマーや言語モデルに興味がある研究者やエンジニアに向いています。特に効率的なモデル設計を探求している方におすすめです。

Variable-Width Transformers
Zhaofeng Wu, Oliver Sieberling, Shawn Tan, Rameswar Panda, Yury Polyanskiy, Yoon Kim