Shapley-Neuron-Values/__init__.py at main · Ali-Vahedifar/Shapley-Neuron-Values · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
"""
Shapley Neuron Valuation (SNV) for Continual Learning

A memory-free continual learning framework that leverages Shapley values
from cooperative game theory to identify and protect important neurons.

Anonymous submission for ICML 2026.
"""

from .snv_core import (
    NeuronMaskManager,
    MeanActivationComputer,
    ShapleyNeuronEstimator,
    SNVContinualLearner
)

from .models import (
    MLP,
    ResNet18,
    BasicBlock,
    ContinualLearningModel,
    create_model,
    count_parameters,
    count_neurons
)

from .datasets import (
    PermutedMNIST,
    ContinualLearningDataset,
    TinyImageNet,
    ContinualLearningBenchmark,
    get_cifar100_transforms,
    get_tinyimagenet_transforms
)

from .metrics import (
    ContinualLearningMetrics,
    compute_capacity,
    compute_per_task_accuracies
)

__version__ = '1.0.0'
__author__ = 'Anonymous'

__all__ = [
    # Core SNV
    'NeuronMaskManager',
    'MeanActivationComputer',
    'ShapleyNeuronEstimator',
    'SNVContinualLearner',

    # Models
    'MLP',
    'ResNet18',
    'BasicBlock',
    'ContinualLearningModel',
    'create_model',
    'count_parameters',
    'count_neurons',

    # Datasets
    'PermutedMNIST',
    'ContinualLearningDataset',
    'TinyImageNet',
    'ContinualLearningBenchmark',
    'get_cifar100_transforms',
    'get_tinyimagenet_transforms',

    # Metrics
    'ContinualLearningMetrics',
    'compute_capacity',
    'compute_per_task_accuracies'
]