nsl.lib.virtual_adv_regularizer
bookmark_borderbookmark
Stay organized with collections
Save and categorize content based on your preferences.
Calculates virtual adversarial loss for the given input.
nsl.lib.virtual_adv_regularizer(
input_layer, embedding_fn, virtual_adv_config, embedding=None
)
Virtual adversarial loss is defined as the distance between the embedding of
the input and that of a slightly perturbed input. Optimizing this loss helps
smooth models locally.
Reference paper: https://arxiv.org/pdf/1704.03976.pdf
Args |
input_layer
|
a dense tensor for input features whose first dimension is the
training batch size.
|
embedding_fn
|
a unary function that computes the embedding for the given
input_layer input.
|
virtual_adv_config
|
an nsl.configs.VirtualAdvConfig object that specifies
parameters for generating adversarial examples and computing the
adversarial loss.
|
embedding
|
(optional) a dense tensor representing the embedding of
input_layer . If not provided, it will be calculated as
embedding_fn(input_layer) .
|
Returns |
virtual_adv_loss
|
a float32 denoting the virtural adversarial loss.
|
Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.
Last updated 2022-10-28 UTC.
[[["Easy to understand","easyToUnderstand","thumb-up"],["Solved my problem","solvedMyProblem","thumb-up"],["Other","otherUp","thumb-up"]],[["Missing the information I need","missingTheInformationINeed","thumb-down"],["Too complicated / too many steps","tooComplicatedTooManySteps","thumb-down"],["Out of date","outOfDate","thumb-down"],["Samples / code issue","samplesCodeIssue","thumb-down"],["Other","otherDown","thumb-down"]],["Last updated 2022-10-28 UTC."],[],[]]