Struct EwcPlusPlus 

pub struct EwcPlusPlus { /* private fields */ }

EWC++ implementation

Implementations

impl EwcPlusPlus

pub fn new(config: EwcConfig) -> Self

Create new EWC++

pub fn update_fisher(&mut self, gradients: &[f32])

Update Fisher information online using EMA

pub fn detect_task_boundary(&self, gradients: &[f32]) -> bool

Detect task boundary using distribution shift

pub fn start_new_task(&mut self)

Start new task - saves current Fisher to memory

pub fn apply_constraints(&self, gradients: &[f32]) -> Vec<f32>

Apply EWC++ constraints to gradients

pub fn regularization_loss(&self, current_weights: &[f32]) -> f32

Compute EWC regularization loss

pub fn set_optimal_weights(&mut self, weights: &[f32])

Update optimal weights reference

pub fn consolidate_all_tasks(&mut self)

Consolidate all tasks (merge Fisher information)

pub fn lambda(&self) -> f32

Get current lambda

pub fn set_lambda(&mut self, lambda: f32)

Set lambda manually

pub fn task_count(&self) -> usize

Get task count

pub fn current_task_id(&self) -> usize

Get current task ID

pub fn samples_seen(&self) -> u64

Get samples seen for current task

pub fn importance_scores(&self) -> Vec<f32>

Get parameter importance scores

Trait Implementations

impl Clone for EwcPlusPlus

fn clone(&self) -> EwcPlusPlus

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for EwcPlusPlus

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for EwcPlusPlus

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for EwcPlusPlus

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.