Struct RandomizedSmoothing 

pub struct RandomizedSmoothing {
    pub sigma: f32,
    pub num_samples: usize,
    pub alpha: f32,
}

Certified defense using randomized smoothing

Fields

sigma: f32

Noise standard deviation

num_samples: usize

Number of samples for certification

alpha: f32

Confidence level

Implementations

impl RandomizedSmoothing

pub fn new(sigma: f32, num_samples: usize, alpha: f32) -> Self

Create new randomized smoothing defense

pub fn predict( &self, input: &Tensor, model_predict: impl Fn(&Tensor) -> Result<usize, String>, ) -> Result<usize, String>

Predict with randomized smoothing

pub fn certify( &self, input: &Tensor, model_predict: impl Fn(&Tensor) -> Result<usize, String>, ) -> Result<(usize, f32), String>

Compute certified radius