SofiTesfay2010 commited on
Commit
ce46efa
·
verified ·
1 Parent(s): 13ef6c0

v0.3: config with auto-tune and profile path

Browse files
Files changed (1) hide show
  1. aria_llm/config.py +15 -9
aria_llm/config.py CHANGED
@@ -1,12 +1,10 @@
1
  """
2
- ARIA Configuration v0.2
3
  ========================
4
 
5
- All thresholds and hyperparameters. v0.2 adds:
6
- - calibration_steps: how long to observe before triggering (default 20)
7
- - sensitivity_k: how many standard deviations above baseline = triggered (default 2.5)
8
- - max_corrections_per_step: budget (default 1 — only strongest signal corrects)
9
- - correction_scale: global multiplier on all correction strengths (default 0.1)
10
  """
11
 
12
  from dataclasses import dataclass, field
@@ -17,14 +15,22 @@ from typing import Optional, List, Dict
17
  class ARIAConfig:
18
  """Configuration for the ARIA reliability attachment.
19
 
20
- v0.2: Calibration-first, budget-limited, statistically-grounded.
21
  """
22
 
23
- # === CALIBRATION (NEW in v0.2) ===
24
  calibration_steps: int = 20
25
  sensitivity_k: float = 2.5
26
 
27
- # === CORRECTION BUDGET (NEW in v0.2) ===
 
 
 
 
 
 
 
 
28
  max_corrections_per_step: int = 1
29
  correction_scale: float = 0.1
30
 
 
1
  """
2
+ ARIA Configuration v0.3
3
  ========================
4
 
5
+ All thresholds and hyperparameters. v0.3 adds:
6
+ - auto_tune_correction_scale: automatically set correction_scale from calibration variance
7
+ - calibration_profile_path: load a pre-saved calibration profile (skip calibration phase)
 
 
8
  """
9
 
10
  from dataclasses import dataclass, field
 
15
  class ARIAConfig:
16
  """Configuration for the ARIA reliability attachment.
17
 
18
+ v0.3: Auto-tuning + calibration profiles.
19
  """
20
 
21
+ # === CALIBRATION ===
22
  calibration_steps: int = 20
23
  sensitivity_k: float = 2.5
24
 
25
+ # === AUTO-TUNE (NEW in v0.3) ===
26
+ auto_tune_correction_scale: bool = True # auto-set correction_scale after calibration
27
+ auto_tune_min_scale: float = 0.02 # floor for auto-tuned scale
28
+ auto_tune_max_scale: float = 0.3 # ceiling for auto-tuned scale
29
+
30
+ # === CALIBRATION PROFILES (NEW in v0.3) ===
31
+ calibration_profile_path: Optional[str] = None # load pre-saved profile
32
+
33
+ # === CORRECTION BUDGET ===
34
  max_corrections_per_step: int = 1
35
  correction_scale: float = 0.1
36