ridgerchu/matmulfreellm
Implementation for MatMul-free LM.
何を解決するか
従来の言語モデルにおける行列積(MatMul)の高い計算コストに対処します。MatMul演算を排除することで、特に神経形態ハードウェアへのデプロイにおいてより効率的なアーキテクチャを目指しています。
動作方法
このプロジェクトは、標準的な行列積をより効率的な演算に置き換える言語モデルアーキテクチャを実装しています。複数のレイヤーで三値重み(ternary weights)を利用しています。Hugging Face Transformersライブラリと互換性があり、標準の AutoModel および AutoModelForCausalLM API を使って既存のAIワークフローに統合できます。
対象ユーザー
効率的なAIアーキテクチャ、神経形態コンピューティングに注力する研究者や開発者、また大規模言語モデルの計算オーバーヘッドを削減したい人向けです。
主な特徴
- 行列積不要アーキテクチャ: 標準的な行列積演算の必要性を完全に排除しています。
- Hugging Face統合: Transformersライブラリと完全互換で、初期化やテキスト生成が容易です。
- スケーリング効率: Transformer++と比較して、計算リソースが増加するにつれてより急峻なスケーリングプロジェクションを示しており、性能向上が期待されます。
- 事前学習済みモデル: 370Mから2.7Bパラメータまでのサイズのモデルズーを提供しています。
関連
- プロジェクト
- プロジェクト
- プロジェクト
- プロジェクト