🐡

【SNN言語モデルv3】BitNet + RWKV + Attentionで43.4%改善!大規模な検証もクリア!

に公開

はじめに

前回の記事では「SNN言語モデルが14.7倍省エネ」という発見を報告しました。

https://zenn.dev/cell_activation/articles/9f89b5396e8230

今回はさらに大きな発見です!

5つの技術を組み合わせた「Ultimate SNN」で、120,000文字の大規模データで20.9%改善を達成しました!

(Ultimate SNNと勝手に名付けてそう呼んでますw)

v3の新発見まとめ

発見 数値
🚀 Ultimate SNN 43.4%改善(標準SNNより)
🔬 大規模検証 120K文字で-20.9%(本物の発見!)
⚡ BitNet融合 乗算不要、標準を超えた
🎯 RWKV統合 36.1%改善
📈 Progressive 28%改善
👁️ Attention 21.5%改善

Ultimate SNNについて

今回実装したUltimate SNNは、以下の5つの技術を統合したSNN言語モデルです!

  1. BitNet三値重み: {-1, 0, 1} で乗算不要
  2. RWKV Time-mixing: 長距離記憶を効率的に保持
  3. Hybrid Readout: スパイク + 膜電位で +39.7%
  4. Progressive Training: 小→大に段階的成長
  5. Attention: 過去の履歴を参照

発見1:各技術の改善効果 🧪

アブレーション実験

Model PPL vs Standard
Progressive 🏆 17.59 -28.0%
Attention 19.17 -21.5%
Standard 24.43 baseline
Multiscale 29.79 +22.0%
Learnable τ 61.33 +151.0%

→ ProgressiveとAttentionが効果的!


発見2:組み合わせの威力 💪

Progressive + Attentionの相乗効果

Model PPL vs Standard
Super Ultimate 🏆 1.21 -32.2%
Progressive+Attention 1.25 -30.1%
Standard 1.79 baseline

→ 組み合わせで最強!


発見3:大規模検証 🔬

本物かどうかの証明

小規模データでの発見が、大規模データでも有効か?

実験設定

  • データ:120,037文字(100K+)
  • モデル:24個並列学習
  • エポック:15

結果

┌────────────────────────┬────────────┬────────────┐
│ Model                  │ PPL (±std) │ vs Standard│
├────────────────────────┼────────────┼────────────┤
│ Super Ultimate (500n)  │ 4.40±0.26  │    -20.9%  │ ✅ WINS!
│ Standard SNN (200n)    │ 5.56±0.17  │   baseline │
└────────────────────────┴────────────┴────────────┘

🎉 VALIDATED! This is a REAL discovery!

→ 大規模データでも効果が確認された!本物の発見!


なぜこれが重要なのか

1. 乗算不要(BitNet)

従来のニューラルネットワークは乗算が多い:

# 従来
output = input @ weight  # 乗算必要

# BitNet SNN
output = sum(input[weight == 1]) - sum(input[weight == -1])  # 加算のみ!

→ エネルギー消費が激減!

2. 長距離記憶(RWKV)

# Time-mixing
mixed = (1 - decay) * current + decay * previous_state

→ Transformerより軽量で長距離記憶!

3. 段階的成長(Progressive)

Phase 1: 125 neurons → Phase 2: 250 neurons → ... → Phase 4: 500 neurons

→ 学習の安定性が向上!


エッジAI向けの特徴

特徴 効果
三値重み 21倍メモリ削減
スパース発火(7.6%) 13倍演算削減
乗算不要 消費電力激減
Progressive 軽量モデルから開始可能

→ スマホやIoTで動くAI言語モデル!


ソースコード

全実験スクリプトはGitHubで公開中!

https://github.com/hafufu-stack/snn-language-model

主要スクリプト:

  • experiments/advanced/snn_lm_ultimate.py - Ultimate SNN
  • experiments/advanced/snn_lm_rwkv.py - RWKV統合
  • experiments/advanced/snn_lm_large_scale.py - 大規模検証
  • experiments/advanced/snn_lm_combined.py - 統合実験

Zenodo論文

本研究はZenodoで公開しています!

v3:https://zenodo.org/records/18304632


まとめ

v3で分かったこと

  1. Ultimate SNN = 5つの技術統合
    • BitNet + RWKV + Hybrid + Progressive + Attention
  2. 大規模検証で-20.9%改善
    • 120K文字で効果確認(本物の発見!)
  3. Progressive + Attentionが最強の組み合わせ
  4. エッジAI向けに最適化
    • 乗算不要、スパース、軽量

次のステップ

  • 🔬 IJCNN 2026 投稿準備
  • 📱 実機検証(Intel Loihi等)
  • 🌍 単語レベルへの拡張

最後まで読んでいただきありがとうございました!

Ultimate SNNの威力、伝わりましたか?⚡🧠

正直私もSNNで実用的なAIを作ることは、正直無謀な気もしてたこともあります・・・(;^_^A

でも段々SNNの可能性が花開いてきて、

本当に実用性でも他のニューラルネットワークと肩を並べていく気がしてます!😊

質問やフィードバックがあれば、コメントでお待ちしています!

Discussion