|
Algorithm 1: Training Procedure of the RC-SEI method. |
|
Require:
K: Number of training iterations;
N: Number of training samples;
C: Number of batches;
B: Number of Batchsize;
: Parameters of neural network and sparse parameters;
: Original features and sparse features;
: Learning rate;
: Coefficient of momentum;
: Sparse factor;
for to K do
for to C do
[Forward propagation]:
Sampling a batch of training samples
Initialize
Extracting the Original features:
Pruning the Original features:
Obtaining the artificial labels:
for to B do
end for
Calculating the loss:
[Backward propagation]:
Updating by Adam algorithm:
Updating by APGD-NAG algorithm:
end for
end for
|