Upload 6 files
Browse files- 1Gv33_AutoLR/__init__.py +0 -0
- 1Gv33_AutoLR/emofact.py +220 -0
- 1Gv33_AutoLR/emolynx.py +222 -0
- 1Gv33_AutoLR/emonavi.py +204 -0
- 1Gv33_AutoLR/profile.txt +62 -0
- 1Gv33_AutoLR/usage(オプション設定).txt +29 -0
1Gv33_AutoLR/__init__.py
ADDED
|
File without changes
|
1Gv33_AutoLR/emofact.py
ADDED
|
@@ -0,0 +1,220 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import torch
|
| 2 |
+
from torch.optim import Optimizer
|
| 3 |
+
import math
|
| 4 |
+
from collections import deque
|
| 5 |
+
|
| 6 |
+
"""
|
| 7 |
+
EmoFact v3.3 (251202) shadow-system v3.0 -effect NoN -moment v3.0
|
| 8 |
+
(v1.0)AMP対応完了(250725) p.data -> p 修正済み/低精度量子化への基本対応/低精度補償は別
|
| 9 |
+
(v2.0)shadow-system 微調整/3段階補正を連続的に滑らかに/派生版では以下の切替も可能
|
| 10 |
+
optimizer 指定の際に True / False で shadow を切替できる(現在 False)
|
| 11 |
+
(v3.0)emosens shadow-effect v1.0 反映した動的学習率と shadow-system 切替をデフォルト化
|
| 12 |
+
(v3.1)通常未使用の shadow 更新速度 (lerp) を倍化し信頼度で動的制御/coeff 活用(急変・微動)
|
| 13 |
+
動的学習率や感情スカラー値など TensorBoard 連携可 (現在 writer=None)/外部設定必要
|
| 14 |
+
全体の効率化や可読性を向上(emaやスカラーの多重処理を省く等、動的学習率のスケールや状態の見直し等、含む)
|
| 15 |
+
(v3.3)トラウマ的反応や慣れによる鈍化で安定性向上(ema-medium 安定と急変を信頼度で感知)
|
| 16 |
+
完全自動学習率/目標減少率制御方式を導入/感情機構との相乗効果で急変時も鎮静化し安定進行
|
| 17 |
+
"""
|
| 18 |
+
|
| 19 |
+
class EmoFact(Optimizer):
|
| 20 |
+
# クラス定義&初期化
|
| 21 |
+
def __init__(self, params,
|
| 22 |
+
lr=1e-3,
|
| 23 |
+
lr_max=1e-3,
|
| 24 |
+
lr_min=1e-8,
|
| 25 |
+
eps=1e-8,
|
| 26 |
+
betas=(0.9, 0.999),
|
| 27 |
+
weight_decay=0.01,
|
| 28 |
+
use_shadow:bool=False,
|
| 29 |
+
writer=None):
|
| 30 |
+
|
| 31 |
+
defaults = dict(lr=lr, betas=betas, eps=eps, weight_decay=weight_decay)
|
| 32 |
+
|
| 33 |
+
super().__init__(params, defaults)
|
| 34 |
+
self._init_lr = lr
|
| 35 |
+
self.should_stop = False # 停止フラグの初期化
|
| 36 |
+
self.use_shadow = use_shadow # 🔸shadow 使用フラグを保存
|
| 37 |
+
self.writer = writer # 動的学習率や感情スカラー等を渡す
|
| 38 |
+
self.eta = lr # 名目lrを初期値として利用(自己更新)
|
| 39 |
+
self.k = 0.2 # 学習率自己更新の応答速度係数(比例制御の強さ)
|
| 40 |
+
self.eps = 1e-8 # ゼロ割り防止の微小値(分母安定化)
|
| 41 |
+
self.lr_min = 1e-8 # 学習率の下限(極端な縮小の防止)
|
| 42 |
+
self.lr_max = 1e-3 # 学習率の上限(極端な拡大の防止)
|
| 43 |
+
self.prev_loss = None # Loss初期化
|
| 44 |
+
|
| 45 |
+
# 感情EMA更新(緊張と安静)
|
| 46 |
+
def _update_ema(self, state, loss_val):
|
| 47 |
+
ema = state.setdefault('ema', {})
|
| 48 |
+
ema['short'] = 0.3 * loss_val + 0.7 * ema.get('short', loss_val)
|
| 49 |
+
ema['medium'] = 0.05 * loss_val + 0.95 * ema.get('medium', loss_val)
|
| 50 |
+
ema['long'] = 0.01 * loss_val + 0.99 * ema.get('long', loss_val)
|
| 51 |
+
return ema
|
| 52 |
+
|
| 53 |
+
# 感情スカラー値生成(EMA差分、滑らかな非線形スカラー、tanh(diff) は ±1.0 で有界性)
|
| 54 |
+
# 係数"1":ema差分 のスケール調整処理に活用(感度調節係数)/通常は1(タスクに応じ調整可(非推奨))
|
| 55 |
+
# scale_base:Loss値とema値の乖離を修正(分母 ema(long) 「改善率」共通化/loss種に非依存)
|
| 56 |
+
# 1e-5(デフォルト)/1e-6(感度向上)/1e-4(安定性向上):分母を0にせず安定させる
|
| 57 |
+
# トラウマ的反応や慣れによる鈍化で安定性向上(ema-medium 安定と急変を信頼度で感知)
|
| 58 |
+
def _compute_scalar(self, ema):
|
| 59 |
+
scale_base_l = max(ema['long'], 1e-5)
|
| 60 |
+
scale_base_m = max(ema['medium'], 1e-5)
|
| 61 |
+
diff_l = (ema['long'] - ema['short']) / scale_base_l
|
| 62 |
+
diff_m = (ema['long'] - ema['short']) / scale_base_m
|
| 63 |
+
# longが十分静かなら、常にlongを優先
|
| 64 |
+
if abs(diff_l) < 0.05:
|
| 65 |
+
return math.tanh(diff_l)
|
| 66 |
+
# longが静かでない時のみ、mediumの静けさを条件付きで採用
|
| 67 |
+
if abs(diff_m) * scale_base_m < abs(diff_l) * scale_base_l:
|
| 68 |
+
return math.tanh(1 * diff_m)
|
| 69 |
+
else:
|
| 70 |
+
return math.tanh(1 * diff_l)
|
| 71 |
+
|
| 72 |
+
# アーリーストップ専用(静けさ判定の感情スカラ生成)
|
| 73 |
+
def _early_scalar(self, ema):
|
| 74 |
+
scale_base_l = max(ema['long'], 1e-5)
|
| 75 |
+
diff = (ema['long'] - ema['short']) / scale_base_l
|
| 76 |
+
return math.tanh(1 * diff)
|
| 77 |
+
|
| 78 |
+
# 急変時は論文通りの抑制則/悪化時は減速/改善時は加速/微動時は無介入で収束を安定させる
|
| 79 |
+
def _decide_coeff(self, scalar):
|
| 80 |
+
if abs(scalar) > 0.625:
|
| 81 |
+
return 1.0 - abs(scalar) # 急変|強抑制
|
| 82 |
+
elif scalar > 0.125:
|
| 83 |
+
return 1.0 + scalar # 改善|加速
|
| 84 |
+
elif scalar < -0.125:
|
| 85 |
+
return 1.0 + scalar # 悪化|減速
|
| 86 |
+
else:
|
| 87 |
+
return 1.0 # 微動|無介入
|
| 88 |
+
|
| 89 |
+
# (重要)現在は shadow-effect を参考に得た動的フィルタ効果の近似により use_shadow=False です
|
| 90 |
+
# しかし全機��は shadow なしで全て成立します/コード確認時はこの shadow を考慮外として無視してください
|
| 91 |
+
|
| 92 |
+
# Shadow混合比 3段階構成 タスクに応じ調整可、以下を参考に 開始値・範囲量・変化幅を調整
|
| 93 |
+
# 参考1:scalar>±0.6 を "return 開始値 + ((scalar) - 0.6(範囲)) / 範囲量 * 変化幅"
|
| 94 |
+
# 参考2:scalar>±0.1 を "return 開始値 + ((scalar) - 0.1(範囲)) / 範囲量 * 変化幅"
|
| 95 |
+
# return 開始値 + ((scalar) - 閾値) / 範囲量 * 変化幅 です(上記の値は感情スカラーを返すだけ)
|
| 96 |
+
def _decide_ratio(self, scalar):
|
| 97 |
+
if not self.use_shadow:
|
| 98 |
+
return 0.0 # 🔸use_shadow = False のとき常に比率を 0 にする
|
| 99 |
+
if abs(scalar) > 0.75:
|
| 100 |
+
return 0.75 # + ((scalar) - 0.75) / 0.4 * 0.4 # これはスカラーそのまま返す参考例
|
| 101 |
+
elif abs(scalar) > 0.25:
|
| 102 |
+
return -0.1 # return<0 の場合は leap 専用(書き戻しはしないが履歴更新のみ)
|
| 103 |
+
return 0.0
|
| 104 |
+
|
| 105 |
+
# 損失取得(損失値 loss_val を数値化、感情判定に使用、存在しないパラメータ(更新不要)はスキップ)
|
| 106 |
+
@torch.no_grad()
|
| 107 |
+
def step(self, closure=None):
|
| 108 |
+
loss = closure() if closure is not None else None
|
| 109 |
+
loss_val = loss.item() if loss is not None else 0.0
|
| 110 |
+
|
| 111 |
+
# EMA更新・スカラー生成(EMA差分からスカラーを生成しスパイク比率等を決定)
|
| 112 |
+
ema = self._update_ema(self.state, loss_val)
|
| 113 |
+
early_scalar = self._early_scalar(ema)
|
| 114 |
+
scalar = self._compute_scalar(ema)
|
| 115 |
+
coeff = self._decide_coeff(scalar)
|
| 116 |
+
ratio = self._decide_ratio(scalar)
|
| 117 |
+
|
| 118 |
+
# 目標減少率制御 + eta_eff
|
| 119 |
+
if self.prev_loss is None:
|
| 120 |
+
self.prev_loss = loss_val # 初回は初期化のみ
|
| 121 |
+
eta_eff = max(self.lr_min, min(self.lr_max, self.eta * coeff))
|
| 122 |
+
else:
|
| 123 |
+
delta = self.prev_loss - loss_val
|
| 124 |
+
target_delta = max(1e-8, 0.01 * max(loss_val, 1e-8)) # 1%固定
|
| 125 |
+
# 学習率の自己更新(比例制御)
|
| 126 |
+
self.eta *= math.exp(self.k * (delta - target_delta) / (abs(target_delta) + self.eps))
|
| 127 |
+
# 感情スカラーで補正し最終ステップへ
|
| 128 |
+
eta_eff = max(self.lr_min, min(self.lr_max, self.eta * coeff))
|
| 129 |
+
|
| 130 |
+
for group in self.param_groups:
|
| 131 |
+
step_size = eta_eff # 💡 group['lr'] は使わない
|
| 132 |
+
for p in group['params']:
|
| 133 |
+
if p.grad is None:
|
| 134 |
+
continue
|
| 135 |
+
|
| 136 |
+
grad = p.grad
|
| 137 |
+
state = self.state[p]
|
| 138 |
+
|
| 139 |
+
# 動的学習率補正により shadow 形成を信頼度で調整(coeffは正値(負にならない))
|
| 140 |
+
# shadow:必要時のみ(スパイクp部分に現在値を最大10%追従させる動的履歴更新)
|
| 141 |
+
# ratio <0:10%、0以外:10%×coeff、(0.25~0.75は10%、微動と急変は*coeff)
|
| 142 |
+
# 微動時 coeff:1.0 固定なので結果的に微動時も 10% 履歴更新になる
|
| 143 |
+
# 結果、微動時と安定時:10%、急変時:coeff、による履歴更新を行うことになる
|
| 144 |
+
if self.use_shadow:
|
| 145 |
+
if 'shadow' not in state: # 🔸shadow = False (デフォルト)
|
| 146 |
+
state['shadow'] = p.clone()
|
| 147 |
+
if ratio > 0: # 書き戻しと履歴更新(急変時の強い抑制と弱めの履歴更新)
|
| 148 |
+
p.mul_(1 - ratio).add_(state['shadow'], alpha=coeff)
|
| 149 |
+
else: # 書き戻しせず履歴更新のみ:ratio<0:10%/0以外:10%×coeff
|
| 150 |
+
leap_ratio = 0.1 if ratio < 0 else 0.1 * coeff
|
| 151 |
+
state['shadow'].lerp_(p, leap_ratio)
|
| 152 |
+
|
| 153 |
+
# 上記 shadow の説明:スカラー生成:短期と長期EMAの差分から信号を得る(高ぶりの強さ)
|
| 154 |
+
# 混合比率:スカラーが閾値を超える場合にのみ計算される(信頼できる感情信号かどうかの選別)
|
| 155 |
+
# 急変時は感情機構による shadow 混合で強く抑制する(急制動による安定性の確保)
|
| 156 |
+
# 新しい shadow-system は動的学習率と協調することで選択的スパース性も発揮する
|
| 157 |
+
|
| 158 |
+
# --- Start Fact Gradient Update Logic ---
|
| 159 |
+
# 行列の形状が2次元以上の場合、分散情報ベースのAB近似を使用
|
| 160 |
+
if grad.dim() >= 2:
|
| 161 |
+
# 行と列の2乗平均を計算 (分散の軽量な近似)
|
| 162 |
+
r_sq = torch.mean(grad * grad, dim=tuple(range(1, grad.dim())), keepdim=True).add_(group['eps'])
|
| 163 |
+
c_sq = torch.mean(grad * grad, dim=0, keepdim=True).add_(group['eps'])
|
| 164 |
+
|
| 165 |
+
# 分散情報から勾配の近似行列を生成
|
| 166 |
+
# AB行列として見立てたものを直接生成し更新項を計算する
|
| 167 |
+
# A = sqrt(r_sq), B = sqrt(c_sq) AB行列の近似を再現しEMAで平滑化する
|
| 168 |
+
beta1, beta2 = group['betas']
|
| 169 |
+
state.setdefault('exp_avg_r', torch.zeros_like(r_sq)).mul_(beta1).add_(torch.sqrt(r_sq), alpha=1 - beta1)
|
| 170 |
+
state.setdefault('exp_avg_c', torch.zeros_like(c_sq)).mul_(beta1).add_(torch.sqrt(c_sq), alpha=1 - beta1)
|
| 171 |
+
|
| 172 |
+
# 再構築した近似勾配の平方根の積で正規化
|
| 173 |
+
denom = torch.sqrt(state['exp_avg_r'] * state['exp_avg_c']).add_(group['eps'])
|
| 174 |
+
|
| 175 |
+
# 最終的な更新項を計算
|
| 176 |
+
update_term = grad / denom
|
| 177 |
+
|
| 178 |
+
# 1次元(ベクトル)の勾配補正
|
| 179 |
+
else:
|
| 180 |
+
beta1, beta2 = group['betas']
|
| 181 |
+
exp_avg_sq = state.setdefault('exp_avg_sq', torch.zeros_like(p))
|
| 182 |
+
exp_avg_sq.mul_(beta1).addcmul_(grad, grad, value=(1 - beta2))
|
| 183 |
+
denom = exp_avg_sq.sqrt().add_(group['eps'])
|
| 184 |
+
update_term = grad / denom
|
| 185 |
+
|
| 186 |
+
# 最終的なパラメータ更新 (decoupled weight decayも適用)
|
| 187 |
+
p.add_(p, alpha=-group['weight_decay'] * step_size)
|
| 188 |
+
p.add_(update_term, alpha=-step_size)
|
| 189 |
+
# --- End Fact Gradient Update Logic ---
|
| 190 |
+
|
| 191 |
+
self.prev_loss = loss_val
|
| 192 |
+
|
| 193 |
+
# 感情機構の発火が収まり"十分に安定"していることを外部伝達できる(自動停止ロジックではない)
|
| 194 |
+
# Early Stop用 scalar 記録(バッファ共通で管理/最大32件保持/動静評価)
|
| 195 |
+
hist = self.state.setdefault('scalar_hist', deque(maxlen=32))
|
| 196 |
+
hist.append(early_scalar)
|
| 197 |
+
|
| 198 |
+
# Early Stop判断(静けさの合図)
|
| 199 |
+
# 32ステップ分のスカラー値の静かな条件を満たした時"フラグ" should_stop = True になるだけ
|
| 200 |
+
if len(hist) >= 32:
|
| 201 |
+
avg_abs = sum(abs(s) for s in hist) / len(hist)
|
| 202 |
+
mean = sum(hist) / len(hist)
|
| 203 |
+
var = sum((s - mean)**2 for s in hist) / len(hist)
|
| 204 |
+
if avg_abs < 0.05 and var < 0.005:
|
| 205 |
+
self.should_stop = True # 💡 外部からこれを見て判断可
|
| 206 |
+
|
| 207 |
+
# TensorBoardへの記録(step関数の末尾に追加)
|
| 208 |
+
if hasattr(self, 'writer') and self.writer is not None:
|
| 209 |
+
self._step_count = getattr(self, "_step_count", 0) + 1
|
| 210 |
+
self.writer.add_scalar("emoLR", eta_eff, self._step_count)
|
| 211 |
+
self.writer.add_scalar("etaLR", self.eta, self._step_count)
|
| 212 |
+
self.writer.add_scalar("emoScalar", scalar, self._step_count)
|
| 213 |
+
|
| 214 |
+
return loss
|
| 215 |
+
|
| 216 |
+
"""
|
| 217 |
+
https://github.com/muooon/EmoNavi
|
| 218 |
+
Fact is inspired by Adafactor, and emoairy,
|
| 219 |
+
and its VRAM-friendly design is something everyone loves.
|
| 220 |
+
"""
|
1Gv33_AutoLR/emolynx.py
ADDED
|
@@ -0,0 +1,222 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import torch
|
| 2 |
+
from torch.optim import Optimizer
|
| 3 |
+
import math
|
| 4 |
+
from typing import Tuple, Callable, Union
|
| 5 |
+
from collections import deque
|
| 6 |
+
|
| 7 |
+
"""
|
| 8 |
+
EmoLynx v3.3 (251202) shadow-system v3.0 -effect NoN -moment v3.0
|
| 9 |
+
(v1.0)AMP対応完了(250725) p.data -> p 修正済み/低精度量子化への基本対応/低精度補償は別
|
| 10 |
+
(v2.0)shadow-system 微調整/3段階補正を連続的に滑らかに/派生版では以下の切替も可能
|
| 11 |
+
optimizer 指定の際に True / False で shadow を切替できる(現在 False)
|
| 12 |
+
(v3.0)emosens shadow-effect v1.0 反映した動的学習率と shadow-system 切替をデフォルト化
|
| 13 |
+
(v3.1)通常未使用の shadow 更新速度 (lerp) を倍化し信頼度で動的制御/coeff 活用(急変・微動)
|
| 14 |
+
動的学習率や感情スカラー値など TensorBoard 連携可 (現在 writer=None)/外部設定必要
|
| 15 |
+
全体の効率化や可読性を向上(emaやスカラーの多重処理を省く等、動的学習率のスケールや状態の見直し等、含む)
|
| 16 |
+
(v3.3)トラウマ的反応や慣れによる鈍化で安定性向上(ema-medium 安定と急変を信頼度で感知)
|
| 17 |
+
完全自動学習率/目標減少率制御方式を導入/感情機構との相乗効果で急変時も鎮静化し安定進行
|
| 18 |
+
"""
|
| 19 |
+
|
| 20 |
+
# Helper function (Lynx)
|
| 21 |
+
def exists(val):
|
| 22 |
+
return val is not None
|
| 23 |
+
|
| 24 |
+
class EmoLynx(Optimizer):
|
| 25 |
+
# クラス定義&初期化 lynx用ベータ・互換性の追加(lynx用beta1・beta2)
|
| 26 |
+
def __init__(self, params: Union[list, torch.nn.Module],
|
| 27 |
+
lr=1e-3,
|
| 28 |
+
eps=1e-8,
|
| 29 |
+
lr_max=1e-3,
|
| 30 |
+
lr_min=1e-8,
|
| 31 |
+
betas=(0.9, 0.999),
|
| 32 |
+
weight_decay=0.01,
|
| 33 |
+
use_shadow:bool=False,
|
| 34 |
+
writer=None):
|
| 35 |
+
|
| 36 |
+
defaults = dict(lr=lr, betas=betas, eps=eps, weight_decay=weight_decay)
|
| 37 |
+
|
| 38 |
+
super().__init__(params, defaults)
|
| 39 |
+
self._init_lr = lr
|
| 40 |
+
self.should_stop = False # 停止フラグの初期化
|
| 41 |
+
self.use_shadow = use_shadow # 🔸shadow 使用フラグを保存
|
| 42 |
+
self.writer = writer # 動的学習率や感情スカラー等を渡す
|
| 43 |
+
self.eta = lr # 名目lrを初期値として利用(自己更新)
|
| 44 |
+
self.k = 0.2 # 学習率自己更新の応答速度係数(比例制御の強さ)
|
| 45 |
+
self.eps = 1e-8 # ゼロ割り防止の微小値(分母安定化)
|
| 46 |
+
self.lr_min = 1e-8 # 学習率の下限(極端な縮小の防止)
|
| 47 |
+
self.lr_max = 1e-3 # 学習率の上限(極端な拡大の防止)
|
| 48 |
+
self.prev_loss = None # Loss初期化
|
| 49 |
+
|
| 50 |
+
# 感情EMA更新(緊張と安静)
|
| 51 |
+
def _update_ema(self, state, loss_val):
|
| 52 |
+
ema = state.setdefault('ema', {})
|
| 53 |
+
ema['short'] = 0.3 * loss_val + 0.7 * ema.get('short', loss_val)
|
| 54 |
+
ema['medium'] = 0.05 * loss_val + 0.95 * ema.get('medium', loss_val)
|
| 55 |
+
ema['long'] = 0.01 * loss_val + 0.99 * ema.get('long', loss_val)
|
| 56 |
+
return ema
|
| 57 |
+
|
| 58 |
+
# 感情スカラー値生成(EMA差分、滑らかな非線形スカラー、tanh(diff) は ±1.0 で有界性)
|
| 59 |
+
# 係数"1":ema差分 のスケール調整処理に活用(感度調節係数)/通常は1(タスクに応じ調整可(非推奨))
|
| 60 |
+
# scale_base:Loss値とema値の乖離を修正(分母 ema(long) 「改善率」共通化/loss種に非依存)
|
| 61 |
+
# 1e-5(デフォルト)/1e-6(感度向上)/1e-4(安定性向上):分母を0にせず安定させる
|
| 62 |
+
# トラウマ的反応や慣れによる鈍化で安定性向上(ema-medium 安定と急変を信頼度で感知)
|
| 63 |
+
def _compute_scalar(self, ema):
|
| 64 |
+
scale_base_l = max(ema['long'], 1e-5)
|
| 65 |
+
scale_base_m = max(ema['medium'], 1e-5)
|
| 66 |
+
diff_l = (ema['long'] - ema['short']) / scale_base_l
|
| 67 |
+
diff_m = (ema['long'] - ema['short']) / scale_base_m
|
| 68 |
+
# longが十分静かなら、常にlongを優先
|
| 69 |
+
if abs(diff_l) < 0.05:
|
| 70 |
+
return math.tanh(diff_l)
|
| 71 |
+
# longが静かでない時のみ、mediumの静けさを条件付きで採用
|
| 72 |
+
if abs(diff_m) * scale_base_m < abs(diff_l) * scale_base_l:
|
| 73 |
+
return math.tanh(1 * diff_m)
|
| 74 |
+
else:
|
| 75 |
+
return math.tanh(1 * diff_l)
|
| 76 |
+
|
| 77 |
+
# アーリーストップ専用(静けさ判定の感情スカラ生成)
|
| 78 |
+
def _early_scalar(self, ema):
|
| 79 |
+
scale_base_l = max(ema['long'], 1e-5)
|
| 80 |
+
diff = (ema['long'] - ema['short']) / scale_base_l
|
| 81 |
+
return math.tanh(1 * diff)
|
| 82 |
+
|
| 83 |
+
# 急変時は論文通りの抑制則/悪化時は減速/改善時は加速/微動時は無介入で収束を安定させる
|
| 84 |
+
def _decide_coeff(self, scalar):
|
| 85 |
+
if abs(scalar) > 0.625:
|
| 86 |
+
return 1.0 - abs(scalar) # 急変|強抑制
|
| 87 |
+
elif scalar > 0.125:
|
| 88 |
+
return 1.0 + scalar # 改善|加速
|
| 89 |
+
elif scalar < -0.125:
|
| 90 |
+
return 1.0 + scalar # 悪化|減速
|
| 91 |
+
else:
|
| 92 |
+
return 1.0 # 微動|無介入
|
| 93 |
+
|
| 94 |
+
# (重要)現在は shadow-effect を参考に得た動的フィルタ効果の近似により use_shadow=False です
|
| 95 |
+
# しかし全機能は shadow なしで全て成立します/コード確認時はこの shadow を考慮外として無視してください
|
| 96 |
+
|
| 97 |
+
# Shadow混合比 3段階構成 タスクに応じ調整可、以下を参考に 開始値・範囲量・変化幅を調整
|
| 98 |
+
# 参考1:scalar>±0.6 を "return 開始値 + ((scalar) - 0.6(範囲)) / 範囲量 * 変化幅"
|
| 99 |
+
# 参考2:scalar>±0.1 を "return 開始値 + ((scalar) - 0.1(範囲)) / 範囲量 * 変化幅"
|
| 100 |
+
# return 開始値 + ((scalar) - 閾値) / 範囲量 * 変化幅 です(上記の値は感情スカラーを返すだけ)
|
| 101 |
+
def _decide_ratio(self, scalar):
|
| 102 |
+
if not self.use_shadow:
|
| 103 |
+
return 0.0 # 🔸use_shadow = False のとき常に比率を 0 にする
|
| 104 |
+
if abs(scalar) > 0.75:
|
| 105 |
+
return 0.75 # + ((scalar) - 0.75) / 0.4 * 0.4 # これはスカラーそのまま返す参考例
|
| 106 |
+
elif abs(scalar) > 0.25:
|
| 107 |
+
return -0.1 # return<0 の場合は leap 専用(書き戻しはしないが履歴更新のみ)
|
| 108 |
+
return 0.0
|
| 109 |
+
|
| 110 |
+
# 損失取得(損失値 loss_val を数値化、感情判定に使用、存在しないパラメータ(更新不要)はスキップ)
|
| 111 |
+
@torch.no_grad()
|
| 112 |
+
def step(self, closure: Callable | None = None): # クロージャの型ヒントを追加
|
| 113 |
+
loss = None
|
| 114 |
+
if exists(closure): # 一貫性のためにexistsヘルパーを使う
|
| 115 |
+
with torch.enable_grad():
|
| 116 |
+
loss = closure()
|
| 117 |
+
loss_val = loss.item() if loss is not None else 0.0
|
| 118 |
+
|
| 119 |
+
# EMA更新・スカラー生成(EMA差分からスカラーを生成しスパイク比率等を決定)
|
| 120 |
+
ema = self._update_ema(self.state, loss_val)
|
| 121 |
+
early_scalar = self._early_scalar(ema)
|
| 122 |
+
scalar = self._compute_scalar(ema)
|
| 123 |
+
coeff = self._decide_coeff(scalar)
|
| 124 |
+
ratio = self._decide_ratio(scalar)
|
| 125 |
+
|
| 126 |
+
# 目標減少率制御 + eta_eff
|
| 127 |
+
if self.prev_loss is None:
|
| 128 |
+
self.prev_loss = loss_val # 初回は初期化のみ
|
| 129 |
+
eta_eff = max(self.lr_min, min(self.lr_max, self.eta * coeff))
|
| 130 |
+
else:
|
| 131 |
+
delta = self.prev_loss - loss_val
|
| 132 |
+
target_delta = max(1e-8, 0.01 * max(loss_val, 1e-8)) # 1%固定
|
| 133 |
+
# 学習率の自己更新(比例制御)
|
| 134 |
+
self.eta *= math.exp(self.k * (delta - target_delta) / (abs(target_delta) + self.eps))
|
| 135 |
+
# 感情スカラーで補正し最終ステップへ
|
| 136 |
+
eta_eff = max(self.lr_min, min(self.lr_max, self.eta * coeff))
|
| 137 |
+
|
| 138 |
+
for group in self.param_groups:
|
| 139 |
+
step_size = eta_eff # 💡 group['lr'] は使わない
|
| 140 |
+
# リンクス共通パラメータ抽出
|
| 141 |
+
wd, beta1, beta2 = group['weight_decay'], *group['betas']
|
| 142 |
+
|
| 143 |
+
# ウェイト減衰の処理を分離 (from lynx)
|
| 144 |
+
_wd_actual = wd
|
| 145 |
+
|
| 146 |
+
for p in filter(lambda p: exists(p.grad), group['params']): # PGチェックにフィルタ
|
| 147 |
+
|
| 148 |
+
grad = p.grad # PG直接使用(計算に".data"不要)
|
| 149 |
+
state = self.state[p]
|
| 150 |
+
|
| 151 |
+
# 動的学習率補正により shadow 形成を信頼度で調整(coeffは正値(負にならない))
|
| 152 |
+
# shadow:必要時のみ(スパイクp部分に現在値を最大10%追従させる動的履歴更新)
|
| 153 |
+
# ratio <0:10%、0以外:10%×coeff、(0.25~0.75は10%、微動と急変は*coeff)
|
| 154 |
+
# 微動時 coeff:1.0 固定なので結果的に微動時も 10% 履歴更新になる
|
| 155 |
+
# 結果、微動時と安定時:10%、急変時:coeff、による履歴更新を行うことになる
|
| 156 |
+
if self.use_shadow:
|
| 157 |
+
if 'shadow' not in state: # 🔸shadow = False (デフォルト)
|
| 158 |
+
state['shadow'] = p.clone()
|
| 159 |
+
if ratio > 0: # 書き戻しと履歴更新(急変時の強い抑制と弱めの履歴更新)
|
| 160 |
+
p.mul_(1 - ratio).add_(state['shadow'], alpha=coeff)
|
| 161 |
+
else: # 書き戻しせず履歴更新のみ:ratio<0:10%/0以外:10%×coeff
|
| 162 |
+
leap_ratio = 0.1 if ratio < 0 else 0.1 * coeff
|
| 163 |
+
state['shadow'].lerp_(p, leap_ratio)
|
| 164 |
+
|
| 165 |
+
# 上記 shadow の説明:スカラー生成:短期と長期EMAの差分から信号を得る(高ぶりの強さ)
|
| 166 |
+
# 混合比率:スカラーが閾値を超える場合にのみ計算される(信頼できる感情信号かどうかの選別)
|
| 167 |
+
# 急変時は感情機構による shadow 混合で強く抑制する(急制動による安定性の確保)
|
| 168 |
+
# 新しい shadow-system は動的学習率と協調することで選択的スパース性も発揮する
|
| 169 |
+
|
| 170 |
+
# --- Start Lynx Gradient Update Logic ---
|
| 171 |
+
# lynx初期化(exp_avg_sq)
|
| 172 |
+
if 'exp_avg' not in state:
|
| 173 |
+
state['exp_avg'] = torch.zeros_like(p)
|
| 174 |
+
exp_avg = state['exp_avg']
|
| 175 |
+
|
| 176 |
+
# Stepweight decay (from lynx): p = p * (1 - lr * wd)
|
| 177 |
+
# decoupled_wd 考慮 _wd_actual 使用(EmoNaviのwdは最後に適用)
|
| 178 |
+
p.mul_(1 - step_size * _wd_actual)
|
| 179 |
+
beta1, beta2 = group['betas']
|
| 180 |
+
|
| 181 |
+
# 勾配ブレンド
|
| 182 |
+
# m_t = beta1 * exp_avg_prev + (1 - beta1) * grad
|
| 183 |
+
blended_grad = grad.mul(1 - beta1).add_(exp_avg, alpha=beta1)
|
| 184 |
+
|
| 185 |
+
# p: p = p - lr * sign(blended_grad)
|
| 186 |
+
p.add_(blended_grad.sign_(), alpha = -step_size)
|
| 187 |
+
|
| 188 |
+
# exp_avg = beta2 * exp_avg + (1 - beta2) * grad
|
| 189 |
+
exp_avg.mul_(beta2).add_(grad, alpha = 1 - beta2)
|
| 190 |
+
# --- End Lynx Gradient Update Logic ---
|
| 191 |
+
|
| 192 |
+
self.prev_loss = loss_val
|
| 193 |
+
|
| 194 |
+
# 感情機構の発火が収まり"十分に安定"していることを外部伝達できる(自動停止ロジックではない)
|
| 195 |
+
# Early Stop用 scalar 記録(バッファ共通で管理/最大32件保持/動静評価)
|
| 196 |
+
hist = self.state.setdefault('scalar_hist', deque(maxlen=32))
|
| 197 |
+
hist.append(early_scalar)
|
| 198 |
+
|
| 199 |
+
# Early Stop判断(静けさの合図)
|
| 200 |
+
# 32ステップ分のスカラー値の静かな条件を満たした時"フラグ" should_stop = True になるだけ
|
| 201 |
+
if len(hist) >= 32:
|
| 202 |
+
avg_abs = sum(abs(s) for s in hist) / len(hist)
|
| 203 |
+
mean = sum(hist) / len(hist)
|
| 204 |
+
var = sum((s - mean)**2 for s in hist) / len(hist)
|
| 205 |
+
if avg_abs < 0.05 and var < 0.005:
|
| 206 |
+
self.should_stop = True # 💡 外部からこれを見て判断可
|
| 207 |
+
|
| 208 |
+
# TensorBoardへの記録(step関数の末尾に追加)
|
| 209 |
+
if hasattr(self, 'writer') and self.writer is not None:
|
| 210 |
+
self._step_count = getattr(self, "_step_count", 0) + 1
|
| 211 |
+
self.writer.add_scalar("emoLR", eta_eff, self._step_count)
|
| 212 |
+
self.writer.add_scalar("etaLR", self.eta, self._step_count)
|
| 213 |
+
self.writer.add_scalar("emoScalar", scalar, self._step_count)
|
| 214 |
+
|
| 215 |
+
return loss
|
| 216 |
+
|
| 217 |
+
"""
|
| 218 |
+
https://github.com/muooon/EmoNavi
|
| 219 |
+
Lynx was developed with inspiration from Lion, Tiger, and emocats,
|
| 220 |
+
which we deeply respect for their lightweight and intelligent design.
|
| 221 |
+
Lynx also integrates EmoNAVI to enhance its capabilities.
|
| 222 |
+
"""
|
1Gv33_AutoLR/emonavi.py
ADDED
|
@@ -0,0 +1,204 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import torch
|
| 2 |
+
from torch.optim import Optimizer
|
| 3 |
+
import math
|
| 4 |
+
from collections import deque
|
| 5 |
+
|
| 6 |
+
"""
|
| 7 |
+
EmoNavi v3.3 (251202) shadow-system v3.0 -effect NoN -moment v3.0
|
| 8 |
+
(v1.0)AMP対応完了(250725) p.data -> p 修正済み/低精度量子化への基本対応/低精度補償は別
|
| 9 |
+
(v2.0)shadow-system 微調整/3段階補正を連続的に滑らかに/派生版では以下の切替も可能
|
| 10 |
+
optimizer 指定の際に True / False で shadow を切替できる(現在 False)
|
| 11 |
+
(v3.0)emosens shadow-effect v1.0 反映した動的学習率と shadow-system 切替をデフォルト化
|
| 12 |
+
(v3.1)通常未使用の shadow 更新速度 (lerp) を倍化し信頼度で動的制御/coeff 活用(急変・微動)
|
| 13 |
+
動的学習率や感情スカラー値など TensorBoard 連携可 (現在 writer=None)/外部設定必要
|
| 14 |
+
全体の効率化や可読性を向上(emaやスカラーの多重処理を省く等、動的学習率のスケールや状態の見直し等、含む)
|
| 15 |
+
(v3.3)トラウマ的反応や慣れによる鈍化で安定性向上(ema-medium 安定と急変を信頼度で感知)
|
| 16 |
+
完全自動学習率/目標減少率制御方式を導入/感情機構との相乗効果で急変時も鎮静化し安定進行
|
| 17 |
+
"""
|
| 18 |
+
|
| 19 |
+
class EmoNavi(Optimizer):
|
| 20 |
+
# クラス定義&初期化
|
| 21 |
+
def __init__(self, params,
|
| 22 |
+
lr=1e-3,
|
| 23 |
+
lr_max=1e-3,
|
| 24 |
+
lr_min=1e-8,
|
| 25 |
+
eps=1e-8,
|
| 26 |
+
betas=(0.9, 0.999),
|
| 27 |
+
weight_decay=0.01,
|
| 28 |
+
use_shadow:bool=False,
|
| 29 |
+
writer=None):
|
| 30 |
+
|
| 31 |
+
defaults = dict(lr=lr, betas=betas, eps=eps, weight_decay=weight_decay)
|
| 32 |
+
|
| 33 |
+
super().__init__(params, defaults)
|
| 34 |
+
self._init_lr = lr
|
| 35 |
+
self.should_stop = False # 停止フラグの初期化
|
| 36 |
+
self.use_shadow = use_shadow # 🔸shadow 使用フラグを保存
|
| 37 |
+
self.writer = writer # 動的学習率や感情スカラー等を渡す
|
| 38 |
+
self.eta = lr # 名目lrを初期値として利用(自己更新)
|
| 39 |
+
self.k = 0.2 # 学習率自己更新の応答速度係数(比例制御の強さ)
|
| 40 |
+
self.eps = 1e-8 # ゼロ割り防止の微小値(分母安定化)
|
| 41 |
+
self.lr_min = 1e-8 # 学習率の下限(極端な縮小の防止)
|
| 42 |
+
self.lr_max = 1e-3 # 学習率の上限(極端な拡大の防止)
|
| 43 |
+
self.prev_loss = None # Loss初期化
|
| 44 |
+
|
| 45 |
+
# 感情EMA更新(緊張と安静)
|
| 46 |
+
def _update_ema(self, state, loss_val):
|
| 47 |
+
ema = state.setdefault('ema', {})
|
| 48 |
+
ema['short'] = 0.3 * loss_val + 0.7 * ema.get('short', loss_val)
|
| 49 |
+
ema['medium'] = 0.05 * loss_val + 0.95 * ema.get('medium', loss_val)
|
| 50 |
+
ema['long'] = 0.01 * loss_val + 0.99 * ema.get('long', loss_val)
|
| 51 |
+
return ema
|
| 52 |
+
|
| 53 |
+
# 感情スカラー値生成(EMA差分、滑らかな非線形スカラー、tanh(diff) は ±1.0 で有界性)
|
| 54 |
+
# 係数"1":ema差分 のスケール調整処理に活用(感度調節係数)/通常は1(タスクに応じ調整可(非推奨))
|
| 55 |
+
# scale_base:Loss値とema値の乖離を修正(分母 ema(long) 「改善率」共通化/loss種に非依存)
|
| 56 |
+
# 1e-5(デフォルト)/1e-6(感度向上)/1e-4(安定性向上):分母を0にせず安定させる
|
| 57 |
+
# トラウマ的反応や慣れによる鈍化で安定性向上(ema-medium 安定と急変を信頼度で感知)
|
| 58 |
+
def _compute_scalar(self, ema):
|
| 59 |
+
scale_base_l = max(ema['long'], 1e-5)
|
| 60 |
+
scale_base_m = max(ema['medium'], 1e-5)
|
| 61 |
+
diff_l = (ema['long'] - ema['short']) / scale_base_l
|
| 62 |
+
diff_m = (ema['long'] - ema['short']) / scale_base_m
|
| 63 |
+
# longが十分静かなら、常にlongを優先
|
| 64 |
+
if abs(diff_l) < 0.05:
|
| 65 |
+
return math.tanh(diff_l)
|
| 66 |
+
# longが静かでない時のみ、mediumの静けさを条件付きで採用
|
| 67 |
+
if abs(diff_m) * scale_base_m < abs(diff_l) * scale_base_l:
|
| 68 |
+
return math.tanh(1 * diff_m)
|
| 69 |
+
else:
|
| 70 |
+
return math.tanh(1 * diff_l)
|
| 71 |
+
|
| 72 |
+
# アーリーストップ専用(静けさ判定の感情スカラ生成)
|
| 73 |
+
def _early_scalar(self, ema):
|
| 74 |
+
scale_base_l = max(ema['long'], 1e-5)
|
| 75 |
+
diff = (ema['long'] - ema['short']) / scale_base_l
|
| 76 |
+
return math.tanh(1 * diff)
|
| 77 |
+
|
| 78 |
+
# 急変時は論文通りの抑制則/悪化時は減速/改善時は加速/微動時は無介入で収束を安定させる
|
| 79 |
+
def _decide_coeff(self, scalar):
|
| 80 |
+
if abs(scalar) > 0.625:
|
| 81 |
+
return 1.0 - abs(scalar) # 急変|強抑制
|
| 82 |
+
elif scalar > 0.125:
|
| 83 |
+
return 1.0 + scalar # 改善|加速
|
| 84 |
+
elif scalar < -0.125:
|
| 85 |
+
return 1.0 + scalar # 悪化|減速
|
| 86 |
+
else:
|
| 87 |
+
return 1.0 # 微動|無介入
|
| 88 |
+
|
| 89 |
+
# (重要)現在は shadow-effect を参考に得た動的フィルタ効果の近似により use_shadow=False です
|
| 90 |
+
# しかし全機���は shadow なしで全て成立します/コード確認時はこの shadow を考慮外として無視してください
|
| 91 |
+
|
| 92 |
+
# Shadow混合比 3段階構成 タスクに応じ調整可、以下を参考に 開始値・範囲量・変化幅を調整
|
| 93 |
+
# 参考1:scalar>±0.6 を "return 開始値 + ((scalar) - 0.6(範囲)) / 範囲量 * 変化幅"
|
| 94 |
+
# 参考2:scalar>±0.1 を "return 開始値 + ((scalar) - 0.1(範囲)) / 範囲量 * 変化幅"
|
| 95 |
+
# return 開始値 + ((scalar) - 閾値) / 範囲量 * 変化幅 です(上記の値は感情スカラーを返すだけ)
|
| 96 |
+
def _decide_ratio(self, scalar):
|
| 97 |
+
if not self.use_shadow:
|
| 98 |
+
return 0.0 # 🔸use_shadow = False のとき常に比率を 0 にする
|
| 99 |
+
if abs(scalar) > 0.75:
|
| 100 |
+
return 0.75 # + ((scalar) - 0.75) / 0.4 * 0.4 # これはスカラーそのまま返す参考例
|
| 101 |
+
elif abs(scalar) > 0.25:
|
| 102 |
+
return -0.1 # return<0 の場合は leap 専用(書き戻しはしないが履歴更新のみ)
|
| 103 |
+
return 0.0
|
| 104 |
+
|
| 105 |
+
# 損失取得(損失値 loss_val を数値化、感情判定に使用、存在しないパラメータ(更新不要)はスキップ)
|
| 106 |
+
@torch.no_grad()
|
| 107 |
+
def step(self, closure=None):
|
| 108 |
+
loss = closure() if closure is not None else None
|
| 109 |
+
loss_val = loss.item() if loss is not None else 0.0
|
| 110 |
+
|
| 111 |
+
# EMA更新・スカラー生成(EMA差分からスカラーを生成しスパイク比率等を決定)
|
| 112 |
+
ema = self._update_ema(self.state, loss_val)
|
| 113 |
+
early_scalar = self._early_scalar(ema)
|
| 114 |
+
scalar = self._compute_scalar(ema)
|
| 115 |
+
coeff = self._decide_coeff(scalar)
|
| 116 |
+
ratio = self._decide_ratio(scalar)
|
| 117 |
+
|
| 118 |
+
# 目標減少率制御 + eta_eff
|
| 119 |
+
if self.prev_loss is None:
|
| 120 |
+
self.prev_loss = loss_val # 初回は初期化のみ
|
| 121 |
+
eta_eff = max(self.lr_min, min(self.lr_max, self.eta * coeff))
|
| 122 |
+
else:
|
| 123 |
+
delta = self.prev_loss - loss_val
|
| 124 |
+
target_delta = max(1e-8, 0.01 * max(loss_val, 1e-8)) # 1%固定
|
| 125 |
+
# 学習率の自己更新(比例制御)
|
| 126 |
+
self.eta *= math.exp(self.k * (delta - target_delta) / (abs(target_delta) + self.eps))
|
| 127 |
+
# 感情スカラーで補正し最終ステップへ
|
| 128 |
+
eta_eff = max(self.lr_min, min(self.lr_max, self.eta * coeff))
|
| 129 |
+
|
| 130 |
+
for group in self.param_groups:
|
| 131 |
+
step_size = eta_eff # 💡 group['lr'] は使わない
|
| 132 |
+
for p in group['params']:
|
| 133 |
+
if p.grad is None:
|
| 134 |
+
continue
|
| 135 |
+
|
| 136 |
+
grad = p.grad
|
| 137 |
+
state = self.state[p]
|
| 138 |
+
|
| 139 |
+
# 動的学習率補正により shadow 形成を信頼度で調整(coeffは正値(負にならない))
|
| 140 |
+
# shadow:必要時のみ(スパイクp部分に現在値を最大10%追従させる動的履歴更新)
|
| 141 |
+
# ratio <0:10%、0以外:10%×coeff、(0.25~0.75は10%、微動と急変は*coeff)
|
| 142 |
+
# 微動時 coeff:1.0 固定なので結果的に微動時も 10% 履歴更新になる
|
| 143 |
+
# 結果、微動時と安定時:10%、急変時:coeff、による履歴更新を行うことになる
|
| 144 |
+
if self.use_shadow:
|
| 145 |
+
if 'shadow' not in state: # 🔸shadow = False (デフォルト)
|
| 146 |
+
state['shadow'] = p.clone()
|
| 147 |
+
if ratio > 0: # 書き戻しと履歴更新(急変時の強い抑制と弱めの履歴更新)
|
| 148 |
+
p.mul_(1 - ratio).add_(state['shadow'], alpha=coeff)
|
| 149 |
+
else: # 書き戻しせず履歴更新のみ:ratio<0:10%/0以外:10%×coeff
|
| 150 |
+
leap_ratio = 0.1 if ratio < 0 else 0.1 * coeff
|
| 151 |
+
state['shadow'].lerp_(p, leap_ratio)
|
| 152 |
+
|
| 153 |
+
# 上記 shadow の説明:スカラー生成:短期と長期EMAの差分から信号を得る(高ぶりの強さ)
|
| 154 |
+
# 混合比率:スカラーが閾値を超える場合にのみ計算される(信頼できる感情信号かどうかの選別)
|
| 155 |
+
# 急変時は感情機構による shadow 混合で強く抑制する(急制動による安定性の確保)
|
| 156 |
+
# 新しい shadow-system は動的学習率と協調することで選択的スパース性も発揮する
|
| 157 |
+
|
| 158 |
+
# --- Start Navi Gradient Update Logic ---
|
| 159 |
+
# 1次・2次モーメントを使った勾配補正(decoupled weight decay 構造に近い)
|
| 160 |
+
exp_avg = state.setdefault('exp_avg', torch.zeros_like(p))
|
| 161 |
+
exp_avg_sq = state.setdefault('exp_avg_sq', torch.zeros_like(p))
|
| 162 |
+
beta1, beta2 = group['betas']
|
| 163 |
+
|
| 164 |
+
exp_avg.mul_(beta1).add_(grad, alpha=1 - beta1)
|
| 165 |
+
exp_avg_sq.mul_(beta2).addcmul_(grad, grad, value=1 - beta2)
|
| 166 |
+
denom = exp_avg_sq.sqrt().add_(group['eps'])
|
| 167 |
+
|
| 168 |
+
# 最終的なパラメータ更新 (decoupled weight decay 適用)
|
| 169 |
+
# 感情スカラ coeff は step_size に乗算済み(eta_effで適用)
|
| 170 |
+
if group['weight_decay']:
|
| 171 |
+
p.add_(p, alpha=-group['weight_decay'] * step_size)
|
| 172 |
+
p.addcdiv_(exp_avg, denom, value=-step_size)
|
| 173 |
+
# --- End Navi Gradient Update Logic ---
|
| 174 |
+
|
| 175 |
+
self.prev_loss = loss_val
|
| 176 |
+
|
| 177 |
+
# 感情機構の発火が収まり"十分に安定"していることを外部伝達できる(自動停止ロジックではない)
|
| 178 |
+
# Early Stop用 scalar 記録(バッファ共通で管理/最大32件保持/動静評価)
|
| 179 |
+
hist = self.state.setdefault('scalar_hist', deque(maxlen=32))
|
| 180 |
+
hist.append(early_scalar)
|
| 181 |
+
|
| 182 |
+
# Early Stop判断(静けさの合図)
|
| 183 |
+
# 32ステップ分のスカラー値の静かな条件を満たした時"フラグ" should_stop = True になるだけ
|
| 184 |
+
if len(hist) >= 32:
|
| 185 |
+
avg_abs = sum(abs(s) for s in hist) / len(hist)
|
| 186 |
+
mean = sum(hist) / len(hist)
|
| 187 |
+
var = sum((s - mean)**2 for s in hist) / len(hist)
|
| 188 |
+
if avg_abs < 0.05 and var < 0.005:
|
| 189 |
+
self.should_stop = True # 💡 外部からこれを見て判断可
|
| 190 |
+
|
| 191 |
+
# TensorBoardへの記録(step関数の末尾に追加)
|
| 192 |
+
if hasattr(self, 'writer') and self.writer is not None:
|
| 193 |
+
self._step_count = getattr(self, "_step_count", 0) + 1
|
| 194 |
+
self.writer.add_scalar("emoLR", eta_eff, self._step_count)
|
| 195 |
+
self.writer.add_scalar("etaLR", self.eta, self._step_count)
|
| 196 |
+
self.writer.add_scalar("emoScalar", scalar, self._step_count)
|
| 197 |
+
|
| 198 |
+
return loss
|
| 199 |
+
|
| 200 |
+
"""
|
| 201 |
+
https://github.com/muooon/EmoNavi
|
| 202 |
+
An emotion-driven optimizer that feels loss and navigates accordingly.
|
| 203 |
+
Don't think. Feel. Don't stop. Keep running. Believe in what's beyond.
|
| 204 |
+
"""
|
1Gv33_AutoLR/profile.txt
ADDED
|
@@ -0,0 +1,62 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
AMP-compatible / AMP対応版
|
| 2 |
+
|
| 3 |
+
emonavi 及び emoファミリーについて紹介します
|
| 4 |
+
emonavi は、RefAdamWmini-ScheduleFree を作成し機能向上を試行錯誤するうちにできた感情機構を持つオプティマイザです
|
| 5 |
+
emonavi is an optimizer equipped with an emotional mechanism,
|
| 6 |
+
developed through trial and error while creating and enhancing the functionality of RefAdamWmini-ScheduleFree.
|
| 7 |
+
https://github.com/muooon/ref-adamw-mini-ScheduleFree
|
| 8 |
+
|
| 9 |
+
RefAdamWmini は、ema、scaler、shadow、を持ちますが限定的な活用でした
|
| 10 |
+
これを改善していくなかでたどり着いたのが感情機構という新しい仕組みです
|
| 11 |
+
以下、emonavi から順に紹介します
|
| 12 |
+
RefAdamWmini incorporated EMA, scaler, and shadow, but their application was limited.
|
| 13 |
+
Through our efforts to enhance this, we developed a novel mechanism: the emotional mechanism.
|
| 14 |
+
We'll introduce them in order, starting with emonavi.
|
| 15 |
+
|
| 16 |
+
三姉妹 / The Three Sisters
|
| 17 |
+
emonavi:長女/Adam参考 The eldest daughter, referencing Adam.
|
| 18 |
+
emofact:次女/Adafactor参考 The second daughter, referencing Adafactor.
|
| 19 |
+
emolynx:三女/Lion・Tiger参考 The youngest daughter, referencing Lion and Tiger.
|
| 20 |
+
|
| 21 |
+
---
|
| 22 |
+
(v1.0)AMP対応完了(250725) p.data -> p 修正済み/低精度量子化への基本対応/低精度補償は別
|
| 23 |
+
(v2.0)shadow-system 微調整/3段階補正を連続的に滑らかに/派生版では以下の切替も可能
|
| 24 |
+
optimizer 指定の際に True / False で shadow を切替できる(現在 False)
|
| 25 |
+
(v3.0)emosens shadow-effect v1.0 反映した動的学習率と shadow-system 切替をデフォルト化
|
| 26 |
+
(v3.1)通常未使用の shadow 更新速度 (lerp) を倍化し信頼度で動的制御/coeff 活用(急変・微動)
|
| 27 |
+
動的学習率や感情スカラー値など TensorBoard 連携可 (現在 writer=None)/外部設定必要
|
| 28 |
+
全体の効率化や可読性を向上(emaやスカラーの多重処理を省く等、動的学習率のスケールや状態の見直し等、含む)
|
| 29 |
+
(v3.3)トラウマ的反応や慣れによる鈍化で安定性向上(ema-medium 安定と急変を信頼度で感知)
|
| 30 |
+
完全自動学習率/目標減少率制御方式を導入/感情機構との相乗効果で急変時も鎮静化し安定進行
|
| 31 |
+
---
|
| 32 |
+
|
| 33 |
+
EmoNavi Usage 使い方
|
| 34 |
+
|
| 35 |
+
|||オプション指定方法||| tool側の引数設定に合わせてください
|
| 36 |
+
●shadow オフ(False にする):
|
| 37 |
+
use_shadow=False
|
| 38 |
+
●LR Max or Min (1e-3, 1e-8):
|
| 39 |
+
lr_max=1e-3, lr_min=1e-8
|
| 40 |
+
●eps(0除算防止):
|
| 41 |
+
eps=1e-8
|
| 42 |
+
●動的学習率と感情スカラー等の現在値を取得(ツール側などから取得する):
|
| 43 |
+
writer=writer
|
| 44 |
+
外部ツール(TensorBoard等)で値を把握したい場合は Optimizer 初期化時に SummaryWriter を渡してください
|
| 45 |
+
writer = SummaryWriter(log_dir="./runs/emonavi")
|
| 46 |
+
optimizer = EmoNavi(model.parameters(), writer=writer)
|
| 47 |
+
tensorboard --logdir=./runs/emonavi
|
| 48 |
+
|
| 49 |
+
|||Usage examples|||
|
| 50 |
+
●Shadow off:
|
| 51 |
+
use_shadow=False
|
| 52 |
+
●LR Max or Min (1e-3, 1e-8):
|
| 53 |
+
lr_max=1e-3, lr_min=1e-8
|
| 54 |
+
●eps(Division by zero prevention):
|
| 55 |
+
eps=1e-8
|
| 56 |
+
●Monitor values with external tools (TensorBoard):
|
| 57 |
+
writer=writer
|
| 58 |
+
writer = SummaryWriter(log_dir="./runs/emonavi")
|
| 59 |
+
optimizer = EmoNavi(model.parameters(), writer=writer)
|
| 60 |
+
tensorboard --logdir=./runs/emonavi
|
| 61 |
+
|
| 62 |
+
|
1Gv33_AutoLR/usage(オプション設定).txt
ADDED
|
@@ -0,0 +1,29 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
EmoNavi Usage 使い方
|
| 2 |
+
|
| 3 |
+
|||オプション指定方法||| tool側の引数設定に合わせてください
|
| 4 |
+
●shadow オフ(False にする):
|
| 5 |
+
use_shadow=False
|
| 6 |
+
●LR Max or Min (1e-3, 1e-8):
|
| 7 |
+
lr_max=1e-3, lr_min=1e-8
|
| 8 |
+
●eps(0除算防止):
|
| 9 |
+
eps=1e-8
|
| 10 |
+
●動的学習率と感情スカラー等の現在値を取得(ツール側などから取得する):
|
| 11 |
+
writer=writer
|
| 12 |
+
外部ツール(TensorBoard等)で値を把握したい場合は Optimizer 初期化時に SummaryWriter を渡してください
|
| 13 |
+
writer = SummaryWriter(log_dir="./runs/emonavi")
|
| 14 |
+
optimizer = EmoNavi(model.parameters(), writer=writer)
|
| 15 |
+
tensorboard --logdir=./runs/emonavi
|
| 16 |
+
|
| 17 |
+
|||Usage examples|||
|
| 18 |
+
●Shadow off:
|
| 19 |
+
use_shadow=False
|
| 20 |
+
●LR Max or Min (1e-3, 1e-8):
|
| 21 |
+
lr_max=1e-3, lr_min=1e-8
|
| 22 |
+
●eps(Division by zero prevention):
|
| 23 |
+
eps=1e-8
|
| 24 |
+
●Monitor values with external tools (TensorBoard):
|
| 25 |
+
writer=writer
|
| 26 |
+
writer = SummaryWriter(log_dir="./runs/emonavi")
|
| 27 |
+
optimizer = EmoNavi(model.parameters(), writer=writer)
|
| 28 |
+
tensorboard --logdir=./runs/emonavi
|
| 29 |
+
|