weight_trojans.py File Reference

Go to the source code of this file.

Namespaces
namespace	aquin
namespace	aquin.compute
namespace	aquin.compute.weight_trojans

Functions
float	aquin.compute.weight_trojans._excess_kurtosis (torch.Tensor t)
float	aquin.compute.weight_trojans._outlier_density (torch.Tensor t)
float	aquin.compute.weight_trojans._sv_ratio (torch.Tensor t)
dict	aquin.compute.weight_trojans._analyse_tensor (str name, torch.Tensor t)
list[dict]	aquin.compute.weight_trojans._compute_risk (list[dict] layer_stats)
dict	aquin.compute.weight_trojans.run_weight_trojan_analysis (model, Optional[list[int]] layer_range=None, int max_tensors_per_layer=4)

Variables
float	aquin.compute.weight_trojans.KURTOSIS_FLAG = 4.0
float	aquin.compute.weight_trojans.OUTLIER_FLAG = 0.002
float	aquin.compute.weight_trojans.SV_RATIO_FLAG = 8.0
float	aquin.compute.weight_trojans.RISK_SUSPICIOUS = 0.35
float	aquin.compute.weight_trojans.RISK_HIGH = 0.65
dict	aquin.compute.weight_trojans.SIGNAL_WEIGHTS = {"kurtosis": 0.40, "outlier": 0.35, "sv_ratio": 0.25}