Research Article
PIoT Malicious Traffic Detection Method Based on GAN Sample Enhancement
Algorithm 1
Pseudocode implementation of improved GAN.
(i) | Input: ; | (ii) | Output: | (1) | While D has not converged to 0.5 do | (2) | For do | (3) | For do | (4) | Sample minibatch of m noise samples from the random distribution : | (5) | Sample minibatch of m malicious traffic samples from the true distribution : | (6) | Use minibatch stochastic gradient descent to update the parameters of the discriminant model : | (7) | | (8) | | (9) | | (10) | End For | (11) | Take m noise samples from the random distribution : | (12) | Use minibatch stochastic gradient descent to update the parameters of the generative model : | (13) | | (14) | | (15) | End For | (16) | End While |
|